Plantait si on lui proposait un mot en dehors d'une partie

[bots/hung.git] / hung.py

diff --git a/hung.py b/hung.py
index b849f7f9ada7a7c69247d42a72cbd9a1036be303..c96f834e5b994f84afb4364babf221ded7a42ad9 100755 (executable)
--- a/hung.py
+++ b/hung.py
@@ -29,7 +29,7 @@ config_play_channels=["#flood"]
 config_quiet_channels=[]
 config_logfile_template="hung.%s.log"
 def get_config_logfile(serveur):
-    serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
+    serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans","localhost":"localhost"}
     return config_logfile_template%(serveurs[serveur])
 config_overops=["[20-100]","[20-100]_","Petite-Peste","PEB"]
 config_ops=[]
@@ -60,6 +60,7 @@ u"B _ E, C R U E L   _ O R L D!",
 u"B Y E, C R U E L   _ O R L D!",
 u"B Y E, C R U E L   W O R L D!",
 ]
+config_leave_messages=["On finira la partie plus tard :p"]
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
@@ -102,6 +103,12 @@ def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case
 def is_tag(chain):
     return is_something(chain,config_tag_triggers)
 
+def is_mot(mot,liste):
+    real_word = "".join([lettre[0] for lettre in liste])
+    real_word = real_word.decode("utf8").lower()
+    mot=remplace_accents(mot.decode("utf8"))
+    return mot==real_word
+
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
         temporary_pseudo=config_irc_pseudo+str(random.randrange(10000,100000))
@@ -115,7 +122,7 @@ class Hung(ircbot.SingleServerIRCBot):
         self.chanlist=config_chanlist
         self.stay_channels=config_stay_channels
         self.play_channels=config_play_channels
-        self.play_status={}
+        self.play_status={i:[None,None,None] for i in self.play_channels}
         self.quiet_channels=config_quiet_channels
 
 
@@ -126,6 +133,7 @@ class Hung(ircbot.SingleServerIRCBot):
         serv.nick(config_irc_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_irc_password))
         log(self.serveur,"Connected")
@@ -175,7 +183,7 @@ class Hung(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(self.serveur,"priv",auteur," ".join(message)+"[successful]")
                     else:
@@ -250,8 +258,7 @@ class Hung(ircbot.SingleServerIRCBot):
         elif cmd=="die":
             if auteur in self.overops:
                 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
-                quit_message=random.choice(config_quit_messages)
-                self.die(message=config_quit_message)
+                self.mourir()
             else:
                 notunderstood=True
         elif cmd=="quiet":
@@ -340,14 +347,14 @@ class Hung(ircbot.SingleServerIRCBot):
             if cmd in ["meurs","die","crève"]:
                 if auteur in self.overops:
                     log(self.serveur,canal,auteur,message+"[successful]")
-                    self.die()
+                    self.mourir()
                 else:
                     serv.privmsg(canal,"%s: crève !"%(auteur))
                     log(self.serveur,canal,auteur,message+"[failed]")
             elif 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))
+                    self.quitter(canal)
                     log(self.serveur,canal,auteur,message+"[successful]")
                     if canal in self.chanlist:
                         self.chanlist.remove(canal)
@@ -398,21 +405,16 @@ class Hung(ircbot.SingleServerIRCBot):
                         serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
             # on essaye de voir si le mot fourni matche la partie en cours
             mot = cmd
-            # bon, ce teste merde et j'arrive pas à trouver pourquoi, alors j'ai craqué, je l'ai mis dans un try
-            # bouh ! beurk ! pas bien ! promis, j'irai me flageller…
-            try:
-                if remplace_accents(mot)==("".join([i[0] for i in self.play_status[canal][0]])).lower():
-                    # on a trouvé le mot
-                    # on regarde combien de lettre il manquait
-                    manquait = sum([not lettre[1] for lettre in self.play_status[canal][0]])
-                    self.add_score({auteur: manquait})
-                    if self.play_status[canal][2].has_key(auteur):
-                        self.play_status[canal][2][auteur]+=manquait
-                    else:
-                        self.play_status[canal][2][auteur]=manquait
-                    self.gagne(serv, canal, bonus=auteur, bonusvalue=manquait)
-            except:
-                pass
+            if self.play_status[canal][0]!=None and is_mot(mot, self.play_status[canal][0]):
+                # on a trouvé le mot
+                # on regarde combien de lettre il manquait
+                manquait = sum([not lettre[1] for lettre in self.play_status[canal][0]])
+                self.add_score({auteur: manquait})
+                if self.play_status[canal][2].has_key(auteur):
+                    self.play_status[canal][2][auteur]+=manquait
+                else:
+                    self.play_status[canal][2][auteur]=manquait
+                self.gagne(serv, canal, bonus=auteur, bonusvalue=manquait)
         else:
             pass
 
@@ -428,7 +430,16 @@ class Hung(ircbot.SingleServerIRCBot):
               "%s : Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…"%(auteur))
             return
         mypseudo=serv.get_nickname()
-        
+    
+    def quitter(self,chan,leave_message=None):
+        if leave_message==None:
+            leave_message=random.choice(config_leave_messages)
+        self.serv.part(chan,message=leave_message.encode("utf8"))
+    
+    def mourir(self):
+        quit_message=random.choice(config_quit_messages)
+        self.die(msg=quit_message.encode("utf8"))
+    
     def get_scores(self):
         f=open(config_scores_file)
         scores=pickle.load(f)
@@ -479,7 +490,8 @@ if __name__=="__main__":
     else:
         debug=False
     serveurs={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
-              "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
+              "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org",
+              "localhost":"localhost"}
     try:
         serveur=serveurs[serveur]
     except KeyError: