get_infos et affichage du compte sous lequel Basile connait quelqu'un

[bots/basile.git] / nk.py

diff --git a/nk.py b/nk.py
index 7cec40aec4bcd2a15924f604ff56213a168fe17f..0aafbc16aff64606b07b40e5d5ccf0e6a6ea6e69 100644 (file)
--- a/nk.py
+++ b/nk.py
@@ -92,25 +92,29 @@ def login(username, password, typ="bdd"):
         raise NKRefused(str(exc))
     return out, sock
 
-def get_solde(sock, idbde, serv, canal):
-    """Récupère le (success, solde, pseudo) de l'utilisateur NK n°``idbde``"""
+def get_infos(sock, idbde, serv, canal):
+    """Récupère les données de l'utilisateur NK n°``idbde``"""
     try:
         sock.write(json.dumps(["compte", idbde]))
         ret = full_read(sock)
         retcode = ret["retcode"]
         if retcode == 0:
-            solde = ret["msg"]["solde"]
-            pseudo = ret["msg"]["pseudo"]
-            return (True, solde, pseudo)
+            return ret["msg"]
         else:
             serv.privmsg(canal, ret["errmsg"].encode("utf-8"))
-            return (False, None, None)
     except Exception as exc:
         trace = traceback.format_exc()
         msg = "failed\n%s" % trace
         for l in msg.split("\n"):
             serv.privmsg(canal, l)
         #log(self.serveur, "priv", auteur, " ".join(message) + "[failed]")
+
+def get_solde(sock, idbde, serv, canal):
+    """Récupère le (success, solde, pseudo) de l'utilisateur NK n°``idbde``"""
+    infos = get_infos(sock, idbde, serv, canal)
+    if infos:
+        return (True, infos["solde"], infos["pseudo"])
+    else:
         return (False, None, None)
 
 def consomme(sock, idbde, conso, serv, canal):