circles/bot.py

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

class Circles(teambot.Handler):
    def __init__(self, bot):
        self.modules = []
        self.cmd = {}
        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.modules.append(m)
    def registerCommand(self, txt, func):
        self.cmd[txt] = func
    def on_connection_established(self):
        self.loadMods()
    def on_pubmessage(self,c,n,m):
        if m.startswith(self.prefix):
            args = m[len(self.prefix):].strip().split()
            cmd = args.pop
            if cmd in self.cmd:
                self.cmd[cmd](self,c,n,m)