# Codé par 20-100 (commencé le 23/04/12)
-# Un bot IRC qui, un jour, s'interfacera avec la Note Kfet 2015
+""" Un bot IRC destiné à s'interfacer avec la Note Kfet 2015 """
-import irclib
-import ircbot
import threading
import random
import time
-import socket, ssl, json
+import json
import pickle
import re
import os
+import signal
+import sys
+
+# Oui, j'ai recodé ma version d'irclib pour pouvoir rattrapper les SIGHUP
+sys.path.insert(0, "/home/vincent/scripts/python-myirclib")
+import irclib
+import ircbot
+
from commands import getstatusoutput as ex
-# on récupère la config
-import config_basile as config
+#: Config de basile
+import config
+#: Module responsable du dialogue avec la NoteKfet2015
+import nk
# la partie qui réfère au fichier lui-même est mieux ici
# sinon on réfère la config et pas le fichier lui-même
def get_filesize():
return ex("ls -s %s"%(config.thisfile))[1].split()[0]
-class NKError(Exception):
- def __init__(self,msg):
- Exception.__init__(self)
- self.msg=msg
- def __str__(self):
- return str(self.msg)
- def __unicode__(self):
- return unicode(self.msg)
-
-class NKRefused(NKError):
- pass
-
-class NKHelloFailed(NKError):
- pass
-
-class NKUnknownError(NKError):
- pass
-
def log(serveur,channel,auteur=None,message=None):
f=open(get_config_logfile(serveur),"a")
if auteur==message==None:
print chain
f.close()
-def connect_NK():
- sock=socket.socket()
- try:
- # On établit la connexion sur port 4242
- sock.connect(("127.0.0.1",4242))
- # On passe en SSL
- sock=ssl.wrap_socket(sock,ca_certs='../keys/ca_.crt')
- # On fait un hello
- sock.write('hello "Basile"')
- # On récupère la réponse du hello
- out=sock.read()
- out=json.loads(out)
- except Exception as exc:
- # Si on a foiré quelque part, c'est que le serveur est down
- raise NKRefused(str(exc))
- if out["retcode"]==0:
- return sock
- elif out["retcode"]==11:
- raise NKHelloFailed(out["errmsg"])
- else:
- raise NKUnknownError(out["errmsg"])
-
-def login_NK(username,password,typ="bdd"):
- sock=connect_NK()
- if typ=="special": # ça c'est pour Basile lui-même
- masque='["note"]'
- elif typ=="bdd":
- masque='[["all"],["all"],false]'
- try:
- # Basile a un compte special user
- commande='login [%s,%s,"%s",%s]'%(json.dumps(username),json.dumps(password),typ,masque)
- sock.write(commande)
- out=sock.read()
- except Exception as exc:
- # Si on a foiré quelque part, c'est que le serveur est down
- raise NKRefused(str(exc))
- # On vérifie ensuite que le login
- return json.loads(out),sock
-
-
def is_something(chain,matches,avant=u".*(?:^| )",apres=u"(?:$|\.| |,|;).*",case_sensitive=False,debug=False):
if case_sensitive:
chain=unicode(chain,"utf8")
class UnicodeBotError(Exception):
pass
+
+class CrashError(Exception):
+ """Pour pouvoir faire crasher Basile, parce que ça a l'air drôle"""
+ def __init__(self, msg=u""):
+ Exception.__init__(self, msg)
+
def bot_unicode(chain):
try:
unicode(chain,"utf8")
except UnicodeDecodeError as exc:
raise UnicodeBotError
+
class Basile(ircbot.SingleServerIRCBot):
def __init__(self,serveur,debug=False):
temporary_pseudo=config.irc_pseudo+str(random.randrange(10000,100000))
def new_connection_NK(self,serv,username,password,typ="bdd"):
+ """Renvoie (``True``, <une socket ouverte et authentifiée sur la NoteKfet2015>)
+ ou bien (``False``, None)"""
try:
- login_result,sock=login_NK(username,password,typ)
+ login_result,sock=nk.login(username,password,typ)
droits,retcode,errmsg=login_result["msg"],login_result["retcode"],login_result["errmsg"]
- except NKRefused as exc:
+ except nk.NKRefused as exc:
for report in self.report_bugs_to:
serv.privmsg(report,"Le Serveur NK2015 est down.")
return (False,None)
- except NKHelloFailed as exc:
+ except nk.NKHelloFailed as exc:
for report in self.report_bugs_to:
serv.privmsg(report,
"La version du site utilisée n'est pas supportée par le serveur NK2015.")
return (False,None)
- except NKUnknownError as exc:
+ except nk.NKUnknownError as exc:
erreurs=["Une fucking erreur inconnue s'est produite"]
erreurs+=str(exc).split("\n")
for report in self.report_bugs_to:
return (False,None)
def give_me_my_pseudo(self,serv):
+ """Récupère le pseudo auprès de NickServ."""
serv.privmsg("NickServ","RECOVER %s %s"%(config.irc_pseudo,config.irc_password))
serv.privmsg("NickServ","RELEASE %s %s"%(config.irc_pseudo,config.irc_password))
time.sleep(0.3)
log(self.serveur,"Connected")
if self.debug:
self.chanlist=["#bot"]
- else:
- serv.privmsg("ChanServ", "INVITE #note-dev")
for c in self.chanlist:
log(self.serveur,"JOIN %s"%(c))
serv.join(c)
Me fait perdre sur le channel.""",None],
"reconnect": [None,"""RECONNECT
Établit à nouveau la connexion avec le serveur NK2015""",None],
+ "reload": [None,"""RELOAD
+ Recharge la configuration.""",None],
"say": [None,None,"""SAY <channel> <message>
Me fait parler sur le channel."""],
"do": [None,None,"""DO <channel> <action>
"kick": [None,None,"""KICK <channel> <pseudo> [<raison>]
Kicke <pseudo> du channel (Il faut bien entendu que j'y sois op)."""],
"die": [None,None,"""DIE
- Me déconnecte du serveur IRC."""]
+ Me déconnecte du serveur IRC."""],
+ "crash": [None,None,"""CRASH
+ Me fait crasher"""]
}
helpmsg_default="Liste des commandes disponibles :\nHELP IDENTIFY DROP SOLDE"
- helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST RECONNECT"
- helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE"
+ helpmsg_ops=" JOIN LEAVE QUIET NOQUIET LOST RECONNECT RELOAD"
+ helpmsg_overops=" SAY DO STAY NOSTAY OPS OVEROPS KICK DIE CRASH"
op,overop=auteur in self.ops, auteur in self.overops
if len(message)==1:
helpmsg=helpmsg_default
self.mourir()
else:
notunderstood=True
+ elif cmd=="crash":
+ if auteur in self.overops:
+ log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+ self.crash("priv", auteur)
+ else:
+ notunderstood=True
+ elif cmd=="reload":
+ if auteur in self.ops:
+ self.reload(auteur)
+ log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
+ else:
+ notunderstood=True
elif cmd=="reconnect":
if auteur in self.ops:
try:
if len(message)==1:
if self.identities.has_key(auteur):
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(self.identities[auteur])))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(self.identities[auteur])))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
serv.privmsg(canal,"Je ne connais pas ton pseudo note.")
elif auteur in self.ops:
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(message[1])))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(message[1])))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
else:
serv.privmsg(canal,("%s: %s"%(auteur,random.choice(config.quit_fail_messages))).encode("utf8"))
log(self.serveur,canal,auteur,message+"[failed]")
-
+ elif cmd == "reload":
+ if auteur in self.ops:
+ log(self.serveur, canal, auteur, message+"[successful]")
+ self.reload(canal)
+ elif cmd == "crash":
+ if auteur in self.overops:
+ self.crash(auteur, message)
elif cmd in ["part","leave","dégage","va-t-en","tut'tiresailleurs,c'estmesgalets"]:
if auteur in self.ops and (not (canal in self.stay_channels)
or auteur in self.overops):
if self.identities.has_key(auteur):
pseudo=self.identities[auteur]
try:
- self.nk.write('search ["x",["pseudo"],%s]'%(json.dumps(pseudo)))
+ self.nk.write('["search", ["x",["pseudo"],%s]]'%(json.dumps(pseudo)))
ret=json.loads(self.nk.read())
solde=ret["msg"][0]["solde"]
pseudo=ret["msg"][0]["pseudo"]
if re.match("ma bite dans ton oreille",message) and not canal in self.quiet_channels:
serv.privmsg(canal,"%s: Seul un olasd peut imiter un olasd dans un de ses grands jours !"%(auteur))
if out and not canal in self.quiet_channels:
- out=out.groups()[0]
+ out = out.groups()[0]
try:
- out=int(out)
- serv.privmsg(canal,"%s: %s !"%(auteur,out+1))
- if out==2147483647:
- serv.privmsg(canal,"%s: Ciel, un maxint ! Heureusement que je suis en python…"%(auteur))
+ iout = int(out)
+ serv.privmsg(canal,"%s: %s !"%(auteur, iout + 1))
+ if iout == 2147483647:
+ serv.privmsg(canal,"%s: Ciel, un maxint ! Heureusement que je suis en python…" % (auteur))
return
- if out+1>1000 and random.randrange(4)==0:
+ if iout + 1 > 1000 and random.randrange(4)==0:
serv.privmsg(canal,"%s: Vous savez, moi et les chiffres…"%(auteur))
return
except Exception as exc:
return self.serv.get_nickname()
nick=property(_getnick)
+ def reload(self, auteur=None):
+ reload(config)
+ if auteur in [None, "SIGHUP"]:
+ towrite = "Config reloaded" + " (SIGHUP received)"*(auteur == "SIGHUP")
+ for to in config.report_bugs_to:
+ self.serv.privmsg(to, towrite)
+ log(self.serveur, towrite)
+ else:
+ self.serv.privmsg(auteur,"Config reloaded")
+
+ def crash(self, chan="nowhere", who="nobody"):
+ where = u"en privé" if chan == "priv" else u"sur le chan %s" % chan
+ raise CrashError(u"Crash demandé par %s %s" % (who, where))
+
+ def start_as_daemon(self, outfile):
+ sys.stderr = Logger(outfile)
+ self.start()
+
+
+class Logger(object):
+ """Pour écrire ailleurs que sur stdout"""
+ def __init__(self, filename="basile.full.log"):
+ self.filename = filename
-if __name__=="__main__":
- import sys
+ def write(self, message):
+ f = open(self.filename, "a")
+ f.write(message)
+ f.close()
+
+def main():
if len(sys.argv)==1:
- print "Usage : basile.py <serveur> [--debug]"
+ print "Usage : basile.py <serveur> [--debug] [--no-output] [--daemon [--pidfile]] [--outfile]"
+ print " --outfile sans --no-output ni --daemon n'a aucun effet"
exit(1)
serveur=sys.argv[1]
+ if "--daemon" in sys.argv:
+ thisfile = os.path.realpath(__file__)
+ thisdirectory = thisfile.rsplit("/", 1)[0]
+ os.chdir(thisdirectory)
+ daemon = True
+ else:
+ daemon = False
if "debug" in sys.argv or "--debug" in sys.argv:
debug=True
else:
config.debug_stdout=False
serveurs={"a♡":"acoeur.crans.org","acoeur":"acoeur.crans.org","acoeur.crans.org":"acoeur.crans.org",
"irc":"irc.crans.org","crans":"irc.crans.org","irc.crans.org":"irc.crans.org"}
+ if "--no-output" in sys.argv or "--daemon" in sys.argv:
+ outfile = "/var/log/bots/basile.full.log"
+ for arg in sys.argv:
+ arg = arg.split("=")
+ if arg[0].strip('-') in ["out", "outfile", "logfile"]:
+ outfile = arg[1]
+ sys.stdout = Logger(outfile)
try:
serveur=serveurs[serveur]
except KeyError:
print "Server Unknown : %s"%(serveur)
exit(404)
basile=Basile(serveur,debug)
- basile.start()
+ # Si on reçoit un SIGHUP, on reload la config
+ def sighup_handler(signum, frame):
+ basile.reload("SIGHUP")
+ signal.signal(signal.SIGHUP, sighup_handler)
+ if daemon:
+ child_pid = os.fork()
+ if child_pid == 0:
+ os.setsid()
+ basile.start_as_daemon(outfile)
+ else:
+ # on enregistre le pid de basile
+ pidfile = "/var/run/bots/basile.pid"
+ for arg in sys.argv:
+ arg = arg.split("=")
+ if arg[0].strip('-') in ["pidfile"]:
+ pidfile = arg[1]
+ f = open(pidfile, "w")
+ f.write("%s\n" % child_pid)
+ f.close()
+ else:
+ basile.start()
+
+if __name__ == "__main__":
+ main()