Cleverbot mod for •AZ•BOT•

_cleverbot.py

#!/usr/bin/python
"""
https://code.google.com/p/pycleverbot
This library lets you open chat session with cleverbot (www.cleverbot.com)

Example of how to use the bindings:

>>> import cleverbot
>>> cb=cleverbot.Session()
>>> print cb.Ask("Hello there")
'Hello.'

"""

import urllib2
import hashlib
import re

class ServerFullError(Exception):
        pass

ReplyFlagsRE = re.compile('<INPUT NAME=(.+?) TYPE=(.+?) VALUE="(.*?)">', re.IGNORECASE | re.MULTILINE)

class Session(object):
        keylist=['stimulus','start','sessionid','vText8','vText7','vText6','vText5','vText4','vText3','vText2','icognoid','icognocheck','prevref','emotionaloutput','emotionalhistory','asbotname','ttsvoice','typing','lineref','fno','sub','islearning','cleanslate']
        headers={}
        headers['User-Agent']='Mozilla/5.0 (Windows NT 6.1; WOW64; rv:7.0.1) Gecko/20100101 Firefox/7.0'
        headers['Accept']='text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8'
        headers['Accept-Language']='en-us;q=0.8,en;q=0.5'
        headers['X-Moz']='prefetch'
        headers['Accept-Charset']='ISO-8859-1,utf-8;q=0.7,*;q=0.7'
        headers['Referer']='http://www.cleverbot.com'
        headers['Cache-Control']='no-cache, no-cache'
        headers['Pragma']='no-cache'

        def __init__(self):
                self.arglist=['','y','','','','','','','','','wsf','','','','','','','','','0','Say','1','false']
                self.MsgList=[]

        def Send(self):
                data=encode(self.keylist,self.arglist)
                digest_txt=data[9:29]
                hash=hashlib.md5(digest_txt).hexdigest()
                self.arglist[self.keylist.index('icognocheck')]=hash
                data=encode(self.keylist,self.arglist)
                req=urllib2.Request("http://www.cleverbot.com/webservicemin",data,self.headers)
                f=urllib2.urlopen(req)
                reply=f.read()
                return reply

        def Ask(self,q):
                self.arglist[self.keylist.index('stimulus')]=q
                if self.MsgList: self.arglist[self.keylist.index('lineref')]='!0'+str(len(self.MsgList)/2)
                asw=self.Send()
                self.MsgList.append(q)
                answer = parseAnswers(asw)
                for k,v in answer.iteritems():
                        try:
                                self.arglist[self.keylist.index(k)] = v
                        except ValueError:
                                pass
                self.arglist[self.keylist.index('emotionaloutput')]=''
                text = answer['ttsText']
                self.MsgList.append(text)
                return text

def parseAnswers(text):
        d = {}
        keys = ["text", "sessionid", "logurl", "vText8", "vText7", "vText6", "vText5", "vText4", "vText3",
                        "vText2", "prevref", "foo", "emotionalhistory", "ttsLocMP3", "ttsLocTXT",
                        "ttsLocTXT3", "ttsText", "lineRef", "lineURL", "linePOST", "lineChoices",
                        "lineChoicesAbbrev", "typingData", "divert"]
        values = text.split("\r")
        i = 0
        for key in keys:
                d[key] = values[i]
                i += 1
        return d

def encode(keylist,arglist):
        text=''
        for i in range(len(keylist)):
                k=keylist[i]; v=quote(arglist[i])
                text+='&'+k+'='+v
        text=text[1:]
        return text

always_safe = ('ABCDEFGHIJKLMNOPQRSTUVWXYZ'
               'abcdefghijklmnopqrstuvwxyz'
               '0123456789' '_.-')
def quote(s, safe = '/'):   #quote('abc def') -> 'abc%20def'
        safe += always_safe
        safe_map = {}
        for i in range(256):
                c = chr(i)
                safe_map[c] = (c in safe) and c or  ('%%%02X' % i)
        res = map(safe_map.__getitem__, s)
        return ''.join(res)


def main():
        import sys
        cb = Session()

        q = ''
        while q != 'bye':
                try:
                        q = raw_input("> ")
                except KeyboardInterrupt:
                        print
                        sys.exit()
                print cb.Ask(q)

if __name__ == "__main__":
        main()

clever.py

"""
Cleverbot mod

Depends on:
    core mods (azbot >= 1.3.0)
    _cleverbot.py

Config options:

cleverbot:
  # Name the cleverbot responds to
  #
  # Required.
  prefix: "clever: "

  # Name the cleverbot speaks to
  #
  # If not present, will default to whomever
  # addressed the bot. If empty, will speak
  # to everyone.
  #talk_to:
"""

from pubsub import pub
import _cleverbot
import _util

config = {}
bots = {}

def scanner(bot, user, channel, message, topic=pub.AUTO_TOPIC):
    coords = {
        'bot': bot,
        'user': user,
        'channel': channel
    }
    
    if 'cleverbot' in bot.config:
        global config
        config = bot.config['cleverbot']
    else:
        print "[!] Clever: not configured."
        return None
    
    if bot.network['name'] == topic.getNameTuple()[-1]:
        if message.startswith(config['prefix']):
            line = message[len(config['prefix']):]
            
            global bots
            if not bot.network['name'] in bots:
                bots[bot.network['name']] = {}
            if not channel in bots[bot.network['name']]:
                bots[bot.network['name']][channel] = _cleverbot.Session()
            
            print "Asking cleverbot..."
            line = bots[bot.network['name']][channel].Ask(line)
            
            if 'talk_to' in config:
                prep = (config['talk_to'], line)
            else:
                prep = (user, line)
            
            _util.answer(coords, "%s: %s" % prep)


pub.subscribe(scanner, 'irc.message')