Skip to content

Commit 4d63136

Browse files
committed
refactor: more type safety in model
1 parent 14de25f commit 4d63136

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tuttle/model.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -593,17 +593,20 @@ class Invoice(SQLModel, table=True):
593593
@property
594594
def sum(self) -> Decimal:
595595
"""Sum over all invoice items."""
596-
return sum([item.subtotal for item in self.items])
596+
s = sum([item.subtotal for item in self.items])
597+
return Decimal(s)
597598

598599
@property
599600
def VAT_total(self) -> Decimal:
600601
"""Sum of VAT over all invoice items."""
601-
return sum(item.VAT for item in self.items)
602+
s = sum(item.VAT for item in self.items)
603+
return Decimal(s)
602604

603605
@property
604606
def total(self) -> Decimal:
605607
"""Total invoiced amount."""
606-
return self.sum + self.VAT_total
608+
t = self.sum + self.VAT_total
609+
return Decimal(t)
607610

608611
def generate_number(self, pattern=None, counter=None) -> str:
609612
"""Generate an invoice number"""

0 commit comments

Comments
 (0)