🐛 don't print None when filename is empty

Author: ianardee

mindee/documents/bank_check.py

@@ -82,7 +82,7 @@ def __str__(self) -> str:
         )
         return (
             "-----Bank check data-----\n"
-            f"Filename: {self.filename}\n"
+            f"Filename: {self.filename or ''}".rstrip() + "\n"
             f"Routing number: {self.routing_number}\n"
             f"Account number: {self.account_number}\n"
             f"Check number: {self.check_number}\n"

mindee/documents/financial_document.py

@@ -126,10 +126,10 @@ def _build_from_api_prediction(
     def __str__(self) -> str:
         return (
             "-----Financial Document data-----\n"
-            "Filename: %s \n"
-            "Invoice number: %s \n"
-            "Total amount including taxes: %s \n"
-            "Total amount excluding taxes: %s \n"
+            f"Filename: {self.filename or ''}".rstrip() + "\n"
+            f"Invoice number: {self.invoice_number.value}\n"
+            f"Total amount including taxes: {self.total_incl.value}\n"
+            f"Total amount excluding taxes: {self.total_excl.value}\n"
             "Date: %s\n"
             "Invoice due date: %s\n"
             "Supplier name: %s\n"
@@ -141,10 +141,6 @@ def __str__(self) -> str:
             "Total taxes: %s\n"
             "----------------------"
             % (
-                self.filename,
-                self.invoice_number.value,
-                self.total_incl.value,
-                self.total_excl.value,
                 self.date.value,
                 self.due_date.value,
                 self.merchant_name.value,

mindee/documents/invoice.py

@@ -130,7 +130,7 @@ def __str__(self) -> str:
         taxes = "\n       ".join(f"{t}" for t in self.taxes)
         return (
             "-----Invoice data-----\n"
-            f"Filename: {self.filename}\n"
+            f"Filename: {self.filename or ''}".rstrip() + "\n"
             f"Invoice number: {self.invoice_number}\n"
             f"Total amount including taxes: {self.total_incl}\n"
             f"Total amount excluding taxes: {self.total_excl}\n"

mindee/documents/passport.py

@@ -95,7 +95,7 @@ def __str__(self) -> str:
         )
         return (
             "-----Passport data-----\n"
-            f"Filename: {self.filename}\n"
+            f"Filename: {self.filename or ''}".rstrip() + "\n"
             f"Full name: {self.full_name}\n"
             f"Given names: {given_names}\n"
             f"Surname: {self.surname}\n"

mindee/documents/receipt.py

@@ -57,7 +57,7 @@ def __str__(self) -> str:
         taxes = "\n       ".join(f"{t}" for t in self.taxes)
         return (
             "-----Receipt data-----\n"
-            f"Filename: {self.filename}\n"
+            f"Filename: {self.filename or ''}".rstrip() + "\n"
             f"Total amount including taxes: {self.total_incl}\n"
             f"Total amount excluding taxes: {self.total_excl}\n"
             f"Date: {self.date}\n"