diff --git a/facho/fe/form.py b/facho/fe/form.py index e1d3de5..5b1e34e 100644 --- a/facho/fe/form.py +++ b/facho/fe/form.py @@ -88,10 +88,14 @@ class Price: @dataclass class PaymentMean: - id: str - code :str - due_at: datetime - payment_id: str + DEBIT = '01' + CREDIT = '02' + + def __init__(self, id: str, code: str, due_at: datetime, payment_id: str): + self.id = id + self.code = code + self.due_at = due_at + self.payment_id = payment_id @dataclass @@ -175,8 +179,8 @@ class Invoice: def set_customer(self, party: Party): self.invoice_customer = party - def set_payment_mean_debit(self, payment_id, code, due_at): - self.invoice_payment_mean = PaymentMean('01', code, due_at, payment_id) + def set_payment_mean(self, payment_mean: PaymentMean): + self.invoice_payment_mean = payment_mean def add_invoice_line(self, line: InvoiceLine): self.invoice_lines.append(line) diff --git a/tests/test_fe_form.py b/tests/test_fe_form.py index 511fade..09860ec 100644 --- a/tests/test_fe_form.py +++ b/tests/test_fe_form.py @@ -20,7 +20,7 @@ def simple_invoice_without_lines(): inv.set_period(datetime.now(), datetime.now()) inv.set_issue(datetime.now()) inv.set_ident('ABC123') - inv.set_payment_mean_debit('1234', '41', datetime.now()) + inv.set_payment_mean(form.PaymentMean(form.PaymentMean.DEBIT, '41', datetime.now(), '1234')) inv.set_supplier(form.Party( name = 'facho-supplier', ident = 123, @@ -43,7 +43,7 @@ def simple_invoice(): inv.set_period(datetime.now(), datetime.now()) inv.set_issue(datetime.now()) inv.set_ident('ABC123') - inv.set_payment_mean_debit('1234', '41', datetime.now()) + inv.set_payment_mean(form.PaymentMean(form.PaymentMean.DEBIT, '41', datetime.now(), ' 1234')) inv.set_supplier(form.Party( name = 'facho-supplier', ident = 123,