bot.py

#!/usr/bin/python2.7

import sys
import socket
import string

import utils
import settings
import privmsg

from handlers import *


def connect(HOST, PORT, NICK, IDENT, REALNAME, PASS, CHANNEL):
    conn.connect((HOST, PORT))
    conn.send('NICK '+NICK+'\r\n')
    conn.send('USER '+IDENT+' '+HOST+' * :'+REALNAME+'\r\n')

    initialPing() # Wait for initial ping

    conn.send('PRIVMSG NickServ IDENTIFY '+PASS+'\r\n')
    conn.send('JOIN '+CHANNEL+'\r\n')
    conn.send('PRIVMSG '+CHANNEL+' :Quack!\r\n')

def sendPing(ping):
    conn.send('PONG '+ping+'\r\n')
    print 'PONG'

def initialPing():
    for line in conn.makefile('r'):
        print line
        if line.split()[0] == 'PING':
            sendPing(line.split()[1])
            print 'Initial PONG sent'
            break

modes = { 'owner': '+q', 'o': '+q',
          'deowner': '-q', 'do': '-q',
          'protected': '+p', 'protect': '+p', 'p': '+p',
          'deprotected': '-p', 'deprotect': '-p', 'dp': '-p',
          'operator': '+o', 'op': '+o',
          'deoperator': '-o', 'deop': '-o', 'dop': '-o',
          'halfop': '+ho', 'hop': '+ho', 'ho': '+ho',
          'dehalfop': '-ho', 'dho': '-ho',
          'voice': '+v', 'v': '+v',
          'devoice': '-v', 'dv': '-v',
          'ban': '+b', 'b': '+b',
          'unban': '-b', 'b': '-b',
        }

def changeMode(conn, msg):
    if utils.isadmin(conn, msg):
        cmd = msg.text.split()[0][1:]
        if len(msg.text.split()) > 1:
            wie = msg.text.split()[1]
        else:
            wie = msg.user
        conn.send('MODE %s %s %s\r\n' % (msg.channel, modes[cmd], wie))
    else:
        conn.send('PRIVMSG %s :Je moet een operator zijn om dit commando te kunnen uitvoeren.\r\n' % msg.user)

actions = {'kick': kick.kick,
           'k': kick.kick,
           'topic': topic.topic,
           't': topic.topic,
           'addquote': quote.addquote,
           'quote': quote.quote,
           'delquote': quote.delquote,
           'tell': tell.tell,
           'assign': definitions.assign,
           'reassign': definitions.reassign,
           'unassign': definitions.unassign,
           'list': definitions.list,
           'dt': dt.dt,
           'dtlist': dt.list,
           'oa': onzinalarm.alarm,
           'help': help.help
        }

def parseMessage(data):

    msg = privmsg.PrivMSG(data)

    tell.active(conn, msg)

    if "Quack" in msg.text:
        conn.send('PRIVMSG %s :Quack, %s!\r\n' % (msg.channel, msg.user))

    if msg.char == '!':
        if msg.cmd[0] in modes:
            changeMode(conn, msg)

        elif msg.cmd[0] in actions:
            actions[msg.cmd[0]](conn, msg)

    elif msg.char == '?':
        definitions.explain(conn, msg)

def listen(channel, conn):
    for line in conn.makefile('r'):
        if len(line) > 0:
            print line
            if line.split()[0] == 'PING':
                sendPing(line.split()[1])
            if "PRIVMSG " + channel in line:
                parseMessage(line)

conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
connect(settings.irc_HOST, settings.irc_PORT, settings.irc_NICK, settings.irc_IDENT, settings.irc_REALNAME, settings.irc_PASS, settings.irc_CHANNEL)
listen(settings.irc_CHANNEL, conn)