murkinabot.py

# -*- coding: utf-8 -*-

"""
yksinkertainen ircbotti

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

import socket
import botcommands
from time import sleep
from config import *


class Ircbot:

    def __init__(self ):

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

        self.users = OWNERS

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

        self.server = SERVER
        self.port = PORT
        self.username = USERNAME
        self.realname = REALNAME
        self.nick = NICKNAME

        # luodaan socket

        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        # haetaan botille komennot

        self.commands = botcommands.command_dict

        # päälooppia toistettan kunnes done = 1

        self.done = 0

        # kanava jolle botti halutaan
        self.channels  = CHANNELS
        # self.channel  = '#murkinatesti'

    def send( self, string, delay=0 ):

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

        self.socket.send( string + '\n' )
        print("[BOT] " + string)
        if(delay > 0 ):
            sleep(delay)

    def sendWithDelay( self, string):
        self.send(string + '\n', 0.3)

    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 0 * :%s' % (self.username, self.realname))

        # liitytään kanavalle

        for channel in self.channels:
            self.send( 'JOIN %s' % channel )

    def check( self, line ):

        print("[SERVER] " + line)
        line = line.split(' ')

        # vastataan pingiin muuten serveri katkaisee yhteyden

        if line[0] == 'PING':

             self.send( 'PONG ' + line[1] )

        else:

            # vastataan komentoihin myös yksityiskeskutelussa

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

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

            # suoritetaan komennot jos niitä on tullut

            # line form :samies!sakrnie@linux.utu.fi PRIVMSG #murkinatesti :!murkinat ict
            # f = open('requests.txt', 'w')
            # kirjoitetaan tiedostoon kuka teki kyselyn ja millä parametrilla
            # f.write("%s: %s" %(line[1],line[2]))

            # TODO: Älä käytä try-exceptiä ohjelma flown hallintaan -> virheitä hankala huomata
            if(len(line) > 3 and line[3] in self.commands.keys()):
                try:
                    self.commands[ line[3] ].main( self , line )

                except Exception as ex:
                    messageTemplate = "[WARNING] Exception of type {0} occured. Argumets:\n{1!r}"
                    message = messageTemplate.format(type(ex).__name__, ex.args)
                    print(message)

    def testing(self):

        # Testaa seuraavia komentoja 
        # self.commands[":!murkinat"].main(self, ['a', 'b', 'c', 'd','Tottis', 'ict', 'perse', 'ict', 'mikro', 'maccis'])
        # self.commands[":!juhannisnimi"].main(self, ['a', 'b', 'c', 'd','Sami'])
        self.commands[":!vitshumor"].main(self, ['a', 'b', 'c', 'd','digit'])

    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()
    # irc.testing()

if __name__ == '__main__' : main()