85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import collections
|
|
import glob
|
|
import os
|
|
import sys
|
|
import traceback
|
|
|
|
if 'mtimes' not in globals():
|
|
mtimes = {}
|
|
|
|
if 'lastfiles' not in globals():
|
|
lastfiles = set()
|
|
|
|
|
|
def format_plug(plug, lpad=0, width=40):
|
|
out = ' ' * lpad + '%s:%s:%s' % (plug[0])
|
|
if len(plug) == 3 and 'hook' in plug[2]:
|
|
out += '%s%s' % (' ' * (width - len(out)), plug[2]['hook'])
|
|
return out
|
|
|
|
|
|
def reload(init=False):
|
|
if init:
|
|
bot.plugs = collections.defaultdict(lambda: [])
|
|
|
|
for filename in glob.glob(os.path.join("core", "*.py")):
|
|
mtime = os.stat(filename).st_mtime
|
|
if mtime != mtimes.get(filename):
|
|
mtimes[filename] = mtime
|
|
try:
|
|
eval(compile(open(filename, 'U').read(), filename, 'exec'),
|
|
globals())
|
|
except Exception:
|
|
traceback.print_exc(Exception)
|
|
if init: # stop if there's a syntax error in a core
|
|
sys.exit() # script on startup
|
|
continue
|
|
|
|
if filename == os.path.join('core', 'reload.py'):
|
|
reload(init=init)
|
|
return
|
|
|
|
fileset = set(glob.glob(os.path.join('plugins', '*py')))
|
|
for name, data in bot.plugs.iteritems(): # remove deleted/moved plugins
|
|
bot.plugs[name] = filter(lambda x: x[0][0] in fileset, data)
|
|
|
|
for filename in fileset:
|
|
mtime = os.stat(filename).st_mtime
|
|
if mtime != mtimes.get(filename):
|
|
mtimes[filename] = mtime
|
|
try:
|
|
code = compile(open(filename, 'U').read(), filename, 'exec')
|
|
namespace = {}
|
|
eval(code, namespace)
|
|
except Exception:
|
|
traceback.print_exc(Exception)
|
|
continue
|
|
|
|
# remove plugins already loaded from this filename
|
|
for name, data in bot.plugs.iteritems():
|
|
|
|
if name == 'tee': # signal tee trampolines to stop
|
|
for csig, func, args in data:
|
|
if csig[0] == filename:
|
|
func._iqueue.put(StopIteration)
|
|
|
|
bot.plugs[name] = filter(lambda x: x[0][0] != filename, data)
|
|
|
|
for obj in namespace.itervalues():
|
|
if hasattr(obj, '_skybot_hook'): # check for magic
|
|
for type, data in obj._skybot_hook:
|
|
bot.plugs[type] += [data]
|
|
|
|
if not init:
|
|
print '### new plugin (type: %s) loaded:' % \
|
|
type, format_plug(data)
|
|
|
|
if init:
|
|
print ' plugin listing:'
|
|
for type, plugs in sorted(bot.plugs.iteritems()):
|
|
print ' %s:' % type
|
|
for plug in plugs:
|
|
out = ' %s:%s:%s' % (plug[0])
|
|
print format_plug(plug, lpad=6)
|
|
print
|