61 lines
2.2 KiB
Python
Executable File
61 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
.. module:: RandQuote
|
|
:synopsis: Log a configurable number of messages and pull up random ones on command
|
|
|
|
.. moduleauthor:: Dave Pedu <dave@davepedu.com>
|
|
|
|
"""
|
|
|
|
from pyircbot.modulebase import ModuleBase, hook, command
|
|
from datetime import datetime
|
|
from pyircbot.modules.ModInfo import info
|
|
|
|
|
|
class RandQuote(ModuleBase):
|
|
def __init__(self, bot, moduleName):
|
|
ModuleBase.__init__(self, bot, moduleName)
|
|
self.db = None
|
|
serviceProviders = self.bot.getmodulesbyservice("sqlite")
|
|
if not serviceProviders:
|
|
self.log.error("RandQuote: Could not find a valid sqlite service provider")
|
|
else:
|
|
self.log.info("RandQuote: Selecting sqlite service provider: %s" % serviceProviders[0])
|
|
self.db = serviceProviders[0].opendb("randquote.db")
|
|
|
|
if not self.db.tableExists("chat"):
|
|
self.log.info("RandQuote: Creating table: chat")
|
|
c = self.db.query("""CREATE TABLE IF NOT EXISTS `chat` (
|
|
`id` INTEGER PRIMARY KEY,
|
|
`date` INTEGER,
|
|
`sender` varchar(64),
|
|
`message` varchar(2048)
|
|
) ;""")
|
|
c.close()
|
|
|
|
@info("randquote", "print a random quote", cmds=["randquote", "randomquote", "rq"])
|
|
@command("randquote", "randomquote", "rq")
|
|
def fetchquotes(self, msg, cmd):
|
|
c = self.db.query("SELECT * FROM `chat` ORDER BY RANDOM() LIMIT 1;")
|
|
row = c.fetchone()
|
|
c.close()
|
|
if row:
|
|
self.bot.act_PRIVMSG(msg.args[0], "<%s> %s" % (row["sender"], row["message"],))
|
|
|
|
@hook("PRIVMSG")
|
|
def logquote(self, msg, cmd):
|
|
self.db.query("INSERT INTO `chat` (`date`, `sender`, `message`) VALUES (?, ?, ?)",
|
|
(int(datetime.now().timestamp()), msg.prefix.nick, msg.trailing)).close()
|
|
# Trim quotes
|
|
c = self.db.query("SELECT * FROM `chat` ORDER BY `date` DESC LIMIT %s, 1000000;" % self.config["limit"])
|
|
while True:
|
|
row = c.fetchone()
|
|
if not row:
|
|
break
|
|
self.db.query("DELETE FROM `chat` WHERE id=?", (row["id"],)).close()
|
|
c.close()
|
|
|
|
def ondisable(self):
|
|
self.db.close()
|
|
|