h/bot.py

138 lines
3.7 KiB
Python
Raw Normal View History

2008-06-09 11:59:56 +00:00
#!/usr/bin/python
2009-03-15 06:51:39 +00:00
network = "irc.synirc.net"
nick = "skybot"
channel = "#cobol"
2008-06-09 11:59:56 +00:00
2009-03-15 03:06:36 +00:00
import sys
import os
import glob
import imp
import re
import thread
import Queue
import collections
2008-06-09 11:59:56 +00:00
2009-03-15 03:06:36 +00:00
import irc
import yaml
2008-06-09 11:59:56 +00:00
2009-03-16 04:51:06 +00:00
os.chdir(sys.path[0]) # do stuff relative to the installation directory
sys.path += ['plugins'] # so 'import hook' works without duplication
2008-06-09 11:59:56 +00:00
class Bot(object):
2009-03-15 20:04:02 +00:00
def __init__(self, nick, channel, network):
self.nick = nick
self.channel = channel
self.network = network
2008-06-09 11:59:56 +00:00
2009-03-15 20:04:02 +00:00
bot = Bot(nick, channel, network)
2008-06-09 11:59:56 +00:00
print 'Loading plugins'
2009-03-15 06:51:39 +00:00
typs = '|'.join('command filter event'.split())
magic_re = re.compile(r'^\s*#!(%s)(?:: +(\S+) *(\S.*)?)?\s*$' % typs)
2009-03-15 04:14:07 +00:00
def reload_plugins(mtime=[0]):
new_mtime = os.stat('plugins')
if new_mtime == mtime[0]:
return
bot.plugs = collections.defaultdict(lambda: [])
2009-03-15 04:14:07 +00:00
for filename in glob.glob("plugins/*.py"):
shortname = os.path.splitext(os.path.basename(filename))[0]
try:
plugin = imp.load_source(shortname, filename)
for obj in vars(plugin).itervalues():
if hasattr(obj, '_skybot_hook'): #check for magic
for type, data in obj._skybot_hook:
bot.plugs[type] += [data]
2009-03-15 04:14:07 +00:00
except Exception, e:
print ' error:', e
2009-03-15 04:14:07 +00:00
mtime[0] = new_mtime
reload_plugins()
2008-06-09 11:59:56 +00:00
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 out,
if len(plug) == 3 and 'hook' in plug[2]:
print '%s%s' % (' ' * (35 - len(out)), plug[2]['hook'])
else:
print
print
2008-06-09 11:59:56 +00:00
print 'Connecting to IRC'
2009-03-15 06:51:39 +00:00
bot.irc = irc.irc(network, nick)
2008-06-09 11:59:56 +00:00
bot.irc.join(channel)
2009-03-15 06:51:39 +00:00
bot.commandprefix = '^(?:\.|'+nick+'[:,]*\s*)'
2008-06-09 11:59:56 +00:00
print 'Running main loop'
class Input(object):
def __init__(self, raw, prefix, command,
params, nick, user, host, paraml, msg):
self.raw = raw
self.prefix = prefix
self.command = command
self.params = params
self.nick = nick
self.user = user
self.host = host
self.paraml = paraml
self.msg = msg
class FakeBot(object):
def __init__(self, bot, input, func):
2008-06-09 11:59:56 +00:00
self.bot = bot
self.input = input
2009-03-15 03:06:36 +00:00
self.msg = bot.irc.msg
self.cmd = bot.irc.cmd
2009-03-15 06:51:39 +00:00
self.join = bot.irc.join
2009-03-15 03:06:36 +00:00
self.func = func
2009-03-15 06:51:39 +00:00
self.doreply = True
2009-03-15 03:06:36 +00:00
if input.command == "PRIVMSG":
self.chan = input.paraml[0]
2009-03-15 06:51:39 +00:00
2008-06-09 11:59:56 +00:00
def say(self, msg):
2009-03-15 04:14:07 +00:00
self.bot.irc.msg(self.input.paraml[0], msg)
2008-06-09 11:59:56 +00:00
def reply(self, msg):
2009-03-15 04:14:07 +00:00
self.say(self.input.nick + ': ' + msg)
2008-06-09 11:59:56 +00:00
2009-03-15 03:06:36 +00:00
def run(self):
ac = self.func.func_code.co_argcount
if ac == 2:
out = self.func(self, self.input)
elif ac == 1:
out = self.func(self.input.inp)
2009-03-15 03:06:36 +00:00
if out is not None:
2009-03-15 04:14:07 +00:00
if self.doreply:
self.reply(unicode(out))
else:
self.say(unicode(out))
2009-03-15 03:06:36 +00:00
2008-06-09 11:59:56 +00:00
while True:
try:
out = bot.irc.out.get(timeout=1)
2009-03-15 04:14:07 +00:00
reload_plugins()
for csig, func, args in (bot.plugs['command'] + bot.plugs['event']):
2008-06-09 11:59:56 +00:00
input = Input(*out)
for fsig, sieve in bot.plugs['sieve']:
2009-03-25 04:45:33 +00:00
try:
input = sieve(bot, input, func, args)
except Exception, e:
print 'filter error:', e
input = None
2009-03-15 03:06:36 +00:00
if input == None:
2008-06-09 11:59:56 +00:00
break
2009-03-15 03:06:36 +00:00
if input == None:
2009-03-15 04:14:07 +00:00
continue
2009-03-15 06:51:39 +00:00
print '<<<', input.raw
thread.start_new_thread(FakeBot(bot, input, func).run, ())
2008-06-09 11:59:56 +00:00
except Queue.Empty:
pass