]> gitweb.pimeys.fr Git - bots/hung.git/blobdiff - hung.py
Fail float/int + compter le bonus qu'on annonce, c'est mieux
[bots/hung.git] / hung.py
diff --git a/hung.py b/hung.py
index 96c8a58a6e4b3dd020f36f5dfdbefb6a18778f45..2c3e7086dbb9955b8b5e970cf8e30a1a5b736911 100755 (executable)
--- a/hung.py
+++ b/hung.py
@@ -32,7 +32,7 @@ def get_config_logfile(serveur):
 
 def log(serveur,channel,auteur=None,message=None):
     f=open(get_config_logfile(serveur),"a")
-    if auteur==message==None:
+    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:
@@ -42,6 +42,17 @@ def log(serveur,channel,auteur=None,message=None):
         print chain
     f.close()
 
+def ignore_event(serv, ev):
+    """Retourne ``True`` si il faut ignorer cet évènement."""
+    for (blackmask, exceptmask) in config.blacklisted_masks:
+        usermask = ev.source()
+        if exceptmask is None:
+            exceptit = False
+        else:
+            exceptit = bool(irclib.mask_matches(usermask, exceptmask))
+        blackit = bool(irclib.mask_matches(usermask, blackmask))
+        if blackit and not exceptit:
+            return True
 
 class UnicodeBotError(Exception):
     pass
@@ -75,7 +86,7 @@ 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
+    return mot.startswith(real_word)
 
 class Hung(ircbot.SingleServerIRCBot):
     def __init__(self,serveur,debug=False):
@@ -91,6 +102,8 @@ class Hung(ircbot.SingleServerIRCBot):
         self.stay_channels=config.stay_channels
         self.play_channels=config.play_channels
         self.play_status={i:[None,None,None] for i in self.play_channels}
+        self.lives={}
+        self.tried_letters={}
         self.quiet_channels=config.quiet_channels
 
 
@@ -122,13 +135,15 @@ class Hung(ircbot.SingleServerIRCBot):
             return (False,message)
 
     def on_privmsg(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         message=ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
             if config.utf8_trigger:
-                serv.privmsg(auteur, config.utf8_fail_answers)
+                serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
             return
         message=message.split()
         cmd=message[0].lower()
@@ -192,8 +207,11 @@ class Hung(ircbot.SingleServerIRCBot):
             if auteur in self.ops:
                 if len(message)>1:
                     if message[1] in self.play_channels:
-                        serv.privmsg(auteur,"Je play déjà sur %s."%(message[1]))
-                        log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
+                        if len(message) > 2:
+                            self.start_partie(serv, message[1], " ".join(message[2:]))
+                        else:
+                            serv.privmsg(auteur,"Je play déjà sur %s."%(message[1]))
+                            log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
                     else:
                         self.play_channels.append(message[1])
                         self.play_status[message[1]]=[None,None,None]
@@ -317,25 +335,33 @@ class Hung(ircbot.SingleServerIRCBot):
             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):
-        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]
+    def start_partie(self, serv, canal, mot=None):
+        if mot is None:
+            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]
+        else:
+            definition = "(custom word)"
         # ' et - sont considérés comme déjà devinés
-        mot = [(lettre,lettre in "'-()") for lettre in list(mot)]
+        mot = mot.upper()
+        mot = [(lettre,lettre in config.non_guess_chars) for lettre in list(mot)]
         self.play_status[canal]=[mot,definition,{}]
+        self.tried_letters[canal] = set ()
+        self.lives[canal] = config.lives
         self.affiche_mot(serv, canal, begin="Devinez")
 
     def on_pubmsg(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         auteur = irclib.nm_to_n(ev.source())
         canal = ev.target()
         message = ev.arguments()[0]
         try:
             test=bot_unicode(message)
         except UnicodeBotError:
-            if not canal in self.quiet_channels and config.utf8_trigger:
-                serv.privmsg(canal, "%s: %s"%(auteur,config.utf8_fail_answers))
+            if config.utf8_trigger and not canal in self.quiet_channels:
+                serv.privmsg(canal, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
             return
         pour_moi,message=self.pourmoi(serv,message)
         if pour_moi and message.split()!=[]:
@@ -377,23 +403,6 @@ class Hung(ircbot.SingleServerIRCBot):
                         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:
@@ -408,8 +417,8 @@ class Hung(ircbot.SingleServerIRCBot):
                     for ligne in answer.split("\n"):
                         serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
             # on essaye de voir si le mot fourni matche la partie en cours
-            mot = cmd
-            if self.play_status[canal][0]!=None and is_mot(mot, self.play_status[canal][0]):
+            mot = message
+            if canal in self.play_channels and 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]])
@@ -419,11 +428,42 @@ class Hung(ircbot.SingleServerIRCBot):
                 else:
                     self.play_status[canal][2][auteur]=manquait
                 self.gagne(serv, canal, bonus=auteur, bonusvalue=manquait)
+                return
+            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))
+                else:
+                    if not giv_let in self.tried_letters[canal]:
+                        # On perd une chance
+                        self.lives[canal] -= 1
+                        if self.lives[canal] > 0:
+                            serv.privmsg(canal, "Pas de %s. Plus que %s chance%s…" % (giv_let, self.lives[canal], "s" * (self.lives[canal] > 1)))
+                        if self.lives[canal] == 0:
+                            serv.privmsg(canal, "Pas de %s." % (giv_let))
+                            self.perd(serv, canal)
+                            return
+                self.tried_letters[canal].add(giv_let)
+                if all([lettre[1] for lettre in listeapres]):
+                    self.gagne(serv, canal)
+                    
         else:
             pass
 
 
     def on_action(self, serv, ev):
+        if ignore_event(serv, ev):
+            return
         action = ev.arguments()[0]
         auteur = irclib.nm_to_n(ev.source())
         channel = ev.target()
@@ -487,17 +527,27 @@ class Hung(ircbot.SingleServerIRCBot):
         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("-","")))
+        serv.privmsg(canal,"Bravo !")
+        realword = self.reveal_word(serv, canal)
+        nlettre=float(len([l for l in realword if not l in config.non_guess_chars]))
         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()}
+        contribs_score={pseudo:int(10.0*contrib/nlettre)+(bonusvalue*(bonus==pseudo)) 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]
+
+    def reveal_word(self, serv, canal):
+        realword="".join([lettre[0] for lettre in self.play_status[canal][0]])
+        serv.privmsg(canal, "C'était %s." % (realword))
+        definition = self.play_status[canal][1]
+        serv.privmsg(canal,definition)
+        return realword
     
+    def perd(self, serv, canal):
+        serv.privmsg(canal,"Pendu !")
+        self.reveal_word(serv, canal)
+        self.play_status[canal]=[None,None,None]
+
     def reload(self, auteur=None):
         reload(config)
         if auteur in [None, "SIGHUP"]: