def __init__(self, cfg=None, usersin=None, plugs=None, botname=None, nick=None, bottype=None, nocbs=None, *args, **kwargs):
logging.debug("type is %s" % str(type(self)))
if cfg: self.cfg = cfg ; botname = botname or self.cfg.name
if not botname: botname = u"default-%s" % str(type(self)).split('.')[-1][:-2]
if not botname: raise Exception("can't determine botname")
self.fleetdir = u'fleet' + os.sep + stripname(botname)
if not self.cfg: self.cfg = Config(self.fleetdir + os.sep + u'config')
self.cfg.name = botname or self.cfg.name
if not self.cfg.name: raise Exception("name is not set in %s config file" % self.fleetdir)
logging.debug("name is %s" % self.cfg.name)
LazyDict.__init__(self)
logging.debug("created bot with config %s" % self.cfg.tojson(full=True))
self.ecounter = 0
self.ids = []
self.aliases = getaliases()
self.reconnectcount = 0
self.plugs = coreplugs
self.gatekeeper = GateKeeper(self.cfg.name)
self.gatekeeper.allow(self.user or self.jid or self.cfg.server or self.cfg.name)
self.starttime = time.time()
self.type = bottype or "base"
self.status = "init"
self.networkname = self.cfg.networkname or self.cfg.name or ""
from jsb.lib.datadir import getdatadir
datadir = getdatadir()
self.datadir = datadir + os.sep + self.fleetdir
self.maincfg = getmainconfig()
self.owner = self.cfg.owner
if not self.owner:
logging.debug(u"owner is not set in %s - using mainconfig" % self.cfg.cfile)
self.owner = self.maincfg.owner
self.users = usersin or getusers()
logging.debug(u"owner is %s" % self.owner)
self.users.make_owner(self.owner)
self.outcache = outcache
self.userhosts = LazyDict()
self.nicks = LazyDict()
self.connectok = threading.Event()
self.reconnectcount = 0
self.cfg.nick = nick or self.cfg.nick or u'jsb'
try:
if not os.isdir(self.datadir): os.mkdir(self.datadir)
except: pass
self.setstate()
self.outputlock = thread.allocate_lock()
try:
self.outqueue = Queue.PriorityQueue()
self.eventqueue = Queue.PriorityQueue()
except AttributeError:
self.outqueue = Queue.Queue()
self.eventqueue = Queue.Queue()
self.laterqueue = Queue.Queue()
self.encoding = self.cfg.encoding or "utf-8"
self.cmndperms = getcmndperms()
self.outputmorphs = outputmorphs
self.inputmorphs = inputmorphs
try:
if nocbs: self.nocbs = nocbs.split(",")
except ValueError: logging.error("cannot determine %s nocbs argument" % self.nocbs)
self.lastiter = 0
def reloadcheck(self, bot, event, target=None):
"""
check if event requires a plugin to be reloaded. if so
reload the plugin.
"""
from boot import getcmndtable
from boot import plugblacklist
plugloaded = None
plugin = None
try:
if not target:
target = event.iscmnd().split()[0]
if not target: target = evemt.txt.split()[0]
except Exception, ex: target = None
if not target: logging.debug("can't find target in %s" % event.txt) ; return
from jsb.lib.aliases import getaliases
aliases = getaliases()
try: target = aliases[target]
except KeyError:
try: target = event.chan.data.aliases[target]
except (AttributeError, KeyError, TypeError): pass
cmndtable = getcmndtable()
if not cmndtable.has_key(target):
try:
short = getshorttable()
if short.has_key(target):
cmndlist = short[target]
if len(cmndlist) == 1: target = cmndlist[0]
except Exception, ex: handle_exception()
if target: target = target.split()[0]
logging.info("checking for reload of %s" % target)
try:
plugin = cmndtable[target]
except KeyError:
logging.warn("no cmnd for %s .. trying REGEX" % target)
try:
retable = getretable()
for regex, mod in retable.iteritems():
if re.search(regex, event.stripcc() or event.txt): plugin = mod ; break
except Exception, ex: handle_exception()
logging.info("plugin is %s" % plugin)
if not plugin: logging.debug("can't find plugin to reload for %s" % target) ; return
if plugin in bot.plugs: logging.info("%s already loaded" % plugin) ; return plugloaded
elif plugin in plugblacklist.data: return plugloaded
elif bot.cfg.loadlist and plugin not in bot.cfg.loadlist: logging.warn("plugin %s is blacklisted" % plugin) ; return plugloaded
logging.info("loaded %s on demand" % plugin)
plugloaded = bot.plugs.reload(plugin)
return plugloaded
def woulddispatch(self, bot, event):
"""
dispatch an event if cmnd exists and user is allowed to exec this
command.
"""
event.bind(bot)
aliased = False
try:
cmnd = event.stripcc().split()[0]
if not cmnd: cmnd = event.execstr.split()[0]
if not cmnd: cmnd = event.txt.split()[0]
except Exception, ex: logging.warn("can't determine command from %s" % event.txt) ; return None
try:
a = event.chan.data.aliases[cmnd]
if a: cmnd = a.split()[0] ; aliased = True
except (KeyError, TypeError):
try:
a = getaliases()[cmnd]
if a: cmnd = a.split()[0] ; aliased = True
except (KeyError, TypeError):
if not self.has_key(cmnd):
try:
short = getshorttable()
if short.has_key(cmnd):
cmndlist = short[cmnd]
if len(cmndlist) == 1: cmnd = cmndlist[0]
else: event.reply("choose one of: ", cmndlist) ; return
except Exception, ex: handle_exception()
logging.info("trying for %s" % cmnd)
result = None
try:
result = self[cmnd]
except KeyError: pass
logging.debug("woulddispatch result: %s" % result)
if result: event.bloh() ; event.makeargs()
if aliased: event.usercmnd = cmnd
return result
class Commands(LazyDict):
"""
the commands object holds all commands of the bot.
"""
regex = []
def add(self,
cmnd,
func,
perms,
threaded=False,
wait=False,
orig=None,
how=None,
speed=None,
regex=False,
needcc=False,
*args,
**kwargs):
""" add a command. """
modname = calledfrom(sys._getframe())
try:
prev = self[cmnd]
except KeyError:
prev = None
target = Command(modname,
cmnd,
func,
perms,
threaded,
wait,
orig,
how,
speed=speed,
needcc=needcc)
if regex:
logging.info("regex command detected - %s" % cmnd)
self.regex.append(target)
target.regex = cmnd
return self
self[cmnd] = target
try:
p = cmnd.split('-')[0]
if not self.pre: self.pre = LazyDict()
if self.pre.has_key(p):
if not self.pre[p]: self.pre[p] = []
if prev in self.pre[p]: self.pre[p].remove(prev)
if target not in self.pre[p]: self.pre[p].append(target)
else: self.pre[p] = [
target,
]
except IndexError:
pass
return self
def checkre(self, bot, event):
gotcc = event.gotcc()
for r in self.regex:
if not r.enable: continue
if r.needcc and not event.gotcc():
logging.debug("RE needs cc but not found")
continue
s = re.search(r.cmnd, event.txt)
if s:
logging.info("regex matches %s" % r.cmnd)
event.groups = list(s.groups())
return r
def wouldmatchre(self, bot, event, cmnd=""):
groups = self.checkre(bot, event)
if groups: return groups
def woulddispatch(self, bot, event):
"""
dispatch an event if cmnd exists and user is allowed to exec this
command.
"""
event.bind(bot)
aliased = False
try:
cmnd = event.stripcc().split()[0]
if not cmnd: cmnd = event.execstr.split()[0]
if not cmnd: cmnd = event.txt.split()[0]
except Exception, ex:
logging.warn("can't determine command from %s" % event.txt)
return None
try:
a = event.chan.data.aliases[cmnd]
if a:
cmnd = a.split()[0]
aliased = True
except (KeyError, TypeError):
try:
a = getaliases()[cmnd]
if a:
cmnd = a.split()[0]
aliased = True
except (KeyError, TypeError):
if not self.has_key(cmnd):
try:
short = getshorttable()
if short.has_key(cmnd):
cmndlist = short[cmnd]
if len(cmndlist) == 1: cmnd = cmndlist[0]
else:
event.reply("choose one of: ", cmndlist)
return
except Exception, ex:
handle_exception()
"""
from boot import getcmndtable
from boot import plugblacklist
plugloaded = None
plugin = None
try:
if not target:
target = event.iscmnd().split()[0]
if not target: target = evemt.txt.split()[0]
except Exception, ex:
target = None
if not target:
logging.debug("can't find target in %s" % event.txt)
return
from jsb.lib.aliases import getaliases
aliases = getaliases()
try:
target = aliases[target]
except KeyError:
try:
target = event.chan.data.aliases[target]
except (AttributeError, KeyError, TypeError):
pass
cmndtable = getcmndtable()
if not cmndtable.has_key(target):
try:
short = getshorttable()
if short.has_key(target):
cmndlist = short[target]
if len(cmndlist) == 1: target = cmndlist[0]
except Exception, ex:
def __init__(self,
cfg=None,
usersin=None,
plugs=None,
botname=None,
nick=None,
bottype=None,
nocbs=None,
*args,
**kwargs):
logging.debug("type is %s" % str(type(self)))
if cfg:
self.cfg = cfg
botname = botname or self.cfg.name
if not botname:
botname = u"default-%s" % str(type(self)).split('.')[-1][:-2]
if not botname: raise Exception("can't determine botname")
self.fleetdir = u'fleet' + os.sep + stripname(botname)
if not self.cfg: self.cfg = Config(self.fleetdir + os.sep + u'config')
self.cfg.name = botname or self.cfg.name
if not self.cfg.name:
raise Exception("name is not set in %s config file" %
self.fleetdir)
logging.debug("name is %s" % self.cfg.name)
LazyDict.__init__(self)
logging.debug("created bot with config %s" %
self.cfg.tojson(full=True))
self.ecounter = 0
self.ids = []
self.aliases = getaliases()
self.reconnectcount = 0
self.plugs = coreplugs
self.gatekeeper = GateKeeper(self.cfg.name)
self.gatekeeper.allow(self.user or self.jid or self.cfg.server
or self.cfg.name)
self.starttime = time.time()
self.type = bottype or "base"
self.status = "init"
self.networkname = self.cfg.networkname or self.cfg.name or ""
from jsb.lib.datadir import getdatadir
datadir = getdatadir()
self.datadir = datadir + os.sep + self.fleetdir
self.maincfg = getmainconfig()
self.owner = self.cfg.owner
if not self.owner:
logging.debug(u"owner is not set in %s - using mainconfig" %
self.cfg.cfile)
self.owner = self.maincfg.owner
self.users = usersin or getusers()
logging.debug(u"owner is %s" % self.owner)
self.users.make_owner(self.owner)
self.outcache = outcache
self.userhosts = LazyDict()
self.nicks = LazyDict()
self.connectok = threading.Event()
self.reconnectcount = 0
self.cfg.nick = nick or self.cfg.nick or u'jsb'
try:
if not os.isdir(self.datadir): os.mkdir(self.datadir)
except:
pass
self.setstate()
self.outputlock = thread.allocate_lock()
try:
self.outqueue = Queue.PriorityQueue()
self.eventqueue = Queue.PriorityQueue()
except AttributeError:
self.outqueue = Queue.Queue()
self.eventqueue = Queue.Queue()
self.laterqueue = Queue.Queue()
self.encoding = self.cfg.encoding or "utf-8"
self.cmndperms = getcmndperms()
self.outputmorphs = outputmorphs
self.inputmorphs = inputmorphs
try:
if nocbs: self.nocbs = nocbs.split(",")
except ValueError:
logging.error("cannot determine %s nocbs argument" % self.nocbs)
self.lastiter = 0
def __init__(self, cfg=None, usersin=None, plugs=None, botname=None, nick=None, *args, **kwargs):
logging.debug("type is %s" % str(type(self)))
if cfg: cfg = LazyDict(cfg)
if cfg and not botname: botname = cfg.name
if not botname: botname = u"default-%s" % str(type(self)).split('.')[-1][:-2]
if not botname: raise Exception("can't determine type")
self.fleetdir = u'fleet' + os.sep + stripname(botname)
self.cfg = Config(self.fleetdir + os.sep + u'config')
if cfg: self.cfg.merge(cfg)
self.cfg.name = botname
if not self.cfg.name: raise Exception(" name is not set in %s config file" % self.fleetdir)
logging.debug("name is %s" % self.cfg.name)
LazyDict.__init__(self)
self.ignore = []
self.ids = []
self.started = False
self.aliases = getaliases()
self.curevent = None
self.inqueue = Queue.Queue()
self.outqueue = Queue.Queue()
self.reconnectcount = 0
self.stopped = False
self.plugs = coreplugs
self.gatekeeper = GateKeeper(self.cfg.name)
self.gatekeeper.allow(self.user or self.jid or self.cfg.server or self.cfg.name)
self.closed = False
try:
import waveapi
self.isgae = True
logging.debug("bot is a GAE bot (%s)" % self.cfg.name)
except ImportError:
self.isgae = False
logging.debug("bot is a shell bot (%s)" % self.cfg.name)
self.starttime = time.time()
self.type = "base"
self.status = "init"
self.networkname = self.cfg.networkname or self.cfg.name or ""
if not self.uuid:
if self.cfg and self.cfg.uuid: self.uuid = self.cfg.uuid
else:
self.uuid = self.cfg.uuid = uuid.uuid4()
self.cfg.save()
if self.cfg and not self.cfg.followlist: self.cfg.followlist = [] ; self.cfg.save()
from jsb.lib.datadir import getdatadir
datadir = getdatadir()
self.datadir = datadir + os.sep + self.fleetdir
self.owner = self.cfg.owner
if not self.owner:
logging.debug(u"owner is not set in %s - using mainconfig" % self.cfg.cfile)
self.owner = getmainconfig().owner
self.setusers(usersin)
logging.debug(u"owner is %s" % self.owner)
self.users.make_owner(self.owner)
self.outcache = outcache
self.userhosts = LazyDict()
self.connectok = threading.Event()
self.reconnectcount = 0
self.cfg.nick = nick or self.cfg.nick or u'jsb'
try:
if not os.isdir(self.datadir): os.mkdir(self.datadir)
except: pass
self.setstate()
self.stopreadloop = False
self.stopoutloop = False
self.outputlock = thread.allocate_lock()
self.outqueues = [Queue.Queue() for i in range(10)]
self.tickqueue = Queue.Queue()
self.encoding = self.cfg.encoding or "utf-8"
self.cmndperms = getcmndperms()
self.outputmorphs = outputmorphs
self.inputmorphs = inputmorphs
if not self.isgae:
defaultrunner.start()
callbackrunner.start()
waitrunner.start()
tickloop.start(self)
def __init__(self,
cfg=None,
usersin=None,
plugs=None,
botname=None,
nick=None,
*args,
**kwargs):
if not botname and cfg and cfg.botname: botname = cfg.botname
if botname: self.botname = botname
else:
self.botname = u"default-%s" % str(type(self)).split('.')[-1][:-2]
logging.info("botbase - name is %s" % self.botname)
self.fleetdir = u'fleet' + os.sep + stripname(self.botname)
if cfg:
self.cfg = Config(self.fleetdir + os.sep + u'config', input=cfg)
else:
self.cfg = Config(self.fleetdir + os.sep + u'config')
LazyDict.__init__(self)
self.update(self.cfg)
self.ignore = []
self.aliases = getaliases()
self.curevent = None
self.inqueue = Queue.Queue()
self.outqueue = Queue.Queue()
self.reconnectcount = 0
self.stopped = False
self.plugs = coreplugs
self.gatekeeper = GateKeeper(self.botname)
self.gatekeeper.allow(self.user or self.jid or self.server
or self.botname)
self.closed = False
try:
import waveapi
self.isgae = True
logging.debug("botbase - bot is a GAE bot (%s)" % self.botname)
except ImportError:
self.isgae = False
logging.debug("botbase - bot is a shell bot (%s)" % self.botname)
self.starttime = time.time()
self.type = "base"
self.status = "init"
self.networkname = self.cfg.networkname or self.botname or ""
if not self.uuid:
if self.cfg and self.cfg.uuid: self.uuid = self.cfg.uuid
else:
self.uuid = self.cfg.uuid = uuid.uuid4()
self.cfg.save()
if self.cfg and not self.cfg.followlist:
self.cfg.followlist = []
self.cfg.save()
from jsb.lib.datadir import getdatadir
datadir = getdatadir()
self.datadir = datadir + os.sep + self.fleetdir
self.name = self.botname
self.owner = self.cfg.owner
if not self.owner:
logging.debug(u"owner is not set in %s - using mainconfig" %
self.cfg.cfile)
self.owner = Config().owner
self.setusers(usersin)
logging.info(u"botbase - owner is %s" % self.owner)
self.users.make_owner(self.owner)
self.outcache = outcache
self.userhosts = {}
self.connectok = threading.Event()
if not self.nick: self.nick = (nick or self.cfg.nick or u'jsb')
try:
if not os.isdir(self.datadir): os.mkdir(self.datadir)
except:
pass
self.setstate()
self.stopreadloop = False
self.stopoutloop = False
self.outputlock = thread.allocate_lock()
self.outqueues = [Queue.Queue() for i in range(10)]
self.tickqueue = Queue.Queue()
self.encoding = self.cfg.encoding or "utf-8"
self.cmndperms = getcmndperms()
self.outputmorphs = outputmorphs
self.inputmorphs = inputmorphs
fleet = getfleet(datadir)
if not fleet.byname(self.name): fleet.bots.append(self)
if not self.isgae:
defaultrunner.start()
tickloop.start(self)
