]> gitweb.pimeys.fr Git - bots/deconnaisseur.git/blobdiff - deconnaisseur.py
Des quit/leave messages
[bots/deconnaisseur.git] / deconnaisseur.py
index 0dc4639749f442e22e8277c312009ed2cc513740..c2bec86881f75bd7f41db5bf0d4458eabc668059 100755 (executable)
@@ -12,6 +12,7 @@ import random
 import time
 import pickle
 import re
+from remplace_accents import remplace_accents
 
 config_password="PatrickSébastien"
 config_pseudo="deconnaisseur"
@@ -21,10 +22,11 @@ config_stay_channels=["#flood","#bot"]
 config_overops=["[20-100]","[20-100]_","PEB"]
 config_ops=["Nit","Eguel","Harry"]
 
-config_source_file_template="deconnaissances.%s.txt" #il faut rajouter le nom du serveur
-def get_config_source_file(serveur):
+config_source_file="deconnaissances.txt"
+config_played_file_template="played.%s.txt" #il faut rajouter le nom du serveur
+def get_config_played_file(serveur):
     serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
-    return config_source_file_template%(serveurs[serveur])
+    return config_played_file_template%(serveurs[serveur])
 ttrig=120 #time trigger (normalement 120, mais diminué pour les tests)
 Ttrig=600 #between two enigms
 config_time_incompressible=15 #on peut pas retrigger en dessous de ce temps (60)
@@ -40,6 +42,19 @@ u"Héhé, try again",
 u"Non, j'ai pas envie",
 u"Peut-être quand toi tu la fermeras, et encore…"]
 
+config_level2=[]
+config_level3=["[20-100]","Petite-Peste","PEB"]
+
+config_debug_stdout = True
+config_logfile_template="deconnaisseur.%s.log"
+def get_config_logfile(serveur):
+    serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
+    return config_logfile_template%(serveurs[serveur])
+
+config_quit_messages=[u"Se décrocher la mâchoire, pas cuit bras d'oiseau ou métalld précieuxld"]
+
+config_leave_messages=[u"On continuera à jouer plus tard ;)"]
+
 class UnicodeBotError(Exception):
     pass
 def bot_unicode(chain):
@@ -48,29 +63,27 @@ def bot_unicode(chain):
     except UnicodeDecodeError:
         raise UnicodeBotError
 
-def log(channel,auteur=None,message=None):
-    #f=open(config_logfile,"a")
-    #if auteur==message==None:
-    #    chain=channel
-    #else:
-    #    chain="%s [%s:%s] %s"%(time.strftime("%T"),channel,auteur,message)
-    #f.write(chain+"\n")
-    #print chain
-    #f.close()
-    a=0 # does nothing
+def log(serveur,channel,auteur=None,message=None):
+    f=open(get_config_logfile(serveur),"a")
+    if auteur==message==None:
+        # alors c'est que c'est pas un channel mais juste une ligne de log
+        chain="%s %s"%(time.strftime("%F %T"),channel)
+    else:
+        chain="%s [%s:%s] %s"%(time.strftime("%F %T"),channel,auteur,message)
+    f.write(chain+"\n")
+    if config_debug_stdout:
+        print chain
+    f.close()
 
 
-def tolere(regexp):
-    """Renvoie une regexp plus tolérante"""
-    reg=unicode(regexp,"utf8").lower()
-    reg=reg.replace(u"á",u"(á|a)").replace(u"à",u"(à|a)").replace(u"â",u"(â|a)").replace(u"ä",u"(ä|a)")
-    reg=reg.replace(u"é",u"(é|e)").replace(u"è",u"(è|e)").replace(u"ê",u"(ê|e)").replace(u"ë",u"(ë|e)")
-    reg=reg.replace(u"í",u"(í|i)").replace(u"ì",u"(ì|i)").replace(u"î",u"(î|i)").replace(u"ï",u"(ï|i)")
-    reg=reg.replace(u"ó",u"(ó|o)").replace(u"ò",u"(ò|o)").replace(u"ô",u"(ô|o)").replace(u"ö",u"(ö|o)")
-    reg=reg.replace(u"ú",u"(ú|u)").replace(u"ù",u"(ù|u)").replace(u"û",u"(û|u)").replace(u"ü",u"(ü|u)")
-    reg=reg.replace(u"ý",u"(ý|y)").replace(u"ỳ",u"(ỳ|y)").replace(u"ŷ",u"(ŷ|y)").replace(u"ÿ",u"(ÿ|y)")
-    reg=reg.replace(u"œ",u"(œ|oe)").replace(u"æ",u"(æ|ae)")
-    return reg
+def reussi(message,answer,answer_regexp,auteur):
+    if auteur in config_level3:
+        return answer in message
+    if auteur in config_level2:
+        return remplace_accents(answer) in message
+    else:
+        if re.match(remplace_accents(answer_regexp).lower(),remplace_accents(message).lower()):
+            return True
 
 def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case_sensitive=False,debug=False):
     if case_sensitive:
@@ -101,7 +114,7 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         self.stay_channels=config_stay_channels
         self.play_channels=config_play_channels
         self.play_status={i:[0] for i in self.play_channels}
-        self.quiet_channels={}
+        self.quiet_channels=[]
 
     def give_me_my_pseudo(self,serv):
         serv.privmsg("NickServ","RECOVER %s %s"%(config_pseudo,config_password))
@@ -110,14 +123,15 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         serv.nick(config_pseudo)
     
     def on_welcome(self, serv, ev):
+        self.serv=serv # ça serv ira :)
         self.give_me_my_pseudo(serv)
         serv.privmsg("NickServ","identify %s"%(config_password))
-        log("Connected")
+        log(self.serveur,"Connected")
         if self.debug:
             self.chanlist=["#bot"]
             self.play_channels=["#bot"]
         for c in self.chanlist:
-            log("JOIN %s"%(c))
+            log(self.serveur,"JOIN %s"%(c))
             serv.join(c)
         for c in self.play_channels:
             token=time.time()-3600
@@ -136,8 +150,8 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                     ok="refuse"
             if ok=="do_it":
                 enigme,indice,answer_reg,answer=self.get_enigme()
-                print "%s; %s; %s; %s"%(enigme, indice, answer_reg, answer)
-                serv.privmsg(channel,enigme)
+                log(self.serveur,channel,u"$Énigme$".encode("utf8"),("%s; %s; %s; %s"%(enigme, indice, answer_reg, answer)).encode("utf8"))
+                serv.privmsg(channel,enigme.encode("utf8"))
                 token=time.time()
                 self.play_status[channel]=[1,enigme,indice,answer_reg,answer,token]
                 serv.execute_delayed(random.randrange(ttrig*3,ttrig*5),self.give_indice,(serv,channel,token))
@@ -151,39 +165,46 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                     token=self.play_status[channel][-1]
             if self.play_status[channel][-1]==token:
                 indice=self.play_status[channel][2]
-                serv.privmsg(channel,"indice : %s"%(indice))
+                serv.privmsg(channel,"indice : %s"%(indice).encode("utf8"))
                 self.play_status[channel][0]=2
                 serv.execute_delayed(random.randrange(ttrig*1,ttrig*3),self.give_answer,(serv,channel,token))
     def give_answer(self,serv,channel,token):
         if self.play_status[channel][0]==2 and self.play_status[channel][-1]==token:
             answer=self.play_status[channel][4]
-            serv.privmsg(channel,"C'était : %s"%(answer))
+            serv.privmsg(channel,"C'était : %s"%(answer).encode("utf8"))
             token=time.time()
             self.play_status[channel]=[0,token]
             serv.execute_delayed(random.randrange(Ttrig*5,Ttrig*10),self.start_enigme,(serv,channel,token))
 
     def get_enigme(self):
-        f=open(get_config_source_file(self.serveur))
+        # on récupère les déconnaissances
+        f=open(config_source_file)
         t=f.read()
+        f.close()
         l=re.findall("%\n(.*)\n(.*)\n(.*)\n(.*)\n(.*)\n",t)
-        l=[list(i) for i in l if len(i)==5]
-        l.sort(lambda x,y: cmp(int(x[4]),int(y[4])))
+        dec={int(i[0]):list(i[1:]) for i in l if len(i)==5}
+        # on va chercher combien de fois elles ont été jouées
+        played_file=get_config_played_file(self.serveur)
+        f=open(played_file)
+        t=f.read()
+        f.close()
+        l=re.findall("(.*):(.*)",t)
+        played={int(i[0]):int(i[1]) for i in l}
         # on récupère le nombre d'occurrences le plus faible
-        mini=l[0][4]
-        # on garde que ceux qui ont le même nombre d'occurrences
-        l_mini=[en for en in l if en[4]==mini]
-        # on tire au hasard dedans
-        choisi=random.randrange(len(l_mini))
-        enigme,indice,answer_reg,answer,_=l_mini[choisi]
-        real_index=l.index(l_mini[choisi])
-        l[real_index][4]=str(int(l[real_index][4])+1)
-        f=open(get_config_source_file(self.serveur),"w")
-        f.write("%\n"+"\n%\n".join(["%s\n%s\n%s\n%s\n%s"%(i[0],i[1],i[2],i[3],i[4]) for i in l])+"\n%")
+        mini=min(played.values())
+        # on choisit un id dans ceux qui ont ce nombre d'occurences
+        id_choisi=random.choice([k for k,v in played.items() if v==mini])
+        enigme,indice,answer_reg,answer=dec[id_choisi]
+        # on incrémente la choisie
+        played[id_choisi]+=1
+        # on enregistre le played_file
+        f=open(played_file,"w")
+        f.write("\n".join(["%-3s : %s"%(k,v) for k,v in played.items()]))
         f.close()
-        return enigme,indice,answer_reg,answer
+        return enigme.decode("utf8"),indice.decode("utf8"),answer_reg.decode("utf8"),answer.decode("utf8")
 
     def pourmoi(self, serv, message):
-        pseudo=serv.get_nickname()
+        pseudo=self.nick
         size=len(pseudo)
         if message[:size]==pseudo and len(message)>size and message[size]==":":
             return (True,message[size+1:].strip(" "))
@@ -237,7 +258,7 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                         serv.join(message[1])
                         self.chanlist.append(message[1])
                         serv.privmsg(auteur,"Channels : "+" ".join(self.chanlist))
-                        log("priv",auteur," ".join(message))
+                        log(self.serveur,"priv",auteur," ".join(message))
                 else:
                     serv.privmsg(auteur,"Channels : "+" ".join(self.chanlist))
             else:
@@ -246,12 +267,12 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
             if auteur in self.ops and len(message)>1:
                 if message[1] in self.chanlist:
                     if not (message[1] in self.stay_channels) or auteur in self.overops:
-                        serv.part(message[1])
+                        self.quitter(message[1]," ".join(message[2:]))
                         self.chanlist.remove(message[1])
-                        log("priv",auteur," ".join(message)+"[successful]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                     else:
                         serv.privmsg(auteur,"Non, je reste !")
-                        log("priv",auteur," ".join(message)+"[failed]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
                 else:
                     serv.privmsg(auteur,"Je ne suis pas sur %s"%(message[1]))
             else:
@@ -261,11 +282,11 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                 if len(message)>1:
                     if message[1] in self.stay_channels:
                         serv.privmsg(auteur,"Je stay déjà sur %s."%(message[1]))
-                        log("priv",auteur," ".join(message)+"[failed]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
                     else:
                         self.stay_channels.append(message[1])
                         serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
-                        log("priv",auteur," ".join(message)+"[successful]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                 else:
                     serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
             else:
@@ -276,10 +297,10 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                     if message[1] in self.stay_channels:
                         self.stay_channels.remove(message[1])
                         serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
-                        log("priv",auteur," ".join(message)+"[successful]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                     else:
                         serv.privmsg(auteur,"Je ne stay pas sur %s."%(message[1]))
-                        log("priv",auteur," ".join(message)+"[failed]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
             else:
                 notunderstood=True
         elif cmd=="play":
@@ -287,12 +308,12 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                 if len(message)>1:
                     if message[1] in self.play_channels:
                         serv.privmsg(auteur,"Je play déjà sur %s."%(message[1]))
-                        log("priv",auteur," ".join(message)+"[failed]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
                     else:
                         self.play_channels.append(message[1])
                         self.play_status[message[1]]=[0,time.time()-3600]
                         serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels))
-                        log("priv",auteur," ".join(message)+"[successful]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                 else:
                     serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels))
             else:
@@ -303,10 +324,10 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                     if message[1] in self.play_channels:
                         self.play_channels.remove(message[1])
                         serv.privmsg(auteur,"Play channels : "+" ".join(self.play_channels))
-                        log("priv",auteur," ".join(message)+"[successful]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                     else:
                         serv.privmsg(auteur,"Je ne play pas sur %s."%(message[1]))
-                        log("priv",auteur," ".join(message)+"[failed]")
+                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
             else:
                 notunderstood=True
         elif cmd=="quiet":
@@ -338,18 +359,19 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         elif cmd in ["states","status"]:
             if auteur in self.overops:
                 for k in self.play_status.keys():
-                    serv.privmsg(auteur,"%s : %s"%(k,"; ".join([str(i) for i in self.play_status[k]])))
+                    serv.privmsg(auteur,(u"%s : %s"%(k,"; ".join([unicode(i) for i in self.play_status[k]]))).encode("utf8") )
         elif cmd=="say":
             if auteur in self.overops and len(message)>2:
                 serv.privmsg(message[1]," ".join(message[2:]))
-                log("priv",auteur," ".join(message))
+                log(self.serveur,"priv",auteur," ".join(message))
             elif len(message)<=2:
                 serv.privmsg(auteur,"Syntaxe : SAY <channel> <message>")
             else:
                 notunderstood=True
         elif cmd=="die":
             if auteur in self.overops:
-                self.die()
+                log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+                self.mourir()
         elif cmd=="score":
             if len(message)>1:
                 if len(message) in [3,4] and message[1].lower()=="transfert":
@@ -438,7 +460,7 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
             if not canal in self.quiet_channels:
                 serv.privmsg(canal,
                    "%s: Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…"%(auteur))
-                return
+            return
         tryother=False
         pour_moi,message=self.pourmoi(serv,message)
         if pour_moi and message.split()!=[]:
@@ -449,29 +471,28 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                 args=""
             if cmd in ["meurs","die","crève"]:
                 if auteur in self.overops:
-                    self.die()
-                    log(canal,auteur,message+"[successful]")
+                    self.mourir()
+                    log(self.serveur,canal,auteur,message+"[successful]")
                 else:
                     serv.privmsg(canal,"%s: crève !"%(auteur))
-                    log(canal,auteur,message+"[failed]")
+                    log(self.serveur,canal,auteur,message+"[failed]")
             if cmd in ["meur", "meurt","meurre","meurres"] and not canal in self.quiet_channels:
                 serv.privmsg(canal,'%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)'%(auteur))
             if cmd in ["part","leave","dégage"]:
                 if auteur in self.ops and (not (canal in self.stay_channels)
                                            or auteur in self.overops):
-                    serv.part(canal,message="Éjecté par %s"%(auteur))
-                    log(canal,auteur,message+"[successful]")
+                    self.quitter(canal)
+                    log(self.serveur,canal,auteur,message+"[successful]")
                     self.chanlist.remove(canal)
                 else:
                     serv.privmsg(canal,"%s: Non, je reste !"%(auteur))
-                    log(canal,auteur,message+"[failed]")
+                    log(self.serveur,canal,auteur,message+"[failed]")
     
             if cmd in ["deviens","pseudo"]:
                 if auteur in self.ops:
                     become=args
                     serv.nick(become)
-                    log(canal,auteur,message+"[successful]")
-    
+                    log(self.serveur,canal,auteur,message+"[successful]")
             if cmd in ["coucou"] and not canal in self.quiet_channels:
                 serv.privmsg(canal,"%s: coucou"%(auteur))
             if cmd in ["ping"] and not canal in self.quiet_channels:
@@ -484,9 +505,20 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
                         except RefuseError:
                             serv.privmsg(canal,"%s: Je peux souffler une minute ?"%(auteur))
                     else:
-                        serv.privmsg(canal,"%s: Rappel : %s"%(auteur,self.play_status[canal][1]))
+                        serv.privmsg(canal,("%s: Rappel : %s"%(auteur,self.play_status[canal][1])).encode("utf8") )
                 else:
                     serv.privmsg(canal,"%s: pas ici…"%(auteur))
+            if cmd in ["score","!score"]:
+                serv.privmsg(auteur,"Votre score : %s"%(self.get_scores().get(auteur,0)) )
+            if cmd in ["scores","!scores"]:
+                scores=self.get_scores().items()
+                # trie par score
+                scores.sort(lambda x,y:cmp(x[1],y[1]))
+                scores.reverse()
+                serv.privmsg(auteur,"Scores by score : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores]))
+                # trie par pseudo
+                scores.sort(lambda x,y:cmp(x[0].lower(),y[0].lower()))
+                serv.privmsg(auteur,"Scores by pseudo : "+" ; ".join(["%s %s"%(i[0],i[1]) for i in scores]))
             if cmd=="indice" and canal in self.play_channels:
                 self.give_indice(serv,canal,None)
             if is_tag(message) and not canal in self.quiet_channels:
@@ -505,13 +537,42 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         if tryother:
             if self.play_status.get(canal,[-1])[0] in [1,2]:
                 answer_regexp=self.play_status[canal][3]
-                if re.match(tolere(answer_regexp),unicode(message,"utf8").lower()):
-                    answer=self.play_status[canal][4]
-                    serv.privmsg(canal,"%s: bravo ! (C'était %s)"%(auteur,answer))
+                answer=self.play_status[canal][4]
+                if reussi(message.decode("utf8"),answer,answer_regexp,auteur):
+                    serv.privmsg(canal,(u"%s: bravo ! (C'était %s)"%(auteur,answer)).encode("utf8"))
+                    log(self.serveur,canal,auteur+"$win",message)
                     self.add_score(auteur,1)
                     token=time.time()
                     self.play_status[canal]=[0,token]
                     serv.execute_delayed(random.randrange(Ttrig*5,Ttrig*10),self.start_enigme,(serv,canal,token))
+    
+    def on_kick(self,serv,ev):
+        auteur = irclib.nm_to_n(ev.source())
+        channel = ev.target()
+        victime = ev.arguments()[0]
+        raison = ev.arguments()[1]
+        if victime==self.nick:
+            log(self.serveur,"%s kické par %s (raison : %s)" %(victime,auteur,raison))
+            time.sleep(5)
+            serv.join(channel)
+            # on ne dit rien au rejoin
+            #l1,l2=config_kick_answers,config_kick_actions
+            #n1,n2=len(l1),len(l2)
+            #i=random.randrange(n1+n2)
+            #if i>=n1:
+            #    serv.action(channel,l2[i-n1].format(auteur).encode("utf8"))
+            #else:
+            #    serv.privmsg(channel,l1[i].format(auteur).encode("utf8"))
+    
+    def quitter(self,chan,leave_message=None):
+        if leave_message==None:
+            leave_message=random.choice(config_leave_messages)
+        self.serv.part(chan,leave_message.encode("utf8"))
+    
+    def mourir(self):
+        quit_message=random.choice(config_quit_messages)
+        self.die(message=config_quit_message)
+    
     def get_scores(self):
         f=open(config_score_file)
         scores=pickle.load(f)
@@ -530,6 +591,10 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         f=open(config_score_file,"w")
         pickle.dump(scores,f)
         f.close()
+    
+    def _getnick(self):
+        return self.serv.get_nickname()
+    nick = property(_getnick)
 
 if __name__=="__main__":
     import sys