chore: minor improvements
This commit is contained in:
@@ -103,5 +103,6 @@ def balance_distribution_over_time(
|
||||
df.loc[filled_at:end_date, trade.pair] -= real_amount
|
||||
df.loc[filled_at:, stake_currency] += stake - fee
|
||||
|
||||
# Round to avoid floating point issues
|
||||
df = df.round(14)
|
||||
return df
|
||||
|
||||
@@ -36,7 +36,7 @@ def convert_bt_wallet_collection(wallet_captures: list[tuple]) -> DataFrame:
|
||||
(date, currency, price, balance).
|
||||
"""
|
||||
if len(wallet_captures) == 0:
|
||||
return []
|
||||
return DataFrame()
|
||||
return DataFrame(
|
||||
wallet_captures,
|
||||
columns=["date", "currency", "price", "balance"],
|
||||
|
||||
@@ -109,7 +109,7 @@ def _migrate_wallet_history(config: Config, exchange: Exchange, starting_balance
|
||||
WalletHistory.session.bulk_save_objects(wallet_entries)
|
||||
WalletHistory.session.commit()
|
||||
KeyValueStore.store_value("wallet_history_migration_date", dt_now())
|
||||
print(f"Successfully created {len(wallet_entries)} wallet balance records")
|
||||
logger.info(f"Successfully created {len(wallet_entries)} wallet balance records")
|
||||
except Exception as e:
|
||||
WalletHistory.session.rollback()
|
||||
print(f"Error saving wallet balance records: {e}")
|
||||
logger.error(f"Error saving wallet balance records: {e}")
|
||||
|
||||
Reference in New Issue
Block a user