X-Git-Url: http://gitweb.pimeys.fr/?p=bots%2Fbasile.git;a=blobdiff_plain;f=basile.py;h=0c6fdc75af2568bd54991d71d1df4c909d1344d3;hp=b6dd64a59376baf7f5583cd71245f4a2d602ab5a;hb=ee7a287b0f564671d100fa29ae0269c482ef4efe;hpb=bf6858eb5f5a79788801fd455837fe8f3f340460 diff --git a/basile.py b/basile.py index b6dd64a..0c6fdc7 100755 --- 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 [ ]") 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))