note.py 2.39 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
#!/usr/bin/python3
# -*- coding:utf-8 -*-

# Codé par PAC , forké de 20-100

""" Module pour dialoguer avec la NoteKfet2015 """

import socket
import json
import ssl
import traceback


def get_response(socket):
    length_str = b''
    char = socket.recv(1)
    while char != b'\n':
        length_str += char
        char = socket.recv(1)
    total = int(length_str)
    return json.loads(socket.recv(total).decode('utf-8'))

def connect(server, port):
    sock = socket.socket()
    try:
        # On établit la connexion sur port 4242
        sock.connect((server, port))
        # On passe en SSL
        sock = ssl.wrap_socket(sock)
        # On fait un hello
        sock.send(b'["hello", "manual"]')
        retcode = get_response(sock)
    except:
        # Si on a foiré quelque part, c'est que le serveur est down
        return (False, sock, "Serveur indisponible")
    return (True, sock, "")

def login(server, port, username, password,  masque = [[], [], True]):
    result, sock, err = connect(server, port)
    if not result:
        return (False, None, err)
    try:
        commande = ["login", [username, password, "bdd", masque]]
        sock.send(json.dumps(commande).encode("utf-8"))
        response = get_response(sock)
        retcode = response['retcode']
        if retcode == 0:
            return (True, sock, "")
        elif retcode == 5:
            return (False, sock, "Login incorrect")
        else:
            return (False, sock, "Erreur inconnue " + str(retcode))
    except:
        # Si on a foiré quelque part, c'est que le serveur est down
        return (False, sock, "Erreur de communication avec le serveur")


def don(sock, montant, id_note, facture):
    """
    Faire faire un don à l'id_note
    """
    try:
        sock.send(json.dumps(["dons", [[id_note], round(montant*100), "Facture : id=%s, designation=%s" % (facture.id, facture.name())]]).encode("utf-8"))
        response = get_response(sock)
        retcode = response['retcode']
        transaction_retcode = response["msg"][0][0]
        if 0 < retcode < 100 or 200 <= retcode or 0 < transaction_retcode < 100 or 200 <= transaction_retcode:
            return (False, "Transaction échouée. (Solde trop négatif ?)")
        elif retcode == 0:
            return (True, "")
        else:
            return (False, "Erreur inconnue " + str(retcode))
    except:
        return (False, "Erreur de communication avec le serveur")