Nouvelle gestion des utilisateurs

[bots/basile.git] / basile.py

diff --git a/basile.py b/basile.py
index b6dd64a59376baf7f5583cd71245f4a2d602ab5a..0c6fdc75af2568bd54991d71d1df4c909d1344d3 100755 (executable)
--- a/basile.py
+++ b/basile.py
@@ -29,6 +29,8 @@ import nk
 import isit
 #: Module définissant les erreurs
 import errors
+#: Module de gestion des utilisateurs
+import users
 
 # 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
@@ -91,10 +93,12 @@ class Basile(ircbot.SingleServerIRCBot):
         self.ops = self.overops + config.ops
         self.report_bugs_to = config.report_bugs_to
         self.chanlist = config.chanlist
-        self.identities = json.load(open(config.identities_file, "r"))
         self.stay_channels = config.stay_channels
         self.quiet_channels = config.quiet_channels
         self.last_perdu = 0
+        # On charge la base de données d'utilisateurs
+        self.users = users.UserDB()
+        self.users.load()
     
     ### Communication NK
     def new_connection_NK(self, serv, username, password, typ="bdd"):
@@ -158,10 +162,10 @@ class Basile(ircbot.SingleServerIRCBot):
         if self.last_perdu + config.time_between_perdu < time.time() or forced:
             if not channel in self.quiet_channels or forced:
                 serv.privmsg(channel, "J'ai perdu !")
-            self.last_perdu=time.time()
-            delay=config.time_between_perdu_trigger
-            delta=config.time_between_perdu_trigger_delta
-            serv.execute_delayed(random.randrange(delay-delta,delay+delta),self.lost,(serv,channel))
+            self.last_perdu = time.time()
+            delay = config.time_between_perdu_trigger
+            delta = config.time_between_perdu_trigger_delta
+            serv.execute_delayed(random.randrange(delay - delta, delay + delta), self.lost, (serv, channel))
     
     def quitter(self, chan, leave_message=None):
         """Quitter un channel avec un message customisable."""
@@ -265,9 +269,9 @@ class Basile(ircbot.SingleServerIRCBot):
                 serv.privmsg(auteur, ligne.encode("utf-8"))
         elif cmd == u"identify":
             if len(message) == 1:
-                if self.identities.has_key(auteur):
+                if self.users.has(auteur):
                     serv.privmsg(auteur, "Je vous connais sous le pseudo note %s." % (
-                                     self.identities[auteur]["pseudo"].encode("utf8")))
+                                     self.users[auteur].pseudonote.encode("utf8")))
                 else:
                     serv.privmsg(auteur, "Je ne connais pas votre pseudo note.")
             elif len(message) >= 3:
@@ -275,9 +279,8 @@ class Basile(ircbot.SingleServerIRCBot):
                 success, info, _ = self.new_connection_NK(serv, username, password)
                 if success:
                     log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
-                    serv.privmsg(auteur, "Identité enregistrée.")
-                    self.identities[auteur] = info
-                    json.dump(self.identities, open(config.identities_file,"w"))
+                    self.users.add(auteur, info["idbde"])
+                    serv.privmsg(auteur, "Pseudo enregistré.")
                 else:
                     log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
                     serv.privmsg(auteur, "Mot de passe invalide. (ou serveur down)")
@@ -285,14 +288,13 @@ class Basile(ircbot.SingleServerIRCBot):
                 serv.privmsg(auteur, "Syntaxe : IDENTIFY [<username> <password>]")
         elif cmd == u"drop":
             if len(message) > 1:
-                if self.identities.has_key(auteur):
+                if self.users.has(auteur):
                     password = " ".join(message[1:])
-                    success, _, _ = self.new_connection_NK(serv, self.identities[auteur], password)
+                    success, _, _ = self.new_connection_NK(serv, "#%s" % self.users[auteur].idbde, password)
                     if success:
-                        del self.identities[auteur]
+                        self.users.drop(auteur)
                         log(self.serveur, "priv", auteur, " ".join(message) + "[successful]")
-                        json.dump(self.identities, open(config.identities_file, "w"))
-                        serv.privmsg(auteur, "Identité oubliée.")
+                        serv.privmsg(auteur, "Pseudo oublié.")
                     else:
                         log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
                         serv.privmsg(auteur, "Mot de passe invalide. (ou serveur down)")
@@ -449,8 +451,8 @@ class Basile(ircbot.SingleServerIRCBot):
                 notunderstood = True
         elif cmd == u"solde":
             if len(message) == 1:
-                if self.identities.has_key(auteur):
-                    success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, auteur)
+                if self.users.has(auteur):
+                    success, solde, pseudo = nk.get_solde(self.nk, self.users[auteur].idbde, serv, auteur)
                     if success:
                         serv.privmsg(auteur, "%.2f (%s)" % (solde/100.0, pseudo.encode("utf8")))
                     log(self.serveur, "priv", auteur, " ".join(message) + ("[successful]" if success else "[failed]"))
@@ -547,14 +549,14 @@ class Basile(ircbot.SingleServerIRCBot):
                 serv.privmsg(canal, "%s: pong" % (auteur))
 
             elif cmd in [u"solde", u"!solde", u"!coca"] or cmd.startswith("!"):
-                if self.identities.has_key(auteur):
-                    idbde = self.identities[auteur]["idbde"]
+                if self.users.has(auteur):
+                    idbde = self.users[auteur].idbde
                     if cmd in [u"solde", u"!solde"]:
-                        success, solde, pseudo = nk.get_solde(self.nk, self.identities[auteur]["idbde"], serv, canal)
+                        success, solde, pseudo = nk.get_solde(self.nk, idbde, serv, canal)
                         if success:
                             serv.privmsg(canal, "%s: %s (%s)" % (auteur, float(solde)/100, pseudo.encode("utf8")))
                     elif cmd in [u"!coca"] or cmd.startswith("!"):
-                        success = nk.consomme(self.nk, self.identities[auteur]["idbde"], message[1:], serv, canal)
+                        success = nk.consomme(self.nk, idbde, message[1:], serv, canal)
                     log(self.serveur, canal, auteur, message + ("[successful]" if success else "[failed]"))
                 else:
                     serv.privmsg(canal, "%s: Je ne connais pas votre pseudo note." % (auteur))