Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-12-03 16:39:25 +01:00
Code Issues Projects Releases Wiki Activity

Added code to allow removal of all factoids associated with a given key.

Browse Source
This commit is contained in:
Jeremy Fincher 2003-09-04 12:14:12 +00:00
parent dab57f3080
commit 062231bb4b
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
plugins/Factoids.py
View File

@ -225,16 +225,20 @@ class Factoids(ChannelDBHandler, callbacks.Privmsg):
irc.error(msg, conf.replyNoCapability % capability) irc.error(msg, conf.replyNoCapability % capability)
def unlearn(self, irc, msg, args): def unlearn(self, irc, msg, args):
"""[<channel>] <key> [<number>] """[<channel>] <key> [<number>|*]
Removes the factoid <key> from the factoids database. If there are Removes the factoid <key> from the factoids database. If there are
more than one factoid with such a key, a number is necessary to more than one factoid with such a key, a number is necessary to
determine which one should be removed. <channel> is only necessary if determine which one should be removed. A * can be used to remove all
factoids associated with a key. <channel> is only necessary if
the message isn't sent in the channel itself. the message isn't sent in the channel itself.
""" """
channel = privmsgs.getChannel(msg, args) channel = privmsgs.getChannel(msg, args)
if args[-1].isdigit(): if args[-1].isdigit():
number = int(args.pop()) number = int(args.pop())
elif args[-1] == '*':
del args[-1]
number = True
else: else:
number = None number = None
key = privmsgs.getArgs(args) key = privmsgs.getArgs(args)
@ -248,7 +252,7 @@ class Factoids(ChannelDBHandler, callbacks.Privmsg):
factoids.key_id=keys.id""", key) factoids.key_id=keys.id""", key)
if cursor.rowcount == 0: if cursor.rowcount == 0:
irc.error(msg, 'There is no such factoid.') irc.error(msg, 'There is no such factoid.')
elif cursor.rowcount == 1: elif cursor.rowcount == 1 or number is True:
(id, _) = cursor.fetchone() (id, _) = cursor.fetchone()
cursor.execute("""DELETE FROM factoids WHERE key_id=%s""", id) cursor.execute("""DELETE FROM factoids WHERE key_id=%s""", id)
cursor.execute("""DELETE FROM keys WHERE key LIKE %s""", key) cursor.execute("""DELETE FROM keys WHERE key LIKE %s""", key)
@ -267,7 +271,8 @@ class Factoids(ChannelDBHandler, callbacks.Privmsg):
irc.reply(msg, conf.replySuccess) irc.reply(msg, conf.replySuccess)
else: else:
irc.error(msg, '%s factoids have that key. ' \ irc.error(msg, '%s factoids have that key. ' \
'Please specify which one to remove.' % \ 'Please specify which one to remove, ' \
'or use * to designate all of them.' % \
cursor.rowcount) cursor.rowcount)
else: else:
irc.error(msg, conf.replyNoCapability % capability) irc.error(msg, conf.replyNoCapability % capability)