sys.path.append(client_config.rsa_path)
import monRSA
-class DeadServer(Exception):
- pass
-
-def full_read(sock):
- """Lit sur la socket jusqu'à ce que l'output soit déJSON-izable"""
- output=""
- i = 0
- while True:
- output+=sock.read()
- i += 1
- try:
- return json.loads(output)
- except:
- pass
- if i==10 and output=="":
- raise DeadServer
+def full_read(socket):
+ """Lit un message complet sur la socket."""
+ # On récupère d'abord la taille du message
+ length_str = ''
+ char = socket.recv(1)
+ while char != '\n':
+ length_str += char
+ char = socket.recv(1)
+ total = int(length_str)
+ # On utilise une memoryview pour recevoir les données chunk par chunk efficacement
+ view = memoryview(bytearray(total))
+ next_offset = 0
+ while total - next_offset > 0:
+ recv_size = socket.recv_into(view[next_offset:], total - next_offset)
+ next_offset += recv_size
+ try:
+ msg = json.loads(view.tobytes())
+ except (TypeError, ValueError) as e:
+ raise NKNotJson("L'objet reçu n'est pas un JSON")
+ return msg
def connect(sock,where,port=None):
if port==None:
return lit(a)
def login(socket,user,mdp="",typ="bdd"):
- if user in ["20-100","moibdd","#1"]:
+ if user in ["#1", "test"]:
a.write('["login", ["#1","plop","bdd",[[],[],false]]]')
elif user in ["vincent","moi"]:
a.write('["login", ["vincent","plop","special",[]]]')
a.write('["login", ["toto","plop","bdd",[[],[],false]]]')
else:
a.write('["login", ["%s","%s","%s",[[],[],false]]]'%(user,mdp,typ))
+ if mdp == "":
+ import getpass
+ mdp = getpass.getpass()
return lit(a)
def search(socket,term):
return obj["msg"]
except ssl.SSLError:
print "Rien à lire"
- except DeadServer:
- print "Serveur down ?"
+
+def do(stuff):
+ """Dirty, 'cause I'm lazy."""
+ global a
+ a.write(json.dumps(stuff))
+
+
+# Shortcuts
+def _toggle_transactions(l, action="valider"):
+ """(Dé)valide plein de transactions si on lui donne la liste d'ids."""
+ global a
+ for i in l:
+ assert isinstance(i, int)
+ print i,
+ do(["%s_transaction" % (action,), i])
+ out = lit(a)
+ if out:
+ print out
+
+def cancel_transactions(l):
+ """Annule plein de transactions. Mange une liste d'ids de transactions."""
+ _toggle_transactions(l, action="devalider")
+
+def validate_transactions(l):
+ """Valide plein de transactions. Mange une liste d'ids de transactions."""
+ _toggle_transactions(l, action="valider")
if __name__=="__main__":
where=raw_input("quel serveur ?")
a.settimeout(2)
#checkidentity(a)
print hello(a,"Python Client alpha")
- print login(a,"20-100","plop")
+ try:
+ import secrets
+ user = secrets.user
+ password = secrets.password
+ except ImportError:
+ print "Vous pouvez enregistrer votre mot de passe dans la variable password dans secrets.py !"
+ user = raw_input("username:")
+ import getpass
+ password = getpass.getpass("password:")
+ print login(a, user, password)
print "Socket créée dans la variable \"a\""
-