rustico/core
2
0
Fork 0
Code Issues 9 Pull Requests Projects Releases Wiki Activity

use getters instead of non-existing attributes

Browse Source
This commit is contained in:
Giulio De Pasquale 2020-12-24 13:48:16 +00:00
parent 296f0abd67
commit 270619a316
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bfxbot/currency.py
View File

@ -59,10 +59,13 @@ class Currency:
def __init__(self, name: str, amount: float, price: float = None): def __init__(self, name: str, amount: float, price: float = None):
self.__name: str = name self.__name: str = name
self.__amount: float = amount self.__amount: float = amount
self.__price: float = price self.__price: Optional[float] = price
def __str__(self): def __str__(self):
if self.__price:
return f"{self.__name} {self.__amount} @ {self.__price}" return f"{self.__name} {self.__amount} @ {self.__price}"
else:
return f"{self.__name} {self.__amount}"
def __repr__(self): def __repr__(self):
return self.__str__() return self.__str__()
@ -100,7 +103,7 @@ class Balance:
self.__quote_equivalent: float = 0.0 self.__quote_equivalent: float = 0.0
self.__wallet: Optional[WalletKind] = wallet self.__wallet: Optional[WalletKind] = wallet
if currency.name == quote.value: if currency.name() == str(quote):
self.__quote_equivalent = currency.amount() self.__quote_equivalent = currency.amount()
else: else:
self.__quote_equivalent = currency.amount() * currency.price() self.__quote_equivalent = currency.amount() * currency.price()
@ -139,7 +142,7 @@ class BalanceGroup:
return self.__balances return self.__balances
def currency_names(self) -> List[str]: def currency_names(self) -> List[str]:
return list(map(lambda x: x.name, self.balances())) return list(map(lambda x: x.currency().name(), self.balances()))
def quote(self) -> Symbol: def quote(self) -> Symbol:
return self.__quote return self.__quote