counterweight/data-xexe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

improve typing

Browse source
This commit is contained in:
Pablo Martin 2024-06-27 18:11:22 +02:00
parent 0bd0d7c862
commit 2ab9ca5456
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
xexe/money_amount.py
View file

@ -12,11 +12,13 @@ DEFAULT_MONEY_PRECISION = Decimal(
class MoneyAmount:
def __init__(self, amount, currency):
def __init__(
self, amount: Union[int, Decimal, str], currency: Union[str, Currency]
) -> "MoneyAmount":
self.amount = self._parse_amount(amount)
self.currency = self._parse_currency(currency)
def _parse_amount(self, amount: Union[int, Decimal, str]):
def _parse_amount(self, amount: Union[int, Decimal, str]) -> Decimal:
if isinstance(amount, (int, Decimal)):
return Decimal(amount).quantize(DEFAULT_MONEY_PRECISION)
elif isinstance(amount, str):
@ -27,7 +29,7 @@ class MoneyAmount:
else:
raise TypeError(f"Amount must be int, Decimal, or str, not {type(amount)}")
def _parse_currency(self, currency):
def _parse_currency(self, currency: Union[str, Currency]) -> Currency:
if isinstance(currency, Currency):
return currency
if isinstance(currency, str):