]>
gitweb.pimeys.fr Git - bots/saturnin.git/blob - saturnin.py
a99fb15cc175011ac8b36d11d8c6a3625bab98a5
2 # -*- encoding: utf-8 -*-
6 # Un bot IRC pour remplacer le canard.
7 # parce que le canard, c'est le bien et que braice ne pong pas
14 import socket
, ssl
, json
18 from commands
import getstatusoutput
as ex
20 # on récupère la config
25 def get_config_logfile(serveur
):
26 serveurs
={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
27 return config
.logfile_template
%(serveurs
[serveur
])
29 def log(serveur
,channel
,auteur
=None,message
=None):
30 f
=open(get_config_logfile(serveur
),"a")
31 if auteur
==message
==None:
32 # alors c'est que c'est pas un channel mais juste une ligne de log
33 chain
="%s %s"%(time
.strftime("%F %T"),channel
)
35 chain
="%s [%s:%s] %s"%(time
.strftime("%F %T"),channel
,auteur
,message
)
37 if config
.debug_stdout
:
41 def is_something(chain
,matches
,avant
=u
".*(?:^| )",apres
=u
"(?:$|\.| |,|;).*",case_sensitive
=False,debug
=False):
43 chain
=unicode(chain
,"utf8")
45 chain
=unicode(chain
,"utf8").lower()
46 allmatches
="("+"|".join(matches
)+")"
47 reg
=(avant
+allmatches
+apres
).lower()
51 regexp_pan
= re
.compile(u
".*(" + "|".join(config
.killwords
) + u
").*")
53 return regexp_pan
.match(unicode(chain
,"utf8").lower())
55 class UnicodeBotError(Exception):
57 def bot_unicode(chain
):
60 except UnicodeDecodeError as exc
:
63 class Saturnin(ircbot
.SingleServerIRCBot
):
64 def __init__(self
,serveur
,debug
=False):
65 temporary_pseudo
=config
.irc_pseudo
+str(random
.randrange(10000,100000))
66 ircbot
.SingleServerIRCBot
.__init
__(self
, [(serveur
, 6667)],
67 temporary_pseudo
,"Coin ? ©braice [mais 'faut frapper 20-100]", 10)
70 self
.overops
=config
.overops
71 self
.ops
=self
.overops
+config
.ops
72 self
.chanlist
=config
.chanlist
73 self
.stay_channels
=config
.stay_channels
74 self
.quiet_channels
=config
.quiet_channels
75 self
.play_channels
=config
.play_channels
76 self
.status
= { chan
: [0, None] for chan
in self
.play_channels
}
77 # 0 : pas de spawn prévu
79 # avec en deuxième paramètre le timestamp du moment où il a été déclenché (pas du moment où il se fera)
83 def give_me_my_pseudo(self
,serv
):
84 serv
.privmsg("NickServ","RECOVER %s %s"%(config
.irc_pseudo
,config
.irc_password
))
85 serv
.privmsg("NickServ","RELEASE %s %s"%(config
.irc_pseudo
,config
.irc_password
))
87 serv
.nick(config
.irc_pseudo
)
89 def on_welcome(self
, serv
, ev
):
90 self
.serv
=serv
# ça serv ira :)
91 self
.give_me_my_pseudo(serv
)
92 serv
.privmsg("NickServ","IDENTIFY %s"%(config
.irc_password
))
93 log(self
.serveur
,"Connected")
95 self
.chanlist
= self
.play_channels
= ["#bot"]
96 self
.status
= { chan
: [0, 0] for chan
in self
.play_channels
}
97 for c
in self
.chanlist
:
98 log(self
.serveur
,"JOIN %s"%(c))
100 if c
in self
.play_channels
:
101 spawn_delay
= random
.randrange(*config
.spawn_delays
)
102 self
.spawn(c
, time
.time(), spawn_delay
)
104 def pourmoi(self
, serv
, message
):
105 """renvoie (False,lemessage) ou (True, le message amputé de "pseudo: ")"""
108 if message
[:size
]==pseudo
and len(message
)>size
and message
[size
]==":":
109 return (True,message
[size
+1:].lstrip(" "))
111 return (False,message
)
113 def on_privmsg(self
, serv
, ev
):
114 message
=ev
.arguments()[0]
115 auteur
= irclib
.nm_to_n(ev
.source())
117 test
=bot_unicode(message
)
118 except UnicodeBotError
:
119 if config
.utf8_trigger
:
120 serv
.privmsg(auteur
, random
.choice(config
.utf8_fail_answers
).encode("utf8"))
122 message
=message
.split()
123 cmd
=message
[0].lower()
126 helpdico
={"help":["""HELP <commande>
127 Affiche de l'aide sur la commande""",None,None],
129 Affiche votre score""", None, None],
131 Afficher tous les scores""", None, None],
132 "join": [None, """JOIN <channel>
133 Me fait rejoindre le channel""",None],
134 "leave": [None,"""LEAVE <channel>
135 Me fait quitter le channel (sauf s'il est dans ma stay_list).""",None],
136 "quiet": [None,"""QUIET <channel>
137 Me rend silencieux sur le channel.""",None],
138 "noquiet": [None,"""NOQUIET <channel>
139 Me rend la parole sur le channel.""",None],
140 "play": [None, """PLAY
141 Passe un channel en mode "jouer" """,None],
142 "noplay": [None, """NOPLAY
143 Passe un channel en mode "ne pas jouer" """,None],
144 "SPAWN": [None, """SPAWN <channel>
145 Me fait spawner sur le channel.""",None],
146 "reload": [None,"""RELOAD
147 Recharge la configuration.""",None],
148 "say": [None,None,"""SAY <channel> <message>
149 Me fait parler sur le channel."""],
150 "do": [None,None,"""DO <channel> <action>
151 Me fait faitre une action (/me) sur le channel."""],
152 "stay": [None,None,"""STAY <channel>
153 Ajoute le channel à ma stay_list."""],
154 "nostay": [None,None,"""NOSTAY <channel>
155 Retire le channel de ma stay_list."""],
156 "ops": [None,None,"""OPS
157 Affiche la liste des ops."""],
158 "overops": [None,None,"""OVEROPS
159 Affiche la liste des overops."""],
160 "kick": [None,None,"""KICK <channel> <pseudo> [<raison>]
161 Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
162 "die": [None,None,"""DIE
163 Me déconnecte du serveur IRC."""]
165 helpmsg_default
="Liste des commandes disponibles :\nHELP SCORE SCORES"
166 helpmsg_ops
=" JOIN LEAVE QUIET NOQUIET PLAY NOPLAY SPAWN"
167 helpmsg_overops
=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE"
168 op
,overop
=auteur
in self
.ops
, auteur
in self
.overops
170 helpmsg
=helpmsg_default
174 helpmsg
+=helpmsg_overops
176 helpmsgs
=helpdico
.get(message
[1].lower(),["Commande inconnue.",None,None])
178 if op
and helpmsgs
[1]:
180 helpmsg
+="\n"+helpmsgs
[1]
183 if overop
and helpmsgs
[2]:
185 helpmsg
+="\n"+helpmsgs
[2]
188 for ligne
in helpmsg
.split("\n"):
189 serv
.privmsg(auteur
,ligne
)
191 if auteur
in self
.ops
:
193 if message
[1] in self
.chanlist
:
194 serv
.privmsg(auteur
,"Je suis déjà sur %s"%(message
[1]))
196 serv
.join(message
[1])
197 self
.chanlist
.append(message
[1])
198 serv
.privmsg(auteur
,"Channels : "+" ".join(self
.chanlist
))
199 log(self
.serveur
,"priv",auteur
," ".join(message
))
201 serv
.privmsg(auteur
,"Channels : "+" ".join(self
.chanlist
))
205 if auteur
in self
.ops
and len(message
)>1:
206 if message
[1] in self
.chanlist
:
207 if not (message
[1] in self
.stay_channels
) or auteur
in self
.overops
:
208 self
.quitter(message
[1]," ".join(message
[2:]))
209 self
.chanlist
.remove(message
[1])
210 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
212 serv
.privmsg(auteur
,"Non, je reste !")
213 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
215 serv
.privmsg(auteur
,"Je ne suis pas sur %s"%(message
[1]))
219 if auteur
in self
.overops
:
221 if message
[1] in self
.stay_channels
:
222 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
223 serv
.privmsg(auteur
,"Je stay déjà sur %s."%(message
[1]))
225 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
226 self
.stay_channels
.append(message
[1])
227 serv
.privmsg(auteur
,"Stay channels : "+" ".join(self
.stay_channels
))
229 serv
.privmsg(auteur
,"Stay channels : "+" ".join(self
.stay_channels
))
233 if auteur
in self
.overops
:
235 if message
[1] in self
.stay_channels
:
236 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
237 self
.stay_channels
.remove(message
[1])
238 serv
.privmsg(auteur
,"Stay channels : "+" ".join(self
.stay_channels
))
240 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
241 serv
.privmsg(auteur
,"Je ne stay pas sur %s."%(message
[1]))
246 if auteur
in self
.overops
:
247 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
252 if auteur
in self
.ops
:
254 if message
[1] in self
.quiet_channels
:
255 serv
.privmsg(auteur
,"Je me la ferme déjà sur %s"%(message
[1]))
256 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
258 self
.quiet_channels
.append(message
[1])
259 serv
.privmsg(auteur
,"Quiet channels : "+" ".join(self
.quiet_channels
))
260 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
262 serv
.privmsg(auteur
,"Quiet channels : "+" ".join(self
.quiet_channels
))
266 if auteur
in self
.ops
:
268 if message
[1] in self
.quiet_channels
:
269 self
.quiet_channels
.remove(message
[1])
270 serv
.privmsg(auteur
,"Quiet channels : "+" ".join(self
.quiet_channels
))
271 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
273 serv
.privmsg(auteur
,"Je ne me la ferme pas sur %s."%(message
[1]))
274 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
278 if auteur
in self
.ops
:
280 if message
[1] in self
.play_channels
:
281 serv
.privmsg(auteur
,"Je play déjà sur %s."%(message
[1]))
282 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
284 self
.play_channels
.append(message
[1])
285 self
.spawn(message
[1], 1)
286 serv
.privmsg(auteur
,"Play channels : "+" ".join(self
.play_channels
))
287 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
289 serv
.privmsg(auteur
,"Play channels : "+" ".join(self
.play_channels
))
293 if auteur
in self
.ops
:
295 if message
[1] in self
.play_channels
:
296 self
.play_channels
.remove(message
[1])
297 serv
.privmsg(auteur
,"Play channels : "+" ".join(self
.play_channels
))
298 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[successful]")
300 serv
.privmsg(auteur
,"Je ne play pas sur %s."%(message
[1]))
301 log(self
.serveur
,"priv",auteur
," ".join(message
)+"[failed]")
305 if auteur
in self
.ops
:
307 if message
[1] in self
.play_channels
:
308 # Le plus pratique pour pas s'embêter c'est de mettre un
309 # delay d'une seconde, comme ça .spawn() fait le boulot
310 self
.spawn(message
[1], time
.time(), 1)
312 serv
.privmsg(auteur
, "Je ne joue pas sur %s" % message
[1])
314 serv
.privmsg(auteur
, "Syntaxe : SPAWN <channel>")
318 if auteur
in self
.overops
and len(message
)>2:
319 serv
.privmsg(message
[1]," ".join(message
[2:]))
320 log(self
.serveur
,"priv",auteur
," ".join(message
))
321 elif len(message
)<=2:
322 serv
.privmsg(auteur
,"Syntaxe : SAY <channel> <message>")
326 if auteur
in self
.overops
and len(message
)>2:
327 serv
.action(message
[1]," ".join(message
[2:]))
328 log(self
.serveur
,"priv",auteur
," ".join(message
))
329 elif len(message
)<=2:
330 serv
.privmsg(auteur
,"Syntaxe : DO <channel> <action>")
334 if auteur
in self
.overops
and len(message
)>2:
335 serv
.kick(message
[1],message
[2]," ".join(message
[3:]))
336 log(self
.serveur
,"priv",auteur
," ".join(message
))
337 elif len(message
)<=2:
338 serv
.privmsg(auteur
,"Syntaxe : KICK <channel> <pseudo> [<raison>]")
342 if auteur
in self
.overops
:
343 serv
.privmsg(auteur
," ".join(self
.ops
))
347 if auteur
in self
.overops
:
348 serv
.privmsg(auteur
," ".join(self
.overops
))
352 if auteur
in self
.ops
:
354 serv
.privmsg(auteur
,"done")
359 if len(message
) in [3,4] and message
[1].lower()=="transfert":
360 scores
=self
.get_scores()
361 de
,to
=auteur
,message
[2]
362 value
=scores
.get(de
,0)
365 asked
=int(message
[3])
367 serv
.privmsg(auteur
,"Syntaxe : SCORE TRANSFERT <pseudo> [<n>]")
372 serv
.privmsg(auteur
,"Vous n'avez pas de points")
375 serv
.privmsg(auteur
,"Bien tenté…")
378 serv
.privmsg(auteur
,"Vous n'avez que %s points"%(value))
381 self
.add_score(de
,-asked
)
382 self
.add_score(to
,asked
)
383 serv
.privmsg(auteur
,"Transfert de %s points de %s à %s"%(asked
,de
,to
))
385 serv
.privmsg(auteur
,"Syntaxe : SCORE TRANSFERT <pseudo> [<n>]")
387 self
.sendscore(auteur
)
390 self
.sendscores(auteur
)
391 elif auteur
in self
.overops
:
392 souscmd
=message
[1].lower()
396 scores
=self
.get_scores()
397 if scores
.has_key(todelete
):
399 self
.save_scores(scores
)
400 serv
.privmsg(auteur
,"Score de %s supprimé"%(todelete))
402 serv
.privmsg(auteur
,"Ce score n'existe pas : %s"%(todelete))
404 serv
.privmsg(auteur
,"Syntaxe : SCORES DEL <pseudo>")
405 elif souscmd
in ["add","sub"]:
407 toadd
,val
=message
[2],message
[3]
411 serv
.privmsg(auteur
,"Syntaxe : SCORES {ADD|SUB} <pseudo> <n>")
415 self
.add_score(toadd
,val
)
416 serv
.privmsg(auteur
,"Done")
418 serv
.privmsg(auteur
,"Syntaxe : SCORES {ADD|SUB} <pseudo> <n>")
420 serv
.privmsg(auteur
,"Syntaxe : SCORES {DEL|ADD|SUB} <pseudo> [<n>]")
426 serv
.privmsg(auteur
,"Je n'ai pas compris. Essayez HELP…")
428 def sendscore(self
, to
):
429 self
.serv
.privmsg(to
, "Votre score : %s"%(self
.get_scores().get(to
,0)) )
431 def sendscores(self
, to
):
432 scores
=self
.get_scores().items()
434 scores
.sort(lambda x
,y
:cmp(x
[1],y
[1]))
436 self
.serv
.privmsg(to
, "Scores by score : "+" ; ".join(["%s %s"%(i
[0],i
[1]) for i
in scores
]))
438 scores
.sort(lambda x
,y
:cmp(x
[0].lower(),y
[0].lower()))
439 self
.serv
.privmsg(to
, "Scores by pseudo : "+" ; ".join(["%s %s"%(i
[0],i
[1]) for i
in scores
]))
441 def on_pubmsg(self
, serv
, ev
):
442 auteur
= irclib
.nm_to_n(ev
.source())
443 channel
= ev
.target()
444 message
= ev
.arguments()[0]
446 test
=bot_unicode(message
)
447 except UnicodeBotError
:
448 if config
.utf8_trigger
and not channel
in self
.quiet_channels
:
449 serv
.privmsg(channel
, (u
"%s: %s"%(auteur
,random
.choice(config
.utf8_fail_answers
))).encode("utf8"))
451 pour_moi
,message
=self
.pourmoi(serv
,message
)
452 if pour_moi
and message
.split()!=[]:
453 cmd
=message
.split()[0].lower()
455 args
=" ".join(message
.split()[1:])
458 if cmd
in ["meurs","die","crève"]:
459 if auteur
in self
.overops
:
460 log(self
.serveur
,channel
,auteur
,message
+"[successful]")
463 serv
.privmsg(channel
,("%s: %s"%(auteur
,random
.choice(config
.quit_fail_messages
))).encode("utf8"))
464 log(self
.serveur
,channel
,auteur
,message
+"[failed]")
465 elif cmd
in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]:
466 if auteur
in self
.ops
and (not (channel
in self
.stay_channels
)
467 or auteur
in self
.overops
):
468 self
.quitter(channel
)
469 log(self
.serveur
,channel
,auteur
,message
+"[successful]")
470 if channel
in self
.chanlist
:
471 self
.chanlist
.remove(channel
)
473 serv
.privmsg(channel
,("%s: %s"%(auteur
,random
.choice(config
.leave_fail_messages
))).encode("utf8"))
474 log(self
.serveur
,channel
,auteur
,message
+"[failed]")
476 self
.sendscore(auteur
)
477 elif cmd
== "scores":
478 self
.sendscores(auteur
)
481 self
.shot(channel
, auteur
)
483 def on_action(self
, serv
, ev
):
484 action
= ev
.arguments()[0]
485 auteur
= irclib
.nm_to_n(ev
.source())
486 channel
= ev
.target()
488 #~ test=bot_unicode(action)
489 #~ except UnicodeBotError:
490 #~ if config.utf8_trigger and not channel in self.quiet_channels:
491 #~ serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
493 #~ mypseudo=self.nick
495 def on_kick(self
,serv
,ev
):
496 auteur
= irclib
.nm_to_n(ev
.source())
497 channel
= ev
.target()
498 victime
= ev
.arguments()[0]
499 raison
= ev
.arguments()[1]
500 if victime
==self
.nick
:
501 log(self
.serveur
,"%s kické de %s par %s (raison : %s)" %(victime
,channel
,auteur
,raison
))
504 #~ l1,l2=config.kick_answers,config.kick_actions
505 #~ n1,n2=len(l1),len(l2)
506 #~ i=random.randrange(n1+n2)
508 #~ serv.action(channel,l2[i-n1].format(auteur).encode("utf8"))
510 #~ serv.privmsg(channel,l1[i].format(auteur).encode("utf8"))
512 def spawn(self
, channel
, timestamp
, delay
=0):
513 if channel
in self
.play_channels
:
515 self
.serv
.execute_delayed(delay
, self
.spawn
, (channel
, timestamp
))
516 self
.status
[channel
] = [1, timestamp
]
518 # on teste le timestamp pour pas s'emmêler dans les spawn
519 infos
= self
.status
.get(channel
, [0,0])
520 if infos
== [1, timestamp
]:
521 spawn_sentence
= random
.choice(config
.canards
) + random
.choice(config
.spawn_sentences
)
522 self
.serv
.privmsg(channel
, spawn_sentence
.encode("utf8"))
523 self
.status
[channel
] = [2, timestamp
]
524 times_up_delay
= random
.randrange(*config
.times_up_delays
)
525 self
.serv
.execute_delayed(times_up_delay
, self
.too_slow
, (channel
, timestamp
))
527 def too_slow(self
, channel
, timestamp
):
528 infos
= self
.status
.get(channel
, [0,0])
529 if infos
== [2, timestamp
]:
530 self
.serv
.privmsg(channel
, random
.choice(config
.times_up_sentences
).encode("utf8"))
531 respawn_delay
= random
.randrange(*config
.spawn_delays
)
532 self
.spawn(channel
, time
.time(), respawn_delay
)
534 def shot(self
, channel
, auteur
):
535 if self
.status
.get(channel
, [0, 0])[0] == 2:
536 succeed
= random
.randrange(0,101) > config
.proba_miss
538 self
.serv
.privmsg(channel
, random
.choice(config
.killed_templates
).format(auteur
).encode("utf8"))
539 self
.add_score(auteur
, 1)
540 if random
.randrange(0, 101) < config
.proba_killed_sentence
:
541 self
.serv
.privmsg(channel
, random
.choice(config
.killed_sentences
).encode("utf8"))
542 respawn_delay
= random
.randrange(*config
.spawn_delays
)
543 self
.spawn(channel
, time
.time(), respawn_delay
)
545 self
.serv
.privmsg(channel
, random
.choice(config
.miss_templates
).format(auteur
).encode("utf8"))
546 if random
.randrange(0,101) < config
.proba_miss_sentence
:
547 self
.serv
.privmsg(channel
, random
.choice(config
.miss_sentences
).encode("utf8"))
549 def quitter(self
,chan
,leave_message
=None):
550 if leave_message
==None:
551 leave_message
=random
.choice(config
.leave_messages
)
552 self
.serv
.part(chan
,message
=leave_message
.encode("utf8"))
555 quit_message
=random
.choice(config
.quit_messages
)
556 self
.die(msg
=quit_message
.encode("utf8"))
558 def get_scores(self
):
559 f
=open(config
.score_file
)
560 scores
=pickle
.load(f
)
564 def add_score(self
, pseudo
, value
):
565 scores
=self
.get_scores()
566 if scores
.has_key(pseudo
):
567 scores
[pseudo
]+=value
570 self
.save_scores(scores
)
572 def save_scores(self
,scores
):
573 f
=open(config
.score_file
,"w")
574 pickle
.dump(scores
,f
)
578 return self
.serv
.get_nickname()
579 nick
=property(_getnick
)
582 if __name__
=="__main__":
585 print "Usage : saturnin.py <serveur> [--debug]"
588 if "debug" in sys
.argv
or "--debug" in sys
.argv
:
592 if "--quiet" in sys
.argv
:
593 config
.debug_stdout
=False
594 serveurs
={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
595 "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
597 serveur
=serveurs
[serveur
]
599 print "Server Unknown : %s"%(serveur)
601 bot
= Saturnin(serveur
,debug
)