def handle_threads(bot, ievent):
""" no arguments - show running threads. """
try: import threading
except ImportError:
ievent.reply("threading is not enabled.")
return
stats = StatDict()
names = Dol()
threadlist = threading.enumerate()
for thread in threadlist:
name = thread.getName()
try: type = thread.type
except AttributeError: type = name
stats.upitem(type)
try: names.add(type, thread.nowrunning)
except AttributeError: pass
if ievent.rest:
result = []
for item, n in names.items():
if not n: continue
res = "%s: " % item
for name in n: res += "%s, " % name
result.append(res[:-2])
result.sort()
ievent.reply("nowrunning: ", result, dot="<br>")
result = []
for item in stats.top(): result.append("%s = %s" % (item[0], item[1]))
result.sort()
ievent.reply(", ".join(result))
def __init__(self):
self.cbs = Dol()
class Callbacks(object):
"""
dict of lists containing callbacks. Callbacks object take care of
dispatching the callbacks based on incoming events. see Callbacks.check()
"""
def __init__(self):
self.cbs = Dol()
def size(self):
""" return number of callbacks. """
return self.cbs.size()
def add(self, what, func, prereq=None, kwargs=None, threaded=False, nr=False, speed=5):
""" add a callback. """
what = what.upper()
modname = calledfrom(sys._getframe())
if not kwargs: kwargs = {}
if nr != False: self.cbs.insert(nr, what, Callback(modname, func, prereq, kwargs, threaded, speed))
else: self.cbs.add(what, Callback(modname, func, prereq, kwargs, threaded, speed))
logging.debug('added %s (%s)' % (what, modname))
return self
register = add
def unload(self, modname):
""" unload all callbacks registered in a plugin. """
unload = []
for name, cblist in self.cbs.items():
index = 0
for item in cblist:
if item.modname == modname: unload.append((name, index))
index += 1
for callback in unload[::-1]:
self.cbs.delete(callback[0], callback[1])
logging.debug(' unloaded %s (%s)' % (callback[0], modname))
def disable(self, plugname):
""" disable all callbacks registered in a plugin. """
unload = []
for name, cblist in self.cbs.items():
index = 0
for item in cblist:
if item.plugname == plugname: item.activate = False
def activate(self, plugname):
""" activate all callbacks registered in a plugin. """
unload = []
for name, cblist in self.cbs.items():
index = 0
for item in cblist:
if item.plugname == plugname: item.activate = True
def whereis(self, cmnd):
""" show where ircevent.CMND callbacks are registered """
result = []
cmnd = cmnd.upper()
for c, callback in self.cbs.items():
if c == cmnd:
for item in callback:
if not item.plugname in result: result.append(item.plugname)
return result
def list(self):
""" show all callbacks. """
result = []
for cmnd, callbacks in self.cbs.items():
for cb in callbacks:
result.append(getname(cb.func))
return result
def check(self, bot, event):
""" check for callbacks to be fired. """
self.reloadcheck(bot, event)
type = event.cbtype or event.cmnd
if 'ALL' in self.cbs:
for cb in self.cbs['ALL']: self.callback(cb, bot, event)
if type in self.cbs:
target = self.cbs[type]
for cb in target: self.callback(cb, bot, event)
def callback(self, cb, bot, event):
""" do the actual callback with provided bot and event as arguments. """
#if event.stop: logging.info("callbacks - event is stopped.") ; return
event.calledfrom = cb.modname
if not event.bonded: event.bind(bot)
try:
if event.status == "done":
if not event.nolog: logging.debug("callback - event is done .. ignoring")
return
if event.chan and cb.plugname in event.chan.data.denyplug:
logging.debug("%s denied in %s - %s" % (cb.modname, event.channel, event.auth))
return
if cb.prereq:
if not event.nolog: logging.debug('executing in loop %s' % str(cb.prereq))
if not cb.prereq(bot, event): return
if not cb.func: return
if event.isremote(): logging.info('%s - executing REMOTE %s - %s' % (bot.cfg.name, getname(cb.func), event.cbtype))
elif not event.nolog: logging.info('%s - executing %s - %s' % (bot.cfg.name, getname(cb.func), event.cbtype))
event.iscallback = True
if not event.createdfrom: event.createdfrom = cb.modname
if not event.nolog: logging.debug("%s - %s - trail - %s" % (bot.cfg.name, getname(cb.func), callstack(sys._getframe())[::-1]))
if cb.threaded:
from .runner import threadrunner
threadrunner.put(event.speed or cb.speed, cb.modname, cb.func, bot, event)
else:
if event.cbtype == "API":
from .runner import apirunner
apirunner.put(event.speed or cb.speed, cb.modname, cb.func, bot, event)
elif event.direct: cb.func(bot, event)
elif not event.dolong:
from .runner import callbackrunner
callbackrunner.put(event.speed or cb.speed, cb.modname, cb.func, bot, event)
else:
from .runner import longrunner
longrunner.put(event.speed or cb.speed, cb.modname, cb.func, bot, event)
return True
except Exception as ex:
handle_exception()
def reloadcheck(self, bot, event, target=None):
""" check if plugin need to be reloaded for callback, """
from .boot import isdisabled
plugloaded = []
done = []
target = target or event.cbtype or event.cmnd
if not event.nolog: logging.debug("%s - checking for %s events" % (bot.cfg.name, target))
try:
from .boot import getcallbacktable
p = getcallbacktable()[target]
except KeyError:
if not event.nolog: logging.debug("can't find plugin to reload for %s" % event.cmnd)
return
if not event.nolog: logging.debug("found %s" % str(p))
for name in p:
if name in bot.plugs: done.append(name) ; continue
if isdisabled(name):
logging.info("%s - %s is disabled" % (bot.cfg.name, name))
continue
elif bot.cfg.loadlist and name not in bot.cfg.loadlist:
logging.info("%s - %s is not in loadlist" % (bot.cfg.name, name))
continue
if not event.nolog: logging.debug("%s - on demand reloading of %s" % (bot.cfg.name, name))
try:
mod = bot.plugs.reload(name, force=True, showerror=True)
if mod: plugloaded.append(mod) ; continue
except Exception as ex: handle_exception(event)
if done and not event.nolog: logging.debug("%s - %s is already loaded" % (bot.cfg.name, str(done)))
return plugloaded
