Forum Thread: Python IRC bot, v3

Beware, the code is quite ratty. A goal in this version was to be able to update/add commands without having to restart the bot. Mission accomplished.

botim.py:

import sys
import socket
import re
import select
from time import sleep
from command import command

#Global Variables
host    = "irc.darchoods.net"   #This is the IRC server variable.
port    = 6667                  #This is the port
nick    = ""             #Bot name
ident   = ""             #ID to NickServ with this name
servername = "bla"
realname= ""             #Bots real name for server identification
password= ""
channel = "#nullbytez"          #This is the channel name

CRLF = '\r\n'
# Examples
# Considered adding something to make it a little more channel independant
"""
    All functions must take one argument: args
    args[0] is the socket
    args[1] is the command object
    args[2] is a list of extra arguments
"""
def pong(args):
    args[0].sendall('PONG :' + args[1] + CRLF)
def b_quit(args):
    args[0].sendall("QUIT :Quit Leaving" + CRLF)
def addUser(args):
    args[1].addSafe('ryoh', args[2][0])
def rmUser(args):
    args[1].rmSafe('ryoh', args[2][0])
def opUser(args):
    args[0].sendall("MODE #nullbytez +o " + args[2][0] + CRLF)
def loadMod(args):
    args[1].load('ryoh', args[2][0])
def refresh(args):
    args[1].refresh('ryoh', args[2][0])

def main():
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP )
    s.connect((host, port))
    s.sendall("NICK %s\r\n" % nick) #Sets the Bot's nick name
    s.sendall("USER %s %s %s :%s\r\n" % (ident, host, servername, realname)) #Logs Bot into IRC and ID's
    # at this point I'm connected and ready to start talking
    f = s.makefile() # I do this to take advantage of readline()
    runner = command(['ryoh', 'L-'])
    runner.add('ryoh', 'pong', pong)
    runner.add('ryoh', '!quit', b_quit)
    runner.add('ryoh', '!add', addUser)
    runner.add('ryoh', '!rm', rmUser)
    runner.add('ryoh', '!op', opUser)
    runner.add('ryoh', '!load', loadMod)
    runner.add('ryoh', '!reload', refresh)
    while 1:
        ready = select.select([0, f], [], [])
        if 0 in ready[0]:
            data = sys.stdin.readline().rstrip().lstrip()
            if data: s.sendall(data + CRLF)

        if f in ready[0]:
            line = f.readline().rstrip().lstrip()
            if 'PING' in line:
                server = line.split(':')[1].lstrip().rstrip()
                runner.do('ryoh', 'pong', s, server)
            if line:
                print line
                if line[0] == ':':
                    try:
                        extras = line.split(':')[2].split(' ')[1:]
                    except IndexError: extras = ''
                    try:
                        if runner.check(line.split(':')[2].split(' ')[0]):
                            runner.do(line.split(':')[1].split('!')[0],
                            line.split(':')[2].split(' ')[0], s, runner, extras)
                    except IndexError: pass
            else: break

if __name__ == '__main__':
    main()

command.py:

class command:

    _COMMANDS = {}

    _safe = []

    def __init__(self, safe):
        self._safe = safe

    def add(self, user, command, handle):
        if user in self._safe:
            if command not in self._COMMANDS:
                self._COMMANDS[command.lower()] = handle

    def rm(self, user, command):
        if user in self._safe:
            if command.lower() in self._COMMANDS:
                del self._COMMANDS[command.lower()]

    def do(self, user, command, *args):
        if user in self._safe:
            self._COMMANDS[command.lower()](args)

    def load(self, user, module):
        self._extract(__import__(module))

    def _extract(self, module):
        for command in dir(module):
            self._COMMANDS["!" + command] = eval("module."+command)

    def refresh(self, user, module):
        m = __import__(module)
        reload(m)
        self._extract(m)

    def getSafe(self, user):
        if user in self._safe:
            return self.safe

    def addSafe(self, user, name):
        if user in self._safe:
            self._safe.append(name)

    def rmSafe(self, user, name):
        if user in self._safe:
            self._safe.remove(name)

    def check(self, command):
        if command in self._COMMANDS:
            return True
        else:
            return False

extras.py:

"""
    All functions must take one argument: args
    args[0] is the socket
    args[1] is the command object
    args[2] is a list of extra arguments
"""
CRLF = '\r\n'
def say(args):
    string = ''
    for c in args[2]:
        string += c + ' '
    args[0].sendall("PRIVMSG #nullbytez :" + string + CRLF)

def nig(args):
    args[0].sendall("PRIVMSG #nullbytez :now I can add shit on the fly" + CRLF)