Keskustelu: Koodit: Python: ircbot

# -*- coding: cp1252 -*-

"""
yksinkertainen ircbotti

komennot:
    !test
    !join
    !quit
    !anagram
"""

import socket
import botcommands

class Ircbot:

    def __init__( self ):

        # määritellään botille pääkäyttäjät

        self.users = [ ':nick!username@host.fi' ]

        # välttämättömiä tietoja

        self.server   = 'irc.irkkailuserverisi.fi'
        self.port     = 6667
        self.username = 's-bot'
        self.realname = 's-bot'
        self.nick     = 's-botti'

        # luodaan socket

        self.socket   = socket.socket()

        # haetaan botille komennot

        self.commands = botcommands.command_dict

        # päälooppia toistettan kunnes done = 1

        self.done     = 0

        # kanava jolle botti halutaan

        self.channel  = '#jokukanavatähän'

    def send( self, string ):

        # tällä lähetetään viestejä

        self.socket.send( string + '\r\n' )

    def connect( self ):

        # yhdistetään serveriin ja läheteään omat tiedot

        self.socket.connect( ( self.server, self.port ) )
        self.send( 'NICK %s' % self.nick )
        self.send( 'USER %s a a :%s' % ( self.username, self.realname ) )

        # liitytään kanavalle

        self.send( 'JOIN %s' % self.channel )

    def check( self, line ):

        print line
        line = line.split(' ')

        # vastataan pingiin muuten serveri katkaisee yhteyden

        if line[0] == 'PING':

             self.send( 'PONG :abc' )

        try:

            # vastataan komentoihin myös yksityiskeskutelussa

            if line[2][0] != '#':

                line[2] = line[0].split( '!' )[0][1:]

            # suoritetaan komennot jos niitä on tullut

            self.commands[ line[3] ].main( self , line )

        except:

            pass

    def mainloop( self ):

    buffer = ''

    while not self.done:

        # vastaanotetaan dataa

        buffer += self.socket.recv( 4096 )
        buffer = buffer.split( '\r\n' )

        for line in buffer[0:-1]:

            self.check( line )

        buffer = buffer[-1]

def main():

    irc = Ircbot()
    irc.connect()
    irc.mainloop()

if __name__ == '__main__': main()

# -*- coding: cp1252 -*-

"""
muutama esimerkki komennoista
"""

import random

# tähän sanastoon lisätään komennot ja niitä vastaavat oliot

command_dict = {}


class Test:

    def main( self, irc, line ):

        irc.send( 'PRIVMSG %s :Hello world!' % line[2] )

command_dict[ ':!test' ] = Test()


class Join:

    def main( self, irc, line):

        irc.send( 'JOIN %s' % ( line[4] ) )

command_dict[ ':!join' ] = Join()


class Quit:

    def main( self, irc , line ):

        # määritellään komento vain pääkäyttäjille

        if line[0] in irc.users:

            irc.send( 'QUIT' )
            irc.socket.close()
            irc.done = 1

command_dict[ ':!quit' ] = Quit()


class Anagram:

    def main( self, irc, line):

        string = list( ' '.join( line[4:] ) )
        random.shuffle( string )
        string = ''.join( string )
        irc.send( 'PRIVMSG %s :%s' % ( line[2], string ) )

command_dict[ ':!anagram' ] = Anagram()