transpub/transWhat/transwhat.py

129 lines
3.5 KiB
Python
Raw Normal View History

2013-06-28 21:10:28 +00:00
#!/usr/bin/python
# use unicode encoding for all literals by default (for python2.x)
from __future__ import unicode_literals
__author__ = "Steffen Vogel"
__copyright__ = "Copyright 2015-2017, Steffen Vogel"
__license__ = "GPLv3"
__maintainer__ = "Steffen Vogel"
__email__ = "post@steffenvogel.de"
2013-08-01 23:45:51 +00:00
"""
2013-08-01 23:45:51 +00:00
This file is part of transWhat
transWhat is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
any later version.
transwhat is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with transWhat. If not, see <http://www.gnu.org/licenses/>.
"""
2013-06-28 21:10:28 +00:00
import argparse
import traceback
2013-06-28 21:10:28 +00:00
import logging
import asyncore
import sys, os
import e4u
import Queue
2017-10-29 18:04:20 +00:00
import transWhat.threadutils
2013-06-28 21:10:28 +00:00
sys.path.insert(0, os.getcwd())
from Spectrum2.iochannel import IOChannel
from Spectrum2.config import SpectrumConfig
2017-10-29 18:04:20 +00:00
from transWhat.whatsappbackend import WhatsAppBackend
2015-06-21 12:01:05 +00:00
from yowsup.common import YowConstants
from yowsup.stacks import YowStack
2013-06-28 21:10:28 +00:00
# Arguments
parser = argparse.ArgumentParser()
parser.add_argument('--debug', action='store_true')
parser.add_argument('--log', type=str)
parser.add_argument('--host', type=str, required=True)
parser.add_argument('--port', type=int, required=True)
parser.add_argument('--service.backend_id', metavar="ID", type=int, required=True)
parser.add_argument('config', type=str)
2016-04-09 22:21:58 +00:00
parser.add_argument('-j', type=str, metavar="JID", required=True)
2013-06-28 21:10:28 +00:00
args, unknown = parser.parse_known_args()
YowConstants.PATH_STORAGE='/var/lib/spectrum2/' + args.j
if args.log is None:
args.log = '/var/log/spectrum2/' + args.j + '/backends/backend.log'
2013-06-28 21:10:28 +00:00
# Logging
logging.basicConfig(
filename = args.log,
format = "%(asctime)-15s %(levelname)s %(name)s: %(message)s",
level = logging.DEBUG if args.debug else logging.INFO
2013-06-28 21:10:28 +00:00
)
2016-04-12 16:21:04 +00:00
if args.config is not None:
specConf = SpectrumConfig(args.config)
2016-04-12 16:21:04 +00:00
else:
specConf = None
2016-04-12 16:21:04 +00:00
2013-06-28 21:10:28 +00:00
# Handler
def handleTransportData(data):
try:
plugin.handleDataRead(data)
except SystemExit as e:
raise e
except:
logger = logging.getLogger('transwhat')
logger.error(traceback.format_exc())
2013-06-28 21:10:28 +00:00
e4u.load()
closed = False
def connectionClosed():
global closed
closed = True
2013-06-28 21:10:28 +00:00
# Main
io = IOChannel(args.host, args.port, handleTransportData, connectionClosed)
2013-06-28 21:10:28 +00:00
2016-04-12 16:21:04 +00:00
plugin = WhatsAppBackend(io, args.j, specConf)
2013-06-28 21:10:28 +00:00
2015-12-18 18:33:32 +00:00
plugin.handleBackendConfig({
'features': [
('send_buddies_on_login', 1),
('muc', 'true'),
],
2015-12-18 18:33:32 +00:00
})
2015-12-01 21:21:38 +00:00
def main():
while True:
try:
asyncore.loop(timeout=1.0, count=10, use_poll = True)
try:
callback = YowStack._YowStack__detachedQueue.get(False) #doesn't block
callback()
except Queue.Empty:
pass
else:
break
if closed:
break
while True:
try:
callback = transWhat.threadutils.eventQueue.get_nowait()
except Queue.Empty:
break
else:
callback()
except SystemExit:
break
except:
logger = logging.getLogger('transwhat')
logger.error(traceback.format_exc())