wrap bot.py in a main() function to be import-safe
This commit is contained in:
parent
62fbe6e237
commit
921b986683
32
bot.py
32
bot.py
|
@ -6,10 +6,6 @@ import sys
|
|||
import traceback
|
||||
import time
|
||||
|
||||
sys.path += ['plugins'] # so 'import hook' works without duplication
|
||||
sys.path += ['lib']
|
||||
os.chdir(sys.path[0] or '.') # do stuff relative to the install directory
|
||||
|
||||
|
||||
class Bot(object):
|
||||
def __init__(self):
|
||||
|
@ -18,29 +14,34 @@ class Bot(object):
|
|||
if not os.path.exists(self.persist_dir):
|
||||
os.mkdir(self.persist_dir)
|
||||
|
||||
bot = Bot()
|
||||
def main():
|
||||
sys.path += ['plugins'] # so 'import hook' works without duplication
|
||||
sys.path += ['lib']
|
||||
os.chdir(sys.path[0] or '.') # do stuff relative to the install directory
|
||||
|
||||
print 'Loading plugins'
|
||||
bot = Bot()
|
||||
|
||||
# bootstrap the reloader
|
||||
eval(compile(open(os.path.join('core', 'reload.py'), 'U').read(),
|
||||
print 'Loading plugins'
|
||||
|
||||
# bootstrap the reloader
|
||||
eval(compile(open(os.path.join('core', 'reload.py'), 'U').read(),
|
||||
os.path.join('core', 'reload.py'), 'exec'))
|
||||
reload(init=True)
|
||||
reload(init=True)
|
||||
|
||||
print 'Connecting to IRC'
|
||||
print 'Connecting to IRC'
|
||||
|
||||
try:
|
||||
try:
|
||||
config()
|
||||
if not hasattr(bot, 'config'):
|
||||
exit()
|
||||
except Exception, e:
|
||||
except Exception, e:
|
||||
print 'ERROR: malformed config file:', e
|
||||
traceback.print_exc()
|
||||
sys.exit()
|
||||
|
||||
print 'Running main loop'
|
||||
print 'Running main loop'
|
||||
|
||||
while True:
|
||||
while True:
|
||||
reload() # these functions only do things
|
||||
config() # if changes have occured
|
||||
|
||||
|
@ -52,3 +53,6 @@ while True:
|
|||
pass
|
||||
while all(conn.out.empty() for conn in bot.conns.itervalues()):
|
||||
time.sleep(.1)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
Loading…
Reference in New Issue