X-Git-Url: http://gitweb.pimeys.fr/?a=blobdiff_plain;f=deconnaisseur.py;h=9726d66ae220149fecc7717b53778dbe776b6dea;hb=d9cc90569cdf10d4883dc00e6904f9529e27d111;hp=a6f31b5eaf49c589e4fb23337dfc8b978b9fe9d0;hpb=9d4849308b2c447f6cbb661883ab26d81a023f0e;p=bots%2Fdeconnaisseur.git

diff --git a/deconnaisseur.py b/deconnaisseur.py
index a6f31b5..9726d66 100755
--- a/deconnaisseur.py
+++ b/deconnaisseur.py
@@ -21,10 +21,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)
@@ -163,22 +164,29 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
             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
 
@@ -196,9 +204,8 @@ class Deconnaisseur(ircbot.SingleServerIRCBot):
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
-            if not canal in self.quiet_channels:
-                serv.privmsg(auteur,
-                  "Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…")
+            serv.privmsg(auteur,
+              "Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…")
             return
         message=message.split()
         cmd=message[0].lower()
@@ -439,7 +446,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()!=[]: