fb.py

# -*-coding:utf-8-*-
import sys
import sleekxmpp
import cleverbot
import logging
logging.basicConfig(level=logging.DEBUG)

reload(sys)
sys.setdefaultencoding('utf8')

user = sys.argv[1] + "@chat.facebook.com"
password = sys.argv[2]
sessions = {}
dont_send = []
only_send = []
spam = ["ios", "app!", "clever", "free"]


class EchoBot(sleekxmpp.ClientXMPP):

    def __init__(self, jid, pwd):
        sleekxmpp.ClientXMPP.__init__(self, jid, pwd)

        self.add_event_handler('session_start', self.start)
        self.add_event_handler('message', self.message)

    def start(self, event):
        self.send_presence()
        self.get_roster()

    def message(self, msg):
        str_from = str(msg['from'])
        str_body = str(msg['body'])

        if len(only_send) > 0:
            if str_from not in only_send:
                return

        if str_from in dont_send:
            return

        if str_body not in sessions:
            sessions[str_from] = cleverbot.Cleverbot()

        msg_back = str(sessions[str_from].ask(str_body))

        print "Msg come"
        while self.check_spam(msg_back):
            msg_back = str(sessions[str_from].ask(str_body))
            print "While, while, while"

        print "\n>>> %s\n >>>> %s\n" % (str_body, msg_back)

        self.send_message(
            mto=str_from,
            mbody=msg_back)

    def check_spam(self, msg):
        spam_available = False
        for word in spam:
            if word in msg.lower():
                spam_available = True
        return spam_available

if __name__ == "__main__":
    xmpp = EchoBot(user, password)
    xmpp.register_plugin('xep_0030')  # service discovery
    xmpp.register_plugin('xep_0004')  # date form
    xmpp.register_plugin('xep_0060')  # pubsub
    xmpp.register_plugin('xep_0199')  # xmpp ping

    if xmpp.connect():
        xmpp.process(threaded=False)