mirror of https://github.com/nealey/firebot
Remove backup file
This commit is contained in:
Neale Pickett
parent
aa75a965e5
commit
a2f454e4f5
451
#firebot.py#
451
#firebot.py#
|
|
@ -1,451 +0,0 @@
|
|||
#! /usr/bin/env python
|
||||
|
||||
import os
|
||||
import sys
|
||||
import re
|
||||
import random
|
||||
from webretriever import WebRetriever
|
||||
import asynchat, asyncore
|
||||
import socket
|
||||
import csv
|
||||
import adns
|
||||
import time
|
||||
|
||||
import procbot
|
||||
import shorturl
|
||||
import infobot
|
||||
|
||||
Runner = procbot.Runner
|
||||
esc = procbot.esc
|
||||
|
||||
URLSERVER = ("", 0)
|
||||
|
||||
class SSLSock:
|
||||
def __init__(self, sock):
|
||||
self.sock = sock
|
||||
self.sock.setblocking(1)
|
||||
self.ssl = socket.ssl(sock)
|
||||
|
||||
def send(self, data):
|
||||
self.ssl.write(data)
|
||||
|
||||
def recv(self, bufsize):
|
||||
return self.ssl.read(bufsize)
|
||||
|
||||
def close(self):
|
||||
self.sock.close()
|
||||
|
||||
class FireBot(infobot.InfoBot, procbot.ProcBot):
|
||||
#debug = True
|
||||
|
||||
bindings = []
|
||||
msg_cat = {}
|
||||
heartbeat_interval = 0.5
|
||||
ping_interval = 120
|
||||
|
||||
def __init__(self, host, nicks, gecos, channels, dbname='info.db', ssl=False, **kwargs):
|
||||
infobot.InfoBot.__init__(self, host, nicks, gecos, channels,
|
||||
**kwargs)
|
||||
self.ssl = ssl
|
||||
self.nosy = True
|
||||
self.seen = {}
|
||||
|
||||
def handle_connect(self):
|
||||
if self.ssl:
|
||||
self.plain_sock = self.socket
|
||||
self.socket = SSLSock(self.socket)
|
||||
infobot.InfoBot.handle_connect(self)
|
||||
|
||||
def send_ping(self):
|
||||
# Send keepalives to the server to see if we've lost
|
||||
# connection. For some reason, using SSL prevents us from
|
||||
# getting a RST.
|
||||
self.write('PING %f' % time.time())
|
||||
self.add_timer(self.ping_interval,
|
||||
self.send_ping)
|
||||
|
||||
def cmd_001(self, sender, forum, addl):
|
||||
infobot.InfoBot.cmd_001(self, sender, forum, addl)
|
||||
self.add_timer(self.ping_interval,
|
||||
self.send_ping)
|
||||
|
||||
def note(self, sender, forum, addl, match):
|
||||
whom = match.group('whom')
|
||||
what = match.group('what')
|
||||
when = time.time()
|
||||
note = "%f:%s:%s" % (when, sender.name(), what)
|
||||
n = self.getall(whom, special="note")
|
||||
n.append(note)
|
||||
self.set(whom, n, special="note")
|
||||
forum.msg(self.gettext('okay', sender=sender.name()))
|
||||
bindings.append((re.compile(r"^\008[:, ]+note (to )?(?P<whom>[^: ]+):? +(?P<what>.*)"),
|
||||
note))
|
||||
bindings.append((re.compile(r"^\008[:, ]+later tell (?P<whom>[^: ]+):? +(?P<what>.*)"),
|
||||
note))
|
||||
|
||||
def cmd_privmsg(self, sender, forum, addl):
|
||||
infobot.InfoBot.cmd_privmsg(self, sender, forum, addl)
|
||||
|
||||
if forum.is_channel():
|
||||
who = sender.name()
|
||||
|
||||
# Update seen
|
||||
text = addl[0]
|
||||
now = time.time()
|
||||
self.seen[who] = (now, text)
|
||||
|
||||
# Deliver notes
|
||||
n = self.getall(who, special="note")
|
||||
if n:
|
||||
notes = ["Welcome back, %s. You have %d notes:" % (who, len(n))]
|
||||
for note in n:
|
||||
when, whom, what = note.split(':', 2)
|
||||
try:
|
||||
notes.append(u"%s: %s <%s> %s" % (who,
|
||||
time.ctime(float(when)),
|
||||
whom,
|
||||
what))
|
||||
except UnicodeDecodeError:
|
||||
notes.append(u"%s" % ((who,
|
||||
time.ctime(note[0]),
|
||||
note[1],
|
||||
note[2]),))
|
||||
self.despool(forum, notes)
|
||||
self.delete(who, special="note")
|
||||
|
||||
##
|
||||
## Firebot stuff
|
||||
##
|
||||
|
||||
def seen(self, sender, forum, addl, match):
|
||||
whom = match.group('whom')
|
||||
if whom == sender.name():
|
||||
forum.msg('Cute, %s.' % whom)
|
||||
return
|
||||
last = self.seen.get(whom)
|
||||
now = time.time()
|
||||
if last:
|
||||
when = now - last[0]
|
||||
units = 'seconds'
|
||||
if when > 120:
|
||||
when /= 60
|
||||
units = 'minutes'
|
||||
if when > 120:
|
||||
when /= 60
|
||||
units = 'hours'
|
||||
if when > 48:
|
||||
when /= 24
|
||||
units = 'days'
|
||||
forum.msg('I last saw %s %d %s ago, saying "%s"' %
|
||||
(whom, when, units, last[1]))
|
||||
else:
|
||||
forum.msg("I've never seen %s!" % (whom))
|
||||
bindings.append((re.compile(r"^seen +(?P<whom>.*)$"),
|
||||
seen))
|
||||
|
||||
def evalstr(self, sender, forum, addl, match):
|
||||
code = match.group('code')
|
||||
if code in (')', '-)'):
|
||||
return True
|
||||
try:
|
||||
ret = repr(eval(code, {"__builtins__": {}}, {}))
|
||||
if len(ret) > 400:
|
||||
ret = ret[:400] + '\026...\026'
|
||||
except:
|
||||
t, v, tb = sys.exc_info()
|
||||
forum.msg(self.gettext('eval', code=code, ret='\002%s\002: %s' % (t, v), sender=sender.name()))
|
||||
else:
|
||||
forum.msg(self.gettext('eval', code=code, ret=ret, sender=sender.name()))
|
||||
#bindings.append((re.compile(r"^\; *(?P<code>.+)$"), evalstr))
|
||||
#msg_cat['eval'] = ('%(code)s ==> %(ret)s',)
|
||||
|
||||
def shorturl(self, sender, forum, addl, match):
|
||||
url = match.group('url')
|
||||
print ('url', url)
|
||||
idx = shorturl.add(url)
|
||||
forum.msg('http://%s:%d/%d' % (URLSERVER[0], URLSERVER[1], idx))
|
||||
bindings.append((re.compile(r".*\b(?P<url>\b[a-z]+://[-a-z0-9_=!?#$@~%&*+/:;.,\w]+[-a-z0-9_=#$@~%&*+/\w])"),
|
||||
shorturl))
|
||||
|
||||
def cdecl(self, sender, forum, addl, match):
|
||||
jibberish = match.group('jibberish')
|
||||
o, i = os.popen2('/usr/bin/cdecl')
|
||||
o.write(jibberish + '\n')
|
||||
o.close()
|
||||
res = i.read().strip()
|
||||
if '\n' in res:
|
||||
forum.msg("Lots of output, sending in private message")
|
||||
self.despool(sender, res.split('\n'))
|
||||
else:
|
||||
forum.msg('cdecl | %s' % res)
|
||||
bindings.append((re.compile(r"^cdecl (?P<jibberish>.*)$"),
|
||||
cdecl))
|
||||
|
||||
def delayed_say(self, sender, forum, addl, match):
|
||||
delay = int(match.group('delay'))
|
||||
unit = match.group('unit')
|
||||
what = match.group('what')
|
||||
|
||||
if not unit or unit[0] == 's':
|
||||
pass
|
||||
elif unit[0] == 'm':
|
||||
delay *= 60
|
||||
elif unit[0] == 'h':
|
||||
delay *= 3600
|
||||
elif unit[0] == 'd':
|
||||
delay *= 86400
|
||||
elif unit[0] == 'w':
|
||||
delay *= 604800
|
||||
else:
|
||||
forum.msg("I don't know what a %s is." % unit)
|
||||
return
|
||||
|
||||
self.add_timer(delay, lambda : forum.msg(what))
|
||||
forum.msg(self.gettext('okay', sender=sender.name()))
|
||||
bindings.append((re.compile(r"^\008[:, ]+in (?P<delay>[0-9]+) ?(?P<unit>[a-z]*) say (?P<what>.*)"),
|
||||
delayed_say))
|
||||
|
||||
msg_cat['nodict'] = ("Sorry, boss, dict returns no lines for %(jibberish)s",)
|
||||
def dict(self, sender, forum, addl, match):
|
||||
jibberish = match.group('jibberish')
|
||||
i = os.popen('/usr/bin/dict %s 2>&1' % esc(jibberish))
|
||||
res = i.readlines()
|
||||
if not res:
|
||||
forum.msg(self.gettext('nodict', jibberish=jibberish))
|
||||
return
|
||||
res = [l.strip() for l in res]
|
||||
if match.group('long'):
|
||||
self.despool(sender, res)
|
||||
else:
|
||||
if len(res) <= 5:
|
||||
self.despool(forum, res)
|
||||
else:
|
||||
del res[:4]
|
||||
short = res[:]
|
||||
while short and ((not short[0]) or (short[0][0] not in '0123456789')):
|
||||
del short[0]
|
||||
if not short:
|
||||
short = res
|
||||
short = ['%s: %s' % (jibberish, r) for r in short[:4]]
|
||||
self.despool(forum, short + ['[truncated: use the --long option to see it all]'])
|
||||
bindings.append((re.compile(r"^dict (?P<long>--?l(ong)? +)?(?P<jibberish>.*)$"),
|
||||
dict))
|
||||
|
||||
def units(self, sender, forum, addl, match):
|
||||
f = match.group('from')
|
||||
t = match.group('to')
|
||||
if f.startswith('a '):
|
||||
f = '1 ' + f[2:]
|
||||
Runner('/usr/bin/units -v %s %s' % (esc(f), esc(t)),
|
||||
lambda l,r: self.proc_cb(None, sender, forum, l, r))
|
||||
bindings.append((re.compile(r"^units +(?P<from>.*) +in +(?P<to>.*)$"),
|
||||
units))
|
||||
bindings.append((re.compile(r"^how many (?P<to>.*) in (?P<from>[^?]*)[?.!]*$"),
|
||||
units))
|
||||
|
||||
def calc(self, sender, forum, addl, match):
|
||||
e = match.group('expr')
|
||||
Runner("echo %s | /usr/bin/bc -l" % procbot.esc(e),
|
||||
lambda l,r: self.proc_cb('%s = ' % e, sender, forum, l, r))
|
||||
bindings.append((re.compile(r"^(?P<expr>[0-9.]+\s*[-+*/^%]\s*[0-9.]+)$"),
|
||||
calc))
|
||||
bindings.append((re.compile(r"^calc (?P<expr>.+)$"),
|
||||
calc))
|
||||
|
||||
def generic_cmd(self, sender, forum, addl, match):
|
||||
cmd = match.group('cmd')
|
||||
args = match.group('args').split(' ')
|
||||
argstr = ' '.join(procbot.lesc(args))
|
||||
Runner('%s %s' % (cmd, argstr),
|
||||
lambda l,r: self.proc_cb(None, sender, forum, l, r))
|
||||
bindings.append((re.compile(r"^(?P<cmd>host) (?P<args>.+)$"),
|
||||
generic_cmd))
|
||||
bindings.append((re.compile(r"^(?P<cmd>whois) (?P<args>.+)$"),
|
||||
generic_cmd))
|
||||
|
||||
def pollen(self, sender, forum, addl, match):
|
||||
forecast_re = re.compile('fimages/std/(?P<count>[0-9]+\.[0-9])\.gif')
|
||||
predom_re = re.compile('Predominant pollen: (?P<pollens>[^<]*)')
|
||||
zip = match.group('zip')
|
||||
def cb(lines):
|
||||
forecast = []
|
||||
predom = ''
|
||||
for line in lines:
|
||||
match = forecast_re.search(line)
|
||||
if match:
|
||||
forecast.append(match.group('count'))
|
||||
match = predom_re.search(line)
|
||||
if match:
|
||||
predom = match.group('pollens')
|
||||
forum.msg('%s: 4-day forecast (out of 12.0): %s; predominant pollen: %s' %
|
||||
(zip, ', '.join(forecast), predom))
|
||||
WebRetriever('http://www.pollen.com/forecast.asp?PostalCode=%s&Logon=Enter' % zip,
|
||||
cb)
|
||||
bindings.append((re.compile('pollen (?P<zip>[0-9]{5})'),
|
||||
pollen))
|
||||
|
||||
|
||||
bindings.append((re.compile('weather (?P<zip>[0-9]{5})'),
|
||||
weather))
|
||||
|
||||
def quote(self, sender, forum, addl, match):
|
||||
def cb(lines):
|
||||
if not lines:
|
||||
forum.msg('oops, no data from server')
|
||||
return
|
||||
c = csv.reader([lines[0].strip()])
|
||||
vals = zip(('symbol', 'value', 'day', 'time', 'change',
|
||||
'open', 'high', 'low', 'volume',
|
||||
'market cap', 'previous close',
|
||||
'percent change', 'open2', 'range',
|
||||
'eps', 'pe_ratio', 'name'),
|
||||
c.next())
|
||||
d = dict(vals)
|
||||
forum.msg(('%(name)s (%(symbol)s)'
|
||||
' last:%(value)s@%(time)s'
|
||||
' vol:%(volume)s'
|
||||
' cap:%(market cap)s'
|
||||
' prev-close:%(previous close)s'
|
||||
' chg:%(change)s(%(percent change)s)'
|
||||
' open:%(open)s'
|
||||
' 1d:%(low)s - %(high)s'
|
||||
' 52wk:%(range)s') %
|
||||
d)
|
||||
|
||||
symbol = match.group('symbol')
|
||||
WebRetriever('http://quote.yahoo.com/d/quotes.csv?s=%s&f=sl1d1t1c1ohgvj1pp2owern&e=.csv' % symbol,
|
||||
cb)
|
||||
bindings.append((re.compile(r"^quote +(?P<symbol>[.a-zA-Z]+)$"),
|
||||
quote))
|
||||
|
||||
def currency(self, sender, forum, addl, match):
|
||||
amt = float(match.group('amt'))
|
||||
frm = match.group('from')
|
||||
to = match.group('to')
|
||||
|
||||
def cb(lines):
|
||||
if not lines:
|
||||
forum.msg('oops, no data from server')
|
||||
return
|
||||
c = csv.reader([lines[0].strip()])
|
||||
vals = zip(('symbol', 'value', 'day', 'time', 'change',
|
||||
'open', 'high', 'low', 'volume',
|
||||
'market cap', 'previous close',
|
||||
'percent change', 'open2', 'range',
|
||||
'eps', 'pe_ratio', 'name'),
|
||||
c.next())
|
||||
d = dict(vals)
|
||||
v = float(d['value'])
|
||||
ans = v * amt
|
||||
forum.msg(('%0.4f %s = %0.4f %s') %
|
||||
(amt, frm, ans, to))
|
||||
|
||||
WebRetriever(('http://quote.yahoo.com/d/quotes.csv?s=%s%s%%3DX&f=sl1d1t1c1ohgvj1pp2owern&e=.csv' %
|
||||
(frm, to)),
|
||||
cb)
|
||||
bindings.append((re.compile(r"^how much is (?P<amt>[0-9.]+) ?(?P<from>[A-Z]{3}) in (?P<to>[A-Z]{3})\??$"),
|
||||
currency))
|
||||
|
||||
def whuffie_mod(self, nick, amt):
|
||||
vs = self.get(nick, "0", special="whuffie")
|
||||
try:
|
||||
val = int(vs)
|
||||
except:
|
||||
val = 0
|
||||
val += amt
|
||||
self.set(nick, [str(val)], special="whuffie")
|
||||
|
||||
def whuffie_modify(self, sender, forum, addl, match):
|
||||
nick = match.group('nick')
|
||||
if nick.lower() == sender.name().lower():
|
||||
forum.msg(self.gettext('whuffie whore', sender=sender.name()))
|
||||
return
|
||||
if match.group('mod') == '++':
|
||||
amt = 1
|
||||
else:
|
||||
amt = -1
|
||||
self.whuffie_mod(nick, amt)
|
||||
bindings.append((re.compile(r"^(?P<nick>\w+)(?P<mod>\+\+|\-\-)[? ]*$"),
|
||||
whuffie_modify))
|
||||
msg_cat['whuffie whore'] = ("Nothing happens.",
|
||||
'A hollow voice says, "Fool."')
|
||||
|
||||
def whuffie(self, sender, forum, addl, match):
|
||||
nick = match.group('nick')
|
||||
val = self.get(nick, special="whuffie")
|
||||
if val and val != "0":
|
||||
forum.msg("%s has whuffie of %s" % (nick, val))
|
||||
else:
|
||||
forum.msg("%s has neutral whuffie" % nick)
|
||||
bindings.append((re.compile(r"^(\008[,:] +)?([Ww]huffie|[Kk]arma) (for )?(?P<nick>\w+)[? ]*$"),
|
||||
whuffie))
|
||||
|
||||
#
|
||||
# This is all stuff that should just be stored in the usual manner.
|
||||
# But I wrote it here before I realized how programmable an Infobot
|
||||
# really is, so here it stays.
|
||||
#
|
||||
|
||||
msg_cat['8ball'] = ("%(sender)s: Outlook good.",
|
||||
"%(sender)s: Outlook not so good.",
|
||||
"%(sender)s: My reply is no.",
|
||||
"%(sender)s: Don't count on it.",
|
||||
"%(sender)s: You may rely on it.",
|
||||
"%(sender)s: Ask again later.",
|
||||
"%(sender)s: Most likely.",
|
||||
"%(sender)s: Cannot predict now.",
|
||||
"%(sender)s: Yes.",
|
||||
"%(sender)s: Yes, definitely.",
|
||||
"%(sender)s: Better not tell you now.",
|
||||
"%(sender)s: It is certain.",
|
||||
"%(sender)s: Very doubtful.",
|
||||
"%(sender)s: It is decidedly so.",
|
||||
"%(sender)s: Concentrate and ask again.",
|
||||
"%(sender)s: Signs point to yes.",
|
||||
"%(sender)s: My sources say no.",
|
||||
"%(sender)s: Without a doubt.",
|
||||
"%(sender)s: Reply hazy, try again.",
|
||||
"%(sender)s: As I see it, yes.")
|
||||
msg_cat['me'] = ('%(sender)s?',
|
||||
'%(sender)s: Yes?',
|
||||
'At your service, %(sender)s.',
|
||||
'May I help you, %(sender)s?')
|
||||
msg_cat['thanks'] = ('It is my pleasure, %(sender)s.',
|
||||
'Of course, %(sender)s.',
|
||||
'I live but to serve, %(sender)s.',
|
||||
"All in a day's work, %(sender)s.")
|
||||
bindings.append((re.compile(r"^(magic )?(8|eight ?)-?ball", re.IGNORECASE),
|
||||
'8ball'))
|
||||
bindings.append((re.compile(r"^\008\?$", re.IGNORECASE),
|
||||
'me'))
|
||||
bindings.append((re.compile(r"^thank(s| you),? *\008", re.IGNORECASE),
|
||||
'thanks'))
|
||||
|
||||
msg_cat.update(infobot.InfoBot.msg_cat)
|
||||
bindings.extend(infobot.InfoBot.bindings)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import irc
|
||||
|
||||
# Short URL server
|
||||
us = shorturl.start(('', 0))
|
||||
URLSERVER = (socket.gethostbyaddr(socket.gethostname())[0],
|
||||
us.getsockname()[1])
|
||||
|
||||
|
||||
NICK = ['hal']
|
||||
INFO = 'Daisy, Daisy...'
|
||||
|
||||
l1 = FireBot(("server1", 6667),
|
||||
NICK,
|
||||
INFO,
|
||||
["#ch1", "#ch2"])
|
||||
l2 = FireBot(('server2', 6667),
|
||||
NICK,
|
||||
INFO,
|
||||
["#ch3"])
|
||||
l1.set_others([l2])
|
||||
l2.set_others([l1])
|
||||
|
||||
irc.run_forever(0.5)
|
||||
Loading…
Reference in New Issue