transpub/transWhat/transwhat.py

#!/usr/bin/python

import argparse
import traceback
import logging
import asyncore
import sys

from Spectrum2.iochannel import IOChannel
from Spectrum2.config import SpectrumConfig
from yowsup.common import YowConstants
from yowsup.stacks import YowStack

from .whatsappbackend import WhatsAppBackend
from . import threadutils

# 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)
parser.add_argument('-j', type=str, metavar="JID", required=True)

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'

# Logging
logging.basicConfig(
    filename = args.log,
    format = "%(asctime)-15s %(levelname)s %(name)s: %(message)s",
    level = logging.DEBUG if args.debug else logging.INFO
)

if args.config is not None:
    specConf = SpectrumConfig(args.config)
else:
    specConf = None

# Handler
def handleTransportData(data):
    try:
        plugin.handleDataRead(data)
    except SystemExit as e:
        raise e
    except:
        logger = logging.getLogger('transwhat')
        logger.error(traceback.format_exc())

closed = False
def connectionClosed():
    global closed
    closed = True

# Main
io = IOChannel(args.host, args.port, handleTransportData, connectionClosed)

plugin = WhatsAppBackend(io, args.j, specConf)

plugin.handleBackendConfig({
    'features': [
        ('send_buddies_on_login', 1),
        ('muc', 'true'),
    ],
})

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())