devicehub-django/dpp/api_dlt.py

import json
import time
import logging

from django.conf import settings
from ereuseapi.methods import API

from dpp.models import Proof, UserDpp


logger = logging.getLogger('django')


# """The code of the status response of api dlt."""
STATUS_CODE = {
    "Success": 201,
    "Notwork": 400
}


ALGORITHM = "sha3-256"


PROOF_TYPE = {
    'Register': 'Register',
    'IssueDPP': 'IssueDPP',
    'proof_of_recycling': 'proof_of_recycling',
    'Erase': 'Erase',
    'EWaste': 'EWaste',
}


def connect_api(user):

    dp = UserDpp.objects.filter(user=user).first()
    if not dp:
        return
    
    api_dlt = settings.API_DLT
    token_dlt = dp.api_keys_dlt

    if not api_dlt or not token_dlt:
        logger.error("NOT POSSIBLE CONNECT WITH API DLT!!!")
        return

    return API(api_dlt, token_dlt, "ethereum")


def register_dlt(api, chid, phid, proof_type=None):
    if proof_type:
        return api.generate_proof(
            chid,
            ALGORITHM,
            phid,
            proof_type,
            settings.ID_FEDERATED
        )

    return api.register_device(
        chid,
        ALGORITHM,
        phid,
        settings.ID_FEDERATED
    )


def issuer_dpp_dlt(api, dpp):
    phid = dpp.split(":")[1]

    return api.issue_passport(
        dpp,
        ALGORITHM,
        phid,
        settings.ID_FEDERATED
    )



def save_proof(signature, ev_uuid, result, proof_type, user):
    if result['Status'] == STATUS_CODE.get("Success"):
        timestamp = result.get('Data', {}).get('data', {}).get('timestamp')

        if not timestamp:
            return

        logger.debug("timestamp: %s", timestamp)
        d = {
            "type": proof_type,
            "timestamp": timestamp,
            "issuer": user.institution,
            "user": user,
            "uuid": ev_uuid,
            "signature": signature,
        }
        Proof.objects.create(**d)


def register_device_dlt(chid, phid, ev_uuid, user):
    cny_a = 1
    while cny_a:
        api = connect_api(user)
        if not api:
            cny_a = 0
            return
        
        result = register_dlt(api, chid, phid)
        try:
            assert result['Status'] == STATUS_CODE.get("Success")
            assert result['Data']['data']['timestamp']
            cny_a = 0
        except Exception:
            if result.get("Data") != "Device already exists":
                logger.error("API return: %s", result)
                time.sleep(10)
            else:
                cny_a = 0

    save_proof(phid, ev_uuid, result, PROOF_TYPE['Register'], user)


    # TODO is neccesary?
    if settings.ID_FEDERATED:
        cny = 1
        while cny:
            try:
                api.add_service(
                    chid,
                    'DeviceHub',
                    settings.ID_FEDERATED,
                    'Inventory service',
                    'Inv',
                )
                cny = 0
            except Exception:
                time.sleep(10)


def register_passport_dlt(chid, phid, ev_uuid, user):
    token_dlt = settings.TOKEN_DLT
    api_dlt = settings.API_DLT
    if not token_dlt or not api_dlt:
        return
    
    dpp = "{chid}:{phid}".format(chid=chid, phid=phid)
    if Proof.objects.filter(signature=dpp, type=PROOF_TYPE['IssueDPP']).exists():
        return

    cny_a = 1
    while cny_a:
        try:
            api = connect_api(user)
            if not api:
                cny_a = 0
                return
        
            result = issuer_dpp_dlt(api, dpp)
            cny_a = 0
        except Exception as err:
            logger.error("ERROR API issue passport return: %s", err)
            time.sleep(10)

    if result['Status'] is not STATUS_CODE.get("Success"):
        logger.error("ERROR API issue passport return: %s", result)
        return
    
    save_proof(phid, ev_uuid, result, PROOF_TYPE['IssueDPP'], user)