satdigitalinvoice.local

Complexity

Total Complexity

31

Size/Duplication

Total Lines	123
Duplicated Lines	0 %

Importance

Changes

0

14 Methods

Rating	Name	Duplication	Size	Complexity
A	LocalDBSatCFDI.status_sat()	0	8	3
A	LocalDB.__init__()	0	2	1
A	LocalDBSatCFDI.pendiente_flip()	0	4	1
A	LocalDBSatCFDI.describe()	0	5	1
A	LocalDBSatCFDI.notificar_flip()	0	4	1
A	LocalDB.status_sat_set()	0	8	3
A	LocalDB.pendiente_set()	0	8	3
A	LocalDB.status_sat()	0	3	1
A	LocalDB.notificar_set()	0	8	3
A	LocalDB.pendiente()	0	3	1
A	LocalDB.notificar()	0	3	1
A	LocalDBSatCFDI.notificar()	0	4	2
A	LocalDBSatCFDI.pendiente()	0	6	4
A	LocalDBSatCFDI.__init__()	0	3	1

import logging
import os
import pickle
from uuid import UUID

import diskcache
from satcfdi.accounting import SatCFDI
from satcfdi.pacs import sat
from .log_tools import print_yaml

EMAIL_NOTIFICADA = 2
STATUS_SAT = 3
PENDIENTE = 4

sat_manager = sat.SAT()

logger = logging.getLogger(__name__)
DATA_DIR = '.data'


class LocalDB:
    def __init__(self):
        self.local_storage = diskcache.Cache(os.path.join(DATA_DIR, 'cache'))

    def pendiente(self, uuid: UUID):
        return self.local_storage.get(
            (PENDIENTE, uuid), True
        )

    def pendiente_set(self, uuid: UUID, value: bool):
        if value:
            try:
                del self.local_storage[(PENDIENTE, uuid)]

The variable PENDIENTE does not seem to be defined.

            except KeyError:
                pass
        else:
            self.local_storage[(PENDIENTE, uuid)] = value

    def notificar(self, uuid: UUID):
        return self.local_storage.get(
            (EMAIL_NOTIFICADA, uuid), True
        )

    def notificar_set(self, uuid: UUID, value: bool):
        if value:
            try:
                del self.local_storage[(EMAIL_NOTIFICADA, uuid)]

The variable EMAIL_NOTIFICADA does not seem to be defined.

            except KeyError:
                pass
        else:
            self.local_storage[(EMAIL_NOTIFICADA, uuid)] = value

    def status_sat(self, uuid: UUID):
        return self.local_storage.get(
            (STATUS_SAT, uuid), {}
        )

    def status_sat_set(self, uuid: UUID, value: dict):
        if value:
            self.local_storage[(STATUS_SAT, uuid)] = value
        else:
            try:
                del self.local_storage[(STATUS_SAT, uuid)]

The variable STATUS_SAT does not seem to be defined.

            except KeyError:
                pass


class LocalDBSatCFDI(LocalDB):
    def __init__(self, config):
        super().__init__()
        self.config = config

    def notificar(self, cfdi: SatCFDI):
        if cfdi["Fecha"] >= self.config['notificar_a_partir']:
            return super().notificar(cfdi.uuid)
        return False

    def notificar_flip(self, cfdi: SatCFDI):
        v = not self.notificar(cfdi)
        self.notificar_set(cfdi.uuid, v)
        return v

    def pendiente(self, cfdi: SatCFDI):
        if cfdi["MetodoPago"] == "PPD" and cfdi.saldo_pendiente == 0:
            return False
        if cfdi["Fecha"] >= self.config['pendiente_a_partir'][cfdi["MetodoPago"]]:
            return super().pendiente(cfdi.uuid)
        return False

    def pendiente_flip(self, cfdi: SatCFDI):
        v = not self.pendiente(cfdi)
        self.pendiente_set(cfdi.uuid, v)
        return v

    def status_sat(self, cfdi: SatCFDI, update=False):
        if update:
            res = sat_manager.status(cfdi)
            if res["ValidacionEFOS"] == "200":
                self.status_sat_set(cfdi.uuid, res)
            return res
        else:
            return super().status_sat(cfdi.uuid)

    def describe(self, cfdi: SatCFDI):
        print_yaml({
            'pendiente': self.pendiente(cfdi),
            'email_notificada': self.notificar(cfdi),
            'status_sat': self.status_sat(cfdi)
        })


def save_data(file, data):
    with open(os.path.join(DATA_DIR, file), 'wb') as f:
        pickle.dump(data, f)


def load_data(file, default=None):
    try:
        with open(os.path.join(DATA_DIR, file), 'rb') as f:
            return pickle.load(f)
    except FileNotFoundError:
        return default