From ebed8039842a22850672fd3e9adfbf1b3cc1b802 Mon Sep 17 00:00:00 2001 From: moyamo Date: Sun, 21 Jun 2015 16:44:20 +0200 Subject: [PATCH] Use YowStackBuilder and the default stack --- session.py | 28 ++++++++++++---------------- 1 file changed, 12 insertions(+), 16 deletions(-) diff --git a/session.py b/session.py index d06ca91..c2bc548 100644 --- a/session.py +++ b/session.py @@ -27,7 +27,7 @@ import logging import urllib import time -from yowsup.stacks import YowStack +from yowsup.stacks import YowStack, YowStackBuilder from yowsup.layers import YowLayerEvent, YowParallelLayer from yowsup.layers.interface import YowInterfaceLayer, ProtocolEntityCallback from yowsup.layers.auth import (YowCryptLayer, YowAuthenticationProtocolLayer, @@ -82,18 +82,10 @@ class Session(): self.bot = Bot(self) env.CURRENT_ENV = env.S40YowsupEnv() - layers = (SpectrumLayer, - YowParallelLayer((YowAuthenticationProtocolLayer, - YowMessagesProtocolLayer, - YowReceiptProtocolLayer, - YowAckProtocolLayer, - YowMediaProtocolLayer)), - YowCoderLayer, - YowCryptLayer, - YowStanzaRegulator, - YowNetworkLayer - ) - self.stack = YowStack(layers) + self.stack = YowStackBuilder()\ + .pushDefaultLayers(False)\ + .push(SpectrumLayer)\ + .build() self.stack.broadcastEvent( YowLayerEvent(SpectrumLayer.EVENT_START, backend = self.backend, @@ -163,13 +155,14 @@ class Session(): elif "-" in sender: # group msg if "/" in sender: room, buddy = sender.split("/") - self.call("message_send", (buddy + "@s.whatsapp.net", message)) + self.call("message_send", to = buddy + "@s.whatsapp.net", + message = message) else: room = sender group = self.groups[room] self.backend.handleMessage(self.user, room, message, group.nick) - self.call("message_send", (room + "@g.us", message)) + self.call("message_send", to = room + "@g.us", message = message) else: # private msg buddy = sender if message == "\\lastseen": @@ -451,18 +444,21 @@ class SpectrumLayer(YowInterfaceLayer): message = layerEvent.getArg('message') messageEntity = TextMessageProtocolEntity(message, to = to) self.toLower(messageEntity) + retval = True elif layerEvent.getName() == 'typing_send': buddy = layerEvent.getArg('buddy') state = OutgoingChatstateProtocolEntity( ChatstateProtocolEntity.STATE_TYPING, buddy ) self.toLower(state) - elif layerEvent.getName() == 'typing_pause': + retval = True + elif layerEvent.getName() == 'typing_paused': buddy = layerEvent.getArg('buddy') state = OutgoingChatstateProtocolEntity( ChatstateProtocolEntity.STATE_PAUSE, buddy ) self.toLower(state) + retval = True self.logger.debug("EVENT %s", layerEvent.getName()) return retval