442 lines
19 KiB
Python
442 lines
19 KiB
Python
###
|
|
# Copyright (c) 2006, Dennis Kaarsemaker
|
|
# All rights reserved.
|
|
#
|
|
#
|
|
###
|
|
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
import supybot.plugins as plugins
|
|
import supybot.ircutils as ircutils
|
|
import supybot.ircmsgs as ircmsgs
|
|
import supybot.callbacks as callbacks
|
|
import sqlite, datetime, time, apt_pkg, commands
|
|
import supybot.registry as registry
|
|
import supybot.ircdb as ircdb
|
|
from email import FeedParser
|
|
import re
|
|
import os
|
|
apt_pkg.init()
|
|
|
|
fallback = ('ubuntu', '#ubuntu')
|
|
datadir = '/home/dennis/ubugtu/data/facts'
|
|
|
|
def r(section):
|
|
if '/' in section:
|
|
return section[:section.find('/')]
|
|
return 'main'
|
|
|
|
class Factoid:
|
|
def __init__(self, name, value, author, added, popularity):
|
|
self.name = name; self.value = value
|
|
self.author = author; self.added = added
|
|
self.popularity = popularity
|
|
|
|
def get_factoid(db, name, channel):
|
|
cur = db.cursor()
|
|
cur.execute("SELECT name, value, author, added, popularity FROM facts WHERE name = %s", '%s-%s' % (name, channel))
|
|
factoid = cur.fetchall()
|
|
if len(factoid):
|
|
f = factoid[0]
|
|
return Factoid(f[0],f[1],f[2],f[3],f[4])
|
|
cur.execute("SELECT name, value, author, added, popularity FROM facts WHERE name = %s", name)
|
|
factoid = cur.fetchall()
|
|
if len(factoid):
|
|
f = factoid[0]
|
|
return Factoid(f[0],f[1],f[2],f[3],f[4])
|
|
return None
|
|
|
|
def resolve_alias(db,factoid,channel,loop=0):
|
|
if loop >= 10:
|
|
return Factoid('','Error: infinite <alias> loop detected','','',0)
|
|
if factoid.value.lower().startswith('<alias>'):
|
|
new_factoid = get_factoid(db,factoid.value[7:].lower().strip(),channel)
|
|
if not new_factoid:
|
|
return Factoid('','Error: unresolvable <alias>','','',0)
|
|
else:
|
|
return resolve_alias(db, new_factoid, channel, loop+1)
|
|
else:
|
|
return factoid
|
|
|
|
class Encyclopedia(callbacks.PluginRegexp):
|
|
"""!factoid: show factoid"""
|
|
threaded = True
|
|
regexps = ['showfactoid', 'addfactoid', 'deletefactoid','info','find','editfactoid','searchfactoid','seen']
|
|
|
|
def __init__(self, irc):
|
|
callbacks.PluginRegexp.__init__(self, irc)
|
|
self.databases = {}
|
|
self.times = {}
|
|
self.seens = {}
|
|
|
|
# Capability check
|
|
def _precheck(self, irc, msg, capability=None, timeout=None):
|
|
channel = msg.args[0].lower()
|
|
inchannel = channel.startswith('#')
|
|
excl = msg.args[1].startswith('!')
|
|
if inchannel and not excl:
|
|
return False
|
|
for c in irc.callbacks:
|
|
comm = msg.args[1].split()[0]
|
|
if c.isCommandMethod(comm) and not c.isDisabled(comm):
|
|
return False
|
|
if capability:
|
|
try:
|
|
_ = ircdb.users.getUser(msg.prefix)
|
|
except KeyError:
|
|
irc.queueMsg(ircmsgs.privmsg('#ubuntu-ops', "In %s, %s said: %s" % (msg.args[0], msg.nick, msg.args[1])))
|
|
irc.reply("Your edit request has been forwarded to #ubuntu-ops. Thank you for your attention to detail",private=True)
|
|
return False
|
|
if not ircdb.checkCapability(msg.prefix, capability):
|
|
irc.queueMsg(ircmsgs.privmsg('#ubuntu-ops', "In %s, %s said: %s" % (msg.args[0], msg.nick, msg.args[1])))
|
|
irc.reply("Your edit request has been forwarded to #ubuntu-ops. Thank you for your attention to detail",private=True)
|
|
return False
|
|
if timeout:
|
|
for key in self.times.keys():
|
|
if self.times[key] < time.time() - 15:
|
|
self.times.pop(key)
|
|
if timeout in self.times:
|
|
return False
|
|
self.times[timeout] = time.time()
|
|
db = self.registryValue('database',channel)
|
|
if not db:
|
|
db,channel = fallback
|
|
if channel not in self.databases:
|
|
self.databases[channel] = sqlite.connect(os.path.join(datadir, '%s.db' % db))
|
|
return self.databases[channel]
|
|
|
|
def searchfactoid(self, irc, msg, match):
|
|
r"^!?search\s+(?P<query>.+)"
|
|
db = self._precheck(irc, msg, timeout=(msg.args[0],match.group('query')))
|
|
if not db: return
|
|
cur = db.cursor()
|
|
query = '%%%s%%' % match.group('query').replace('%','').replace('*','%')
|
|
try:
|
|
cur.execute("SELECT name FROM facts WHERE (value LIKE %s OR name LIKE %s ) AND value NOT LIKE '<alias>%%'", (query, query))
|
|
data = cur.fetchall()
|
|
all = [x[0] for x in data]
|
|
cur.execute("SELECT value FROM facts WHERE name LIKE %s AND value LIKE '<alias>%%'", query)
|
|
data = cur.fetchall()
|
|
all += [x[0][7:].strip() for x in data]
|
|
all = list(set(all))
|
|
|
|
if len(all) > 10:
|
|
irc.reply("Found: %s (and %d more)" % (', '.join(all[:10]), len(all)-10))
|
|
elif len(all):
|
|
irc.reply("Found: %s" % ', '.join(all))
|
|
else:
|
|
irc.reply("Found nothing")
|
|
except:
|
|
irc.error('An error occured (code 561)')
|
|
|
|
def showfactoid(self, irc, msg, match):
|
|
r"^(!|ubotu\S?\s)?(?P<noalias>-)?\s*(tell\s+(?P<nick>\S+)\s+about\s+)?(?P<factoid>\S.*?)(>\s*(?P<nick2>\S+))?$"
|
|
db = self._precheck(irc, msg, timeout=(msg.args[0], match.group('nick'), match.group('factoid'), match.group('nick2')))
|
|
if not db: return
|
|
to = channel = msg.args[0]
|
|
if channel[0] != '#':
|
|
to = msg.nick
|
|
cur = db.cursor()
|
|
retmsg = ''
|
|
|
|
noalias = match.group('noalias')
|
|
factoid = match.group('factoid').lower().strip()
|
|
if ' is ' in match.group(0) or \
|
|
'=~' in match.group(0) or \
|
|
'<sed>' in match.group(0) or \
|
|
factoid.startswith('forget ') or \
|
|
factoid.startswith('info ') or \
|
|
factoid.startswith('find ') or \
|
|
factoid.startswith('search ') or \
|
|
factoid.startswith('seen'):
|
|
return
|
|
|
|
#if channel.startswith('#'):
|
|
if True:
|
|
nick = match.group('nick')
|
|
if match.group('nick2'): nick = match.group('nick2')
|
|
if nick == 'me': nick = msg.nick
|
|
if nick:
|
|
for chan in irc.state.channels:
|
|
if nick in irc.state.channels[chan].users and\
|
|
msg.nick in irc.state.channels[chan].users:
|
|
retmsg = '%s wants you to know: ' % msg.nick
|
|
to = nick
|
|
break
|
|
else:
|
|
irc.error("That person could not be found in any channel you're in")
|
|
return
|
|
|
|
# Retrieve factoid
|
|
try:
|
|
factoid = get_factoid(db, factoid, channel)
|
|
if not factoid:
|
|
irc.reply('I know nothing about %s' % match.group('factoid'))
|
|
return
|
|
# Output factoid
|
|
if not noalias:
|
|
factoid = resolve_alias(db,factoid,channel)
|
|
else:
|
|
cur.execute("SELECT name FROM facts WHERE value = %s", '<alias> ' + factoid.name)
|
|
data = cur.fetchall()
|
|
if(len(data)):
|
|
irc.queueMsg(ircmsgs.privmsg(to, "%s aliases: %s" % (factoid.name, ', '.join([x[0].strip() for x in data]))))
|
|
return
|
|
# Do timing
|
|
if not self._precheck(irc, msg, timeout=(to,factoid.name)):
|
|
return
|
|
cur.execute("UPDATE FACTS SET popularity = %d WHERE name = %s", factoid.popularity+1, factoid.name)
|
|
db.commit()
|
|
if factoid.value.startswith('<reply>'):
|
|
irc.queueMsg(ircmsgs.privmsg(to, '%s%s' % (retmsg, factoid.value[7:].strip())))
|
|
else:
|
|
irc.queueMsg(ircmsgs.privmsg(to, '%s%s is %s' % (retmsg, factoid.name, factoid.value.strip())))
|
|
except:
|
|
raise
|
|
irc.error('An error occured (code 813)')
|
|
|
|
def addfactoid(self, irc, msg, match):
|
|
r"^!?(?P<no>no,?\s+)?(?P<factoid>\S.*?)\s+is\s+(?P<fact>\S.*)"
|
|
db = self._precheck(irc, msg, capability='editfactoids', timeout=(msg.args[0],match.group(0)))
|
|
if not db: return
|
|
channel = msg.args[0]
|
|
cur = db.cursor()
|
|
|
|
factoid = match.group('factoid').lower().strip()
|
|
fact = match.group('fact').strip()
|
|
if '<sed>' in match.group(0) or \
|
|
'=~' in match.group(0) or \
|
|
factoid.startswith('forget') or \
|
|
factoid.startswith('info') or \
|
|
factoid.startswith('find') or \
|
|
factoid.startswith('search'):
|
|
return
|
|
|
|
try:
|
|
# See if the alias exists and resolve it...
|
|
old_factoid = get_factoid(db, factoid, channel)
|
|
if old_factoid:
|
|
if not fact.startswith('<alias>'):
|
|
old_factoid = resolve_alias(db, old_factoid, channel)
|
|
# Unresolvable alias
|
|
if not old_factoid.name:
|
|
irc.reply(old_factoid.value)
|
|
return
|
|
if match.group('no'):
|
|
if fact.startswith('<alias>'):
|
|
cur.execute("SELECT COUNT(*) FROM facts WHERE value = %s", '<alias> ' + factoid)
|
|
num = cur.fetchall()[0][0]
|
|
if num:
|
|
irc.reply("Can't turn factoid with aliases into an alias")
|
|
return
|
|
alias_factoid = get_factoid(db, fact[7:].lower().strip(), channel)
|
|
if not alias_factoid:
|
|
alias_factoid = Factoid('','Error: unresolvable <alias>','','',0)
|
|
else:
|
|
alias_factoid = resolve_alias(db, alias_factoid, channel)
|
|
if not alias_factoid.name:
|
|
irc.reply(alias_factoid.value)
|
|
return
|
|
fact = '<alias> %s' % alias_factoid.name
|
|
fact = fact.lower()
|
|
cur.execute("""UPDATE facts SET value=%s, author=%s, added=%s WHERE name=%s""",
|
|
(fact, msg.prefix, str(datetime.datetime.now()), old_factoid.name))
|
|
db.commit()
|
|
irc.reply("I'll remember that")
|
|
else:
|
|
irc.reply('%s is already known...' % factoid)
|
|
else:
|
|
if fact.lower().startswith('<alias>'):
|
|
old_factoid = get_factoid(db, fact[7:].lower().strip(), channel)
|
|
if not old_factoid:
|
|
old_factoid = Factoid('','Error: unresolvable <alias>','','',0)
|
|
else:
|
|
old_factoid = resolve_alias(db, old_factoid, channel)
|
|
if not old_factoid.name:
|
|
irc.reply(old_factoid.value)
|
|
return
|
|
fact = '<alias> %s' % old_factoid.name
|
|
fact = fact.lower()
|
|
cur.execute("""INSERT INTO facts (name, value, author, added) VALUES
|
|
(%s, %s, %s, %s)""", (factoid, fact, msg.prefix, str(datetime.datetime.now())))
|
|
db.commit()
|
|
irc.reply("I'll remember that")
|
|
except:
|
|
irc.error('An error occured (code 735)')
|
|
|
|
def editfactoid(self, irc, msg, match):
|
|
r"^!?(?P<factoid>.*?)\s*(=~|(\s+is\s*)<sed>)\s*s?(?P<regex>.*)"
|
|
db = self._precheck(irc, msg, capability='editfactoids', timeout=(msg.args[0],match.group(0)))
|
|
if not db: return
|
|
channel = msg.args[0]
|
|
cur = db.cursor()
|
|
|
|
factoid = match.group('factoid').lower().strip()
|
|
regex = match.group('regex').strip()
|
|
if factoid.startswith('forget') or \
|
|
factoid.startswith('info') or \
|
|
factoid.startswith('find') or \
|
|
factoid.startswith('search'): return
|
|
# Store factoid if nonexistant or 'no' is given
|
|
try:
|
|
# See if the alias exists and resolve it...
|
|
factoid = get_factoid(db, factoid, channel)
|
|
if factoid:
|
|
factoid = resolve_alias(db, factoid, channel)
|
|
# Unresolvable alias
|
|
if not factoid.name:
|
|
irc.reply(old_factoid.value)
|
|
return
|
|
delim = regex[0]
|
|
if regex[-1] != delim:
|
|
irc.reply("Missing end delimiter")
|
|
return
|
|
data = regex.split(delim)[1:-1]
|
|
if len(data) != 2:
|
|
irc.reply("You used the delimiter too often. Maybe try another one?")
|
|
return
|
|
regex, change = data
|
|
if '<alias>' in change.lower():
|
|
irc.reply("Can't turn factoids into aliases this way")
|
|
return
|
|
try:
|
|
regex = re.compile(regex)
|
|
except:
|
|
irc.reply("Malformed regex")
|
|
return
|
|
newval = regex.sub(change, factoid.value, 1)
|
|
if newval != factoid.value:
|
|
cur.execute("""UPDATE facts SET value=%s, author=%s, added=%s WHERE name=%s""",
|
|
(newval, msg.prefix, str(datetime.datetime.now()), factoid.name))
|
|
db.commit()
|
|
irc.reply("I'll remember that")
|
|
else:
|
|
irc.reply("No changes, not saving")
|
|
else:
|
|
irc.reply('I know nothing about %s' % match.group('factoid'))
|
|
except:
|
|
irc.error('An error occured (code 735)')
|
|
|
|
def deletefactoid(self, irc, msg, match):
|
|
r"^!?forget\s+(?P<factoid>\S.*)"
|
|
db = self._precheck(irc, msg, capability='editfactoids', timeout=(msg.args[0],match.group('factoid')))
|
|
if not db: return
|
|
channel = msg.args[0]
|
|
cur = db.cursor()
|
|
try:
|
|
cur.execute("SELECT COUNT(*) FROM facts WHERE value = %s", '<alias> ' + match.group('factoid'))
|
|
num = cur.fetchall()[0][0]
|
|
if num:
|
|
irc.reply("Can't forget factoids with aliases")
|
|
else:
|
|
cur.execute("DELETE FROM facts WHERE name = %s", match.group('factoid'))
|
|
db.commit()
|
|
irc.reply("I've forgotten it")
|
|
except:
|
|
raise
|
|
irc.error('An error occured (code 124)')
|
|
|
|
|
|
aptcommand = """apt-cache\\
|
|
-o"Dir::State::Lists=/home/dennis/ubugtu/data/%s"\\
|
|
-o"Dir::etc::sourcelist=/home/dennis/ubugtu/data/%s.list"\\
|
|
-o"Dir::State::status=/home/dennis/ubugtu/data/%s.status"\\
|
|
-o"Dir::Cache=/home/dennis/ubugtu/data/cache"\\
|
|
%s %s"""
|
|
def info(self, irc, msg, match):
|
|
r"^!?info\s+(?P<package>\S+)(\s+(?P<distro>\S+))?"
|
|
if not self._precheck(irc, msg, timeout=(msg.args[0],match.group('package'), match.group('distro'))):
|
|
return
|
|
distro = 'dapper'
|
|
if (match.group('distro') in ('warty','hoary','breezy','dapper','edgy')):
|
|
distro = match.group('distro')
|
|
data = commands.getoutput(self.aptcommand % (distro, distro, distro, 'show', match.group('package')))
|
|
if not data or 'E: No packages found' in data:
|
|
irc.reply('Package %s does not exist in %s' % (match.group('package'), distro))
|
|
else:
|
|
maxp = {'Version': '0'}
|
|
packages = [x.strip() for x in data.split('\n\n')]
|
|
for p in packages:
|
|
if not p.strip():
|
|
continue
|
|
parser = FeedParser.FeedParser()
|
|
parser.feed(p)
|
|
p = parser.close()
|
|
if apt_pkg.VersionCompare(maxp['Version'], p['Version']) < 0:
|
|
maxp = p
|
|
del parser
|
|
irc.reply("%s: %s. In repository %s, is %s. Version %s (%s), package size %s kB, installed size %s kB" %
|
|
(maxp['Package'], maxp['Description'].split('\n')[0], r(maxp['Section']),
|
|
maxp['Priority'], maxp['Version'], distro, int(maxp['Size'])/1024, maxp['Installed-Size']))
|
|
|
|
def find(self, irc, msg, match):
|
|
r"^!?find\s+(?P<package>\S+)(\s+(?P<distro>\S+))?"
|
|
if not self._precheck(irc, msg, timeout=(msg.args[0],match.group('package'), match.group('distro'),2)):
|
|
return
|
|
distro = 'dapper'
|
|
if (match.group('distro') in ('warty','hoary','breezy','dapper','edgy')):
|
|
distro = match.group('distro')
|
|
data = commands.getoutput(self.aptcommand % (distro, distro, distro, 'search -n', match.group('package')))
|
|
if not data:
|
|
irc.reply("No packages matching '%s' could be found" % match.group('package'))
|
|
else:
|
|
pkgs = [x.split()[0] for x in data.split('\n')]
|
|
if len(pkgs) > 5:
|
|
irc.reply("Found: %s (and %d others)" % (', '.join(pkgs[:5]), len(pkgs) -5))
|
|
else:
|
|
irc.reply("Found: %s" % ', '.join(pkgs[:5]))
|
|
|
|
def seen(self, irc, msg, match):
|
|
r"^!?seen\s+(?P<nick>\S+)"
|
|
if not self._precheck(irc, msg, timeout=(msg.args[0],match.group('nick'))):
|
|
return
|
|
to = msg.args[0]
|
|
if msg.args[0][0] != '#':
|
|
to = msg.nick
|
|
self.seens[match.group('nick')] = (to, time.time())
|
|
irc.queueMsg(ircmsgs.privmsg('seenserv', "seen %s" % match.group('nick')))
|
|
|
|
def doNotice(self, irc, msg):
|
|
if msg.nick.lower() == 'seenserv':
|
|
resp = msg.args[1]
|
|
for n in self.seens.keys():
|
|
if self.seens[n][1] < time.time() - 10:
|
|
self.seens.pop(n)
|
|
for n in self.seens.keys():
|
|
if n.lower() in resp.lower():
|
|
irc.queueMsg(ircmsgs.privmsg(self.seens[n][0], resp))
|
|
self.seens.pop(n)
|
|
|
|
def addeditor(self, irc, msg, args, name):
|
|
self._precheck(irc, msg, capability='addeditors')
|
|
try:
|
|
u = ircdb.users.getUser(name)
|
|
except:
|
|
irc.error('User %s is not registered' % name)
|
|
else:
|
|
u.addCapability('editfactoids')
|
|
addeditor = wrap(addeditor, ['text'])
|
|
|
|
def editors(self, irc, msg, args):
|
|
irc.reply(', '.join([ircdb.users.getUser(u).name for u in ircdb.users.users \
|
|
if 'editfactoids' in ircdb.users.getUser(u).capabilities]))
|
|
editors = wrap(editors)
|
|
def moderators(self, irc, msg, args):
|
|
irc.reply(', '.join([ircdb.users.getUser(u).name for u in ircdb.users.users \
|
|
if 'addeditors' in ircdb.users.getUser(u).capabilities]))
|
|
moderators = wrap(moderators)
|
|
|
|
def removeeditor(self, irc, msg, args, name):
|
|
self._precheck(irc, msg, capability='addeditors')
|
|
try:
|
|
u = ircdb.users.getUser(name)
|
|
except:
|
|
irc.error('User %s is not registered' % name)
|
|
else:
|
|
u.removeCapability('editfactoids')
|
|
removeeditor = wrap(removeeditor, ['text'])
|
|
|
|
Class = Encyclopedia
|
|
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|