Added lasturl command.

This commit is contained in:
Jeremy Fincher 2003-08-19 09:10:41 +00:00
parent 8b093f3b2b
commit dcacf9c7d5

View File

@ -39,6 +39,7 @@ from baseplugin import *
import re
import time
import getopt
import urlparse
import sqlite
@ -116,13 +117,13 @@ class URLSnarfer(callbacks.Privmsg, ChannelDBHandler):
Returns a random URL from the URL database. <channel> is only required
if the message isn't sent in the channel itself.
"""
channel = msg.args[0]
channel = privmsgs.getChannel(msg, args)
db = self.getDb(channel)
cursor = db.cursor()
cursor.execute("""SELECT * FROM urls ORDER BY random() LIMIT 1""")
(id, url, added, addedBy, _, _, _, _, _, _) = cursor.fetchone()
when = time.ctime(int(added))
s = '%s: %s (added by %s on %s)' % (id, url, addedBy, when)
s = '%s: <%s> (added by %s on %s)' % (id, url, addedBy, when)
irc.reply(msg, s)
def numurls(self, irc, msg, args):
@ -131,13 +132,61 @@ class URLSnarfer(callbacks.Privmsg, ChannelDBHandler):
Returns the number of URLs in the URL database. <channel> is only
required if the message isn't sent in the channel itself.
"""
channel = msg.args[0]
channel = privmsgs.getChannel(msg, args)
db = self.getDb(channel)
cursor = db.cursor()
cursor.execute("""SELECT COUNT(*) FROM urls""")
(count,) = cursor.fetchone()
irc.reply(msg, 'I have %s %s in my database.' % \
(count, int(count) == 1 and 'URL' or 'URLs'))
def lasturl(self, irc, msg, args):
"""[<channel>] [--{from,with,at,proto,near}=<value>]
Gives the last URL matching the given criteria. --from is from whom
the URL came; --at is the site of the URL; --proto is the protocol the
URL used; --with is something inside the URL; --near is a string in the
messages before and after the link. <channel> is only necessary if the
message isn't sent in the channel itself.
"""
channel = privmsgs.getChannel(msg, args)
(optlist, rest) = getopt.getopt(args, '', ['from=', 'with=', 'at=',
'proto=', 'near='])
criteria = ['1=1']
formats = []
for (option, argument) in optlist:
option = option[2:] # Strip off the --.
if option == 'from':
criteria.append('added_by LIKE %s')
formats.append(argument)
elif option == 'with':
criteria.append('url LIKE %s')
formats.append(argument)
elif option == 'at':
criteria.append('site LIKE %s')
formats.append(argument)
elif option == 'proto':
criteria.append('protocol=%s')
formats.append(argument)
elif option == 'near':
criteria.append("""(previous_msg LIKE %s OR
next_msg LIKE %s OR
current_msg LIKE %s)""")
if '%' not in argument:
argument = '%%%s%%' % argument
formats.append(argument)
formats.append(argument)
formats.append(argument)
db = self.getDb(channel)
cursor = db.cursor()
criterion = ' AND '.join(criteria)
sql = 'SELECT url FROM urls WHERE %s ORDER BY id DESC' % criterion
cursor.execute(sql, *formats)
if cursor.rowcount == 0:
irc.reply(msg, 'No URLs matched that criteria.')
else:
(url,) = cursor.fetchone()
irc.reply(msg, url)
Class = URLSnarfer