Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2025-02-19 07:00:57 +01:00
Code Issues Projects Releases Wiki Activity

Made lookups persistent with the registry.

Browse Source
This commit is contained in:
Jeremy Fincher 2004-02-04 07:11:47 +00:00
parent 03c6efcb7f
commit be31cac273
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
plugins/Lookup.py
View File

@ -47,6 +47,7 @@ import string
import conf import conf
import utils import utils
import privmsgs import privmsgs
import registry
import callbacks import callbacks
try: try:
@ -58,6 +59,7 @@ except ImportError:
def configure(advanced): def configure(advanced):
from questions import expect, anything, something, yn from questions import expect, anything, something, yn
conf.registerPlugin('Lookup', True) conf.registerPlugin('Lookup', True)
lookups = conf.supybot.plugins.Lookup.lookups
output("""This module allows you to define commands that do a simple key output("""This module allows you to define commands that do a simple key
lookup and return some simple value. It has a command "add" lookup and return some simple value. It has a command "add"
that takes a command name and a file from the data dir and adds a that takes a command name and a file from the data dir and adds a
@ -85,6 +87,7 @@ def configure(advanced):
'line #%s is malformed.' % counter) 'line #%s is malformed.' % counter)
continue continue
command = something('What would you like the command to be?') command = something('What would you like the command to be?')
conf.registerGlobalValue(lookups,command, registry.String(filename,''))
onStart.append('lookup add %s %s' % (command, filename)) onStart.append('lookup add %s %s' % (command, filename))
@ -101,6 +104,19 @@ class Lookup(callbacks.Privmsg):
self.lookupDomains = sets.Set() self.lookupDomains = sets.Set()
dataDir = conf.supybot.directories.data() dataDir = conf.supybot.directories.data()
self.dbHandler = LookupDB(name=os.path.join(dataDir, 'Lookup')) self.dbHandler = LookupDB(name=os.path.join(dataDir, 'Lookup'))
for (name, value) in registry._cache.iteritems():
name = name.lower()
if name.startswith('supybot.plugins.lookup.lookups.'):
name = name[len('supybot.plugins.lookup.lookups.'):]
if '.' in name:
continue
self.addRegistryValue(name, value())
group = conf.supybot.plugins.Lookup.lookups
for (name, value) in group.getValues(fullNames=False):
name = name.lower() # Just in case.
filename = value()
self.addDatabase(name, filename)
self.addCommand(name)
def _shrink(self, s): def _shrink(self, s):
return utils.ellipsisify(s, 50) return utils.ellipsisify(s, 50)
@ -150,7 +166,6 @@ class Lookup(callbacks.Privmsg):
try: try:
cursor.execute("""SELECT * FROM %s LIMIT 1""" % name) cursor.execute("""SELECT * FROM %s LIMIT 1""" % name)
self.addCommand(name) self.addCommand(name)
irc.replySuccess()
except sqlite.DatabaseError: except sqlite.DatabaseError:
try: try:
self.addDatabase(name, filename) self.addDatabase(name, filename)
@ -158,9 +173,14 @@ class Lookup(callbacks.Privmsg):
irc.error('Could not open %s: %s' % (filename, e.args[1])) irc.error('Could not open %s: %s' % (filename, e.args[1]))
return return
self.addCommand(name) self.addCommand(name)
irc.replySuccess('(lookup %s added)' % name) self.addRegistryValue(name, filename)
irc.replySuccess('Lookup %s added.' % name)
add = privmsgs.checkCapability(add, 'admin') add = privmsgs.checkCapability(add, 'admin')
def addRegistryValue(self, name, filename):
v = registry.String(filename, '')
conf.supybot.plugins.Lookup.lookups.register(name, v)
def addDatabase(self, name, filename): def addDatabase(self, name, filename):
db = self.dbHandler.getDb() db = self.dbHandler.getDb()
cursor = db.cursor() cursor = db.cursor()