Fixed bug in calling function get_balance and added enhancements to use to_symbol argument (#2877)

Co-authored-by: Colin Delahunty <72827203+colin99d@users.noreply.github.com>
Co-authored-by: James Maslek <jmaslek11@gmail.com>
This commit is contained in:
0xDareDevil
2022-10-17 22:29:37 +05:30
committed by GitHub
parent d7a5b33f13
commit 7c68217eb4
2 changed files with 11 additions and 4 deletions

View File

@@ -194,15 +194,22 @@ def get_balance(
client = Client(cfg.API_BINANCE_KEY, cfg.API_BINANCE_SECRET)
pair = from_symbol + to_symbol
current_balance = client.get_asset_balance(asset=pair)
current_balance = client.get_asset_balance(asset=from_symbol)
if current_balance is None:
console.print("Check loaded coin\n")
return pd.DataFrame()
last_price = client.get_ticker(symbol=pair).get("lastPrice")
amounts = [float(current_balance["free"]), float(current_balance["locked"])]
df = pd.DataFrame(amounts).apply(lambda x: str(float(x)))
df = pd.DataFrame(amounts)
df.columns = ["Amount"]
df.index = ["Free", "Locked"]
df["Percent"] = df.div(df.sum(axis=0), axis=1).round(3)
if last_price:
last_price = float(last_price)
df[f"Amount [{to_symbol}]"] = df["Amount"].mul(last_price)
df["Percent"] = df["Amount"].mul(100).div(df["Amount"].sum(axis=0)).round(3)
return df

View File

@@ -953,7 +953,7 @@ class DueDiligenceController(CryptoBaseController):
if ns_parser:
binance_view.display_balance(
to_symbol=coin, from_symbol=ns_parser.vs, export=ns_parser.export
from_symbol=coin, to_symbol=ns_parser.vs, export=ns_parser.export
)
@log_start_end(log=logger)