4 # Codé par 20-100 (commencé le 23/04/12)
6 """ Un bot IRC destiné à s'interfacer avec la Note Kfet 2015 """
17 # Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
18 sys
.path
.insert(0, "/home/vincent/scripts/python-myirclib")
20 # On veut réagir sur la partie du whois qui dit qu'un nick est registered
21 irclib
.numeric_events
['307'] = "whoisregnick"
24 from commands
import getstatusoutput
as ex
28 #: Module responsable du dialogue avec la NoteKfet2015
30 #: Module de réponse aux questions de base
32 #: Module définissant les erreurs
34 #: Module de gestion des utilisateurs
37 # la partie qui réfère au fichier lui-même est mieux ici
38 # sinon on réfère la config et pas le fichier lui-même
39 config
.thisfile
= os
.path
.realpath(__file__
)
41 def get_config_logfile(serveur
):
42 """Renvoie le nom du fichier de log en fonction du ``serveur`` et de la config."""
43 serveurs
= {"acoeur.crans.org" : "acoeur",
44 "irc.crans.org" : "crans"}
45 return config
.logfile_template
% (serveurs
[serveur
],)
48 """Récupère la taille de ce fichier."""
49 return ex("ls -s %s" % (config
.thisfile
))[1].split()[0]
51 def log(serveur
, channel
, auteur
=None, message
=None):
52 """Enregistre une ligne de log."""
53 if auteur
== message
== None:
54 # alors c'est que c'est pas un channel mais juste une ligne de log
55 chain
= u
"%s %s" % (time
.strftime("%F %T"), channel
)
57 chain
= u
"%s [%s:%s] %s" % (time
.strftime("%F %T"), channel
, auteur
, message
)
58 f
= open(get_config_logfile(serveur
), "a")
59 f
.write((chain
+ u
"\n").encode("utf-8"))
61 if config
.debug_stdout
:
62 print chain
.encode("utf-8")
64 def ignore_event(serv
, ev
):
65 """Retourne ``True`` si il faut ignorer cet évènement."""
66 for (blackmask
, exceptlist
) in config
.blacklisted_masks
:
67 usermask
= ev
.source()
68 blackit
= bool(irclib
.mask_matches(usermask
, blackmask
))
69 exceptit
= any([bool(irclib
.mask_matches(usermask
, exceptmask
)) for exceptmask
in exceptlist
])
70 if exceptit
: # Il est exempté
73 if blackit
: # Il n'est pas exempté et matche la blacklist
77 def bot_unicode(chain
):
78 """Essaye de décoder ``chain`` en UTF-8.
79 Lève une py:class:`errors.UnicodeBotError` en cas d'échec."""
81 return chain
.decode("utf8")
82 except UnicodeDecodeError as exc
:
83 raise errors
.UnicodeBotError
86 class Basile(ircbot
.SingleServerIRCBot
):
87 """Classe principale : définition du bot Basile."""
88 def __init__(self
, serveur
, debug
=False):
89 temporary_pseudo
= config
.irc_pseudo
+ str(random
.randrange(10000,100000))
90 ircbot
.SingleServerIRCBot
.__init
__(self
, [(serveur
, 6667)],
91 temporary_pseudo
, "Basile, le bot irc. [Codé par 20-100]", 10)
93 self
.serveur
= serveur
94 self
.overops
= config
.overops
95 self
.ops
= self
.overops
+ config
.ops
96 self
.report_bugs_to
= config
.report_bugs_to
97 self
.chanlist
= config
.chanlist
98 self
.stay_channels
= config
.stay_channels
99 self
.quiet_channels
= config
.quiet_channels
101 # On charge la base de données d'utilisateurs
102 self
.users
= users
.UserDB()
106 def new_connection_NK(self
, serv
, username
, password
, typ
="bdd"):
107 """Renvoie (``True``, <une socket ouverte et authentifiée sur la NoteKfet2015>)
108 ou bien (``False``, None)"""
110 login_result
, sock
= nk
.login(username
, password
, typ
)
111 info
, retcode
, errmsg
= login_result
["msg"], login_result
["retcode"], login_result
["errmsg"]
112 except nk
.NKRefused
as exc
:
113 for report
in self
.report_bugs_to
:
114 serv
.privmsg(report
, "Le Serveur NK2015 est down.")
115 return (False, None, None)
116 except nk
.NKHelloFailed
as exc
:
117 for report
in self
.report_bugs_to
:
119 "La version du protocole utilisée n'est pas supportée par le serveur NK2015.")
120 return (False, None, None)
121 except nk
.NKUnknownError
as exc
:
122 erreurs
= ["Une fucking erreur inconnue s'est produite"]
123 erreurs
+= str(exc
).split("\n")
124 for report
in self
.report_bugs_to
:
126 serv
.privmsg(report
, err
)
127 return (False, None, None)
128 except Exception as exc
:
129 # Exception qui ne vient pas de la communication avec le serveur NK2015
130 log(self
.serveur
, "Erreur dans new_connection_NK\n" + str(exc
))
131 return (False, None, None)
133 return (True, info
, sock
)
135 return (False, None, None)
139 """Récuère le nick effectif du bot sur le serveur."""
140 return self
.serv
.get_nickname()
141 nick
= property(_getnick
)
143 def give_me_my_pseudo(self
, serv
):
144 """Récupère le pseudo auprès de NickServ."""
145 serv
.privmsg("NickServ", "RECOVER %s %s" % (config
.irc_pseudo
, config
.irc_password
))
146 serv
.privmsg("NickServ", "RELEASE %s %s" % (config
.irc_pseudo
, config
.irc_password
))
148 serv
.nick(config
.irc_pseudo
)
150 def pourmoi(self
, serv
, message
):
151 """Renvoie (False, lemessage) ou (True, le message amputé de "pseudo: ")"""
153 pseudo
= pseudo
.decode("utf-8")
155 if message
[:size
] == pseudo
and len(message
) > size
and message
[size
] == ":":
156 return (True, message
[size
+1:].lstrip(" "))
158 return (False, message
)
160 ### Exécution d'actions
161 def lost(self
, serv
, channel
, forced
=False):
162 """Réaction à un trigger de perdu.
163 Annonce "J'ai perdu" sur le channel si on n'a pas perdu depuis un certain temps."""
164 if self
.last_perdu
+ config
.time_between_perdu
< time
.time() or forced
:
165 if not channel
in self
.quiet_channels
or forced
:
166 serv
.privmsg(channel
, "J'ai perdu !")
167 self
.last_perdu
= time
.time()
168 delay
= config
.time_between_perdu_trigger
169 delta
= config
.time_between_perdu_trigger_delta
170 serv
.execute_delayed(random
.randrange(delay
- delta
, delay
+ delta
), self
.lost
, (serv
, channel
))
172 def quitter(self
, chan
, leave_message
=None):
173 """Quitter un channel avec un message customisable."""
174 if leave_message
== None:
175 leave_message
= random
.choice(config
.leave_messages
)
176 self
.serv
.part(chan
, message
=leave_message
.encode("utf8"))
179 """Se déconnecter du serveur IRC avec un message customisable."""
180 quit_message
= random
.choice(config
.quit_messages
)
181 self
.die(msg
=quit_message
.encode("utf8"))
183 def execute_reload(self
, auteur
=None):
184 """Recharge la config."""
186 isit
.regexp_compile()
187 if auteur
in [None, "SIGHUP"]:
188 towrite
= "Config reloaded" + " (SIGHUP received)" * (auteur
== "SIGHUP")
189 for to
in config
.report_bugs_to
:
190 self
.serv
.privmsg(to
, towrite
)
191 log(self
.serveur
, towrite
)
194 return True, u
"Config reloaded"
196 def crash(self
, who
="nobody", chan
="nowhere"):
197 """Fait crasher le bot."""
198 where
= "en privé" if chan
== "priv" else "sur le chan %s" % chan
199 raise errors
.CrashError((u
"Crash demandé par %s %s" % (who
, where
)).encode("utf-8"))
202 "reload" : execute_reload
,
205 def execute_something(self
, something
, params
, place
=None, auteur
=None):
206 """Exécute une action et répond son résultat à ``auteur``
207 sur un chan ou en privé en fonction de ``place``"""
208 action
= self
.ACTIONS
[something
]
209 success
, message
= action(self
, **params
)
211 if irclib
.is_channel(place
):
212 message
= "%s: %s" % (auteur
, message
.encode("utf-8"))
213 self
.serv
.privmsg(place
, message
)
214 log(self
.serveur
, place
, auteur
, something
+ "%r" % params
+ ("[successful]" if success
else "[failed]"))
216 def whois(self
, pseudo
, askedwhy
, askedby
):
217 """Demande un whois sur ``pseudo``. La réponse sera handled par une autre fonction."""
218 self
.users
.pending_whois
[pseudo
] = ["pending", askedwhy
, askedby
, None]
219 self
.serv
.whois([pseudo
])
221 ### Surcharge des events du Bot
222 def on_welcome(self
, serv
, ev
):
223 """À l'arrivée sur le serveur."""
224 self
.serv
= serv
# ça serv ira :)
225 self
.give_me_my_pseudo(serv
)
226 serv
.privmsg("NickServ", "IDENTIFY %s" % (config
.irc_password
))
227 log(self
.serveur
, "Connected")
229 self
.chanlist
= ["#bot"]
230 for c
in self
.chanlist
:
231 log(self
.serveur
, "JOIN %s" % (c
))
233 # on ouvre la connexion note de Basile, special user
234 self
.nk
= self
.new_connection_NK(serv
, config
.note_pseudo
, config
.note_password
, "special")[2]
236 for report
in self
.report_bugs_to
:
237 serv
.privmsg(report
, "Connection to NK2015 failed, invalid password ?")
239 def on_whoisregnick(self
, serv
, ev
):
240 """Appelée sur une réponse à un whois au moment où ça dit "is a registered nick".
241 J'ai vérifié, "is a registered nick" ça inclu le fiat qu'il est identified correctement.
243 On stocke l'information comme quoi cette personne est registered, et quand c'était."""
244 pseudo
, phrase
= ev
.arguments()
245 if phrase
== 'is a registered nick':
246 # Le whois n'est plus "pending"
247 if self
.users
.pending_whois
.has_key(pseudo
):
248 self
.users
.pending_whois
[pseudo
][0] = "registered"
249 self
.users
.pending_whois
[pseudo
][3] = time
.time()
250 _
, askedwhy
, askedby
, _
= self
.users
.pending_whois
[pseudo
]
251 if askedwhy
== "cmd WHOIS":
252 # Ce whois a été demandé par quelqu'un, on lui répond
253 self
.serv
.privmsg(askedby
, "%s is a registered nick" % (pseudo
,))
255 def on_endofwhois(self
, serv
, ev
):
256 """Si on arrive à la fin du whois, on va voir si on n'a pas reçu "is a registered nick"
257 c'est que le pseudo n'est pas identifié. """
258 pseudo
= ev
.arguments()[0]
259 # On laisse le temps au bot de souffler un coup
260 serv
.execute_delayed(config
.whois_timeout
, self
.fail_whoisregnick
, (pseudo
,))
262 def fail_whoisregnick(self
, pseudo
):
263 """Maintenant qu'on a laissé quelques secondes au bot pour gérer les affaires courantes,
264 on considère que le pseudo n'est pas registered. """
265 # Attention, parce qu'il se pourrait qu'on n'ait pas sollicité ce whois
266 # et que donc pending_whois n'ai pas été peuplé en conséquence
267 if self
.users
.pending_whois
.has_key(pseudo
):
268 status
, askedwhy
, askedby
, _
= self
.users
.pending_whois
[pseudo
]
269 if status
== "pending":
270 # Si le status est encore pending, on n'a pas eu de réponse positive, donc elle est négative
271 self
.users
.pending_whois
[pseudo
] = "notregistered"
272 if askedwhy
== "cmd WHOIS":
273 self
.serv
.privmsg(askedby
, "%s is NOT a registered nick" % (pseudo
,))
275 def on_privmsg(self
, serv
, ev
):
276 """À la réception d'un message en privé."""
277 if ignore_event(serv
, ev
):
279 message
= ev
.arguments()[0]
280 auteur
= irclib
.nm_to_n(ev
.source())
282 message
= bot_unicode(message
)
283 except errors
.UnicodeBotError
:
284 if config
.utf8_trigger
:
285 serv
.privmsg(auteur
, random
.choice(config
.utf8_fail_answers
).encode("utf8"))
287 message
= message
.split()
288 cmd
= message
[0].lower()
289 notunderstood
= False
291 op
,overop
=auteur
in self
.ops
, auteur
in self
.overops
293 helpmsg
= config
.helpmsg_default
295 helpmsg
+= config
.helpmsg_ops
297 helpmsg
+= config
.helpmsg_overops
299 helpmsgs
= config
.helpdico
.get(message
[1].lower(), ["Commande inconnue.", None, None])
300 helpmsg
= helpmsgs
[0]
301 if op
and helpmsgs
[1]:
303 helpmsg
+= "\n" + helpmsgs
[1]
305 helpmsg
= helpmsgs
[1]
306 if overop
and helpmsgs
[2]:
308 helpmsg
+= "\n" + helpmsgs
[2]
310 helpmsg
= helpmsgs
[2]
311 for ligne
in helpmsg
.split("\n"):
312 serv
.privmsg(auteur
, ligne
.encode("utf-8"))
313 elif cmd
== u
"identify":
314 if len(message
) == 1:
315 if self
.users
.has(auteur
):
316 infos
= self
.users
[auteur
].get_infos(self
.nk
, serv
, auteur
)
317 serv
.privmsg(auteur
, (u
"Vous avez le compte note n°%(idbde)s, pseudo : %(pseudo)s." % infos
320 serv
.privmsg(auteur
, "Je ne connais pas votre note.")
321 elif len(message
) >= 3:
322 username
, password
= message
[1], " ".join(message
[2:])
323 success
, info
, _
= self
.new_connection_NK(serv
, username
, password
)
325 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
326 self
.users
.add(auteur
, info
["idbde"])
327 serv
.privmsg(auteur
, "Pseudo enregistré.")
329 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
330 serv
.privmsg(auteur
, "Mot de passe invalide. (ou serveur down)")
332 serv
.privmsg(auteur
, "Syntaxe : IDENTIFY [<username> <password>]")
335 if self
.users
.has(auteur
):
336 password
= " ".join(message
[1:])
337 success
, _
, _
= self
.new_connection_NK(serv
, "#%s" % self
.users
[auteur
].idbde
, password
)
339 self
.users
.drop(auteur
)
340 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
341 serv
.privmsg(auteur
, "Pseudo oublié.")
343 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
344 serv
.privmsg(auteur
, "Mot de passe invalide. (ou serveur down)")
346 serv
.privmsg(auteur
, "Je ne connais pas votre note.")
348 serv
.privmsg(auteur
, "Syntaxe : DROP <password>")
350 if auteur
in self
.ops
:
352 if message
[1] in self
.chanlist
:
353 serv
.privmsg(auteur
, (u
"Je suis déjà sur %s" % (message
[1])).encode("utf-8"))
355 serv
.join(message
[1])
356 self
.chanlist
.append(message
[1])
357 serv
.privmsg(auteur
, "Channels : " + " ".join(self
.chanlist
))
358 log(self
.serveur
, "priv", auteur
, " ".join(message
))
360 serv
.privmsg(auteur
, "Channels : " + " ".join(self
.chanlist
))
363 elif cmd
== u
"leave":
364 if auteur
in self
.ops
and len(message
) > 1:
365 if message
[1] in self
.chanlist
:
366 if not (message
[1] in self
.stay_channels
) or auteur
in self
.overops
:
367 self
.quitter(message
[1].encode("utf-8"), " ".join(message
[2:]))
368 self
.chanlist
.remove(message
[1])
369 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
371 serv
.privmsg(auteur
, "Non, je reste !")
372 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
374 serv
.privmsg(auteur
, "Je ne suis pas sur %s" % (message
[1]))
378 if auteur
in self
.overops
:
380 if message
[1] in self
.stay_channels
:
381 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
382 serv
.privmsg(auteur
, "Je stay déjà sur %s." % (message
[1]))
384 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
385 self
.stay_channels
.append(message
[1])
386 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
388 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
391 elif cmd
== u
"nostay":
392 if auteur
in self
.overops
:
394 if message
[1] in self
.stay_channels
:
395 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
396 self
.stay_channels
.remove(message
[1])
397 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
399 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
400 serv
.privmsg(auteur
, "Je ne stay pas sur %s." % (message
[1]))
404 if auteur
in self
.overops
:
405 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
409 elif cmd
== u
"crash":
410 if auteur
in self
.overops
:
411 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
412 self
.crash(auteur
, "priv")
415 elif cmd
== u
"reload":
416 if auteur
in self
.ops
:
417 self
.execute_something("reload", {"auteur" : auteur
}, place
=auteur
, auteur
=auteur
)
420 elif cmd
== u
"reconnect":
421 if auteur
in self
.ops
:
423 self
.nk
= self
.new_connection_NK(serv
, config
.note_pseudo
,
424 config
.note_password
, "special")[2]
425 except Exception as exc
:
427 log(self
.serveur
, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc
))
429 serv
.privmsg(auteur
, "%s: done" % (auteur
))
430 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
432 serv
.privmsg(auteur
, "%s: failed" % (auteur
))
433 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
434 for report
in self
.report_bugs_to
:
435 serv
.privmsg(report
, "Connection to NK2015 failed, invalid password ? Server dead ?")
438 elif cmd
== u
"quiet":
439 if auteur
in self
.ops
:
441 if message
[1] in self
.quiet_channels
:
442 serv
.privmsg(auteur
, "Je me la ferme déjà sur %s" % (message
[1]))
443 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
445 self
.quiet_channels
.append(message
[1])
446 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
447 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
449 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
452 elif cmd
== u
"noquiet":
453 if auteur
in self
.ops
:
455 if message
[1] in self
.quiet_channels
:
456 self
.quiet_channels
.remove(message
[1])
457 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
458 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
460 serv
.privmsg(auteur
, "Je ne me la ferme pas sur %s." % (message
[1]))
461 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
465 if auteur
in self
.overops
and len(message
) > 2:
466 serv
.privmsg(message
[1].encode("utf-8"), (u
" ".join(message
[2:])).encode("utf-8"))
467 log(self
.serveur
, "priv", auteur
, " ".join(message
))
468 elif len(message
) <= 2:
469 serv
.privmsg(auteur
, "Syntaxe : SAY <channel> <message>")
473 if auteur
in self
.overops
and len(message
) > 2:
474 serv
.action(message
[1], " ".join(message
[2:]))
475 log(self
.serveur
, "priv", auteur
, " ".join(message
))
476 elif len(message
) <= 2:
477 serv
.privmsg(auteur
, "Syntaxe : DO <channel> <action>")
481 if auteur
in self
.overops
and len(message
) > 2:
482 serv
.kick(message
[1].encode("utf-8"), message
[2].encode("utf-8"), " ".join(message
[3:]).encode("utf-8"))
483 log(self
.serveur
, "priv", auteur
, " ".join(message
))
484 elif len(message
) <= 2:
485 serv
.privmsg(auteur
, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
489 if auteur
in self
.ops
and len(message
) > 1:
490 serv
.privmsg(message
[1], "J'ai perdu !")
491 log(self
.serveur
, "priv", auteur
, " ".join(message
))
492 elif len(message
) <= 1:
493 serv
.privmsg(auteur
, "Syntaxe : LOST <channel>")
496 elif cmd
== u
"solde":
497 if len(message
) == 1:
498 if self
.users
.has(auteur
):
499 success
, solde
, pseudo
= nk
.get_solde(self
.nk
, self
.users
[auteur
].idbde
, serv
, auteur
)
501 serv
.privmsg(auteur
, "%.2f (%s)" % (solde
/100.0, pseudo
.encode("utf8")))
502 log(self
.serveur
, "priv", auteur
, " ".join(message
) + ("[successful]" if success
else "[failed]"))
504 serv
.privmsg(auteur
, "Je ne connais pas votre note.")
506 if auteur
in self
.overops
:
507 serv
.privmsg(auteur
, " ".join(self
.ops
))
510 elif cmd
== u
"overops":
511 if auteur
in self
.overops
:
512 serv
.privmsg(auteur
, " ".join(self
.overops
))
515 elif cmd
== u
"whois":
516 if auteur
in self
.ops
and len(message
) > 1:
517 self
.whois(message
[1], askedwhy
="cmd WHOIS", askedby
=auteur
)
523 serv
.privmsg(auteur
, "Je n'ai pas compris. Essayez HELP…")
525 def on_pubmsg(self
, serv
, ev
):
526 """À la réception d'un message sur un channel."""
527 if ignore_event(serv
, ev
):
529 auteur
= irclib
.nm_to_n(ev
.source())
531 message
= ev
.arguments()[0]
533 message
= bot_unicode(message
)
534 except errors
.UnicodeBotError
:
535 if config
.utf8_trigger
and not canal
in self
.quiet_channels
:
536 serv
.privmsg(canal
, (u
"%s: %s"% ( auteur
, random
.choice(config
.utf8_fail_answers
))).encode("utf8"))
538 pour_moi
, message
= self
.pourmoi(serv
, message
)
539 if pour_moi
and message
.split()!=[]:
540 cmd
= message
.split()[0].lower()
542 args
= " ".join(message
.split()[1:])
545 if cmd
in [u
"meurs", u
"die", u
"crève"]:
546 if auteur
in self
.overops
:
547 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
550 serv
.privmsg(canal
,(u
"%s: %s"%(auteur
, random
.choice(config
.quit_fail_messages
))).encode("utf8"))
551 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
552 elif cmd
== u
"reload":
553 if auteur
in self
.ops
:
554 self
.execute_something("reload", {"auteur" : auteur
}, place
=canal
, auteur
=auteur
)
555 elif cmd
== u
"crash":
556 if auteur
in self
.overops
:
557 self
.crash(auteur
, canal
)
558 elif cmd
in [u
"part", u
"leave", u
"dégage", u
"va-t-en", u
"tut'tiresailleurs,c'estmesgalets"]:
559 if auteur
in self
.ops
and (not (canal
in self
.stay_channels
)
560 or auteur
in self
.overops
):
562 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
563 if canal
in self
.chanlist
:
564 self
.chanlist
.remove(canal
)
566 serv
.privmsg(canal
,(u
"%s: %s" % (auteur
, random
.choice(config
.leave_fail_messages
))).encode("utf8"))
567 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
569 elif cmd
== u
"reconnect":
570 if auteur
in self
.ops
:
572 self
.nk
= self
.new_connection_NK(serv
, config
.note_pseudo
,
573 config
.note_password
, "special")[2]
574 except Exception as exc
:
576 log(self
.serveur
, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc
))
578 serv
.privmsg(canal
, "%s: done" % (auteur
))
579 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
581 serv
.privmsg(canal
, "%s: failed" % (auteur
))
582 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
583 for report
in self
.report_bugs_to
:
584 serv
.privmsg(report
, "Connection to NK2015 failed, invalid password ? Server dead ?")
586 serv
.privmsg(canal
, "%s: %s" % (auteur
, random
.choice(config
.pas_programme_pour_tobeir
).encode("utf8")))
587 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
589 elif cmd
in [u
"deviens", u
"pseudo"]:
590 if auteur
in self
.ops
:
593 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
595 if cmd
in [u
"meur", u
"meurt", u
"meurre", u
"meurres"] and not canal
in self
.quiet_channels
:
596 serv
.privmsg(canal
, '%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)' % (auteur
))
597 elif cmd
in [u
"ping"] and not canal
in self
.quiet_channels
:
598 serv
.privmsg(canal
, "%s: pong" % (auteur
))
600 elif cmd
in [u
"solde", u
"!solde", u
"!coca"] or cmd
.startswith("!"):
601 if self
.users
.has(auteur
):
602 idbde
= self
.users
[auteur
].idbde
603 if cmd
in [u
"solde", u
"!solde"]:
604 success
, solde
, pseudo
= nk
.get_solde(self
.nk
, idbde
, serv
, canal
)
606 serv
.privmsg(canal
, "%s: %s (%s)" % (auteur
, float(solde
)/100, pseudo
.encode("utf8")))
607 elif cmd
in [u
"!coca"] or cmd
.startswith("!"):
608 success
= nk
.consomme(self
.nk
, idbde
, message
[1:], serv
, canal
)
609 log(self
.serveur
, canal
, auteur
, message
+ ("[successful]" if success
else "[failed]"))
611 serv
.privmsg(canal
, "%s: Je ne connais pas votre pseudo note." % (auteur
))
612 log(self
.serveur
, canal
, auteur
, message
+ "[unknown]")
613 elif (re
.match("(pain au chocolat|chocolatine)", message
.lower())
614 and not canal
in self
.quiet_channels
):
615 serv
.action(canal
, "sert un pain au chocolat à %s" % (auteur
))
616 elif re
.match("manzana",message
.lower()) and not canal
in self
.quiet_channels
:
617 if auteur
in config
.manzana
:
618 serv
.action(canal
, "sert une bouteille de manzana à %s" % (auteur
))
619 elif auteur
in config
.manzana_bis
:
620 serv
.action(canal
, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur
))
622 serv
.action(canal
, "sert un verre de manzana à %s" % (auteur
))
623 if isit
.is_insult(message
) and not canal
in self
.quiet_channels
:
624 if isit
.is_not_insult(message
):
625 answer
= random
.choice(config
.compliment_answers
)
626 for ligne
in answer
.split("\n"):
627 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
629 answer
= random
.choice(config
.insultes_answers
)
630 for ligne
in answer
.split("\n"):
631 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
632 elif isit
.is_compliment(message
) and not canal
in self
.quiet_channels
:
633 answer
= random
.choice(config
.compliment_answers
)
634 for ligne
in answer
.split("\n"):
635 serv
.privmsg(canal
, "%s: %s" % (auteur
,ligne
.encode("utf8")))
636 gros_match
= isit
.is_gros(message
)
637 if gros_match
and not canal
in self
.quiet_channels
:
638 taille
= get_filesize()
639 answer
= u
"Mais non, je ne suis pas %s, %sKo tout au plus…" % (gros_match
.groups()[0], taille
)
640 serv
.privmsg(canal
, "%s: %s"%(auteur
, answer
.encode("utf8")))
641 if isit
.is_tesla(message
) and not canal
in self
.quiet_channels
:
642 l1
, l2
= config
.tesla_answers
, config
.tesla_actions
643 n1
, n2
= len(l1
), len(l2
)
644 i
= random
.randrange(n1
+ n2
)
646 serv
.action(canal
, l2
[i
- n1
].encode("utf8"))
648 serv
.privmsg(canal
, "%s: %s" % (auteur
, l1
[i
].encode("utf8")))
649 if isit
.is_tamere(message
) and not canal
in self
.quiet_channels
:
650 answer
= random
.choice(config
.tamere_answers
)
651 for ligne
in answer
.split("\n"):
652 serv
.privmsg(canal
, "%s: %s"%(auteur
, ligne
.encode("utf8")))
653 if isit
.is_tag(message
) and not canal
in self
.quiet_channels
:
654 if auteur
in self
.ops
:
655 action
= random
.choice(config
.tag_actions
)
656 serv
.action(canal
, action
.encode("utf8"))
657 self
.quiet_channels
.append(canal
)
659 answer
= random
.choice(config
.tag_answers
)
660 for ligne
in answer
.split("\n"):
661 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
662 if isit
.is_merci(message
):
663 answer
= random
.choice(config
.merci_answers
)
664 for ligne
in answer
.split("\n"):
665 serv
.privmsg(canal
, "%s: %s"%(auteur
, ligne
.encode("utf8")))
666 out
= re
.match(ur
"^([A-Z[]|\\|[0-9]+|(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+)(?:| \?| !)$", message
.upper())
667 if re
.match("ma bite dans ton oreille", message
) and not canal
in self
.quiet_channels
:
668 serv
.privmsg(canal
, "%s: Seul un olasd peut imiter un olasd dans un de ses grands jours !" % (auteur
))
669 if out
and not canal
in self
.quiet_channels
:
670 out
= out
.groups()[0]
673 serv
.privmsg(canal
, "%s: %s !" % (auteur
, iout
+ 1))
674 if iout
== 2147483647:
675 serv
.privmsg(canal
, "%s: Ciel, un maxint ! Heureusement que je suis en python…" % (auteur
))
677 if iout
+ 1 > 1000 and random
.randrange(4) == 0:
678 serv
.privmsg(canal
, "%s: Vous savez, moi et les chiffres…" % (auteur
))
680 except Exception as exc
:
682 if re
.match("[A-Y]", out
):
683 alphabet
= "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
684 serv
.privmsg(canal
, "%s: %s !"%(auteur
, alphabet
[alphabet
.index(out
) + 1]))
686 serv
.privmsg(canal
, "%s: Je ne vous remercie pas, j'ai l'air idiot ainsi… [ ?" % (auteur
))
688 serv
.privmsg(canal
, "%s: Nous devrions nous en tenir là, ça va finir par poser des problèmes…" % (auteur
))
689 elif re
.match(ur
"(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+", out
):
691 return "".join([{u
"⁰¹²³⁴⁵⁶⁷⁸⁹0123456789"[i
]:u
"0123456789⁰¹²³⁴⁵⁶⁷⁸⁹"[i
]
692 for i
in range(20)}[j
]
694 out
= int(translate(out
))
695 serv
.privmsg(canal
,"%s: %s !" % (auteur
, translate(str(out
+ 1)).encode("utf8")))
696 if isit
.is_bonjour(message
) and not canal
in self
.quiet_channels
:
698 answer
= random
.choice(config
.night_answers
)
700 answer
= random
.choice(config
.bonjour_answers
)
702 answer
= random
.choice(config
.bonsoir_answers
)
703 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
704 if isit
.is_bonne_nuit(message
) and not canal
in self
.quiet_channels
:
705 answer
= random
.choice(config
.bonne_nuit_answers
)
706 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
707 if isit
.is_pan(message
) and not canal
in self
.quiet_channels
:
708 serv
.privmsg(canal
, "%s: ce n'est pas sur moi qu'il faut tirer, même si je sais que j'attire l'œil !" % (auteur
))
710 if message
in [u
"!pain au chocolat", u
"!chocolatine"] and not canal
in self
.quiet_channels
:
711 serv
.action(canal
, "sert un pain au chocolat à %s" % (auteur
))
712 if message
in [u
"!manzana"] and not canal
in self
.quiet_channels
:
713 if auteur
in config
.manzana
:
714 serv
.action(canal
, "sert une bouteille de manzana à %s" % (auteur
))
715 elif auteur
in config
.manzana_bis
:
716 serv
.action(canal
, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur
))
718 serv
.action(canal
, "sert un verre de manzana à %s" % (auteur
))
719 if re
.match(config
.buffer_fail_regexp
, message
, flags
=re
.UNICODE
) and not canal
in self
.quiet_channels
:
720 failanswers
= config
.buffer_fail_answers
721 answer
= random
.choice(failanswers
)
722 serv
.privmsg(canal
, ("%s: %s"%(auteur
,answer
)).encode("utf8"))
723 if not canal
in self
.quiet_channels
:
725 if re
.match((u
"^(" + u
"|".join(config
.bonjour_triggers
)
726 + ur
")( {}| all| tout le monde| (à )?tous)(\.| ?!)?$"
727 ).format(mypseudo
).lower(), message
.strip().lower()):
728 answer
= random
.choice(config
.bonjour_answers
)
729 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
730 if (isit
.is_perdu(message
) and not canal
in self
.quiet_channels
):
731 # proba de perdre sur trigger :
732 # avant 30min (enfin, config) : 0
733 # ensuite, +25%/30min, linéairement
734 deltat
= time
.time() - self
.last_perdu
735 barre
= (deltat
- config
.time_between_perdu
)/(2*3600.0)
736 if random
.uniform(0, 1) < barre
:
737 serv
.privmsg(canal
, "%s: J'ai perdu !" % (auteur
))
738 self
.last_perdu
= time
.time()
740 def on_action(self
, serv
, ev
):
741 """À la réception d'une action."""
742 if ignore_event(serv
, ev
):
744 action
= ev
.arguments()[0]
745 auteur
= irclib
.nm_to_n(ev
.source())
746 channel
= ev
.target()
748 action
= bot_unicode(action
)
749 except errors
.UnicodeBotError
:
750 if config
.utf8_trigger
and not channel
in self
.quiet_channels
:
751 serv
.privmsg(channel
, (u
"%s: %s"%(auteur
,random
.choice(config
.utf8_fail_answers
))).encode("utf8"))
755 if isit
.is_bad_action_trigger(action
, mypseudo
) and not channel
in self
.quiet_channels
:
756 l1
, l2
= config
.bad_action_answers
, config
.bad_action_actions
757 n1
, n2
= len(l1
), len(l2
)
758 i
= random
.randrange(n1
+ n2
)
760 serv
.action(channel
, l2
[i
- n1
].format(auteur
).encode("utf8"))
762 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
763 if isit
.is_good_action_trigger(action
, mypseudo
) and not channel
in self
.quiet_channels
:
764 l1
, l2
= config
.good_action_answers
, config
.good_action_actions
765 n1
, n2
= len(l1
), len(l2
)
766 i
= random
.randrange(n1
+ n2
)
768 serv
.action(channel
, l2
[i
-n1
].format(auteur
).encode("utf8"))
770 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
772 def on_kick(self
, serv
, ev
):
773 """À la réception d'une action."""
774 auteur
= irclib
.nm_to_n(ev
.source())
775 channel
= ev
.target()
776 victime
= ev
.arguments()[0]
777 raison
= ev
.arguments()[1]
778 if victime
== self
.nick
:
779 log(self
.serveur
, ("%s kické de %s par %s (raison : %s)" % (victime
, channel
, auteur
, raison
)).decode("utf-8"))
782 l1
, l2
= config
.kick_answers
, config
.kick_actions
783 n1
, n2
= len(l1
), len(l2
)
784 i
= random
.randrange(n1
+ n2
)
786 serv
.action(channel
, l2
[i
-n1
].format(auteur
).encode("utf8"))
788 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
791 def start_as_daemon(self
, outfile
):
792 sys
.stderr
= Logger(outfile
)
796 class Logger(object):
797 """Pour écrire ailleurs que sur stdout"""
798 def __init__(self
, filename
="basile.full.log"):
799 self
.filename
= filename
801 def write(self
, message
):
802 f
= open(self
.filename
, "a")
807 """Exécution principal : lecture des paramètres et lancement du bot."""
808 if len(sys
.argv
) == 1:
809 print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
810 print " --outfile sans --no-output ni --daemon n'a aucun effet"
812 serveur
= sys
.argv
[1]
813 if "--daemon" in sys
.argv
:
814 thisfile
= os
.path
.realpath(__file__
)
815 thisdirectory
= thisfile
.rsplit("/", 1)[0]
816 os
.chdir(thisdirectory
)
820 if "debug" in sys
.argv
or "--debug" in sys
.argv
:
824 if "--quiet" in sys
.argv
:
825 config
.debug_stdout
= False
826 serveurs
= {"a♡" : "acoeur.crans.org",
827 "acoeur" : "acoeur.crans.org",
828 "acoeur.crans.org" : "acoeur.crans.org",
829 "irc" : "irc.crans.org",
830 "crans" : "irc.crans.org",
831 "irc.crans.org" : "irc.crans.org"}
832 if "--no-output" in sys
.argv
or "--daemon" in sys
.argv
:
833 outfile
= "/var/log/bots/basile.full.log"
836 if arg
[0].strip('-') in ["out", "outfile", "logfile"]:
838 sys
.stdout
= Logger(outfile
)
840 serveur
= serveurs
[serveur
]
842 print "Server Unknown : %s" % (serveur
)
844 basile
= Basile(serveur
,debug
)
845 # Si on reçoit un SIGHUP, on reload la config
846 def sighup_handler(signum
, frame
):
847 basile
.execute_reload(auteur
="SIGHUP")
848 signal
.signal(signal
.SIGHUP
, sighup_handler
)
851 child_pid
= os
.fork()
854 basile
.start_as_daemon(outfile
)
856 # on enregistre le pid de basile
857 pidfile
= "/var/run/bots/basile.pid"
860 if arg
[0].strip('-') in ["pidfile"]:
862 f
= open(pidfile
, "w")
863 f
.write("%s\n" % child_pid
)
868 if __name__
== "__main__":