Example #1
0
    def __init__(self, configfile='team.cfg'):
        self.configuration = TeamConfiguration(configfile)

        self.Jid, self.Password = self.configuration.getBotId()

        sleekxmpp.ClientXMPP.__init__(self, self.Jid, self.Password)

        self.register_plugin('xep_0030') # Service Discovery
        self.register_plugin('xep_0199') # XMPP Ping

        self.logger = StatsLog()

        self.staffstate = StaffState()

        # The session_start event will be triggered when
        # the bot establishes its connection with the server
        # and the XML streams are ready for use. We want to
        # listen for this event so that we we can initialize
        # our roster.
        self.add_event_handler('connected', self.connected)
        self.add_event_handler('disconnected', self.disconnected)

        self.add_event_handler('presence_available', self.available)
        self.add_event_handler('presence_unavailable', self.unavailable)

        self.add_event_handler("session_start", self.sessionStart, threaded=True)
        #self.add_event_handler("message", self.messageCB())
        self.add_event_handler("message", self.message)
Example #2
0
class TeamBot(sleekxmpp.ClientXMPP):
    #GtalkServer = '74.125.142.125'

    ''' use DepInj so we can easy test '''
    def __init__(self, configfile='team.cfg'):
        self.configuration = TeamConfiguration(configfile)

        self.Jid, self.Password = self.configuration.getBotId()

        sleekxmpp.ClientXMPP.__init__(self, self.Jid, self.Password)

        self.register_plugin('xep_0030') # Service Discovery
        self.register_plugin('xep_0199') # XMPP Ping

        self.logger = StatsLog()

        self.staffstate = StaffState()

        # The session_start event will be triggered when
        # the bot establishes its connection with the server
        # and the XML streams are ready for use. We want to
        # listen for this event so that we we can initialize
        # our roster.
        self.add_event_handler('connected', self.connected)
        self.add_event_handler('disconnected', self.disconnected)

        self.add_event_handler('presence_available', self.available)
        self.add_event_handler('presence_unavailable', self.unavailable)

        self.add_event_handler("session_start", self.sessionStart, threaded=True)
        #self.add_event_handler("message", self.messageCB())
        self.add_event_handler("message", self.message)

    def connected(self, event=None):
        self.logger.logConsole('connected...')

    def disconnected(self, event=None):
        self.logger.logConsole('disconnected...re-connected GTalk server')
        self.connectToGtalk()

    def available(self, presence):
        pto = presence['to'].bare
        pfrom = presence['from'].bare
        self.logger.logConsole('presence available : ' + pto + ' from: ' + pfrom)

    def unavailable(self, presence):
        pto = presence['to'].bare
        pfrom = presence['from'].bare
        self.logger.logConsole('presence unavailable : ' + pto + ' from: ' + pfrom)

    """ connect to GTalk thru tls http proxying wwwgate0.mot """
    def connectToGtalk(self):
        gtalk = self.configuration.getGtalkServer()
        self.use_proxy = self.configuration.useProxy()
        if self.use_proxy:
            print 'use proxy', self.use_proxy
            #raise AssertionError, 'use proxy wrong'
            h,p,u,pwd = self.configuration.getProxyInfo()
            self.proxy_config = {
                'host': h,
                'port': int(p),
                'username': u,
                'password': pwd}

        if self.connect((gtalk, 5222), True, True, False):
            self.process(block=True)
            print("Done")
        else:
            print("Unable to connect.")

    """ the main entry point of bot logic """
    def sessionStart(self, event):
        """
        Process the session_start event.

        Typical actions for the session_start event are
        requesting the roster and broadcasting an initial
        presence stanza.
        """
        self.send_presence()
        self.get_roster()

        day = self.configuration.getDayOfWeek()
        hour = self.configuration.getHour()
        minute = self.configuration.getMinute()

        # add schedule cron job
        sched.add_cron_job(self.collectStats, day_of_week=day, hour=hour, minute=minute)

        # Using wait=True ensures that the send queue will be
        # emptied before ending the session.
        #self.disconnect(wait=True)
        self.collectStats()

    #@sched.cron_schedule(day_of_week='mon-sat', hour='10-17', minute=12)
    def collectStats(self):
        memlist = self.configuration.getTeamMember()
        for name,jid in memlist:
            self.staffstate.updateQuestionState(name, '1')   # first question
            self.sendMsg(jid, self.configuration.getQuestion('1'))

    # send msg to jid
    def sendMsg(self, toid, msg):
        self.logger.logConsole('sending msg to : ' + toid + ' : ' + msg)
        self.send_message(mto=toid, mbody=msg, mtype='chat')

    def message(self, msg):
        """
        Process incoming message stanzas. Be aware that this also
        includes MUC messages and error messages. It is usually
        a good idea to check the messages's type before processing
        or sending replies.

        Arguments:
            msg -- The received message stanza. See the documentation
                   for stanza objects and the Message stanza to see
                   how it may be used.
        """
        if msg['type'] in ('chat', 'normal'):
            from_id = str(msg['from']).split('@')[0]
            to_id = msg['to']
            body = msg['body']
            print 'message : ', from_id, ' -> ', to_id, ' :: ', body
            self.logger.logFile(from_id, body)
            #msg.reply("Thanks for sending status!\n%(body)s" % msg).send()
            nextqidx, nextq = self.getNextQuestion(from_id)
            if nextq is not None:
                self.staffstate.updateQuestionState(from_id, nextqidx)   # first question
                msg.reply("Thanks!\n%s" % nextq).send()

    def messageCB(self):
        def cb(msg):
            if msg['type'] in ('chat', 'normal'):
                from_id = str(msg['from']).split('@')[0]  # need to convert to string
                to_id = msg['to']
                body = msg['body']
                print 'cb log msg: ', from_id, ' -> ', to_id, ' :: ', body
                self.logger.logFile(from_id, body)
                msg.reply("Thanks for sending status!\n%(body)s" % msg).send()
        return cb

    def getNextQuestion(self, name):
        qidx = int(self.staffstate.getQuestionState(name))
        return str(qidx+1), self.configuration.getQuestion(str(qidx+1))