circles/bot.py

#!/usr/bin/env python3
import teambot, sys, os

class Circles(teambot.Handler):
    def __init__(self, bot):
        print('initilized')
        self.modules = []
        self.cmd = {}
        self.raw = {}
        self.help = {}
        self.prefix = "."
    def loadMods(self):
        for i in [s for s in os.listdir('modules') if ".py" in s]:
            i = i[:-3]
            print('loading',i)
            m = __import__("modules."+i)
            m = eval('m.'+i)
            m.init(self)
            self.modules.append(m)
    def registerCommand(self, txt, func):
        self.cmd[txt] = func
    def regosterRaw(self, txt, func):
        self.raw[txt] = func
    def registerHelp(self, cmd, ht):
        self.help[cmd] = ht
    def on_connection_established(self):
        print('Connected!')
        self.loadMods()
        print('Ready!')
    def on_pubmessage(self,c,n,m):
        print(m)
        if m.startswith(self.prefix):
            args = m[len(self.prefix):].strip().split()
            cmd = args.pop()
            if cmd in self.cmd:
                try:
                    self.cmd[cmd](self,c,n,m)
                except TypeError:
                    self.say(c,n+": Usage: "+self.help[cmd])
        if cmd in self.raw:
            try:
                self.raw[cmd](self,c,n,m)
            except TypeError:
                self.say(c,n+": Usage: "+self.help[cmd])

if __name__=="__main__":
    chans = ['#chaos']
    bot = teambot.TeamBot(chans, "circles", "localhost", chandler=Circles)
    bot.start()