X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/061081bc194645190b57616ad56fe50105309820..5f42c250f797d8cf1bca89d431134764d97c63fd:/bot.py?ds=sidebyside diff --git a/bot.py b/bot.py index 5f9f26f..fa4eaf1 100644 --- a/bot.py +++ b/bot.py @@ -3,7 +3,7 @@ # Erebus IRC bot - Author: John Runyon # "Bot" and "BotConnection" classes (handling a specific "arm") -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): @@ -80,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)) @@ -94,14 +94,15 @@ class Bot(object): elif pieces[1] == "NICK": oldnick = pieces[0].split('!')[0][1:] newnick = pieces[2][1:] - self.parent.users[newnick.lower()] = self.parent.users[oldnick.lower()] - del self.parent.users[oldnick.lower()] - self.parent.user(newnick).nick(newnick) + 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: @@ -175,8 +176,6 @@ class Bot(object): def __repr__(self): return "" % (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 = '' @@ -186,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)) @@ -204,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]