]> gitweb.pimeys.fr Git - bots/basile.git/blobdiff - basile.py
Début de séparation de ce qui concerne NK2015 dans un fichier nk.py
[bots/basile.git] / basile.py
index d30ed46dbba04b143a5d8d07fbb7a85ca6c4aa1c..96641c120badf7e01959faa6d3351b017c898ebe 100755 (executable)
--- a/basile.py
+++ b/basile.py
@@ -3,12 +3,12 @@
 
 # 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 threading
 import random
 import time
-import socket, ssl, json
+import json
 import pickle
 import re
 import os
@@ -22,8 +22,10 @@ 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
@@ -37,24 +39,6 @@ def get_config_logfile(serveur):
 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:
@@ -67,46 +51,6 @@ def log(serveur,channel,auteur=None,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")
@@ -169,7 +113,8 @@ class UnicodeBotError(Exception):
 
 class CrashError(Exception):
     """Pour pouvoir faire crasher Basile, parce que ça a l'air drôle"""
-    pass
+    def __init__(self, msg=u""):
+        Exception.__init__(self, msg)
 
 def bot_unicode(chain):
     try:
@@ -196,19 +141,21 @@ class Basile(ircbot.SingleServerIRCBot):
 
 
     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:
@@ -225,6 +172,7 @@ class Basile(ircbot.SingleServerIRCBot):
             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)
@@ -237,8 +185,6 @@ class Basile(ircbot.SingleServerIRCBot):
         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)
@@ -449,7 +395,7 @@ class Basile(ircbot.SingleServerIRCBot):
         elif cmd=="crash":
             if auteur in self.overops:
                 log(self.serveur,"priv",auteur," ".join(message)+"[successful]")
-                self.crash()
+                self.crash("priv", auteur)
             else:
                 notunderstood=True
         elif cmd=="reload":
@@ -538,7 +484,7 @@ class Basile(ircbot.SingleServerIRCBot):
             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"]
@@ -553,7 +499,7 @@ class Basile(ircbot.SingleServerIRCBot):
                     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"]
@@ -607,7 +553,7 @@ class Basile(ircbot.SingleServerIRCBot):
                     self.reload(canal)
             elif cmd == "crash":
                 if auteur in self.overops:
-                    self.crash()
+                    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):
@@ -654,7 +600,7 @@ class Basile(ircbot.SingleServerIRCBot):
                 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"]
@@ -725,14 +671,14 @@ class Basile(ircbot.SingleServerIRCBot):
             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:
@@ -864,8 +810,9 @@ class Basile(ircbot.SingleServerIRCBot):
         else:
             self.serv.privmsg(auteur,"Config reloaded")
     
-    def crash(self):
-        raise CrashError
+    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)