pyircbot/pyircbot/modules/Inventory.py

#!/usr/bin/env python
"""
.. module:: Inventory
	:synopsis: Lets the bot hold random items

.. moduleauthor:: Dave Pedu <dave@davepedu.com>

"""

from pyircbot.modulebase import ModuleBase,ModuleHook
from datetime import datetime

class Inventory(ModuleBase):
	def __init__(self, bot, moduleName):
		ModuleBase.__init__(self, bot, moduleName);
		self.hooks=[]
		self.db = None
		serviceProviders = self.bot.getmodulesbyservice("sqlite")
		if len(serviceProviders)==0:
			self.log.error("Inventory: Could not find a valid sqlite service provider")
		else:
			self.log.info("Inventory: Selecting sqlite service provider: %s" % serviceProviders[0])
			self.db = serviceProviders[0].opendb("inventory.db")
		
		if not self.db.tableExists("inventory"):
			self.log.info("Inventory: Creating table: inventory")
			c = self.db.query("""CREATE TABLE IF NOT EXISTS `inventory` (
			`id` INTEGER PRIMARY KEY,
			`date` INTEGER,
			`donor` varchar(64),
			`item` varchar(64)
			) ;""")
			c.close()
		
		self.hooks=[ModuleHook("PRIVMSG", self.checkInv)]
	
	def checkInv(self, args, prefix, trailing):
		if not args[0][0]=="#":
			return
		prefixObj = self.bot.decodePrefix(prefix)
		cmd = self.bot.messageHasCommand([".have"], trailing, True)
		if cmd:
			if len(cmd.args)<1:
				return
			adjective = None
			if cmd.args[0] in self.config["adjectives"]:
				cmd.args_str = cmd.args_str[len(cmd.args[0]):].strip()
				adjective = cmd.args[0]
			newItem = cmd.args_str
			
			if self.has_item(newItem):
				self.bot.act_PRIVMSG(args[0], self.config["dupe_msg"] % 
					{"item":newItem,})
				return
			
			dropped = self.add_item(prefixObj.nick, newItem)
			if len(dropped) > 0:
				self.bot.act_PRIVMSG(args[0], self.config["swap_msg"] % 
					{"adjective":(adjective+" ") if adjective else "", "recv_item":newItem, "drop_item":", ".join(dropped)})
			else:
				self.bot.act_PRIVMSG(args[0], self.config["recv_msg"] % 
					{"item":newItem, "adjective":"these " if newItem[-1:]=="s" else "this "})
		
		cmd = self.bot.messageHasCommand([".inventory", ".inv"], trailing)
		if cmd:
			inv = self.getinventory()
			if len(inv)==0:
				self.bot.act_PRIVMSG(args[0], self.config["inv_msg"] % {"itemlist":"nothing!"})
			else:
				self.bot.act_PRIVMSG(args[0], self.config["inv_msg"] % {"itemlist":", ".join(inv)})
		
	
	def has_item(self, itemName):
		c = self.db.query("SELECT COUNT(*) as `num` FROM `inventory` WHERE `item`=? COLLATE NOCASE", (itemName,)) # 
		row = c.fetchone()
		c.close()
		return row["num"]>0
	
	def add_item(self, donor, itemName):
		dropped = []
		c = self.db.query("SELECT * FROM `inventory` ORDER BY RANDOM() LIMIT %s,1000000" % self.config["limit"])
		while True:
			row = c.fetchone()
			if row == None:
				break
			dropped.append(row["item"])
			self.db.query("DELETE FROM `inventory` WHERE `id`=?", (row["id"],)).close()
		c.close()
		
		self.db.query("INSERT INTO `inventory` (`date`, `donor`, `item`) VALUES (?, ?, ?)",
			(int(datetime.now().timestamp()), donor, itemName)).close()
		return dropped
	
	def getinventory(self):
		inv = []
		c = self.db.query("SELECT * FROM `inventory`")
		while True:
			row = c.fetchone()
			if row == None:
				break
			inv.append(row["item"])
		c.close()
		return inv
	
	def ondisable(self):
		self.db.close()