trivia - add last win to topic

[erebus.git] / bot.py

diff --git a/bot.py b/bot.py
index fe9be344525fa60a2a7bbdd47f53eb382f3a7236..fa4eaf1b6eb123a77a0b6c780db3dde2164a202f 100644 (file)
--- a/bot.py
+++ b/bot.py
@@ -3,9 +3,7 @@
 # Erebus IRC bot - Author: John Runyon
 # "Bot" and "BotConnection" classes (handling a specific "arm")
 
-#TODO: error checking
-
-import socket, sys
+import socket, sys, time
 
 #bots = {'erebus': bot.Bot(nick='Erebus', user='erebus', bind='', server='irc.quakenet.org', port=6667, realname='Erebus')}
 class Bot(object):
@@ -82,7 +80,7 @@ class Bot(object):
                elif pieces[1] == "PART":
                        nick = pieces[0].split('!')[0][1:]
                        chan = self.parent.channel(pieces[2])
-                       
+
                        if nick != self.nick:
                                self.parent.user(nick).part(chan)
                                chan.userpart(self.parent.user(nick))
@@ -93,10 +91,18 @@ class Bot(object):
                                self.parent.user(nick).quit()
                                del self.parent.users[nick.lower()]
 
-               elif pieces[1] == "MODE": #TODO
+               elif pieces[1] == "NICK":
+                       oldnick = pieces[0].split('!')[0][1:]
+                       newnick = pieces[2][1:]
+                       if newnick.lower() != oldnick.lower():
+                               self.parent.users[newnick.lower()] = self.parent.users[oldnick.lower()]
+                               del self.parent.users[oldnick.lower()]
+                       self.parent.users[newnick.lower()].nickchange(newnick)
+
+               elif pieces[1] == "MODE": #TODO parse for ops/voices (at least)
                        pass
 
-       
+
        def parsemsg(self, user, target, msg):
                chan = None
                if len(msg) == 0:
@@ -120,7 +126,7 @@ class Bot(object):
                                                pieces.pop(1)
 
                else: # message was sent to a channel
-                       chan = self.parent.channel(target) #TODO check if bot's on channel --- in Erebus.channel() maybe?
+                       chan = self.parent.channel(target)
                        try:
                                if msg[0] == '*': # message may be addressed to bot by "*BOTNICK" trigger?
                                        if pieces[0][1:].lower() == self.nick.lower():
@@ -170,8 +176,6 @@ class Bot(object):
        def __repr__(self): return "<Bot %r>" % (self.nick)
 
 class BotConnection(object):
-       state = 0 # 0=disconnected, 1=registering, 2=connected
-
        def __init__(self, parent, bind, server, port):
                self.parent = parent
                self.buffer = ''
@@ -181,6 +185,8 @@ class BotConnection(object):
                self.server = server
                self.port = int(port)
 
+               self.state = 0 # 0=disconnected, 1=registering, 2=connected
+
        def connect(self):
                self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                self.socket.bind((self.bind, 0))
@@ -199,19 +205,21 @@ class BotConnection(object):
 
        #TODO: rewrite send() to queue
        def send(self, line):
-               print self.parent.nick, '[O]', str(line)
-               self.write(line)
+               print "%05.3f %s [O] %s" % (time.time() % 100000, self.parent.nick, line)
+#              print (time.time() % 1466400000), self.parent.nick, '[O]', str(line)
+               self._write(line)
 
-       def write(self, line):
+       def _write(self, line):
                self.socket.sendall(line+"\r\n")
 
        def read(self):
                self.buffer += self.socket.recv(8192)
                lines = []
 
-               while '\r\n' in self.buffer:
-                       pieces = self.buffer.split('\r\n', 1)
-                       print self.parent.nick, '[I]', pieces[0]
+               while "\r\n" in self.buffer:
+                       pieces = self.buffer.split("\r\n", 1)
+                       print "%05.3f %s [I] %s" % (time.time() % 100000, self.parent.nick, pieces[0])
+#                      print (time.time() % 1460000000), self.parent.nick, '[I]', pieces[0]
                        lines.append(pieces[0])
                        self.buffer = pieces[1]