]> gitweb.pimeys.fr Git - bots/basile.git/blob - basile.py
Amélioration de l'erreur de Crash. (Ça fait quand même pas ce que je veux)
[bots/basile.git] / basile.py
1 #!/usr/bin/python
2 # -*- coding:utf8 -*-
3
4 # Codé par 20-100 (commencé le 23/04/12)
5
6 # Un bot IRC qui, un jour, s'interfacera avec la Note Kfet 2015
7
8 import threading
9 import random
10 import time
11 import socket, ssl, json
12 import pickle
13 import re
14 import os
15 import signal
16 import sys
17
18 # Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
19 sys.path.insert(0, "/home/vincent/scripts/python-myirclib")
20 import irclib
21 import ircbot
22
23 from commands import getstatusoutput as ex
24
25 # on récupère la config
26 import config
27
28 # la partie qui réfère au fichier lui-même est mieux ici
29 # sinon on réfère la config et pas le fichier lui-même
30 import os
31 config.thisfile= os.path.realpath( __file__ )
32
33 def get_config_logfile(serveur):
34 serveurs={"acoeur.crans.org":"acoeur","irc.crans.org":"crans"}
35 return config.logfile_template%(serveurs[serveur])
36
37 def get_filesize():
38 return ex("ls -s %s"%(config.thisfile))[1].split()[0]
39
40 class NKError(Exception):
41 def __init__(self,msg):
42 Exception.__init__(self)
43 self.msg=msg
44 def __str__(self):
45 return str(self.msg)
46 def __unicode__(self):
47 return unicode(self.msg)
48
49 class NKRefused(NKError):
50 pass
51
52 class NKHelloFailed(NKError):
53 pass
54
55 class NKUnknownError(NKError):
56 pass
57
58 def log(serveur,channel,auteur=None,message=None):
59 f=open(get_config_logfile(serveur),"a")
60 if auteur==message==None:
61 # alors c'est que c'est pas un channel mais juste une ligne de log
62 chain="%s %s"%(time.strftime("%F %T"),channel)
63 else:
64 chain="%s [%s:%s] %s"%(time.strftime("%F %T"),channel,auteur,message)
65 f.write(chain+"\n")
66 if config.debug_stdout:
67 print chain
68 f.close()
69
70 def connect_NK():
71 sock=socket.socket()
72 try:
73 # On établit la connexion sur port 4242
74 sock.connect(("127.0.0.1",4242))
75 # On passe en SSL
76 sock=ssl.wrap_socket(sock,ca_certs='../keys/ca_.crt')
77 # On fait un hello
78 sock.write('hello "Basile"')
79 # On récupère la réponse du hello
80 out=sock.read()
81 out=json.loads(out)
82 except Exception as exc:
83 # Si on a foiré quelque part, c'est que le serveur est down
84 raise NKRefused(str(exc))
85 if out["retcode"]==0:
86 return sock
87 elif out["retcode"]==11:
88 raise NKHelloFailed(out["errmsg"])
89 else:
90 raise NKUnknownError(out["errmsg"])
91
92 def login_NK(username,password,typ="bdd"):
93 sock=connect_NK()
94 if typ=="special": # ça c'est pour Basile lui-même
95 masque='[]'
96 elif typ=="bdd":
97 masque='[[], [], true]'
98 try:
99 # Basile a un compte special user
100 commande='login [%s,%s,"%s",%s]'%(json.dumps(username),json.dumps(password),typ,masque)
101 sock.write(commande)
102 out=sock.read()
103 except Exception as exc:
104 # Si on a foiré quelque part, c'est que le serveur est down
105 raise NKRefused(str(exc))
106 # On vérifie ensuite que le login
107 return json.loads(out),sock
108
109
110 def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case_sensitive=False,debug=False):
111 if case_sensitive:
112 chain=unicode(chain,"utf8")
113 else:
114 chain=unicode(chain,"utf8").lower()
115 allmatches="("+"|".join(matches)+")"
116 reg=(avant+allmatches+apres).lower()
117 o=re.match(reg,chain)
118 return o
119
120 def is_insult(chain,debug=True):
121 return is_something(chain,config.insultes,avant=".*(?:^| |')")
122 def is_not_insult(chain):
123 chain=unicode(chain,"utf8")
124 insult_regexp=u"("+u"|".join(config.insultes)+u")"
125 middle_regexp=u"(une? (?:(?:putain|enfoiré) d(?:e |'))*|)(?:| super )(?: (?:gros|petit|grand|énorme) |)"
126 reg=".*pas %s%s.*"%(middle_regexp,insult_regexp)
127 if re.match(reg,chain):
128 return True
129 else:
130 return False
131 def is_compliment(chain,debug=True):
132 return is_something(chain,config.compliment_triggers,avant=".*(?:^| |')")
133 def is_perdu(chain):
134 return is_something(chain,config.perdu)
135 def is_tag(chain):
136 return is_something(chain,config.tag_triggers)
137 def is_gros(chain):
138 return is_something(chain,config.gros)
139 def is_tesla(chain):
140 return is_something(chain,config.tesla_triggers,avant=u"^",apres=u"$",debug=True)
141 def is_merci(chain):
142 return is_something(chain,config.merci_triggers)
143 def is_tamere(chain):
144 return is_something(chain,config.tamere_triggers)
145 def is_bad_action_trigger(chain,pseudo):
146 return is_something(chain,config.bad_action_triggers,avant=u"^",
147 apres="(?: [a-z]*ment)? %s($|\.| |,|;).*"%(pseudo))
148 def is_good_action_trigger(chain,pseudo):
149 return is_something(chain,config.good_action_triggers,avant=u"^",
150 apres="(?: [a-z]*ment)? %s($|\.| |,|;).*"%(pseudo))
151 def is_bonjour(chain):
152 return is_something(chain,config.bonjour_triggers,avant=u"^")
153 def is_bonne_nuit(chain):
154 return is_something(chain,config.bonne_nuit_triggers,avant=u"^")
155 def is_pan(chain):
156 return re.match(u"^(pan|bim|bang)( .*)?$",unicode(chain,"utf8").lower().strip())
157
158 def is_time(conf):
159 _,_,_,h,m,s,_,_,_=time.localtime()
160 return (conf[0],0,0)<(h,m,s)<(conf[1],0,0)
161 def is_day():
162 return is_time(config.daytime)
163 def is_night():
164 return is_time(config.nighttime)
165
166
167 class UnicodeBotError(Exception):
168 pass
169
170 class CrashError(Exception):
171 """Pour pouvoir faire crasher Basile, parce que ça a l'air drôle"""
172 def __init__(self, msg=u""):
173 Exception.__init__(self, msg)
174
175 def bot_unicode(chain):
176 try:
177 unicode(chain,"utf8")
178 except UnicodeDecodeError as exc:
179 raise UnicodeBotError
180
181
182 class Basile(ircbot.SingleServerIRCBot):
183 def __init__(self,serveur,debug=False):
184 temporary_pseudo=config.irc_pseudo+str(random.randrange(10000,100000))
185 ircbot.SingleServerIRCBot.__init__(self, [(serveur, 6667)],
186 temporary_pseudo,"Basile, le bot irc.[Codé par 20-100, fouettez-le]", 10)
187 self.debug=debug
188 self.serveur=serveur
189 self.overops=config.overops
190 self.ops=self.overops+config.ops
191 self.report_bugs_to=config.report_bugs_to
192 self.chanlist=config.chanlist
193 self.identities=pickle.load(open("identities.pickle","r"))
194 self.stay_channels=config.stay_channels
195 self.quiet_channels=config.quiet_channels
196 self.last_perdu=0
197
198
199 def new_connection_NK(self,serv,username,password,typ="bdd"):
200 try:
201 login_result,sock=login_NK(username,password,typ)
202 droits,retcode,errmsg=login_result["msg"],login_result["retcode"],login_result["errmsg"]
203 except NKRefused as exc:
204 for report in self.report_bugs_to:
205 serv.privmsg(report,"Le Serveur NK2015 est down.")
206 return (False,None)
207 except NKHelloFailed as exc:
208 for report in self.report_bugs_to:
209 serv.privmsg(report,
210 "La version du site utilisée n'est pas supportée par le serveur NK2015.")
211 return (False,None)
212 except NKUnknownError as exc:
213 erreurs=["Une fucking erreur inconnue s'est produite"]
214 erreurs+=str(exc).split("\n")
215 for report in self.report_bugs_to:
216 for err in erreurs:
217 serv.privmsg(report,err)
218 return (False,None)
219 except Exception as exc:
220 # Exception qui ne vient pas de la communication avec le serveur NK2015
221 log(self.serveur,"Erreur dans new_connection_NK\n"+str(exc))
222 return (False,None)
223 if retcode==0:
224 return (True,sock)
225 else:
226 return (False,None)
227
228 def give_me_my_pseudo(self,serv):
229 serv.privmsg("NickServ","RECOVER %s %s"%(config.irc_pseudo,config.irc_password))
230 serv.privmsg("NickServ","RELEASE %s %s"%(config.irc_pseudo,config.irc_password))
231 time.sleep(0.3)
232 serv.nick(config.irc_pseudo)
233
234 def on_welcome(self, serv, ev):
235 self.serv=serv # ça serv ira :)
236 self.give_me_my_pseudo(serv)
237 serv.privmsg("NickServ","identify %s"%(config.irc_password))
238 log(self.serveur,"Connected")
239 if self.debug:
240 self.chanlist=["#bot"]
241 for c in self.chanlist:
242 log(self.serveur,"JOIN %s"%(c))
243 serv.join(c)
244 # on ouvre la connexion note de Basile, special user
245 self.nk=self.new_connection_NK(serv,config.note_pseudo,config.note_password,"special")[1]
246 if self.nk==None:
247 for report in self.report_bugs_to:
248 serv.privmsg(report,"Connection to NK2015 failed, invalid password ?")
249
250 def lost(self,serv,channel,forced=False):
251 if self.last_perdu+config.time_between_perdu<time.time() or forced:
252 if not channel in self.quiet_channels or forced:
253 serv.privmsg(channel,"J'ai perdu !")
254 self.last_perdu=time.time()
255 delay=config.time_between_perdu_trigger
256 delta=config.time_between_perdu_trigger_delta
257 serv.execute_delayed(random.randrange(delay-delta,delay+delta),self.lost,(serv,channel))
258
259 def pourmoi(self, serv, message):
260 """renvoie (False,lemessage) ou (True, le message amputé de "pseudo: ")"""
261 pseudo=self.nick
262 size=len(pseudo)
263 if message[:size]==pseudo and len(message)>size and message[size]==":":
264 return (True,message[size+1:].lstrip(" "))
265 else:
266 return (False,message)
267
268 def on_privmsg(self, serv, ev):
269 message=ev.arguments()[0]
270 auteur = irclib.nm_to_n(ev.source())
271 try:
272 test=bot_unicode(message)
273 except UnicodeBotError:
274 if config.utf8_trigger:
275 serv.privmsg(auteur, random.choice(config.utf8_fail_answers).encode("utf8"))
276 return
277 message=message.split()
278 cmd=message[0].lower()
279 notunderstood=False
280 if cmd=="help":
281 helpdico={"help":["""HELP <commande>
282 Affiche de l'aide sur la commande""",None,None],
283 "identify": ["""IDENTIFY <username> <password>
284 Vérifie le mot de passe et me permet de savoir à l'avenir quel est votre pseudo note kfet.
285 Sans paramètre, je vous précise sous quel pseudo je vous connais.""",None,None],
286 "drop":["""DROP <password>
287 Vérifie le mot de passe et me fait d'oublier votre pseudo note kfet.""",None,None],
288 "solde": ["""SOLDE
289 Affiche votre solde, si je connais votre pseudo note kfet.""",
290 """SOLDE <pseudo>
291 Affiche le solde de la personne désignée (par son pseudo note).""",None],
292 "join": [None, """JOIN <channel>
293 Me fait rejoindre le channel""",None],
294 "leave": [None,"""LEAVE <channel>
295 Me fait quitter le channel (sauf s'il est dans ma stay_list).""",None],
296 "quiet": [None,"""QUIET <channel>
297 Me rend silencieux sur le channel.""",None],
298 "noquiet": [None,"""NOQUIET <channel>
299 Me rend la parole sur le channel.""",None],
300 "lost": [None,"""LOST <channel>
301 Me fait perdre sur le channel.""",None],
302 "reconnect": [None,"""RECONNECT
303 Établit à nouveau la connexion avec le serveur NK2015""",None],
304 "reload": [None,"""RELOAD
305 Recharge la configuration.""",None],
306 "say": [None,None,"""SAY <channel> <message>
307 Me fait parler sur le channel."""],
308 "do": [None,None,"""DO <channel> <action>
309 Me fait faitre une action (/me) sur le channel."""],
310 "stay": [None,None,"""STAY <channel>
311 Ajoute le channel à ma stay_list."""],
312 "nostay": [None,None,"""NOSTAY <channel>
313 Retire le channel de ma stay_list."""],
314 "ops": [None,None,"""OPS
315 Affiche la liste des ops."""],
316 "overops": [None,None,"""OVEROPS
317 Affiche la liste des overops."""],
318 "kick": [None,None,"""KICK <channel> <pseudo> [<raison>]
319 Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
320 "die": [None,None,"""DIE
321 Me déconnecte du serveur IRC."""],
322 "crash": [None,None,"""CRASH
323 Me fait crasher"""]
324 }
325 helpmsg_default="Liste des commandes disponibles :\nHELP IDENTIFY DROP SOLDE"
326 helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST RECONNECT RELOAD"
327 helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE CRASH"
328 op,overop=auteur in self.ops, auteur in self.overops
329 if len(message)==1:
330 helpmsg=helpmsg_default
331 if op:
332 helpmsg+=helpmsg_ops
333 if overop:
334 helpmsg+=helpmsg_overops
335 else:
336 helpmsgs=helpdico.get(message[1].lower(),["Commande inconnue.",None,None])
337 helpmsg=helpmsgs[0]
338 if op and helpmsgs[1]:
339 if helpmsg:
340 helpmsg+="\n"+helpmsgs[1]
341 else:
342 helpmsg=helpmsgs[1]
343 if overop and helpmsgs[2]:
344 if helpmsg:
345 helpmsg+="\n"+helpmsgs[2]
346 else:
347 helpmsg=helpmsgs[2]
348 for ligne in helpmsg.split("\n"):
349 serv.privmsg(auteur,ligne)
350 elif cmd=="identify":
351 if len(message)==1:
352 if self.identities.has_key(auteur):
353 serv.privmsg(auteur,"Je vous connais sous le pseudo note %s."%(
354 self.identities[auteur].encode("utf8")))
355 else:
356 serv.privmsg(auteur,"Je ne connais pas votre pseudo note.")
357 elif len(message)>=3:
358 username,password=message[1],unicode(" ".join(message[2:]),"utf8")
359 success,_=self.new_connection_NK(serv,username,password)
360 if success:
361 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
362 serv.privmsg(auteur,"Identité enregistrée.")
363 self.identities[auteur]=username
364 pickle.dump(Xself.identities,open("identities.pickle","w"))
365 else:
366 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
367 serv.privmsg(auteur,"Mot de passe invalide. (ou serveur down)")
368 else:
369 serv.privmsg(auteur,u"Syntaxe : IDENTIFY [<username> <password>]")
370 elif cmd=="drop":
371 if len(message)>1:
372 if self.identities.has_key(auteur):
373 password=" ".join(message[1:])
374 success,_=self.new_connection_NK(serv,self.identities[auteur],password)
375 if success:
376 del self.identities[auteur]
377 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
378 pickle.dump(self.identities,open("identities.pickle","w"))
379 serv.privmsg(auteur,"Identité oubliée.")
380 else:
381 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
382 serv.privmsg(auteur,"Mot de passe invalide. (ou serveur down)")
383 else:
384 serv.privmsg(auteur,"Je ne connais pas ton pseudo note.")
385 else:
386 serv.privmsg(auteur,"Syntaxe : DROP <password>")
387 elif cmd=="join":
388 if auteur in self.ops:
389 if len(message)>1:
390 if message[1] in self.chanlist:
391 serv.privmsg(auteur,"Je suis déjà sur %s"%(message[1]))
392 else:
393 serv.join(message[1])
394 self.chanlist.append(message[1])
395 serv.privmsg(auteur,"Channels : "+" ".join(self.chanlist))
396 log(self.serveur,"priv",auteur," ".join(message))
397 else:
398 serv.privmsg(auteur,"Channels : "+" ".join(self.chanlist))
399 else:
400 notunderstood=True
401 elif cmd=="leave":
402 if auteur in self.ops and len(message)>1:
403 if message[1] in self.chanlist:
404 if not (message[1] in self.stay_channels) or auteur in self.overops:
405 self.quitter(message[1]," ".join(message[2:]))
406 self.chanlist.remove(message[1])
407 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
408 else:
409 serv.privmsg(auteur,"Non, je reste !")
410 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
411 else:
412 serv.privmsg(auteur,"Je ne suis pas sur %s"%(message[1]))
413 else:
414 notunderstood=True
415 elif cmd=="stay":
416 if auteur in self.overops:
417 if len(message)>1:
418 if message[1] in self.stay_channels:
419 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
420 serv.privmsg(auteur,"Je stay déjà sur %s."%(message[1]))
421 else:
422 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
423 self.stay_channels.append(message[1])
424 serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
425 else:
426 serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
427 else:
428 notunderstood=True
429 elif cmd=="nostay":
430 if auteur in self.overops:
431 if len(message)>1:
432 if message[1] in self.stay_channels:
433 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
434 self.stay_channels.remove(message[1])
435 serv.privmsg(auteur,"Stay channels : "+" ".join(self.stay_channels))
436 else:
437 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
438 serv.privmsg(auteur,"Je ne stay pas sur %s."%(message[1]))
439
440 else:
441 notunderstood=True
442 elif cmd=="die":
443 if auteur in self.overops:
444 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
445 self.mourir()
446 else:
447 notunderstood=True
448 elif cmd=="crash":
449 if auteur in self.overops:
450 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
451 self.crash("priv", auteur)
452 else:
453 notunderstood=True
454 elif cmd=="reload":
455 if auteur in self.ops:
456 self.reload(auteur)
457 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
458 else:
459 notunderstood=True
460 elif cmd=="reconnect":
461 if auteur in self.ops:
462 try:
463 self.nk=self.new_connection_NK(serv,config.note_pseudo,
464 config.note_password,"special")[1]
465 except Exception as exc:
466 self.nk=None
467 log(self.serveur,"""Erreur dans on_pubmsg/"cmd in ["reconnect"]\n"""+str(exc))
468 if self.nk!=None:
469 serv.privmsg(auteur,"%s: done"%(auteur))
470 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
471 else:
472 serv.privmsg(auteur,"%s: failed"%(auteur))
473 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
474 for report in self.report_bugs_to:
475 serv.privmsg(report,"Connection to NK2015 failed, invalid password ? Server dead ?")
476 else:
477 notunderstood=True
478 elif cmd=="quiet":
479 if auteur in self.ops:
480 if len(message)>1:
481 if message[1] in self.quiet_channels:
482 serv.privmsg(auteur,"Je me la ferme déjà sur %s"%(message[1]))
483 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
484 else:
485 self.quiet_channels.append(message[1])
486 serv.privmsg(auteur,"Quiet channels : "+" ".join(self.quiet_channels))
487 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
488 else:
489 serv.privmsg(auteur,"Quiet channels : "+" ".join(self.quiet_channels))
490 else:
491 notunderstood=True
492 elif cmd=="noquiet":
493 if auteur in self.ops:
494 if len(message)>1:
495 if message[1] in self.quiet_channels:
496 self.quiet_channels.remove(message[1])
497 serv.privmsg(auteur,"Quiet channels : "+" ".join(self.quiet_channels))
498 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
499 else:
500 serv.privmsg(auteur,"Je ne me la ferme pas sur %s."%(message[1]))
501 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
502 else:
503 notunderstood=True
504 elif cmd=="say":
505 if auteur in self.overops and len(message)>2:
506 serv.privmsg(message[1]," ".join(message[2:]))
507 log(self.serveur,"priv",auteur," ".join(message))
508 elif len(message)<=2:
509 serv.privmsg(auteur,"Syntaxe : SAY <channel> <message>")
510 else:
511 notunderstood=True
512 elif cmd=="do":
513 if auteur in self.overops and len(message)>2:
514 serv.action(message[1]," ".join(message[2:]))
515 log(self.serveur,"priv",auteur," ".join(message))
516 elif len(message)<=2:
517 serv.privmsg(auteur,"Syntaxe : DO <channel> <action>")
518 else:
519 notunderstood=True
520 elif cmd=="kick":
521 if auteur in self.overops and len(message)>2:
522 serv.kick(message[1],message[2]," ".join(message[3:]))
523 log(self.serveur,"priv",auteur," ".join(message))
524 elif len(message)<=2:
525 serv.privmsg(auteur,"Syntaxe : KICK <channel> <pseudo> [<raison>]")
526 else:
527 notunderstood=True
528 elif cmd=="lost":
529 if auteur in self.ops and len(message)>1:
530 serv.privmsg(message[1],"J'ai perdu !")
531 log(self.serveur,"priv",auteur," ".join(message))
532 elif len(message)<=1:
533 serv.privmsg(auteur,"Syntaxe : LOST <channel>")
534 else:
535 notunderstood=True
536 elif cmd=="solde":
537 if len(message)==1:
538 if self.identities.has_key(auteur):
539 try:
540 self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(self.identities[auteur])))
541 ret=json.loads(self.nk.read())
542 solde=ret["msg"][0]["solde"]
543 pseudo=ret["msg"][0]["pseudo"]
544 except Exception as exc:
545 print exc
546 serv.privmsg(auteur,"failed")
547 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
548 return
549 serv.privmsg(auteur,"%s (%s)"%(float(solde)/100,pseudo.encode("utf8")))
550 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
551 else:
552 serv.privmsg(canal,"Je ne connais pas ton pseudo note.")
553 elif auteur in self.ops:
554 try:
555 self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(message[1])))
556 ret=json.loads(self.nk.read())
557 solde=ret["msg"][0]["solde"]
558 pseudo=ret["msg"][0]["pseudo"]
559 except Exception as exc:
560 serv.privmsg(auteur,"failed")
561 log(self.serveur,"priv",auteur," ".join(message)+"[failed]")
562 return
563 serv.privmsg(auteur,"%s (%s)"%(float(solde)/100,pseudo.encode("utf8")))
564 elif cmd=="ops":
565 if auteur in self.overops:
566 serv.privmsg(auteur," ".join(self.ops))
567 else:
568 notunderstood=True
569 elif cmd=="overops":
570 if auteur in self.overops:
571 serv.privmsg(auteur," ".join(self.overops))
572 else:
573 notunderstood=True
574 else:
575 notunderstood=True
576 if notunderstood:
577 serv.privmsg(auteur,"Je n'ai pas compris. Essayez HELP…")
578
579 def on_pubmsg(self, serv, ev):
580 auteur = irclib.nm_to_n(ev.source())
581 canal = ev.target()
582 message = ev.arguments()[0]
583 try:
584 test=bot_unicode(message)
585 except UnicodeBotError:
586 if config.utf8_trigger and not canal in self.quiet_channels:
587 serv.privmsg(canal, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
588 return
589 pour_moi,message=self.pourmoi(serv,message)
590 if pour_moi and message.split()!=[]:
591 cmd=message.split()[0].lower()
592 try:
593 args=" ".join(message.split()[1:])
594 except:
595 args=""
596 if cmd in ["meurs","die","crève"]:
597 if auteur in self.overops:
598 log(self.serveur,canal,auteur,message+"[successful]")
599 self.mourir()
600 else:
601 serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8"))
602 log(self.serveur,canal,auteur,message+"[failed]")
603 elif cmd == "reload":
604 if auteur in self.ops:
605 log(self.serveur, canal, auteur, message+"[successful]")
606 self.reload(canal)
607 elif cmd == "crash":
608 if auteur in self.overops:
609 self.crash(auteur, message)
610 elif cmd in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]:
611 if auteur in self.ops and (not (canal in self.stay_channels)
612 or auteur in self.overops):
613 self.quitter(canal)
614 log(self.serveur,canal,auteur,message+"[successful]")
615 if canal in self.chanlist:
616 self.chanlist.remove(canal)
617 else:
618 serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.leave_fail_messages))).encode("utf8"))
619 log(self.serveur,canal,auteur,message+"[failed]")
620
621 elif cmd in ["reconnect"]:
622 if auteur in self.ops:
623 try:
624 self.nk=self.new_connection_NK(serv,config.note_pseudo,
625 config.note_password,"special")[1]
626 except Exception as exc:
627 self.nk=None
628 log(self.serveur,"""Erreur dans on_pubmsg/"cmd in ["reconnect"]\n"""+str(exc))
629 if self.nk!=None:
630 serv.privmsg(canal,"%s: done"%(auteur))
631 log(self.serveur,canal,auteur,message+"[successful]")
632 else:
633 serv.privmsg(canal,"%s: failed"%(auteur))
634 log(self.serveur,canal,auteur,message+"[failed]")
635 for report in self.report_bugs_to:
636 serv.privmsg(report,"Connection to NK2015 failed, invalid password ? Server dead ?")
637 else:
638 serv.privmsg(canal,"%s: %s"%(auteur,random.choice(config.pas_programme_pour_tobeir).encode("utf8")))
639 log(self.serveur,canal,auteur,message+"[failed]")
640
641 elif cmd in ["deviens","pseudo"]:
642 if auteur in self.ops:
643 become=args
644 serv.nick(become)
645 log(self.serveur,canal,auteur,message+"[successful]")
646
647 if cmd in ["meur", "meurt","meurre","meurres"] and not canal in self.quiet_channels:
648 serv.privmsg(canal,'%s: Mourir, impératif, 2ème personne du singulier : "meurs" (de rien)'%(auteur))
649 elif cmd in ["ping"] and not canal in self.quiet_channels:
650 serv.privmsg(canal,"%s: pong"%(auteur))
651
652 elif cmd in ["solde","!solde"]:
653 if self.identities.has_key(auteur):
654 pseudo=self.identities[auteur]
655 try:
656 self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(pseudo)))
657 ret=json.loads(self.nk.read())
658 solde=ret["msg"][0]["solde"]
659 pseudo=ret["msg"][0]["pseudo"]
660 except Exception as exc:
661 serv.privmsg(canal,"%s: failed"%(auteur))
662 log(self.serveur,canal,auteur,message+"[failed]")
663 else:
664 serv.privmsg(canal,"%s: %s (%s)"%(auteur,float(solde)/100,pseudo.encode("utf8")))
665 log(self.serveur,canal,auteur,message+"[successful]")
666 else:
667 serv.privmsg(canal,"%s: Je ne connais pas votre pseudo note."%(auteur))
668 log(self.serveur,canal,auteur,message+"[unknown]")
669 elif (re.match("!?(pain au chocolat|chocolatine)",message.lower())
670 and not canal in self.quiet_channels):
671 serv.action(canal,"sert un pain au chocolat à %s"%(auteur))
672 elif re.match("!?manzana",message.lower()) and not canal in self.quiet_channels:
673 if auteur in config.manzana:
674 serv.action(canal,"sert une bouteille de manzana à %s"%(auteur))
675 elif auteur in config.manzana_bis:
676 serv.action(canal,"sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas."%(auteur))
677 else:
678 serv.action(canal,"sert un verre de manzana à %s"%(auteur))
679 if is_insult(message) and not canal in self.quiet_channels:
680 if is_not_insult(message):
681 answer=random.choice(config.compliment_answers)
682 for ligne in answer.split("\n"):
683 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
684 else:
685 answer=random.choice(config.insultes_answers)
686 for ligne in answer.split("\n"):
687 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
688 elif is_compliment(message) and not canal in self.quiet_channels:
689 answer=random.choice(config.compliment_answers)
690 for ligne in answer.split("\n"):
691 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
692 gros_match=is_gros(message)
693 if gros_match and not canal in self.quiet_channels:
694 taille=get_filesize()
695 answer=u"Mais non, je ne suis pas %s, %sKo tout au plus…"%(gros_match.groups()[0],taille)
696 serv.privmsg(canal,"%s: %s"%(auteur,answer.encode("utf8")))
697 if is_tesla(message) and not canal in self.quiet_channels:
698 l1,l2=config.tesla_answers,config.tesla_actions
699 n1,n2=len(l1),len(l2)
700 i=random.randrange(n1+n2)
701 if i>=n1:
702 serv.action(canal,l2[i-n1].encode("utf8"))
703 else:
704 serv.privmsg(canal,"%s: %s"%(auteur,l1[i].encode("utf8")))
705 if is_tamere(message) and not canal in self.quiet_channels:
706 answer=random.choice(config.tamere_answers)
707 for ligne in answer.split("\n"):
708 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
709 if is_tag(message) and not canal in self.quiet_channels:
710 if auteur in self.ops:
711 action=random.choice(config.tag_actions)
712 serv.action(canal,action.encode("utf8"))
713 self.quiet_channels.append(canal)
714 else:
715 answer=random.choice(config.tag_answers)
716 for ligne in answer.split("\n"):
717 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
718 if is_merci(message):
719 answer=random.choice(config.merci_answers)
720 for ligne in answer.split("\n"):
721 serv.privmsg(canal,"%s: %s"%(auteur,ligne.encode("utf8")))
722 out=re.match(ur"^([A-Z[]|\\|[0-9]+|(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+)(?:| \?| !)$",
723 unicode(message.upper(),"utf8"))
724 if re.match("ma bite dans ton oreille",message) and not canal in self.quiet_channels:
725 serv.privmsg(canal,"%s: Seul un olasd peut imiter un olasd dans un de ses grands jours !"%(auteur))
726 if out and not canal in self.quiet_channels:
727 out=out.groups()[0]
728 try:
729 out=int(out)
730 serv.privmsg(canal,"%s: %s !"%(auteur,out+1))
731 if out==2147483647:
732 serv.privmsg(canal,"%s: Ciel, un maxint ! Heureusement que je suis en python…"%(auteur))
733 return
734 if out+1>1000 and random.randrange(4)==0:
735 serv.privmsg(canal,"%s: Vous savez, moi et les chiffres…"%(auteur))
736 return
737 except Exception as exc:
738 pass
739 if re.match("[A-Y]",out):
740 alphabet="ABCDEFGHIJKLMNOPQRSTUVWXYZ"
741 serv.privmsg(canal,"%s: %s !"%(auteur,alphabet[alphabet.index(out)+1]))
742 elif out=="Z":
743 serv.privmsg(canal,"%s: Je ne vous remercie pas, j'ai l'air idiot ainsi… [ ?"%(auteur))
744 elif out in "[\\":
745 serv.privmsg(canal,"%s: Nous devrions nous en tenir là, ça va finir par poser des problèmes…"%(auteur))
746 elif re.match(ur"(¹|²|³|⁴|⁵|⁶|⁷|⁸|⁹|⁰)+",out):
747 def translate(mess):
748 return "".join([{u"⁰¹²³⁴⁵⁶⁷⁸⁹0123456789"[i]:u"0123456789⁰¹²³⁴⁵⁶⁷⁸⁹"[i]
749 for i in range(20)}[j]
750 for j in mess])
751 out=int(translate(out))
752 serv.privmsg(canal,"%s: %s !"%(auteur,translate(str(out+1)).encode("utf8")))
753 if is_bonjour(message) and not canal in self.quiet_channels:
754 if is_night():
755 answer=random.choice(config.night_answers)
756 elif is_day():
757 answer=random.choice(config.bonjour_answers)
758 else:
759 answer=random.choice(config.bonsoir_answers)
760 serv.privmsg(canal,answer.format(auteur).encode("utf8"))
761 if is_bonne_nuit(message) and not canal in self.quiet_channels:
762 answer=random.choice(config.bonne_nuit_answers)
763 serv.privmsg(canal,answer.format(auteur).encode("utf8"))
764 if is_pan(message) and not canal in self.quiet_channels:
765 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))
766 else:
767 if message in ["!pain au chocolat","!chocolatine"] and not canal in self.quiet_channels:
768 serv.action(canal,"sert un pain au chocolat à %s"%(auteur))
769 if message in ["!manzana"] and not canal in self.quiet_channels:
770 if auteur in config.manzana:
771 serv.action(canal,"sert une bouteille de manzana à %s"%(auteur))
772 elif auteur in config.manzana_bis:
773 serv.action(canal,"sert un grand verre de jus de pomme à %s : tout le monde sait qu'il ne boit pas."%(auteur))
774 else:
775 serv.action(canal,"sert un verre de manzana à %s"%(auteur))
776 if re.match(u'^ *(.|§|!|/|/|:|)(w|b) [0-9]+$',message.decode("utf8")) and not canal in self.quiet_channels:
777 failanswers=config.buffer_fail_answers
778 answer=random.choice(failanswers)
779 serv.privmsg(canal,("%s: %s"%(auteur,answer)).encode("utf8"))
780 if not canal in self.quiet_channels:
781 mypseudo=self.nick
782 if re.match((u"^("+u"|".join(config.bonjour_triggers)
783 +ur")( {}| all| tout le monde| (à )?tous)(\.| ?!)?$"
784 ).format(mypseudo).lower(), message.decode("utf8").strip().lower()):
785 answer=random.choice(config.bonjour_answers)
786 serv.privmsg(canal,answer.format(auteur).encode("utf8"))
787 if (is_perdu(message) and not canal in self.quiet_channels):
788 # proba de perdre sur trigger :
789 # avant 30min (enfin, config) : 0
790 # ensuite, +25%/30min, linéairement
791 deltat=time.time()-self.last_perdu
792 barre=(deltat-config.time_between_perdu)/(2*3600.0)
793 if random.uniform(0,1)<barre:
794 serv.privmsg(canal,"%s: J'ai perdu !"%(auteur))
795 self.last_perdu=time.time()
796
797 def on_action(self, serv, ev):
798 action = ev.arguments()[0]
799 auteur = irclib.nm_to_n(ev.source())
800 channel = ev.target()
801 try:
802 test=bot_unicode(action)
803 except UnicodeBotError:
804 if config.utf8_trigger and not channel in self.quiet_channels:
805 serv.privmsg(channel, (u"%s: %s"%(auteur,random.choice(config.utf8_fail_answers))).encode("utf8"))
806 return
807 mypseudo=self.nick
808
809 if is_bad_action_trigger(action,mypseudo) and not channel in self.quiet_channels:
810 l1,l2=config.bad_action_answers,config.bad_action_actions
811 n1,n2=len(l1),len(l2)
812 i=random.randrange(n1+n2)
813 if i>=n1:
814 serv.action(channel,l2[i-n1].format(auteur).encode("utf8"))
815 else:
816 serv.privmsg(channel,l1[i].format(auteur).format(auteur).encode("utf8"))
817 if is_good_action_trigger(action,mypseudo) and not channel in self.quiet_channels:
818 l1,l2=config.good_action_answers,config.good_action_actions
819 n1,n2=len(l1),len(l2)
820 i=random.randrange(n1+n2)
821 if i>=n1:
822 serv.action(channel,l2[i-n1].format(auteur).format(auteur).encode("utf8"))
823 else:
824 serv.privmsg(channel,l1[i].format(auteur).format(auteur).encode("utf8"))
825
826 def on_kick(self,serv,ev):
827 auteur = irclib.nm_to_n(ev.source())
828 channel = ev.target()
829 victime = ev.arguments()[0]
830 raison = ev.arguments()[1]
831 if victime==self.nick:
832 log(self.serveur,"%s kické de %s par %s (raison : %s)" %(victime,channel,auteur,raison))
833 time.sleep(2)
834 serv.join(channel)
835 l1,l2=config.kick_answers,config.kick_actions
836 n1,n2=len(l1),len(l2)
837 i=random.randrange(n1+n2)
838 if i>=n1:
839 serv.action(channel,l2[i-n1].format(auteur).encode("utf8"))
840 else:
841 serv.privmsg(channel,l1[i].format(auteur).encode("utf8"))
842
843 def quitter(self,chan,leave_message=None):
844 if leave_message==None:
845 leave_message=random.choice(config.leave_messages)
846 self.serv.part(chan,message=leave_message.encode("utf8"))
847
848 def mourir(self):
849 quit_message=random.choice(config.quit_messages)
850 self.die(msg=quit_message.encode("utf8"))
851
852 def _getnick(self):
853 return self.serv.get_nickname()
854 nick=property(_getnick)
855
856 def reload(self, auteur=None):
857 reload(config)
858 if auteur in [None, "SIGHUP"]:
859 towrite = "Config reloaded" + " (SIGHUP received)"*(auteur == "SIGHUP")
860 for to in config.report_bugs_to:
861 self.serv.privmsg(to, towrite)
862 log(self.serveur, towrite)
863 else:
864 self.serv.privmsg(auteur,"Config reloaded")
865
866 def crash(self, chan="nowhere", who="nobody"):
867 where = u"en privé" if chan == "priv" else u"sur le chan %s" % chan
868 raise CrashError(u"Crash demandé par %s %s" % (who, where))
869
870 def start_as_daemon(self, outfile):
871 sys.stderr = Logger(outfile)
872 self.start()
873
874
875 class Logger(object):
876 """Pour écrire ailleurs que sur stdout"""
877 def __init__(self, filename="basile.full.log"):
878 self.filename = filename
879
880 def write(self, message):
881 f = open(self.filename, "a")
882 f.write(message)
883 f.close()
884
885 def main():
886 if len(sys.argv)==1:
887 print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
888 print " --outfile sans --no-output ni --daemon n'a aucun effet"
889 exit(1)
890 serveur=sys.argv[1]
891 if "--daemon" in sys.argv:
892 thisfile = os.path.realpath(__file__)
893 thisdirectory = thisfile.rsplit("/", 1)[0]
894 os.chdir(thisdirectory)
895 daemon = True
896 else:
897 daemon = False
898 if "debug" in sys.argv or "--debug" in sys.argv:
899 debug=True
900 else:
901 debug=False
902 if "--quiet" in sys.argv:
903 config.debug_stdout=False
904 serveurs={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
905 "irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
906 if "--no-output" in sys.argv or "--daemon" in sys.argv:
907 outfile = "/var/log/bots/basile.full.log"
908 for arg in sys.argv:
909 arg = arg.split("=")
910 if arg[0].strip('-') in ["out", "outfile", "logfile"]:
911 outfile = arg[1]
912 sys.stdout = Logger(outfile)
913 try:
914 serveur=serveurs[serveur]
915 except KeyError:
916 print "Server Unknown : %s"%(serveur)
917 exit(404)
918 basile=Basile(serveur,debug)
919 # Si on reçoit un SIGHUP, on reload la config
920 def sighup_handler(signum, frame):
921 basile.reload("SIGHUP")
922 signal.signal(signal.SIGHUP, sighup_handler)
923 if daemon:
924 child_pid = os.fork()
925 if child_pid == 0:
926 os.setsid()
927 basile.start_as_daemon(outfile)
928 else:
929 # on enregistre le pid de basile
930 pidfile = "/var/run/bots/basile.pid"
931 for arg in sys.argv:
932 arg = arg.split("=")
933 if arg[0].strip('-') in ["pidfile"]:
934 pidfile = arg[1]
935 f = open(pidfile, "w")
936 f.write("%s\n" % child_pid)
937 f.close()
938 else:
939 basile.start()
940
941 if __name__ == "__main__":
942 main()