From 4a66302303531e1d90fea90866a8acd5ffdc93c7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Richard=20Kub=C3=AD=C4=8Dek?= <rici.kubicek@gmail.com>
Date: Tue, 4 Mar 2025 13:59:01 +0100
Subject: [PATCH] add_inovice: add some protection clauses

---
 examples/add_invoice.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/examples/add_invoice.py b/examples/add_invoice.py
index 400f27f..d292a5c 100644
--- a/examples/add_invoice.py
+++ b/examples/add_invoice.py
@@ -21,7 +21,7 @@ Dependencies:
     - superfaktura.utils.data_types.Date
 """
 
-from superfaktura.bank_account import BankAccount
+from superfaktura.bank_account import BankAccount, NoDefaultBankAccountException
 from superfaktura.client_contacts import ClientContactModel
 from superfaktura.enumerations.currency import Currencies
 from superfaktura.enumerations.language import Language
@@ -41,6 +41,12 @@ def main():
     """
     invoice = Invoice()
     bank = BankAccount()
+    try:
+        # Get default bank account
+        bank_account = bank.default().as_dict()
+    except NoDefaultBankAccountException as e:
+        print(f"Error getting default bank account: {e}")
+        bank_account = {}
     resp = invoice.add(
         invoice_model=InvoiceModel(
             type=InvoiceType.INVOICE,
@@ -48,7 +54,7 @@ def main():
             due=Date("2025-04-01"),
             invoice_currency=Currencies.EUR,
             header_comment="We invoice you for services",
-            bank_accounts=[bank.default().as_dict()],
+            bank_accounts=[bank_account],
         ),
         items=[
             InvoiceItem(