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
]
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 _
, askedwhy
, askedby
= self
.users
.pending_whois
[pseudo
]
250 if askedwhy
== "cmd WHOIS":
251 # Ce whois a été demandé par quelqu'un, on lui répond
252 self
.serv
.privmsg(askedby
, "%s is a registered nick" % (pseudo
,))
254 def on_endofwhois(self
, serv
, ev
):
255 """Si on arrive à la fin du whois, on va voir si on n'a pas reçu "is a registered nick"
256 c'est que le pseudo n'est pas identifié. """
257 pseudo
= ev
.arguments()[0]
258 # On laisse le temps au bot de souffler un coup
259 serv
.execute_delayed(5, self
.fail_whoisregnick
, (pseudo
,))
261 def fail_whoisregnick(self
, pseudo
):
262 """Maintenant qu'on a laissé 5 secondes au bot pour gérer les affaires courantes,
263 on considère que le pseudo n'est pas registered. """
264 # Attention, parce qu'il se pourrait qu'on n'ait pas sollicité ce whois
265 # et que donc pending_whois n'ai pas été peuplé en conséquence
266 if self
.users
.pending_whois
.has_key(pseudo
):
267 status
, askedwhy
, askedby
= self
.users
.pending_whois
[pseudo
]
268 if status
== "pending":
269 # Si le status est encore pending, on n'a pas eu de réponse positive, donc elle est négative
270 self
.users
.pending_whois
[pseudo
] = "notregistered"
271 if askedwhy
== "cmd WHOIS":
272 self
.serv
.privmsg(askedby
, "%s is NOT a registered nick" % (pseudo
,))
274 def on_privmsg(self
, serv
, ev
):
275 """À la réception d'un message en privé."""
276 if ignore_event(serv
, ev
):
278 message
= ev
.arguments()[0]
279 auteur
= irclib
.nm_to_n(ev
.source())
281 message
= bot_unicode(message
)
282 except errors
.UnicodeBotError
:
283 if config
.utf8_trigger
:
284 serv
.privmsg(auteur
, random
.choice(config
.utf8_fail_answers
).encode("utf8"))
286 message
= message
.split()
287 cmd
= message
[0].lower()
288 notunderstood
= False
290 op
,overop
=auteur
in self
.ops
, auteur
in self
.overops
292 helpmsg
= config
.helpmsg_default
294 helpmsg
+= config
.helpmsg_ops
296 helpmsg
+= config
.helpmsg_overops
298 helpmsgs
= config
.helpdico
.get(message
[1].lower(), ["Commande inconnue.", None, None])
299 helpmsg
= helpmsgs
[0]
300 if op
and helpmsgs
[1]:
302 helpmsg
+= "\n" + helpmsgs
[1]
304 helpmsg
= helpmsgs
[1]
305 if overop
and helpmsgs
[2]:
307 helpmsg
+= "\n" + helpmsgs
[2]
309 helpmsg
= helpmsgs
[2]
310 for ligne
in helpmsg
.split("\n"):
311 serv
.privmsg(auteur
, ligne
.encode("utf-8"))
312 elif cmd
== u
"identify":
313 if len(message
) == 1:
314 if self
.users
.has(auteur
):
315 infos
= self
.users
[auteur
].get_infos(self
.nk
, serv
, auteur
)
316 serv
.privmsg(auteur
, (u
"Vous avez le compte note n°%(idbde)s, pseudo : %(pseudo)s." % infos
319 serv
.privmsg(auteur
, "Je ne connais pas votre note.")
320 elif len(message
) >= 3:
321 username
, password
= message
[1], " ".join(message
[2:])
322 success
, info
, _
= self
.new_connection_NK(serv
, username
, password
)
324 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
325 self
.users
.add(auteur
, info
["idbde"])
326 serv
.privmsg(auteur
, "Pseudo enregistré.")
328 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
329 serv
.privmsg(auteur
, "Mot de passe invalide. (ou serveur down)")
331 serv
.privmsg(auteur
, "Syntaxe : IDENTIFY [<username> <password>]")
334 if self
.users
.has(auteur
):
335 password
= " ".join(message
[1:])
336 success
, _
, _
= self
.new_connection_NK(serv
, "#%s" % self
.users
[auteur
].idbde
, password
)
338 self
.users
.drop(auteur
)
339 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
340 serv
.privmsg(auteur
, "Pseudo oublié.")
342 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
343 serv
.privmsg(auteur
, "Mot de passe invalide. (ou serveur down)")
345 serv
.privmsg(auteur
, "Je ne connais pas votre note.")
347 serv
.privmsg(auteur
, "Syntaxe : DROP <password>")
349 if auteur
in self
.ops
:
351 if message
[1] in self
.chanlist
:
352 serv
.privmsg(auteur
, (u
"Je suis déjà sur %s" % (message
[1])).encode("utf-8"))
354 serv
.join(message
[1])
355 self
.chanlist
.append(message
[1])
356 serv
.privmsg(auteur
, "Channels : " + " ".join(self
.chanlist
))
357 log(self
.serveur
, "priv", auteur
, " ".join(message
))
359 serv
.privmsg(auteur
, "Channels : " + " ".join(self
.chanlist
))
362 elif cmd
== u
"leave":
363 if auteur
in self
.ops
and len(message
) > 1:
364 if message
[1] in self
.chanlist
:
365 if not (message
[1] in self
.stay_channels
) or auteur
in self
.overops
:
366 self
.quitter(message
[1].encode("utf-8"), " ".join(message
[2:]))
367 self
.chanlist
.remove(message
[1])
368 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
370 serv
.privmsg(auteur
, "Non, je reste !")
371 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
373 serv
.privmsg(auteur
, "Je ne suis pas sur %s" % (message
[1]))
377 if auteur
in self
.overops
:
379 if message
[1] in self
.stay_channels
:
380 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
381 serv
.privmsg(auteur
, "Je stay déjà sur %s." % (message
[1]))
383 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
384 self
.stay_channels
.append(message
[1])
385 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
387 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
390 elif cmd
== u
"nostay":
391 if auteur
in self
.overops
:
393 if message
[1] in self
.stay_channels
:
394 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
395 self
.stay_channels
.remove(message
[1])
396 serv
.privmsg(auteur
, "Stay channels : " + " ".join(self
.stay_channels
))
398 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
399 serv
.privmsg(auteur
, "Je ne stay pas sur %s." % (message
[1]))
403 if auteur
in self
.overops
:
404 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
408 elif cmd
== u
"crash":
409 if auteur
in self
.overops
:
410 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
411 self
.crash(auteur
, "priv")
414 elif cmd
== u
"reload":
415 if auteur
in self
.ops
:
416 self
.execute_something("reload", {"auteur" : auteur
}, place
=auteur
, auteur
=auteur
)
419 elif cmd
== u
"reconnect":
420 if auteur
in self
.ops
:
422 self
.nk
= self
.new_connection_NK(serv
, config
.note_pseudo
,
423 config
.note_password
, "special")[2]
424 except Exception as exc
:
426 log(self
.serveur
, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc
))
428 serv
.privmsg(auteur
, "%s: done" % (auteur
))
429 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
431 serv
.privmsg(auteur
, "%s: failed" % (auteur
))
432 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
433 for report
in self
.report_bugs_to
:
434 serv
.privmsg(report
, "Connection to NK2015 failed, invalid password ? Server dead ?")
437 elif cmd
== u
"quiet":
438 if auteur
in self
.ops
:
440 if message
[1] in self
.quiet_channels
:
441 serv
.privmsg(auteur
, "Je me la ferme déjà sur %s" % (message
[1]))
442 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
444 self
.quiet_channels
.append(message
[1])
445 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
446 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
448 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
451 elif cmd
== u
"noquiet":
452 if auteur
in self
.ops
:
454 if message
[1] in self
.quiet_channels
:
455 self
.quiet_channels
.remove(message
[1])
456 serv
.privmsg(auteur
, "Quiet channels : " + " ".join(self
.quiet_channels
))
457 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[successful]")
459 serv
.privmsg(auteur
, "Je ne me la ferme pas sur %s." % (message
[1]))
460 log(self
.serveur
, "priv", auteur
, " ".join(message
) + "[failed]")
464 if auteur
in self
.overops
and len(message
) > 2:
465 serv
.privmsg(message
[1].encode("utf-8"), (u
" ".join(message
[2:])).encode("utf-8"))
466 log(self
.serveur
, "priv", auteur
, " ".join(message
))
467 elif len(message
) <= 2:
468 serv
.privmsg(auteur
, "Syntaxe : SAY <channel> <message>")
472 if auteur
in self
.overops
and len(message
) > 2:
473 serv
.action(message
[1], " ".join(message
[2:]))
474 log(self
.serveur
, "priv", auteur
, " ".join(message
))
475 elif len(message
) <= 2:
476 serv
.privmsg(auteur
, "Syntaxe : DO <channel> <action>")
480 if auteur
in self
.overops
and len(message
) > 2:
481 serv
.kick(message
[1].encode("utf-8"), message
[2].encode("utf-8"), " ".join(message
[3:]).encode("utf-8"))
482 log(self
.serveur
, "priv", auteur
, " ".join(message
))
483 elif len(message
) <= 2:
484 serv
.privmsg(auteur
, "Syntaxe : KICK <channel> <pseudo> [<raison>]")
488 if auteur
in self
.ops
and len(message
) > 1:
489 serv
.privmsg(message
[1], "J'ai perdu !")
490 log(self
.serveur
, "priv", auteur
, " ".join(message
))
491 elif len(message
) <= 1:
492 serv
.privmsg(auteur
, "Syntaxe : LOST <channel>")
495 elif cmd
== u
"solde":
496 if len(message
) == 1:
497 if self
.users
.has(auteur
):
498 success
, solde
, pseudo
= nk
.get_solde(self
.nk
, self
.users
[auteur
].idbde
, serv
, auteur
)
500 serv
.privmsg(auteur
, "%.2f (%s)" % (solde
/100.0, pseudo
.encode("utf8")))
501 log(self
.serveur
, "priv", auteur
, " ".join(message
) + ("[successful]" if success
else "[failed]"))
503 serv
.privmsg(canal
, "Je ne connais pas votre note.")
505 if auteur
in self
.overops
:
506 serv
.privmsg(auteur
, " ".join(self
.ops
))
509 elif cmd
== u
"overops":
510 if auteur
in self
.overops
:
511 serv
.privmsg(auteur
, " ".join(self
.overops
))
514 elif cmd
== u
"whois":
515 if auteur
in self
.ops
and len(message
) > 1:
516 self
.whois(message
[1], askedwhy
="cmd WHOIS", askedby
=auteur
)
522 serv
.privmsg(auteur
, "Je n'ai pas compris. Essayez HELP…")
524 def on_pubmsg(self
, serv
, ev
):
525 """À la réception d'un message sur un channel."""
526 if ignore_event(serv
, ev
):
528 auteur
= irclib
.nm_to_n(ev
.source())
530 message
= ev
.arguments()[0]
532 message
= bot_unicode(message
)
533 except errors
.UnicodeBotError
:
534 if config
.utf8_trigger
and not canal
in self
.quiet_channels
:
535 serv
.privmsg(canal
, (u
"%s: %s"% ( auteur
, random
.choice(config
.utf8_fail_answers
))).encode("utf8"))
537 pour_moi
, message
= self
.pourmoi(serv
, message
)
538 if pour_moi
and message
.split()!=[]:
539 cmd
= message
.split()[0].lower()
541 args
= " ".join(message
.split()[1:])
544 if cmd
in [u
"meurs", u
"die", u
"crève"]:
545 if auteur
in self
.overops
:
546 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
549 serv
.privmsg(canal
,(u
"%s: %s"%(auteur
, random
.choice(config
.quit_fail_messages
))).encode("utf8"))
550 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
551 elif cmd
== u
"reload":
552 if auteur
in self
.ops
:
553 self
.execute_something("reload", {"auteur" : auteur
}, place
=canal
, auteur
=auteur
)
554 elif cmd
== u
"crash":
555 if auteur
in self
.overops
:
556 self
.crash(auteur
, canal
)
557 elif cmd
in [u
"part", u
"leave", u
"dégage", u
"va-t-en", u
"tut'tiresailleurs,c'estmesgalets"]:
558 if auteur
in self
.ops
and (not (canal
in self
.stay_channels
)
559 or auteur
in self
.overops
):
561 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
562 if canal
in self
.chanlist
:
563 self
.chanlist
.remove(canal
)
565 serv
.privmsg(canal
,(u
"%s: %s" % (auteur
, random
.choice(config
.leave_fail_messages
))).encode("utf8"))
566 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
568 elif cmd
== u
"reconnect":
569 if auteur
in self
.ops
:
571 self
.nk
= self
.new_connection_NK(serv
, config
.note_pseudo
,
572 config
.note_password
, "special")[2]
573 except Exception as exc
:
575 log(self
.serveur
, 'Erreur dans on_pubmsg/"cmd in ["reconnect"]\n' + str(exc
))
577 serv
.privmsg(canal
, "%s: done" % (auteur
))
578 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
580 serv
.privmsg(canal
, "%s: failed" % (auteur
))
581 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
582 for report
in self
.report_bugs_to
:
583 serv
.privmsg(report
, "Connection to NK2015 failed, invalid password ? Server dead ?")
585 serv
.privmsg(canal
, "%s: %s" % (auteur
, random
.choice(config
.pas_programme_pour_tobeir
).encode("utf8")))
586 log(self
.serveur
, canal
, auteur
, message
+ "[failed]")
588 elif cmd
in [u
"deviens", u
"pseudo"]:
589 if auteur
in self
.ops
:
592 log(self
.serveur
, canal
, auteur
, message
+ "[successful]")
594 if cmd
in [u
"meur", u
"meurt", u
"meurre", u
"meurres"] and not canal
in self
.quiet_channels
:
595 serv
.privmsg(canal
, '%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)' % (auteur
))
596 elif cmd
in [u
"ping"] and not canal
in self
.quiet_channels
:
597 serv
.privmsg(canal
, "%s: pong" % (auteur
))
599 elif cmd
in [u
"solde", u
"!solde", u
"!coca"] or cmd
.startswith("!"):
600 if self
.users
.has(auteur
):
601 idbde
= self
.users
[auteur
].idbde
602 if cmd
in [u
"solde", u
"!solde"]:
603 success
, solde
, pseudo
= nk
.get_solde(self
.nk
, idbde
, serv
, canal
)
605 serv
.privmsg(canal
, "%s: %s (%s)" % (auteur
, float(solde
)/100, pseudo
.encode("utf8")))
606 elif cmd
in [u
"!coca"] or cmd
.startswith("!"):
607 success
= nk
.consomme(self
.nk
, idbde
, message
[1:], serv
, canal
)
608 log(self
.serveur
, canal
, auteur
, message
+ ("[successful]" if success
else "[failed]"))
610 serv
.privmsg(canal
, "%s: Je ne connais pas votre pseudo note." % (auteur
))
611 log(self
.serveur
, canal
, auteur
, message
+ "[unknown]")
612 elif (re
.match("(pain au chocolat|chocolatine)", message
.lower())
613 and not canal
in self
.quiet_channels
):
614 serv
.action(canal
, "sert un pain au chocolat à %s" % (auteur
))
615 elif re
.match("manzana",message
.lower()) and not canal
in self
.quiet_channels
:
616 if auteur
in config
.manzana
:
617 serv
.action(canal
, "sert une bouteille de manzana à %s" % (auteur
))
618 elif auteur
in config
.manzana_bis
:
619 serv
.action(canal
, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur
))
621 serv
.action(canal
, "sert un verre de manzana à %s" % (auteur
))
622 if isit
.is_insult(message
) and not canal
in self
.quiet_channels
:
623 if isit
.is_not_insult(message
):
624 answer
= random
.choice(config
.compliment_answers
)
625 for ligne
in answer
.split("\n"):
626 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
628 answer
= random
.choice(config
.insultes_answers
)
629 for ligne
in answer
.split("\n"):
630 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
631 elif isit
.is_compliment(message
) and not canal
in self
.quiet_channels
:
632 answer
= random
.choice(config
.compliment_answers
)
633 for ligne
in answer
.split("\n"):
634 serv
.privmsg(canal
, "%s: %s" % (auteur
,ligne
.encode("utf8")))
635 gros_match
= isit
.is_gros(message
)
636 if gros_match
and not canal
in self
.quiet_channels
:
637 taille
= get_filesize()
638 answer
= u
"Mais non, je ne suis pas %s, %sKo tout au plus…" % (gros_match
.groups()[0], taille
)
639 serv
.privmsg(canal
, "%s: %s"%(auteur
, answer
.encode("utf8")))
640 if isit
.is_tesla(message
) and not canal
in self
.quiet_channels
:
641 l1
, l2
= config
.tesla_answers
, config
.tesla_actions
642 n1
, n2
= len(l1
), len(l2
)
643 i
= random
.randrange(n1
+ n2
)
645 serv
.action(canal
, l2
[i
- n1
].encode("utf8"))
647 serv
.privmsg(canal
, "%s: %s" % (auteur
, l1
[i
].encode("utf8")))
648 if isit
.is_tamere(message
) and not canal
in self
.quiet_channels
:
649 answer
= random
.choice(config
.tamere_answers
)
650 for ligne
in answer
.split("\n"):
651 serv
.privmsg(canal
, "%s: %s"%(auteur
, ligne
.encode("utf8")))
652 if isit
.is_tag(message
) and not canal
in self
.quiet_channels
:
653 if auteur
in self
.ops
:
654 action
= random
.choice(config
.tag_actions
)
655 serv
.action(canal
, action
.encode("utf8"))
656 self
.quiet_channels
.append(canal
)
658 answer
= random
.choice(config
.tag_answers
)
659 for ligne
in answer
.split("\n"):
660 serv
.privmsg(canal
, "%s: %s" % (auteur
, ligne
.encode("utf8")))
661 if isit
.is_merci(message
):
662 answer
= random
.choice(config
.merci_answers
)
663 for ligne
in answer
.split("\n"):
664 serv
.privmsg(canal
, "%s: %s"%(auteur
, ligne
.encode("utf8")))
665 out
= re
.match(ur
"^([A-Z[]|\\|[0-9]+|(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+)(?:| \?| !)$", message
.upper())
666 if re
.match("ma bite dans ton oreille", message
) and not canal
in self
.quiet_channels
:
667 serv
.privmsg(canal
, "%s: Seul un olasd peut imiter un olasd dans un de ses grands jours !" % (auteur
))
668 if out
and not canal
in self
.quiet_channels
:
669 out
= out
.groups()[0]
672 serv
.privmsg(canal
, "%s: %s !" % (auteur
, iout
+ 1))
673 if iout
== 2147483647:
674 serv
.privmsg(canal
, "%s: Ciel, un maxint ! Heureusement que je suis en python…" % (auteur
))
676 if iout
+ 1 > 1000 and random
.randrange(4) == 0:
677 serv
.privmsg(canal
, "%s: Vous savez, moi et les chiffres…" % (auteur
))
679 except Exception as exc
:
681 if re
.match("[A-Y]", out
):
682 alphabet
= "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
683 serv
.privmsg(canal
, "%s: %s !"%(auteur
, alphabet
[alphabet
.index(out
) + 1]))
685 serv
.privmsg(canal
, "%s: Je ne vous remercie pas, j'ai l'air idiot ainsi… [ ?" % (auteur
))
687 serv
.privmsg(canal
, "%s: Nous devrions nous en tenir là, ça va finir par poser des problèmes…" % (auteur
))
688 elif re
.match(ur
"(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+", out
):
690 return "".join([{u
"⁰¹²³⁴⁵⁶⁷⁸⁹0123456789"[i
]:u
"0123456789⁰¹²³⁴⁵⁶⁷⁸⁹"[i
]
691 for i
in range(20)}[j
]
693 out
= int(translate(out
))
694 serv
.privmsg(canal
,"%s: %s !" % (auteur
, translate(str(out
+ 1)).encode("utf8")))
695 if isit
.is_bonjour(message
) and not canal
in self
.quiet_channels
:
697 answer
= random
.choice(config
.night_answers
)
699 answer
= random
.choice(config
.bonjour_answers
)
701 answer
= random
.choice(config
.bonsoir_answers
)
702 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
703 if isit
.is_bonne_nuit(message
) and not canal
in self
.quiet_channels
:
704 answer
= random
.choice(config
.bonne_nuit_answers
)
705 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
706 if isit
.is_pan(message
) and not canal
in self
.quiet_channels
:
707 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
))
709 if message
in [u
"!pain au chocolat", u
"!chocolatine"] and not canal
in self
.quiet_channels
:
710 serv
.action(canal
, "sert un pain au chocolat à %s" % (auteur
))
711 if message
in [u
"!manzana"] and not canal
in self
.quiet_channels
:
712 if auteur
in config
.manzana
:
713 serv
.action(canal
, "sert une bouteille de manzana à %s" % (auteur
))
714 elif auteur
in config
.manzana_bis
:
715 serv
.action(canal
, "sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas." % (auteur
))
717 serv
.action(canal
, "sert un verre de manzana à %s" % (auteur
))
718 if re
.match(config
.buffer_fail_regexp
, message
, flags
=re
.UNICODE
) and not canal
in self
.quiet_channels
:
719 failanswers
= config
.buffer_fail_answers
720 answer
= random
.choice(failanswers
)
721 serv
.privmsg(canal
, ("%s: %s"%(auteur
,answer
)).encode("utf8"))
722 if not canal
in self
.quiet_channels
:
724 if re
.match((u
"^(" + u
"|".join(config
.bonjour_triggers
)
725 + ur
")( {}| all| tout le monde| (à )?tous)(\.| ?!)?$"
726 ).format(mypseudo
).lower(), message
.strip().lower()):
727 answer
= random
.choice(config
.bonjour_answers
)
728 serv
.privmsg(canal
, answer
.format(auteur
).encode("utf8"))
729 if (isit
.is_perdu(message
) and not canal
in self
.quiet_channels
):
730 # proba de perdre sur trigger :
731 # avant 30min (enfin, config) : 0
732 # ensuite, +25%/30min, linéairement
733 deltat
= time
.time() - self
.last_perdu
734 barre
= (deltat
- config
.time_between_perdu
)/(2*3600.0)
735 if random
.uniform(0, 1) < barre
:
736 serv
.privmsg(canal
, "%s: J'ai perdu !" % (auteur
))
737 self
.last_perdu
= time
.time()
739 def on_action(self
, serv
, ev
):
740 """À la réception d'une action."""
741 if ignore_event(serv
, ev
):
743 action
= ev
.arguments()[0]
744 auteur
= irclib
.nm_to_n(ev
.source())
745 channel
= ev
.target()
747 action
= bot_unicode(action
)
748 except errors
.UnicodeBotError
:
749 if config
.utf8_trigger
and not channel
in self
.quiet_channels
:
750 serv
.privmsg(channel
, (u
"%s: %s"%(auteur
,random
.choice(config
.utf8_fail_answers
))).encode("utf8"))
754 if isit
.is_bad_action_trigger(action
, mypseudo
) and not channel
in self
.quiet_channels
:
755 l1
, l2
= config
.bad_action_answers
, config
.bad_action_actions
756 n1
, n2
= len(l1
), len(l2
)
757 i
= random
.randrange(n1
+ n2
)
759 serv
.action(channel
, l2
[i
- n1
].format(auteur
).encode("utf8"))
761 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
762 if isit
.is_good_action_trigger(action
, mypseudo
) and not channel
in self
.quiet_channels
:
763 l1
, l2
= config
.good_action_answers
, config
.good_action_actions
764 n1
, n2
= len(l1
), len(l2
)
765 i
= random
.randrange(n1
+ n2
)
767 serv
.action(channel
, l2
[i
-n1
].format(auteur
).encode("utf8"))
769 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
771 def on_kick(self
, serv
, ev
):
772 """À la réception d'une action."""
773 auteur
= irclib
.nm_to_n(ev
.source())
774 channel
= ev
.target()
775 victime
= ev
.arguments()[0]
776 raison
= ev
.arguments()[1]
777 if victime
== self
.nick
:
778 log(self
.serveur
, ("%s kické de %s par %s (raison : %s)" % (victime
, channel
, auteur
, raison
)).decode("utf-8"))
781 l1
, l2
= config
.kick_answers
, config
.kick_actions
782 n1
, n2
= len(l1
), len(l2
)
783 i
= random
.randrange(n1
+ n2
)
785 serv
.action(channel
, l2
[i
-n1
].format(auteur
).encode("utf8"))
787 serv
.privmsg(channel
, l1
[i
].format(auteur
).encode("utf8"))
790 def start_as_daemon(self
, outfile
):
791 sys
.stderr
= Logger(outfile
)
795 class Logger(object):
796 """Pour écrire ailleurs que sur stdout"""
797 def __init__(self
, filename
="basile.full.log"):
798 self
.filename
= filename
800 def write(self
, message
):
801 f
= open(self
.filename
, "a")
806 """Exécution principal : lecture des paramètres et lancement du bot."""
807 if len(sys
.argv
) == 1:
808 print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
809 print " --outfile sans --no-output ni --daemon n'a aucun effet"
811 serveur
= sys
.argv
[1]
812 if "--daemon" in sys
.argv
:
813 thisfile
= os
.path
.realpath(__file__
)
814 thisdirectory
= thisfile
.rsplit("/", 1)[0]
815 os
.chdir(thisdirectory
)
819 if "debug" in sys
.argv
or "--debug" in sys
.argv
:
823 if "--quiet" in sys
.argv
:
824 config
.debug_stdout
= False
825 serveurs
= {"a♡" : "acoeur.crans.org",
826 "acoeur" : "acoeur.crans.org",
827 "acoeur.crans.org" : "acoeur.crans.org",
828 "irc" : "irc.crans.org",
829 "crans" : "irc.crans.org",
830 "irc.crans.org" : "irc.crans.org"}
831 if "--no-output" in sys
.argv
or "--daemon" in sys
.argv
:
832 outfile
= "/var/log/bots/basile.full.log"
835 if arg
[0].strip('-') in ["out", "outfile", "logfile"]:
837 sys
.stdout
= Logger(outfile
)
839 serveur
= serveurs
[serveur
]
841 print "Server Unknown : %s" % (serveur
)
843 basile
= Basile(serveur
,debug
)
844 # Si on reçoit un SIGHUP, on reload la config
845 def sighup_handler(signum
, frame
):
846 basile
.execute_reload(auteur
="SIGHUP")
847 signal
.signal(signal
.SIGHUP
, sighup_handler
)
850 child_pid
= os
.fork()
853 basile
.start_as_daemon(outfile
)
855 # on enregistre le pid de basile
856 pidfile
= "/var/run/bots/basile.pid"
859 if arg
[0].strip('-') in ["pidfile"]:
861 f
= open(pidfile
, "w")
862 f
.write("%s\n" % child_pid
)
867 if __name__
== "__main__":