From 7e396685c88b5213e2f44aadd0bf2eb1557332ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Richard=20Kub=C3=AD=C4=8Dek?= Date: Thu, 16 Jan 2025 13:11:28 +0100 Subject: [PATCH] invoice: add InvoiceType enum --- superfaktura/invoice.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/superfaktura/invoice.py b/superfaktura/invoice.py index 2110c69..34fa038 100644 --- a/superfaktura/invoice.py +++ b/superfaktura/invoice.py @@ -1,5 +1,4 @@ from dataclasses import dataclass, asdict -from pprint import pprint from typing import Optional, List import json @@ -88,6 +87,11 @@ class InvoiceItem: return data +class InvoiceType: + PROFORMA = "proforma" + INVOICE = "regular" + + class Invoice(SuperFakturaAPI): def __init__(self): super().__init__() @@ -113,7 +117,7 @@ if __name__ == "__main__": bank = BankAccount() invoice.add( invoice=InvoiceModel( - type="proforma", + type=InvoiceType.PROFORMA, name="Invoice 3", due=Date("2025-02-01"), invoice_currency=Currencies.CZK,