[IRC] bot python.

[Junky]

Bonjour,

En parallèle d'une journée merdique pour un admin sys linux (maj openssl du parc plus installation d'n windows 7) je vous poste le début du code de mon BOT irc que je suis en train de me faire pour le fun.

Code PYTHON :


#!/usr/bin/python2

import socket
import time
import string
import re

VERSION = "PythonBot: 0.1 powered by Snorky"

DEBUG = False


LIST=["!help", "!quit", "!conn"]


def init_conn(channel, nickname, server, port):
    s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.connect((server, port))
    return s

def send_data(s, msg):
    msg = msg + "\r\n"
    s.send(msg)

def ping(s, msg):
    send_data(s, "PONG %s" % msg)

def conn(s, channel, nickname, host):
    ident = "aa"
    realname = "bb"
    #Send data for connection
    send_data(s, "NICK %s" % nickname)
    send_data(s, "USER %s %s oxy :%s" % (ident, host, realname))
    #Waiting
    time.sleep(2)
    #Join the channel
    send_data(s, "JOIN %s" % channel)

def cctp(s, line):
    nick, msg_type, chan = info(line)
    if (re.findall(ur"^.+VERSION.+$", line[3])):
        #[':Snorky!Silentd@snorky', 'PRIVMSG', 'Oxy', ':\x01VERSION\x01']
        send_data(s, (("NOTICE %s \001VERSION %s\001")% (nick, VERSION)))
    else:
        send_data(s, (("NOTICE %s \001THE_GAME %s\001")% (nick, "www.tukif.com")))

def info(line):
    nick = line[0].split("!")[0][1:]
    msg_type = line[1]
    channel = line[2]
    return nick, msg_type, channel


def menu(s, channel):
    for cmd in LIST:
        send_data(s, ("PRIVMSG %s %s" % (channel, cmd)))
        time.sleep(0.2)

def do_command(s, line, connected, channel):
    #[':Snorky!Silentd@snorky', 'PRIVMSG', '#Snorky_land', ':zz']
    #:Snorky!Silentd@snorky PRIVMSG #Snorky_land :zz
    regex = ur"^:\x01.+\x01$"
    victim = line[0].split("!")[0][1:]
    if DEBUG: print line

    if (connected == 0):
        print ' '.join(map(str, line))
        if (line[1] == "JOIN"):
            connected = 1
    #CTCP VERSION
    elif (re.findall(regex, line[3])):
        cctp(s, line)
    #Menu
    elif (line[3]==":!menu"):
        menu(s, channel)
    #Chanserv Notification
    elif(victim == "ChanServ"):
        print line
    else:
        print line[0].split("!")[0][1:], "\t| ", line[3][1:]
    return connected

def master(s, channel, nickname, server, port):
    readbuffer = ""
    connected = 0
    while True:
        readbuffer=readbuffer+s.recv(1024)
        temp=string.split(readbuffer, "\n")
        readbuffer=temp.pop( )

        for line in temp:
            line=string.rstrip(line)
            line=string.split(line)
            if(line[0]=="PING"):
                ping(s, line[1])
            else :
                connected = do_command(s, line, connected, channel)
#        msg = raw_input(">>> ")
#        if (connected == 1):
#            send_data(s, ("PRIVMSG %s %s" %(channel, msg)))
        time.sleep(0.2)

def main():
    import sys
    if len(sys.argv) != 4:
        print "Usage: testbot <server[:port]> <channel> <nickname>"
        sys.exit(1)

    s = sys.argv[1].split(":", 1)
    server = s[0]
    if len(s) == 2:
        try:
            port = int(s[1])
        except ValueError:
            print "Error: Erroneous port."
            sys.exit(1)
    else:
        port = 6667
    channel = sys.argv[2]
    nickname = sys.argv[3]

    s = init_conn(channel, nickname, server, port)
    conn(s, channel, nickname, server)

    master(s, channel, nickname, server, port)


if __name__ == "__main__":
    main()

 

Pour le lancer:

Code :

#./prog.py SERVER_NAME CHAN NICK
./oxy.py irc.big-daddy.fr \#bot Oxy

#Attention, bien échapper le '#' du chan sinon marche pas et vous devriez comprendre pk. :)

Voilou, je retourne a mes MAJ + install .

Junky,