X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/963f2522b39c57a38f9cc05aa66c9d504162a54d..b367d0c5869793453f81614d320a65b169926794:/erebus.py

diff --git a/erebus.py b/erebus.py
index e6c0459..90e1389 100644
--- a/erebus.py
+++ b/erebus.py
@@ -3,11 +3,11 @@
 # Erebus IRC bot - Author: John Runyon
 # main startup code
 
-import os, sys, select, MySQLdb, MySQLdb.cursors, time, random
+import os, sys, select, MySQLdb, MySQLdb.cursors, time, random, gc
 import bot, config, ctlmod
 
-class Erebus(object):
-	APIVERSION = 1
+class Erebus(object): #singleton to pass around
+	APIVERSION = 2
 	RELEASE = 0
 
 	bots = {}
@@ -28,6 +28,8 @@ class Erebus(object):
 
 		def msg(self, *args, **kwargs):
 			main.randbot().msg(self, *args, **kwargs)
+		def slowmsg(self, *args, **kwargs):
+			main.randbot().slowmsg(self, *args, **kwargs)
 		def fastmsg(self, *args, **kwargs):
 			main.randbot().fastmsg(self, *args, **kwargs)
 
@@ -55,12 +57,14 @@ class Erebus(object):
 			return self.glevel
 
 		def join(self, chan):
-			self.chans.append(chan)
+			if chan not in self.chans: self.chans.append(chan)
 		def part(self, chan):
-			self.chans.remove(chan)
-		def quit(self):
-			for chan in self.chans:
+			try:
 				self.chans.remove(chan)
+			except: pass
+			return len(self.chans) == 0
+		def quit(self):
+			pass
 		def nickchange(self, newnick):
 			self.nick = newnick
 
@@ -87,6 +91,8 @@ class Erebus(object):
 
 		def msg(self, *args, **kwargs):
 			self.bot.msg(self, *args, **kwargs)
+		def slowmsg(self, *args, **kwargs):
+			self.bot.slowmsg(self, *args, **kwargs)
 		def fastmsg(self, *args, **kwargs):
 			self.bot.fastmsg(self, *args, **kwargs)
 
@@ -159,18 +165,20 @@ class Erebus(object):
 	def randbot(self): #get Bot() randomly
 		return self.bots[random.choice(self.bots.keys())]
 
-	def user(self, _nick, justjoined=False):
+	def user(self, _nick, justjoined=False, create=True):
 		nick = _nick.lower()
 		if nick in self.users:
 			return self.users[nick]
-		else:
+		elif create:
 			user = self.User(_nick)
 			self.users[nick] = user
 
 			if justjoined:
-				self.randbot().conn.send("WHO %s n%%ant,2" % (nick))
+				self.randbot().conn.send("WHO %s n%%ant,1" % (nick))
 
 			return user
+		else:
+			return None
 	def channel(self, name): #get Channel() by name
 		if name.lower() in self.chans:
 			return self.chans[name.lower()]
@@ -196,6 +204,12 @@ class Erebus(object):
 	def module(self, name):
 		return ctlmod.modules[name]
 
+	def log(self, source, level, message):
+		print "%09.3f %s [%s] %s" % (time.time() % 100000, source, level, message)
+
+	def getuserbyauth(self, auth):
+		return [u for u in self.users.itervalues() if u.auth == auth.lower()]
+
 	#bind functions
 	def hook(self, word, handler):
 		try:
@@ -239,12 +253,21 @@ class Erebus(object):
 
 class MyCursor(MySQLdb.cursors.DictCursor):
 	def execute(self, *args, **kwargs):
-		print "%05.3f [SQL] [#] MyCursor.execute(self, %s, %s)" % (time.time() % 100000, ', '.join([repr(i) for i in args]), ', '.join([str(key)+"="+repr(kwargs[key]) for key in kwargs]))
+		if 'norecurse' in kwargs:
+			norecurse = kwargs['norecurse']
+			del kwargs['norecurse']
+		else:
+			norecurse = False
+		main.log("[SQL]", "?", "MyCursor.execute(self, %s, %s)" % (', '.join([repr(i) for i in args]), ', '.join([str(key)+"="+repr(kwargs[key]) for key in kwargs])))
+#		print "%09.3f [SQL] [#] MyCursor.execute(self, %s, %s)" % (time.time() % 100000, ', '.join([repr(i) for i in args]), ', '.join([str(key)+"="+repr(kwargs[key]) for key in kwargs]))
 		try:
-			super(self.__class__, self).execute(*args, **kwargs)
+			return super(self.__class__, self).execute(*args, **kwargs)
 		except MySQLdb.MySQLError as e:
-			print "%05.3f [SQL] [!] MySQL error! %r" % (time.time() % 100000, e)
-			dbsetup()
+			main.log("[SQL]", "!", "MySQL error! %r" % (e))
+#			print "%09.3f [SQL] [!] MySQL error! %r" % (time.time() % 100000, e)
+			if not norecurse:
+				dbsetup()
+				return self.execute(norecurse=True, *args, **kwargs)
 			return False
 		return True
 
@@ -256,7 +279,10 @@ def dbsetup():
 def setup():
 	global cfg, main
 
-	cfg = config.Config('bot.config')
+	cfg = config.setup('bot.config')
+
+	if int(cfg.get('debug', 'gc', default=0)) == 1:
+		gc.set_debug(gc.DEBUG_LEAK)
 
 	pidfile = open(cfg.pidfile, 'w')
 	pidfile.write(str(os.getpid()))
@@ -286,7 +312,7 @@ def loop():
 if __name__ == '__main__':
 	try: os.rename('logfile', 'oldlogs/%s' % (time.time()))
 	except: pass
-	sys.stdout = open('logfile', 'w')
+	sys.stdout = open('logfile', 'w', 1)
 	sys.stderr = sys.stdout
 	setup()
 	while True: loop()