Merge pull request #17 from kalaspuff/feature/currency-to-money

Added money() method to currency objects

stockholm/currency.py

@@ -1,6 +1,8 @@
 import sys
 from typing import Any, Dict, List, Optional, Set, Tuple, Type, Union, cast
 
+from decimal import Decimal
+
 
 class MetaCurrency(type):
     ticker: str
@@ -31,6 +33,27 @@ def __new__(cls, name: str, bases: Tuple[type, ...], attributedict: Dict) -> "Me
         result: Type[BaseCurrency] = type.__new__(cls, name, bases, attributedict)
         return result
 
+    def money(
+        self,
+        amount: Optional[Union["Money", Decimal, int, float, str, object]] = None,
+        from_sub_units: Optional[bool] = None,
+        units: Optional[int] = None,
+        nanos: Optional[int] = None,
+        currency_code: Optional[str] = None,
+        **kwargs: Any,
+    ) -> "Money":
+        kwargs.pop("currency", None)
+
+        return Money(
+            amount,
+            currency=cast(BaseCurrency, self),
+            from_sub_units=from_sub_units,
+            units=units,
+            nanos=nanos,
+            currency_code=currency_code,
+            **kwargs,
+        )
+
     def __setattr__(self, *args: Any) -> None:
         raise AttributeError("Attributes of currencies cannot be changed")
 
@@ -127,6 +150,28 @@ def __init__(
         object.__setattr__(self, "_meta", False)
         object.__setattr__(self, "as_string", lambda: str(self))
         object.__setattr__(self, "as_str", lambda: str(self))
+        object.__setattr__(self, "money", lambda *args, **kwargs: self._money(*args, **kwargs))
+
+    def _money(
+        self,
+        amount: Optional[Union["Money", Decimal, int, float, str, object]] = None,
+        from_sub_units: Optional[bool] = None,
+        units: Optional[int] = None,
+        nanos: Optional[int] = None,
+        currency_code: Optional[str] = None,
+        **kwargs: Any,
+    ) -> "Money":
+        kwargs.pop("currency", None)
+
+        return Money(
+            amount,
+            currency=self,
+            from_sub_units=from_sub_units,
+            units=units,
+            nanos=nanos,
+            currency_code=currency_code,
+            **kwargs,
+        )
 
     def __setattr__(self, *args: Any) -> None:
         raise AttributeError("Attributes of currencies cannot be changed")
@@ -1476,3 +1521,6 @@ def _load_currencies(currencies: List[str]) -> None:
 
 
 Currency._load_currencies(all_currencies())
+
+
+from stockholm.money import Money  # noqa

tests/test_currency.py

@@ -1,6 +1,7 @@
 from decimal import Decimal
 import pytest
 
+import stockholm
 from stockholm import BaseCurrency, Currency, Money
 from stockholm.currency import Bitcoin, CLF, DOGE, DogeCoin, Ethereum, IQD, JPY, USD, XBT, get_currency
 
@@ -292,6 +293,16 @@ def test_currency_types() -> None:
     assert str(Money.from_sub_units(Money("133742", Currency.SEK, from_sub_units=True).to_sub_units())) == "1337.42 SEK"
 
 
+def test_currency_to_money():
+    currency = Currency("NEW")
+
+    assert str(currency.money(100)) == "100.00 NEW"
+    assert str(Currency.SEK.money(100)) == "100.00 SEK"
+    assert str(Currency.JPY.money(100)) == "100 JPY"
+    assert str(JPY.money(100)) == "100 JPY"
+    assert str(stockholm.currency.Bitcoin.money("1.523")) == "1.523 BTC"
+
+
 def test_currency_constructors():
     currency = Currency("NEW")
     assert currency