]>
gitweb.pimeys.fr Git - bots/hung.git/blob - hung.py
4 # Codé par 20-100 le 23/04/12
6 # Un test de bot irc, parce que c'est cool
13 import socket
, ssl
, json
17 from commands
import getstatusoutput
as ex
20 config_debug_stdout
= True
21 if "--quiet" in sys
. argv
:
22 config_debug_stdout
= False
24 config_irc_password
= "I'mAHungMan"
25 config_irc_pseudo
= "Hung"
26 config_chanlist
=[ "#bot" , "#flood" ]
27 config_stay_channels
=[ "#bot" , "#flood" ]
28 config_play_channels
=[ "#flood" ]
29 config_quiet_channels
=[]
30 config_logfile_template
= "hung. %s .log"
31 def get_config_logfile ( serveur
):
32 serveurs
={ "acoeur.crans.org" : "acoeur" , "irc.crans.org" : "crans" }
33 return config_logfile_template
%( serveurs
[ serveur
])
34 config_overops
=[ "[20-100]" , "[20-100]_" , "Petite-Peste" , "PEB" ]
36 config_report_bugs_to
=[ "[20-100]" ]
38 config_dico_mots
= "mots.txt"
39 config_dico_defs
= "definitions.txt"
41 config_scores_file
= "scores.pickle"
43 config_tag_triggers
=[ u
"t(|a)g" , u
"ta gueule" , u
"la ferme" , u
"ferme( |-)la" , u
"tais-toi" , u
"chut" ]
44 config_tag_actions
=[ u
"se tait" , u
"ferme sa gueule" , u
"se la ferme" , u
"la ferme" ]
45 config_tag_answers
=[ u
"J'me tais si j'veux !" ,
46 u
"Je t'entends pas :°" ,
48 u
"Non, j'ai pas envie" ,
49 u
"Peut-être quand toi tu la fermeras, et encore…" ]
51 config_quit_messages
=[ u
"_ _ _, _ _ _ _ _ _ _ _ _ _!" ,
52 u
"_ _ E, _ _ _ E _ _ _ _ _ _!" ,
53 u
"_ _ E, _ _ _ E _ _ O _ _ _!" ,
54 u
"_ _ E, _ _ U E _ _ O _ _ _!" ,
55 u
"_ _ E, _ R U E _ _ O R _ _!" ,
56 u
"_ _ E, _ R U E L _ O R L _!" ,
57 u
"B _ E, _ R U E L _ O R L _!" ,
58 u
"B _ E, C R U E L _ O R L _!" ,
59 u
"B _ E, C R U E L _ O R L D!" ,
60 u
"B Y E, C R U E L _ O R L D!" ,
61 u
"B Y E, C R U E L W O R L D!" ,
63 config_leave_messages
=[ "On finira la partie plus tard :p" ]
65 def log ( serveur
, channel
, auteur
= None , message
= None ):
66 f
= open ( get_config_logfile ( serveur
), "a" )
67 if auteur
== message
== None :
68 # alors c'est que c'est pas un channel mais juste une ligne de log
69 chain
= " %s %s " %( time
. strftime ( " %F %T" ), channel
)
71 chain
= " %s [ %s : %s ] %s " %( time
. strftime ( " %F %T" ), channel
, auteur
, message
)
73 if config_debug_stdout
:
78 class UnicodeBotError ( Exception ):
80 def bot_unicode ( chain
):
83 except UnicodeDecodeError as exc
:
86 def remplace_accents ( chaine
):
88 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" }
89 for avant
, apres
in remplacements
. items ():
90 chaine
= chaine
. replace ( avant
, apres
)
93 def is_something ( chain
, matches
, avant
= u
".*(?:^| )" , apres
= u
"(?:$|\.| |,|;).*" , case_sensitive
= False , debug
= False ):
95 chain
= unicode ( chain
, "utf8" )
97 chain
= unicode ( chain
, "utf8" ). lower ()
98 allmatches
= "(" + "|" . join ( matches
)+ ")"
99 reg
=( avant
+ allmatches
+ apres
). lower ()
100 o
= re
. match ( reg
, chain
)
104 return is_something ( chain
, config_tag_triggers
)
106 class Hung ( ircbot
. SingleServerIRCBot
):
107 def __init__ ( self
, serveur
, debug
= False ):
108 temporary_pseudo
= config_irc_pseudo
+ str ( random
. randrange ( 10000 , 100000 ))
109 ircbot
. SingleServerIRCBot
.__ init
__ ( self
, [( serveur
, 6667 )],
110 temporary_pseudo
, "Bot irc pour jouer au pendu" , 10 )
113 self
. overops
= config_overops
114 self
. ops
= self
. overops
+ config_ops
115 self
. report_bugs_to
= config_report_bugs_to
116 self
. chanlist
= config_chanlist
117 self
. stay_channels
= config_stay_channels
118 self
. play_channels
= config_play_channels
120 self
. quiet_channels
= config_quiet_channels
123 def give_me_my_pseudo ( self
, serv
):
124 serv
. privmsg ( "NickServ" , "RECOVER %s %s " %( config_irc_pseudo
, config_irc_password
))
125 serv
. privmsg ( "NickServ" , "RELEASE %s %s " %( config_irc_pseudo
, config_irc_password
))
127 serv
. nick ( config_irc_pseudo
)
129 def on_welcome ( self
, serv
, ev
):
130 self
. give_me_my_pseudo ( serv
)
131 serv
. privmsg ( "NickServ" , "IDENTIFY %s " %(config_irc_password))
132 log ( self
. serveur
, "Connected" )
134 self
. chanlist
=[ "#bot" ]
135 self
. play_channels
=[ "#bot" ]
136 for c
in self
. chanlist
:
137 log ( self
. serveur
, "JOIN %s " %(c))
140 def pourmoi ( self
, serv
, message
):
141 """renvoie (False,lemessage) ou (True, le message amputé de "pseudo: ")"""
142 pseudo
= serv
. get_nickname ()
144 if message
[: size
]== pseudo
and len ( message
)> size
and message
[ size
]== ":" :
145 return ( True , message
[ size
+ 1 :]. lstrip ( " " ))
147 return ( False , message
)
149 def on_privmsg ( self
, serv
, ev
):
150 message
= ev
. arguments ()[ 0 ]
151 auteur
= irclib
. nm_to_n ( ev
. source ())
153 test
= bot_unicode ( message
)
154 except UnicodeBotError
:
156 "Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…" )
158 message
= message
. split ()
159 cmd
= message
[ 0 ]. lower ()
162 if auteur
in self
. ops
:
164 if message
[ 1 ] in self
. chanlist
:
165 serv
. privmsg ( auteur
, "Je suis déjà sur %s " %( message
[ 1 ]))
167 serv
. join ( message
[ 1 ])
168 self
. chanlist
. append ( message
[ 1 ])
169 serv
. privmsg ( auteur
, "Channels : " + " " . join ( self
. chanlist
))
170 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
))
172 serv
. privmsg ( auteur
, "Channels : " + " " . join ( self
. chanlist
))
176 if auteur
in self
. ops
and len ( message
)> 1 :
177 if message
[ 1 ] in self
. chanlist
:
178 if not ( message
[ 1 ] in self
. stay_channels
) or auteur
in self
. overops
:
179 self
. quitter ( message
[ 1 ], " " . join ( message
[ 2 :]))
180 self
. chanlist
. remove ( message
[ 1 ])
181 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
183 serv
. privmsg ( auteur
, "Non, je reste !" )
184 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
186 serv
. privmsg ( auteur
, "Je ne suis pas sur %s " %( message
[ 1 ]))
190 if auteur
in self
. ops
:
192 if message
[ 1 ] in self
. play_channels
:
193 serv
. privmsg ( auteur
, "Je play déjà sur %s ." %( message
[ 1 ]))
194 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
196 self
. play_channels
. append ( message
[ 1 ])
197 self
. play_status
[ message
[ 1 ]]=[ None , None , None ]
198 serv
. privmsg ( auteur
, "Play channels : " + " " . join ( self
. play_channels
))
199 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
201 serv
. privmsg ( auteur
, "Play channels : " + " " . join ( self
. play_channels
))
205 if auteur
in self
. ops
:
207 if message
[ 1 ] in self
. play_channels
:
208 self
. play_channels
. remove ( message
[ 1 ])
209 serv
. privmsg ( auteur
, "Play channels : " + " " . join ( self
. play_channels
))
210 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
212 serv
. privmsg ( auteur
, "Je ne play pas sur %s ." %( message
[ 1 ]))
213 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
217 if auteur
in self
. overops
:
219 if message
[ 1 ] in self
. stay_channels
:
220 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
221 serv
. privmsg ( auteur
, "Je stay déjà sur %s ." %( message
[ 1 ]))
223 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
224 self
. stay_channels
. append ( message
[ 1 ])
225 serv
. privmsg ( auteur
, "Stay channels : " + " " . join ( self
. stay_channels
))
227 serv
. privmsg ( auteur
, "Stay channels : " + " " . join ( self
. stay_channels
))
231 if auteur
in self
. overops
:
233 if message
[ 1 ] in self
. stay_channels
:
234 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
235 self
. stay_channels
. remove ( message
[ 1 ])
236 serv
. privmsg ( auteur
, "Stay channels : " + " " . join ( self
. stay_channels
))
238 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
239 serv
. privmsg ( auteur
, "Je ne stay pas sur %s ." %( message
[ 1 ]))
243 elif cmd
in [ "states" , "status" ]:
244 if auteur
in self
. overops
:
245 for k
in self
. play_status
. keys ():
246 if self
. play_status
[ k
]==[ None , None , None ]:
247 serv
. privmsg ( auteur
, "None" )
249 serv
. privmsg ( auteur
, " %s : %s ( %s ) [ %s ]" %( k
, "" . join ([ str ( i
[ 0 ]) for i
in self
. play_status
[ k
][ 0 ]])
250 , self
. play_status
[ k
][ 1 ], self
. play_status
[ k
][ 2 ]))
252 if auteur
in self
. overops
:
253 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
258 if auteur
in self
. ops
:
260 if message
[ 1 ] in self
. quiet_channels
:
261 serv
. privmsg ( auteur
, "Je me la ferme déjà sur %s " %( message
[ 1 ]))
262 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
264 self
. quiet_channels
. append ( message
[ 1 ])
265 serv
. privmsg ( auteur
, "Quiet channels : " + " " . join ( self
. quiet_channels
))
266 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
268 serv
. privmsg ( auteur
, "Quiet channels : " + " " . join ( self
. quiet_channels
))
272 if auteur
in self
. ops
:
274 if message
[ 1 ] in self
. quiet_channels
:
275 self
. quiet_channels
. remove ( message
[ 1 ])
276 serv
. privmsg ( auteur
, "Quiet channels : " + " " . join ( self
. quiet_channels
))
277 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[successful]" )
279 serv
. privmsg ( auteur
, "Je ne me la ferme pas sur %s ." %( message
[ 1 ]))
280 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
)+ "[failed]" )
284 if auteur
in self
. overops
and len ( message
)> 2 :
285 serv
. privmsg ( message
[ 1 ], " " . join ( message
[ 2 :]))
286 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
))
287 elif len ( message
)<= 2 :
288 serv
. privmsg ( auteur
, "Syntaxe : SAY <channel> <message>" )
292 if auteur
in self
. overops
and len ( message
)> 2 :
293 serv
. action ( message
[ 1 ], " " . join ( message
[ 2 :]))
294 log ( self
. serveur
, "priv" , auteur
, " " . join ( message
))
295 elif len ( message
)<= 2 :
296 serv
. privmsg ( auteur
, "Syntaxe : DO <channel> <action>" )
299 elif cmd
in [ "score" , "scores" ]:
300 self
. send_scores ( serv
, auteur
)
304 serv
. privmsg ( auteur
, "Je n'ai pas compris. Essaye HELP…" )
306 def affiche_mot ( self
, serv
, canal
, begin
= "Mot courant" ):
307 if self
. play_status
. has_key ( canal
):
308 mot
= self
. play_status
[ canal
][ 0 ]
309 obfuskated
= " " . join ([ lettre
[ 0 ] if lettre
[ 1 ] else "_" for lettre
in mot
])
310 serv
. privmsg ( canal
, " %s : %s " %( begin
, obfuskated
))
312 def start_partie ( self
, serv
, canal
):
313 mots
=[ mot
. strip () for mot
in open ( config_dico_mots
). readlines ()]
314 defs
=[ defi
. strip () for defi
in open ( config_dico_defs
). readlines ()]
315 indice
= random
. randrange ( 0 , len ( mots
))
316 mot
, definition
= mots
[ indice
], defs
[ indice
]
317 # ' et - sont considérés comme déjà devinés
318 mot
= [( lettre
, lettre
in "'-()" ) for lettre
in list ( mot
)]
319 self
. play_status
[ canal
]=[ mot
, definition
,{}]
320 self
. affiche_mot ( serv
, canal
, begin
= "Devinez" )
322 def on_pubmsg ( self
, serv
, ev
):
323 auteur
= irclib
. nm_to_n ( ev
. source ())
325 message
= ev
. arguments ()[ 0 ]
327 test
= bot_unicode ( message
)
328 except UnicodeBotError
:
329 if not canal
in self
. quiet_channels
:
331 " %s : Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…" %(auteur))
333 pour_moi
, message
= self
. pourmoi ( serv
, message
)
334 if pour_moi
and message
. split ()!=[]:
335 cmd
= message
. split ()[ 0 ]. lower ()
337 args
= " " . join ( message
. split ()[ 1 :])
340 if cmd
in [ "meurs" , "die" , "crève" ]:
341 if auteur
in self
. overops
:
342 log ( self
. serveur
, canal
, auteur
, message
+ "[successful]" )
345 serv
. privmsg ( canal
, " %s : crève !" %(auteur))
346 log ( self
. serveur
, canal
, auteur
, message
+ "[failed]" )
347 elif cmd
in [ "part" , "leave" , "dégage" ]:
348 if auteur
in self
. ops
and ( not ( canal
in self
. stay_channels
)
349 or auteur
in self
. overops
):
351 log ( self
. serveur
, canal
, auteur
, message
+ "[successful]" )
352 if canal
in self
. chanlist
:
353 self
. chanlist
. remove ( canal
)
355 serv
. privmsg ( canal
, " %s : Non, je reste !" %(auteur))
356 log ( self
. serveur
, canal
, auteur
, message
+ "[failed]" )
357 elif cmd
in [ "play" , "jeu" , "encore" , "again" , "partie" , "pendu" , "game" , "mot" , "go" , "allez" ]:
358 if not canal
in self
. quiet_channels
and canal
in self
. play_channels
:
359 if self
. play_status
. has_key ( canal
):
360 if self
. play_status
[ canal
]==[ None , None , None ]:
361 self
. start_partie ( serv
, canal
)
363 self
. affiche_mot ( serv
, canal
, begin
= " %s : Rappel" %(auteur))
365 self
. play_status
[ canal
]=[ None , None , None ]
366 self
. start_partie ( serv
, canal
)
367 elif not canal
in self
. play_channels
:
368 serv
. privmsg ( canal
, " %s : pas ici…" %(auteur))
369 elif ( cmd
in list ( "azertyuiopqsdfghjklmwxcvbn" ) and canal
in self
. play_channels
370 and self
. play_status
. has_key ( canal
) and self
. play_status
[ canal
]!=[ None , None , None ]):
372 liste
= self
. play_status
[ canal
][ 0 ]
373 listeapres
=[( lettre
[ 0 ], lettre
[ 1 ] or lettre
[ 0 ]== giv_let
) for lettre
in liste
]
374 if liste
!= listeapres
:
375 nbtrouvees
=( sum ([ lettre
[ 1 ] for lettre
in listeapres
if not lettre
[ 0 ] in "'-()" ])
376 - sum ([ lettre
[ 1 ] for lettre
in liste
if not lettre
[ 0 ] in "'-()" ]))
377 if self
. play_status
[ canal
][ 2 ]. has_key ( auteur
):
378 self
. play_status
[ canal
][ 2 ][ auteur
]+= nbtrouvees
380 self
. play_status
[ canal
][ 2 ][ auteur
] = nbtrouvees
381 self
. play_status
[ canal
][ 0 ]= listeapres
382 self
. affiche_mot ( serv
, canal
, begin
= " %s placé" %(giv_let))
383 if all ([ lettre
[ 1 ] for lettre
in listeapres
]):
384 self
. gagne ( serv
, canal
)
386 elif cmd
in [ "score" , "scores" , "!score" , "!scores" ]:
387 self
. send_scores ( serv
, auteur
)
388 if cmd
in [ "meur" , "meurt" , "meurre" , "meurres" ] and not canal
in self
. quiet_channels
:
389 serv
. privmsg ( canal
, ' %s : Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)' %(auteur))
390 if is_tag ( message
) and not canal
in self
. quiet_channels
:
391 if auteur
in self
. ops
:
392 action
= random
. choice ( config_tag_actions
)
393 serv
. action ( canal
, action
. encode ( "utf8" ))
394 self
. quiet_channels
. append ( canal
)
396 answer
= random
. choice ( config_tag_answers
)
397 for ligne
in answer
. split ( " \n " ):
398 serv
. privmsg ( canal
, " %s : %s " %( auteur
, ligne
. encode ( "utf8" )))
399 # on essaye de voir si le mot fourni matche la partie en cours
401 # bon, ce teste merde et j'arrive pas à trouver pourquoi, alors j'ai craqué, je l'ai mis dans un try
402 # bouh ! beurk ! pas bien ! promis, j'irai me flageller…
404 if remplace_accents ( mot
)==( "" . join ([ i
[ 0 ] for i
in self
. play_status
[ canal
][ 0 ]])). lower ():
406 # on regarde combien de lettre il manquait
407 manquait
= sum ([ not lettre
[ 1 ] for lettre
in self
. play_status
[ canal
][ 0 ]])
408 self
. add_score ({ auteur
: manquait
})
409 if self
. play_status
[ canal
][ 2 ]. has_key ( auteur
):
410 self
. play_status
[ canal
][ 2 ][ auteur
]+= manquait
412 self
. play_status
[ canal
][ 2 ][ auteur
]= manquait
413 self
. gagne ( serv
, canal
, bonus
= auteur
, bonusvalue
= manquait
)
420 def on_action ( self
, serv
, ev
):
421 action
= ev
. arguments ()[ 0 ]
422 auteur
= irclib
. nm_to_n ( ev
. source ())
423 channel
= ev
. target ()
425 test
= bot_unicode ( action
)
426 except UnicodeBotError
:
427 serv
. privmsg ( channel
,
428 " %s : Euh, tu fais de la merde avec ton encodage là, j'ai failli crasher…" %(auteur))
430 mypseudo
= serv
. get_nickname ()
432 def quitter ( self
, chan
, leave_message
= None ):
433 if leave_message
== None :
434 leave_message
= random
. choice ( config_leave_messages
)
435 self
. serv
. part ( chan
, leave_message
. encode ( "utf8" ))
438 quit_message
= random
. choice ( config_quit_messages
)
439 self
. die ( message
= config_quit_message
)
441 def get_scores ( self
):
442 f
= open ( config_scores_file
)
443 scores
= pickle
. load ( f
)
446 def save_scores ( self
, scores
):
447 f
= open ( config_scores_file
, 'w' )
448 pickle
. dump ( scores
, f
)
450 def add_score ( self
, dico
):
451 scores
= self
. get_scores ()
452 for k
, v
in dico
. items ():
453 if scores
. has_key ( k
):
457 self
. save_scores ( scores
)
458 def send_scores ( self
, serv
, destinataire
):
459 scores
= self
. get_scores ()
460 scores
= scores
. items ()
461 scores
. sort ( lambda x
, y
: cmp ( x
[ 1 ], y
[ 1 ]))
463 serv
. privmsg ( destinataire
, "Scores by score : " + " ; " . join ([ " %s %s " %( k
, v
) for ( k
, v
) in scores
]) )
464 scores
. sort ( lambda x
, y
: cmp ( x
[ 0 ]. lower (), y
[ 0 ]. lower ()))
465 serv
. privmsg ( destinataire
, "Scores by pseudo : " + " ; " . join ([ " %s %s " %( k
, v
) for ( k
, v
) in scores
]) )
467 def gagne ( self
, serv
, canal
, bonus
= None , bonusvalue
= 2 ):
468 realword
= "" . join ([ lettre
[ 0 ] for lettre
in self
. play_status
[ canal
][ 0 ]])
469 definition
= self
. play_status
[ canal
][ 1 ]
470 serv
. privmsg ( canal
, "Bravo ! C'était %s " %(realword))
471 serv
. privmsg ( canal
, definition
)
472 nlettre
= float ( len ( realword
. replace ( "'" , "" ). replace ( "-" , "" )))
473 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 ()]
474 contribs_score
={ pseudo
: int ( 10 * contrib
/ nlettre
) for pseudo
, contrib
in self
. play_status
[ canal
][ 2 ]. items ()}
475 self
. add_score ( contribs_score
)
476 serv
. privmsg ( canal
, "Contributions : %s " %( " " . join ( contribs
)) )
477 self
. play_status
[ canal
]=[ None , None , None ]
480 if __name__
== "__main__" :
483 print "Usage : hung.py <serveur> [--debug]"
486 if "debug" in sys
. argv
or "--debug" in sys
. argv
:
490 serveurs
={ "a♡" : "acoeur.crans.org" , "acoeur" : "acoeur.crans.org" , "acoeur.crans.org" : "acoeur.crans.org" ,
491 "irc" : "irc.crans.org" , "crans" : "irc.crans.org" , "irc.crans.org" : "irc.crans.org" }
493 serveur
= serveurs
[ serveur
]
495 print "Server Unknown : %s " %(serveur)
497 hung
= Hung ( serveur
, debug
)