nemubot/nemubot.py

#!/usr/bin/python2.7
# coding=utf-8

#import signal
import sys
import socket
import string
import os
import re
import thread
from datetime import date
from datetime import datetime
from datetime import timedelta

import norme
import newyear
import ontime
import watchWebsite

if len(sys.argv) > 1:
    sys.exit(0)

HOST='127.0.0.1'
PORT=2770
#HOST='irc.rezosup.org'
#PORT=6667
NICK='nemubot'
IDENT='nemubot'
REALNAME='nemubot'
OWNER='nemunaire' #The bot owner's nick
#CHANLIST='#nemutest'
CHANLIST='#42sh #nemutest #tc'
readbuffer='' #Here we store all the messages from server

birthdays = {"maxence23": datetime(1991, 8, 11),
             "nemunaire": datetime(1991, 12, 4, 6, 42),
             "xetal": datetime(1991, 2, 23),
             "bob": datetime(1991, 2, 1),
             "test": datetime(1991, 1, 25),
             "colona": datetime(1991, 7, 16)}

s = socket.socket( ) #Create the socket
s.connect((HOST, PORT)) #Connect to server
s.send("PASS %s\r\n" % "McsuapTesbuf")
s.send("NICK %s\r\n" % NICK)
s.send("USER %s %s bla :%s\r\n" % (IDENT, HOST, REALNAME))
#s.send("JOIN %s\r\n" % CHANLIST)

print("Welcome on Nemubot. I operate on %s. My PID is %i" % (CHANLIST, os.getpid()))

def parsemsg(msg):
    global birthdays
    complete = msg[1:].split(':',1) #Parse the message into useful data
    info = complete[0].split(' ')
    msgpart = complete[1]
    sender = info[0].split('!')

    if CHANLIST.find(info[2]) != -1 and re.match(".*(norme|coding style).*", msgpart) is not None and re.match(".*(please|give|obtenir|now|plz|stp|svp|s'il (te|vous) pla.t|check).*", msgpart) is not None:
        norme.launch (s, sender, msgpart)

    elif CHANLIST.find(info[2]) != -1 and msgpart.find("%s:"%NICK) == 0: #Treat all messages starting with 'nemubot:' as distinct commands
        msgpartl = msgpart.lower()
        print msgpartl
        if re.match(".*(m[' ]?entends?[ -]+tu|h?ear me|ping).*", msgpartl) is not None:
            s.send("PRIVMSG %s :%s: pong\r\n"%(info[2], sender[0]))
        elif re.match(".*di[st] (a|à) ([a-zA-Z0-9_]+) (.+)$", msgpart) is not None:
            result = re.match(".*di[st] (a|à) ([a-zA-Z0-9_]+) (qu(e |'))?(.+)$", msgpart)
            s.send("PRIVMSG %s :%s: %s\r\n"%(info[2], result.group(2), result.group(5)))
        elif re.match(".*di[st] (.+) (a|à) ([a-zA-Z0-9_]+)$", msgpart) is not None:
            result = re.match(".*di[st] (.+) (à|a) ([a-zA-Z0-9_]+)$", msgpart)
            s.send("PRIVMSG %s :%s: %s\r\n"%(info[2], result.group(3), result.group(1)))
        elif re.match(".*(date de naissance|birthday|geburtstag|née?|nee? le|born on).*", msgpartl) is not None:
            result = re.match("[^0-9]+(([0-9]{1,4})[^0-9]+([0-9]{1,2}|janvier|january|fevrier|février|february|mars|march|avril|april|mai|maï|may|juin|juni|juillet|july|jully|august|aout|août|septembre|september|october|obtobre|novembre|november|decembre|décembre|december)([^0-9]+([0-9]{1,4}))?)[^0-9]+(([0-9]{1,2})[^0-9]*[h':][^0-9]*([0-9]{1,2}))?.*", msgpartl)

#            if re.match(".*(je|i|I).*") is not None:
#                target = sender[0]
#            elif re.match(".*().*") is not None:

            if result is None:
                s.send("PRIVMSG %s :%s: je ne reconnais pas le format de ta date de naissance :(\r\n"%(info[2], sender[0]))
            else:
                day = result.group(2)
                if len(day) == 4:
                    year = day
                    day = 0
                month = result.group(3)
                if month == "janvier" or month == "january" or month == "januar":
                    month = 1
                elif month == "fevrier" or month == "février" or month == "february":
                    month = 2
                elif month == "mars" or month == "march":
                    month = 3
                elif month == "avril" or month == "april":
                    month = 4
                elif month == "mai" or month == "may" or month == "maï":
                    month = 5
                elif month == "juin" or month == "juni" or month == "junni":
                    month = 6
                elif month == "juillet" or month == "jully" or month == "july":
                    month = 7
                elif month == "aout" or month == "août" or month == "august":
                    month = 8
                elif month == "september" or month == "septembre":
                    month = 9
                elif month == "october" or month == "october" or month == "oktober":
                    month = 10
                elif month == "november" or month == "novembre":
                    month = 11
                elif month == "december" or month == "decembre" or month == "décembre":
                    month = 12
                if day == 0:
                    day = result.group(5)
                else:
                    year = result.group(5)

                hour = result.group(7)
                minute = result.group(8)

                print "Chaîne reconnue : %s/%s/%s %s:%s"%(day, month, year, hour, minute)
                if year == None:
                    year = date.today().year
                if hour == None:
                    hour = 0
                if minute == None:
                    minute = 0

                try:
                    newdate = datetime(int(year), int(month), int(day), int(hour), int(minute))
                    birthdays[sender[0].lower()] = newdate
                    s.send("PRIVMSG %s :%s: ok, c'est noté, ta date de naissance est le %s\r\n"%(info[2], sender[0], newdate.strftime("%A %d %B %Y à %H:%M")))
                except ValueError:
                    s.send("PRIVMSG %s :%s: ta date de naissance me paraît peu probable...\r\n"%(info[2], sender[0]))                

    elif CHANLIST.find(info[2]) != -1 and msgpart[0] == '!': #Treat all messages starting with '!' as command
        cmd = msgpart[1:].split(' ')
        if cmd[0] == 'help' or cmd[0] == 'h':
            s.send("PRIVMSG %s :Pour me demander quelque chose, commencer votre message par !.\nVoici ce dont je suis capable :\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !new-year !newyear !ny : Affiche le temps restant avant la nouvelle année\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !end-of-world !worldend !eow : Temps restant avant la fin du monde\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !weekend !week-end !we : Affiche le temps restant avant le prochain week-end\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !partiels : Affiche le temps restant avant les prochains partiels\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !vacs !vacances !holidays !free-time : Affiche le temps restant avant les prochaines vacances\r\n"%(sender[0]))
            s.send("PRIVMSG %s :  - !jpo !next-jpo : Affiche le temps restant avant la prochaine JPO\r\n"%(sender[0]))

        if cmd[0] == 'new-year' or cmd[0] == 'newyear' or cmd[0] == 'ny':
            #What is the next year?
            nyear = datetime.today().year + 1
            ndate = datetime(nyear, 1, 1, 0, 0, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant la nouvelle année", "Nous faisons déjà la fête depuis%s"], cmd)
        if cmd[0] == 'end-of-world' or cmd[0] == 'endofworld' or cmd[0] == 'worldend' or cmd[0] == 'eow':
            ndate = datetime(2012, 12, 12, 12, 12, 13)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant la fin du monde", "Non, cela ne peut pas arriver, la fin du monde s'est produite il y a maintenant %s. Vous n'êtes vraiment pas mort ? :("], cmd)
        if cmd[0] == 'weekend' or cmd[0] == 'week-end' or cmd[0] == 'we':
            ndate = datetime.today() + timedelta(5 - datetime.today().weekday())
            ndate = datetime(ndate.year, ndate.month, ndate.day, 0, 0, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant le week-end, courrage ;)", "Youhou, on est en week-end depuis%s"], cmd)
        if cmd[0] == 'google' or cmd[0] == 'eog':
            ndate = datetime(2012, 3, 1, 0, 0, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s pour fermer tous ces comptes Google, hop hop hop, il y a du boulot !", "Nous ne pouvons plus utiliser les services de Google depuis%s"], cmd)
        if cmd[0] == 'endweekend' or cmd[0] == 'end-week-end' or cmd[0] == 'endwe' or cmd[0] == 'eowe':
            ndate = datetime.today() + timedelta(7 - datetime.today().weekday())
            ndate = datetime(ndate.year, ndate.month, ndate.day, 0, 0, 1)
            if datetime.today().weekday() >= 5:
                newyear.launch (s, info[2], ndate, ["Plus que%s avant la fin du week-end :(", ""], cmd)
            else:
                newyear.launch (s, info[2], ndate, ["Encore%s avant la prochaine semaine", ""], cmd)
        if cmd[0] == 'partiels' or cmd[0] == 'partiel':
            ndate = datetime(2012, 3, 26, 9, 0, 1)
            #ndate = datetime(2012, 1, 23, 9, 0, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant les partiels :-o", "Les partiels ont commencés depuis maintenant%s"], cmd)
        if cmd[0] == 'vacs' or cmd[0] == 'vacances' or cmd[0] == 'holiday' or cmd[0] == 'holidays' or cmd[0] == 'free-time':
            ndate = datetime(2012, 3, 30, 18, 0, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant les vacances :)", "Profitons, c'est les vacances depuis%s"], cmd)
        if cmd[0] == 'katy' or cmd[0] == 'album':
            ndate = datetime(2012, 3, 26, 8, 0, 0)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant la sortie du prochain album de Katy Perry :)", "Vite, courrons s'acheter le nouvel album de Katy Perry, il est sorti depuis%s"], cmd)
        if cmd[0] == 'anniv':
            if len(cmd) < 2 or cmd[1].lower() == "moi":
                name = sender[0].lower()
            else:
                name = cmd[1].lower()

            matches = []

            if name in birthdays:
                matches.append(name)
            else:
                for k in birthdays.keys():
                    if k.find(name) == 0:
                        matches.append(k)

            if len(matches) == 1:
                (n, d) = (matches[0], birthdays[matches[0]])
                tyd = d
                tyd = datetime(date.today().year, tyd.month, tyd.day)

                if tyd.day == datetime.today().day and tyd.month == datetime.today().month:
                    newyear.launch (s, info[2], d, ["", "C'est aujourd'hui l'anniversaire de %s ! Il a%s. Joyeux anniversaire :)" % (n, "%s")], cmd)
                else:
                    if tyd < datetime.today():
                        tyd = datetime(date.today().year + 1, tyd.month, tyd.day)

                    newyear.launch (s, info[2], tyd, ["Il reste%s avant l'anniversaire de %s !" % ("%s", n), ""], cmd)
            else:
                s.send("PRIVMSG %s :%s: désolé, je ne connais pas la date d'anniversaire de %s. Quand est-il né ?\r\n"%(info[2], sender[0], name))

        if cmd[0] == 'jpo' or cmd[0] == 'JPO' or cmd[0] == 'next-jpo':
            #ndate = datetime(2012, 5, 12, 8, 30, 1)
            #ndate = datetime(2012, 3, 31, 8, 30, 1)
            ndate = datetime(2012, 3, 17, 8, 30, 1)
            #ndate = datetime(2012, 2, 4, 8, 30, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant la prochaine JPO... We want you!", "Nous somme en pleine JPO depuis%s"], cmd)
        if cmd[0] == 'professional-project' or cmd[0] == 'project-professionnel' or cmd[0] == 'projet-professionnel' or cmd[0] == 'project-professionel' or cmd[0] == 'tc' or cmd[0] == 'next-rendu' or cmd[0] == 'rendu':
            ndate = datetime(2012, 3, 4, 23, 42, 1)
            newyear.launch (s, info[2], ndate, ["Il reste%s avant la fermeture du rendu de TC-4, vite au boulot !", "À %s près, il aurait encore été possible de rendre"], cmd)


    elif msgpart[0] == '`' and sender[0] == OWNER and CHANLIST.find(info[2]) != -1: #Treat all messages starting with '`' as command
        cmd=msgpart[1:].split(' ')
        if cmd[0]=='op':
            s.send("MODE %s +o %s\r\n" % (info[2], cmd[1]))
        if cmd[0]=='deop':
            s.send("MODE %s -o %s\r\n" % (info[2], cmd[1]))
        if cmd[0]=='voice':
            s.send('MODE '+info[2]+' +v '+cmd[1]+'n')
        if cmd[0]=='devoice':
            s.send('MODE '+info[2]+' -v '+cmd[1]+'n')
        if cmd[0]=='restart':
            print "Restarting thread"
            launch(s)
        if cmd[0]=='stop':
            print "Bye!"
            s.send("PRIVMSG %s :Bye!\r\n"%(info[2]))
            sys.exit (0)
        if cmd[0]=='sys':
            syscmd(msgpart[1:],info[2])

    if msgpart[0]=='-' and sender[0]==OWNER : #Treat msgs with - as explicit command to send to server
        cmd=msgpart[1:]
        #s.send(cmd+'n')
        print 'cmd='+cmd

def read():
    global s, readbuffer
    while 1:
        readbuffer = readbuffer + s.recv(1024) #recieve server messages
        temp = readbuffer.split("\n")
        readbuffer = temp.pop( )
        #signal.signal(signal.SIGHUP, onSignal)

        for line in temp:
            print line
            line = line.rstrip() #remove trailing 'rn'

            if line.find('PRIVMSG') != -1: #Call a parsing function
                parsemsg(line)

            line = line.split()

            if(line[0] == 'PING'): #If server pings then pong
                s.send("PONG %s\r\n" % line[1])

def launch(s):
#    thread.start_new_thread(ontime.startThread, (s,datetime(2012, 1, 18, 6, 0, 1),["Il reste%s avant la fin de Wikipédia", "C'est fini, Wikipédia a sombrée depuis%s"],CHANLIST))
    thread.start_new_thread(watchWebsite.startThread, (s, "you.p0m.fr", "", "#42sh", "Oh, quelle est cette nouvelle image sur http://you.p0m.fr/ ? :p"))
    thread.start_new_thread(watchWebsite.startThread, (s, "intra.nbr23.com", "", "#42sh", "Oh, quel est ce nouveau film sur http://intra.nbr23.com/ ? :p"))
    print "Launched successfuly"

launch(s)
read()