X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=hung.py;h=9582bc0fe27cc763ad154e15e7f23e361a8aa60e;hb=d8dc10f1a6c439f8f21b1fe0b1a044a6b517c609;hp=7f25389af3c85e6605bbcd487435b6c857599bae;hpb=5ffa82df2494fa37b81390f0d31f4d36b600e40a;p=bots%2Fhung.git

diff --git a/hung.py b/hung.py
index 7f25389..9582bc0 100755
--- a/hung.py
+++ b/hung.py
@@ -31,11 +31,15 @@ config_logfile_template="hung.%s.log"
 def get_config_logfile(serveur):
     serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
     return config_logfile_template%(serveurs[serveur])
-config_overops=["[20-100]","[20-100]_"]
+config_overops=["[20-100]","[20-100]_","Petite-Peste","PEB"]
 config_ops=[]
 config_report_bugs_to=["[20-100]"]
 
-config_dico_file="dico"
+config_dico_mots="mots.txt"
+config_dico_defs="definitions.txt"
+
+config_scores_file="scores.pickle"
+
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
@@ -57,11 +61,19 @@ def bot_unicode(chain):
     except UnicodeDecodeError as exc:
         raise UnicodeBotError
 
+def remplace_accents(chaine):
+    chaine=chaine.lower()
+    remplacements = {u"á":u"a",u"à":u"a",u"â":u"a",u"ä":u"a",u"é":u"e",u"è":u"e",u"ê":u"e",u"ë":u"e",u"í":u"i",u"ì":u"i",u"î":u"i",u"ï":u"i",u"ó":u"o",	u"ò":u"o",u"ô":u"o",u"ö":u"o",u"ú":u"u",u"ù":u"u",u"û":u"u",u"ü":u"u",u"ý":u"y",u"ỳ":u"y",u"ŷ":u"y",u"ÿ":u"y",u"œ":u"oe",u"æ":u"ae"}
+    for avant,apres in remplacements.items():
+        chaine=chaine.replace(avant,apres)
+    return chaine
+
+
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
         temporary_pseudo=config_irc_pseudo+str(random.randrange(10000,100000))
         ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
-                              temporary_pseudo,"Bot irc pour embêter Shadobot", 10)
+                              temporary_pseudo,"Bot irc pour jouer au pendu", 10)
         self.debug=debug
         self.serveur=serveur
         self.overops=config_overops
@@ -145,12 +157,12 @@ class Hung(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]]=[None]
+                        self.play_status[message[1]]=[None,None,None]
                         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:
@@ -161,10 +173,10 @@ class Hung(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=="stay":
@@ -197,7 +209,11 @@ class Hung(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]])))
+                    if self.play_status[k]==[None,None,None]:
+                        serv.privmsg(auteur,"None")
+                    else:
+                        serv.privmsg(auteur,"%s : %s (%s) [%s]"%(k,"".join([str(i[0]) for i in self.play_status[k][0]])
+                                                     ,self.play_status[k][1], self.play_status[k][2]))
         elif cmd=="die":
             if auteur in self.overops:
                 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
@@ -246,6 +262,8 @@ class Hung(ircbot.SingleServerIRCBot):
                 serv.privmsg(auteur,"Syntaxe : DO <channel> <action>")
             else:
                 notunderstood=True
+        elif cmd in ["score","scores"]:
+            self.send_scores(serv,auteur)
         else:
             notunderstood=True
         if notunderstood:
@@ -253,16 +271,18 @@ class Hung(ircbot.SingleServerIRCBot):
 
     def affiche_mot(self, serv, canal, begin="Mot courant"):
         if self.play_status.has_key(canal):
-            mot = self.play_status[canal]
+            mot = self.play_status[canal][0]
             obfuskated=" ".join([lettre[0] if lettre[1] else "_" for lettre in mot])
             serv.privmsg(canal,"%s : %s"%(begin,obfuskated))
 
     def start_partie(self, serv, canal):
-        dico=[mot.strip() for mot in open(config_dico_file).readlines()]
-        mot = random.choice(dico)
+        mots=[mot.strip() for mot in open(config_dico_mots).readlines()]
+        defs=[defi.strip() for defi in open(config_dico_defs).readlines()]
+        indice = random.randrange(0,len(mots))
+        mot,definition=mots[indice],defs[indice]
         # ' et - sont considérés comme déjà devinés
-        mot = [(lettre,lettre in "'-") for lettre in list(mot)]
-        self.play_status[canal]=mot
+        mot = [(lettre,lettre in "'-()") for lettre in list(mot)]
+        self.play_status[canal]=[mot,definition,{}]
         self.affiche_mot(serv, canal, begin="Devinez")
 
     def on_pubmsg(self, serv, ev):
@@ -300,23 +320,60 @@ class Hung(ircbot.SingleServerIRCBot):
                 else:
                     serv.privmsg(canal,"%s: Non, je reste !"%(auteur))
                     log(self.serveur,canal,auteur,message+"[failed]")
-            elif cmd in ["play","jeu","encore","again","partie","pendu","game","mot"]:
+            elif cmd in ["play","jeu","encore","again","partie","pendu","game","mot","go","allez"]:
                 if not canal in self.quiet_channels and canal in self.play_channels:
                     if self.play_status.has_key(canal):
-                        if self.play_status[canal]==[None]:
+                        if self.play_status[canal]==[None,None,None]:
                             self.start_partie(serv, canal)
                         else:
                             self.affiche_mot(serv, canal, begin="%s: Rappel"%(auteur))
                     else:
-                        self.play_status[canal]=[None]
+                        self.play_status[canal]=[None,None,None]
                         self.start_partie(serv, canal)
                 elif not canal in self.play_channels:
                     serv.privmsg(canal,"%s: pas ici…"%(auteur))
+            elif (cmd in list("azertyuiopqsdfghjklmwxcvbn") and canal in self.play_channels 
+                  and self.play_status.has_key(canal) and self.play_status[canal]!=[None,None,None]):
+                giv_let=cmd.upper()
+                liste=self.play_status[canal][0]
+                listeapres=[(lettre[0],lettre[1] or lettre[0]==giv_let) for lettre in liste]
+                if liste!=listeapres:
+                    nbtrouvees=(sum([lettre[1] for lettre in listeapres if not lettre[0] in "'-()"])
+                                - sum([lettre[1] for lettre in liste if not lettre[0] in "'-()"]))
+                    if self.play_status[canal][2].has_key(auteur):
+                        self.play_status[canal][2][auteur]+= nbtrouvees
+                    else:
+                        self.play_status[canal][2][auteur] = nbtrouvees
+                    self.play_status[canal][0]=listeapres
+                    self.affiche_mot(serv, canal, begin="%s placé"%(giv_let))
+                if all([lettre[1] for lettre in listeapres]):
+                    self.gagne(serv, canal)
+                    
+            elif cmd in ["score","scores","!score","!scores"]:
+                self.send_scores(serv,auteur)
             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))
+            # 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
         else:
             pass
 
+
     def on_action(self, serv, ev):
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
@@ -329,6 +386,45 @@ class Hung(ircbot.SingleServerIRCBot):
             return
         mypseudo=serv.get_nickname()
         
+    def get_scores(self):
+        f=open(config_scores_file)
+        scores=pickle.load(f)
+        f.close()
+        return scores
+    def save_scores(self,scores):
+        f=open(config_scores_file,'w')
+        pickle.dump(scores,f)
+        f.close()
+    def add_score(self,dico):
+        scores=self.get_scores()
+        for k,v in dico.items():
+            if scores.has_key(k):
+                scores[k]+=v
+            else:
+                scores[k]=v
+        self.save_scores(scores)
+    def send_scores(self, serv, destinataire):
+        scores=self.get_scores()
+        scores=scores.items()
+        scores.sort(lambda x,y:cmp(x[1],y[1]))
+        scores.reverse()
+        serv.privmsg(destinataire,"Scores by score : "+" ; ".join(["%s %s"%(k,v) for (k,v) in scores]) )
+        scores.sort(lambda x,y:cmp(x[0].lower(),y[0].lower()))
+        serv.privmsg(destinataire,"Scores by pseudo : "+" ; ".join(["%s %s"%(k,v) for (k,v) in scores]) )
+
+    def gagne(self, serv, canal, bonus=None, bonusvalue=2):
+        realword="".join([lettre[0] for lettre in self.play_status[canal][0]])
+        definition = self.play_status[canal][1]
+        serv.privmsg(canal,"Bravo ! C'était %s"%(realword))
+        serv.privmsg(canal,definition)
+        nlettre=float(len(realword.replace("'","").replace("-","")))
+        contribs=["%s:%s%%%s"%(pseudo,str(int(100*contrib/nlettre)),("+bonus(%s)"%(bonusvalue))*(bonus==pseudo)) for pseudo,contrib in self.play_status[canal][2].items()]
+        contribs_score={pseudo:int(10*contrib/nlettre) for pseudo,contrib in self.play_status[canal][2].items()}
+        self.add_score(contribs_score)
+        serv.privmsg(canal,"Contributions : %s"%("  ".join(contribs)) )
+        self.play_status[canal]=[None,None,None]
+
+
 if __name__=="__main__":
     import sys
     if len(sys.argv)==1: