script for daily info

scripts/daily_data.py

@@ -0,0 +1,31 @@
+import logging
+from utils import measure_execution_time
+from pull_data import DATA_DIR
+from tools import DEFAULT_FILENAME as TOOLS_FILENAME, generate_tools_file
+from profitability import (
+    analyse_all_traders,
+    label_trades_by_staking,
+)
+import pandas as pd
+
+logging.basicConfig(level=logging.INFO)
+
+
+@measure_execution_time
+def prepare_live_metrics(
+    tools_filename="new_tools.parquet", trades_filename="new_fpmmTrades.parquet"
+):
+    fpmmTrades = pd.read_parquet(DATA_DIR / trades_filename)
+    tools = pd.read_parquet(DATA_DIR / tools_filename)
+    print("Analysing trades...")
+    all_trades_df = analyse_all_traders(fpmmTrades, tools, daily_info=True)
+
+    # staking label
+    label_trades_by_staking(all_trades_df)
+
+    # save into a separate file
+    all_trades_df.to_parquet(DATA_DIR / "daily_info.parquet", index=False)
+
+
+if __name__ == "__main__":
+    prepare_live_metrics()