bit4bit@riseup.net
f4c2282e3d
* facho/fe/form/query.py: utilidades. FossilOrigin-Name: a1486421fcfcf6bda1e3b4901f3fc3fe86b7156852571aad74274d5b7ce3778a
540 lines
15 KiB
Python
540 lines
15 KiB
Python
# This file is part of facho. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
|
|
import hashlib
|
|
from functools import reduce
|
|
import copy
|
|
from dataclasses import dataclass
|
|
from datetime import datetime, date
|
|
from collections import defaultdict
|
|
import decimal
|
|
from decimal import Decimal
|
|
import typing
|
|
|
|
from ..data.dian import codelist
|
|
|
|
DECIMAL_PRECISION = 6
|
|
|
|
class AmountCurrencyError(TypeError):
|
|
pass
|
|
|
|
@dataclass
|
|
class Currency:
|
|
code: str
|
|
|
|
def __eq__(self, other):
|
|
return self.code == other.code
|
|
|
|
def __str__(self):
|
|
return self.code
|
|
|
|
class Collection:
|
|
|
|
def __init__(self, array):
|
|
self.array = array
|
|
|
|
def filter(self, filterer):
|
|
new_array = filter(filterer, self.array)
|
|
return self.__class__(new_array)
|
|
|
|
def map(self, mapper):
|
|
new_array = map(mapper, self.array)
|
|
return self.__class__(new_array)
|
|
|
|
def sum(self):
|
|
return sum(self.array)
|
|
|
|
class AmountCollection(Collection):
|
|
|
|
def sum(self):
|
|
total = Amount(0)
|
|
for v in self.array:
|
|
total += v
|
|
return total
|
|
|
|
class Amount:
|
|
def __init__(self, amount: int or float or Amount, currency: Currency = Currency('COP')):
|
|
|
|
#DIAN 1.7.-2020: 1.2.3.1
|
|
if amount < 0:
|
|
raise ValueError('amount must be positive >= 0')
|
|
|
|
if isinstance(amount, Amount):
|
|
self.amount = amount.amount
|
|
self.currency = amount.currency
|
|
else:
|
|
self.amount = Decimal(amount, decimal.Context(prec=DECIMAL_PRECISION,
|
|
#DIAN 1.7.-2020: 1.2.1.1
|
|
rounding=decimal.ROUND_HALF_EVEN ))
|
|
self.currency = currency
|
|
|
|
def __round__(self, prec):
|
|
return round(self.amount, prec)
|
|
|
|
def __str__(self):
|
|
return '%.06f' % self.amount
|
|
|
|
def __eq__(self, other):
|
|
if not self.is_same_currency(other):
|
|
raise AmountCurrencyError()
|
|
return round(self.amount, DECIMAL_PRECISION) == round(other.amount, DECIMAL_PRECISION)
|
|
|
|
def __add__(self, other):
|
|
if not self.is_same_currency(other):
|
|
raise AmountCurrencyError()
|
|
return Amount(self.amount + other.amount, self.currency)
|
|
|
|
def __sub__(self, other):
|
|
if not self.is_same_currency(other):
|
|
raise AmountCurrencyError()
|
|
return Amount(self.amount - other.amount, self.currency)
|
|
|
|
def __mul__(self, other):
|
|
if not self.is_same_currency(other):
|
|
raise AmountCurrencyError()
|
|
return Amount(self.amount * other.amount, self.currency)
|
|
|
|
def is_same_currency(self, other):
|
|
return self.currency == other.currency
|
|
|
|
|
|
@dataclass
|
|
class Item:
|
|
description: str
|
|
id: str
|
|
|
|
|
|
@dataclass
|
|
class StandardItem(Item):
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class Country:
|
|
code: str
|
|
name: str = ''
|
|
|
|
def __post_init__(self):
|
|
if self.code not in codelist.Paises:
|
|
raise ValueError("code [%s] not found" % (self.code))
|
|
self.name = codelist.Paises[self.code]['name']
|
|
|
|
@dataclass
|
|
class CountrySubentity:
|
|
code: str
|
|
name: str = ''
|
|
|
|
def __post_init__(self):
|
|
if self.code not in codelist.Departamento:
|
|
raise ValueError("code [%s] not found" % (self.code))
|
|
self.name = codelist.Departamento[self.code]['name']
|
|
|
|
@dataclass
|
|
class City:
|
|
code: str
|
|
name: str = ''
|
|
|
|
def __post_init__(self):
|
|
if self.code not in codelist.Municipio:
|
|
raise ValueError("code [%s] not found" % (self.code))
|
|
self.name = codelist.Municipio[self.code]['name']
|
|
|
|
@dataclass
|
|
class Address:
|
|
name: str
|
|
street: str = ''
|
|
city: City = City('05001')
|
|
country: Country = Country('CO')
|
|
countrysubentity: CountrySubentity = CountrySubentity('05')
|
|
|
|
@dataclass
|
|
class PartyIdentification:
|
|
number: str
|
|
dv: str
|
|
type_fiscal: str
|
|
|
|
def __str__(self):
|
|
return self.number
|
|
|
|
def __eq__(self, other):
|
|
return str(self) == str(other)
|
|
|
|
def full(self):
|
|
return "%s%s" % [self.number, self.dv]
|
|
|
|
def __post_init__(self):
|
|
if self.type_fiscal not in codelist.TipoIdFiscal:
|
|
raise ValueError("type_fiscal [%s] not found" % (self.type_fiscal))
|
|
|
|
@dataclass
|
|
class Responsability:
|
|
codes: list
|
|
|
|
def __str__(self):
|
|
return ';'.join(self.codes)
|
|
|
|
def __eq__(self, other):
|
|
return str(self) == str(other)
|
|
|
|
def __iter__(self):
|
|
return iter(self.codes)
|
|
|
|
def __post_init__(self):
|
|
for code in self.codes:
|
|
if code not in codelist.TipoResponsabilidad:
|
|
raise ValueError("code %s not found" % (code))
|
|
|
|
|
|
@dataclass
|
|
class TaxScheme:
|
|
code: str
|
|
name: str = ''
|
|
|
|
|
|
def __post_init__(self):
|
|
if self.code not in codelist.TipoImpuesto:
|
|
raise ValueError("code not found")
|
|
self.name = codelist.TipoImpuesto[self.code]['name']
|
|
|
|
@dataclass
|
|
class Party:
|
|
name: str
|
|
ident: str
|
|
responsability_code: typing.List[Responsability]
|
|
responsability_regime_code: str
|
|
organization_code: str
|
|
tax_scheme: TaxScheme = TaxScheme('01')
|
|
|
|
phone: str = ''
|
|
address: Address = Address('')
|
|
email: str = ''
|
|
legal_name: str = ''
|
|
legal_company_ident: str = ''
|
|
legal_address: str = ''
|
|
|
|
def __post_init__(self):
|
|
if self.organization_code not in codelist.TipoOrganizacion:
|
|
raise ValueError("organization_code not found")
|
|
|
|
@dataclass
|
|
class TaxSubTotal:
|
|
percent: float
|
|
tax_scheme_ident: str = '01'
|
|
tax_scheme_name: str = 'IVA'
|
|
|
|
tax_amount: Amount = Amount(0.0)
|
|
taxable_amount: Amount = Amount(0.0)
|
|
|
|
def calculate(self, invline):
|
|
self.tax_amount = invline.total_amount * Amount(self.percent / 100)
|
|
self.taxable_amount = invline.total_amount
|
|
|
|
|
|
@dataclass
|
|
class TaxTotal:
|
|
subtotals: list
|
|
tax_amount: Amount = Amount(0.0)
|
|
taxable_amount: Amount = Amount(0.0)
|
|
|
|
def calculate(self, invline):
|
|
for subtax in self.subtotals:
|
|
subtax.calculate(invline)
|
|
self.tax_amount += subtax.tax_amount
|
|
self.taxable_amount += subtax.taxable_amount
|
|
|
|
|
|
@dataclass
|
|
class Price:
|
|
amount: Amount
|
|
type_code: str
|
|
type: str
|
|
|
|
def __post_init__(self):
|
|
if self.type_code not in codelist.CodigoPrecioReferencia:
|
|
raise ValueError("type_code [%s] not found" % (self.type_code))
|
|
|
|
|
|
@dataclass
|
|
class PaymentMean:
|
|
DEBIT = '01'
|
|
CREDIT = '02'
|
|
|
|
def __init__(self, id: str, code: str, due_at: datetime, payment_id: str):
|
|
if code not in codelist.MediosPago:
|
|
raise ValueError("code not found")
|
|
|
|
self.id = id
|
|
self.code = code
|
|
self.due_at = due_at
|
|
self.payment_id = payment_id
|
|
|
|
|
|
@dataclass
|
|
class PrePaidPayment:
|
|
#DIAN 1.7.-2020: FBD03
|
|
paid_amount: Amount = Amount(0.0)
|
|
|
|
|
|
@dataclass
|
|
class BillingReference:
|
|
ident: str
|
|
uuid: str
|
|
date: date
|
|
|
|
class CreditNoteDocumentReference(BillingReference):
|
|
"""
|
|
ident: Prefijo + Numero de la factura relacionada
|
|
uuid: CUFE de la factura electronica
|
|
date: fecha de emision de la factura relacionada
|
|
"""
|
|
|
|
|
|
class DebitNoteDocumentReference(BillingReference):
|
|
"""
|
|
ident: Prefijo + Numero de la factura relacionada
|
|
uuid: CUFE de la factura electronica
|
|
date: fecha de emision de la factura relacionada
|
|
"""
|
|
|
|
class InvoiceDocumentReference(BillingReference):
|
|
"""
|
|
ident: Prefijo + Numero de la nota credito relacionada
|
|
uuid: CUDE de la nota credito relacionada
|
|
date: fecha de emision de la nota credito relacionada
|
|
"""
|
|
|
|
@dataclass
|
|
class InvoiceLine:
|
|
# RESOLUCION 0004: pagina 155
|
|
quantity: int
|
|
description: str
|
|
item: Item
|
|
price: Price
|
|
# TODO mover a Invoice
|
|
# ya que al reportar los totales es sobre
|
|
# la factura y el percent es unico por type_code
|
|
# de subtotal
|
|
tax: TaxTotal
|
|
|
|
@property
|
|
def total_amount(self):
|
|
return Amount(self.quantity) * self.price.amount
|
|
|
|
@property
|
|
def total_tax_inclusive_amount(self):
|
|
return self.tax.taxable_amount + self.tax.tax_amount
|
|
|
|
@property
|
|
def total_tax_exclusive_amount(self):
|
|
return self.tax.taxable_amount
|
|
|
|
@property
|
|
def tax_amount(self):
|
|
return self.tax.tax_amount
|
|
|
|
@property
|
|
def taxable_amount(self):
|
|
return self.tax.taxable_amount
|
|
|
|
def calculate(self):
|
|
self.tax.calculate(self)
|
|
|
|
|
|
@dataclass
|
|
class LegalMonetaryTotal:
|
|
line_extension_amount: Amount = Amount(0.0)
|
|
tax_exclusive_amount: Amount = Amount(0.0)
|
|
tax_inclusive_amount: Amount = Amount(0.0)
|
|
charge_total_amount: Amount = Amount(0.0)
|
|
allowance_total_amount: Amount = Amount(0.0)
|
|
payable_amount: Amount = Amount(0.0)
|
|
prepaid_amount: Amount = Amount(0.0)
|
|
|
|
|
|
@dataclass
|
|
class AllowanceChargeReason:
|
|
code: str
|
|
reason: str
|
|
|
|
def __post_init__(self):
|
|
if self.code not in codelist.CodigoDescuento:
|
|
raise ValueError("code [%s] not found" % (self.code))
|
|
|
|
|
|
@dataclass
|
|
class AllowanceCharge:
|
|
#DIAN 1.7.-2020: FAQ03
|
|
charge_indicator: bool = True
|
|
amount: Amount = Amount(0.0)
|
|
reason: AllowanceChargeReason = None
|
|
|
|
def isCharge(self):
|
|
return self.charge_indicator == True
|
|
|
|
def isDiscount(self):
|
|
return self.charge_indicator == False
|
|
|
|
def asCharge(self):
|
|
self.charge_indicator = True
|
|
|
|
def asDiscount(self):
|
|
self.charge_indicator = False
|
|
|
|
def hasReason(self):
|
|
return self.reason is not None
|
|
|
|
class NationalSalesInvoiceDocumentType(str):
|
|
def __str__(self):
|
|
# 6.1.3
|
|
return '01'
|
|
|
|
class CreditNoteDocumentType(str):
|
|
def __str__(self):
|
|
# 6.1.3
|
|
return '91'
|
|
|
|
class DebitNoteDocumentType(str):
|
|
def __str__(self):
|
|
# 6.1.3
|
|
return '92'
|
|
|
|
class Invoice:
|
|
def __init__(self, type_code: str):
|
|
if str(type_code) not in codelist.TipoDocumento:
|
|
raise ValueError("type_code [%s] not found")
|
|
|
|
self.invoice_period_start = None
|
|
self.invoice_period_end = None
|
|
self.invoice_issue = None
|
|
self.invoice_ident = None
|
|
self.invoice_operation_type = None
|
|
self.invoice_legal_monetary_total = LegalMonetaryTotal()
|
|
self.invoice_customer = None
|
|
self.invoice_supplier = None
|
|
self.invoice_payment_mean = None
|
|
self.invoice_payments = []
|
|
self.invoice_lines = []
|
|
self.invoice_allowance_charge = []
|
|
self.invoice_prepaid_payment = []
|
|
self.invoice_billing_reference = None
|
|
self.invoice_type_code = str(type_code)
|
|
|
|
|
|
def set_period(self, startdate, enddate):
|
|
self.invoice_period_start = startdate
|
|
self.invoice_period_end = enddate
|
|
|
|
def set_issue(self, dtime: datetime):
|
|
if dtime.tzname() not in ['UTC-05:00', '-05', None]:
|
|
raise ValueError("dtime must be UTC-05:00")
|
|
|
|
self.invoice_issue = dtime
|
|
|
|
def set_ident(self, ident: str):
|
|
self.invoice_ident = ident
|
|
|
|
def set_supplier(self, party: Party):
|
|
self.invoice_supplier = party
|
|
|
|
def set_customer(self, party: Party):
|
|
self.invoice_customer = party
|
|
|
|
def set_payment_mean(self, payment_mean: PaymentMean):
|
|
self.invoice_payment_mean = payment_mean
|
|
|
|
def _get_codelist_tipo_operacion(self):
|
|
return codelist.TipoOperacionF
|
|
|
|
def set_operation_type(self, operation):
|
|
if operation not in self._get_codelist_tipo_operacion():
|
|
raise ValueError("operation not found")
|
|
|
|
self.invoice_operation_type = operation
|
|
|
|
def add_allownace_charge(self, charge: AllowanceCharge):
|
|
self.invoice_allowance_charge.append(charge)
|
|
|
|
def add_invoice_line(self, line: InvoiceLine):
|
|
self.invoice_lines.append(line)
|
|
|
|
def add_prepaid_payment(self, paid: PrePaidPayment):
|
|
self.invoice_prepaid_payment.append(paid)
|
|
|
|
def set_billing_reference(self, billing_reference: BillingReference):
|
|
self.invoice_billing_reference = billing_reference
|
|
|
|
def accept(self, visitor):
|
|
visitor.visit_payment_mean(self.invoice_payment_mean)
|
|
visitor.visit_customer(self.invoice_customer)
|
|
visitor.visit_supplier(self.invoice_supplier)
|
|
for payment in self.invoice_payments:
|
|
visitor.visit_payment(payment)
|
|
for invline in self.invoice_lines:
|
|
visitor.visit_invoice_line(invline)
|
|
|
|
def _calculate_legal_monetary_total(self):
|
|
for invline in self.invoice_lines:
|
|
self.invoice_legal_monetary_total.line_extension_amount += invline.total_amount
|
|
self.invoice_legal_monetary_total.tax_exclusive_amount += invline.total_tax_exclusive_amount
|
|
#DIAN 1.7.-2020: FAU6
|
|
self.invoice_legal_monetary_total.tax_inclusive_amount += invline.total_tax_inclusive_amount
|
|
|
|
#DIAN 1.7.-2020: FAU08
|
|
self.invoice_legal_monetary_total.allowance_total_amount = AmountCollection(self.invoice_allowance_charge)\
|
|
.filter(lambda charge: charge.isDiscount())\
|
|
.map(lambda charge: charge.amount)\
|
|
.sum()
|
|
|
|
#DIAN 1.7.-2020: FAU10
|
|
self.invoice_legal_monetary_total.charge_total_amount = AmountCollection(self.invoice_allowance_charge)\
|
|
.filter(lambda charge: charge.isCharge())\
|
|
.map(lambda charge: charge.amount)\
|
|
.sum()
|
|
|
|
#DIAN 1.7.-2020: FAU12
|
|
self.invoice_legal_monetary_total.prepaid_amount = AmountCollection(self.invoice_prepaid_payment)\
|
|
.map(lambda paid: paid.paid_amount)\
|
|
.sum()
|
|
|
|
#DIAN 1.7.-2020: FAU14
|
|
self.invoice_legal_monetary_total.payable_amount = \
|
|
self.invoice_legal_monetary_total.tax_inclusive_amount \
|
|
+ self.invoice_legal_monetary_total.allowance_total_amount \
|
|
+ self.invoice_legal_monetary_total.charge_total_amount \
|
|
- self.invoice_legal_monetary_total.prepaid_amount
|
|
|
|
|
|
def calculate(self):
|
|
for invline in self.invoice_lines:
|
|
invline.calculate()
|
|
self._calculate_legal_monetary_total()
|
|
|
|
|
|
class NationalSalesInvoice(Invoice):
|
|
def __init__(self):
|
|
super().__init__(NationalSalesInvoiceDocumentType())
|
|
|
|
|
|
class CreditNote(Invoice):
|
|
def __init__(self, invoice_document_reference: BillingReference):
|
|
super().__init__(CreditNoteDocumentType())
|
|
|
|
if not isinstance(invoice_document_reference, BillingReference):
|
|
raise TypeError('invoice_document_reference invalid type')
|
|
self.invoice_billing_reference = invoice_document_reference
|
|
|
|
def _get_codelist_tipo_operacion(self):
|
|
return codelist.TipoOperacionNC
|
|
|
|
|
|
class DebitNote(Invoice):
|
|
def __init__(self, invoice_document_reference: BillingReference):
|
|
super().__init__(DebitNoteDocumentType())
|
|
|
|
if not isinstance(invoice_document_reference, BillingReference):
|
|
raise TypeError('invoice_document_reference invalid type')
|
|
self.invoice_billing_reference = invoice_document_reference
|
|
|
|
def _get_codelist_tipo_operacion(self):
|
|
return codelist.TipoOperacionND
|