rams/ftdt-quant-lab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: duckdb_load stats requires init_schema before querying tables

Browse Source
This commit is contained in:
ramseshk
2026-08-11 12:41:16 +08:00
parent 20ee340cef
commit 189691e8f2
1 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/duckdb_load.py
+21 -9
View File
@@ -434,16 +434,28 @@ class DuckDBLoader:
def stats(self) -> dict: def stats(self) -> dict:
"""Get current database statistics.""" """Get current database statistics."""
self.init_schema()
l2_count = self.conn.execute(
"SELECT COUNT(*) FROM l2_snapshots"
).fetchone()[0]
trade_count = self.conn.execute(
"SELECT COUNT(*) FROM trades"
).fetchone()[0]
funding_count = self.conn.execute(
"SELECT COUNT(*) FROM funding"
).fetchone()[0]
coins_result = self.conn.execute(
"SELECT DISTINCT coin FROM l2_snapshots"
).fetchall()
date_result = self.conn.execute(
"SELECT MIN(exchange_ts_ms), MAX(exchange_ts_ms) FROM l2_snapshots"
).fetchone()
return { return {
"l2_snapshots": self.conn.execute("SELECT COUNT(*) FROM l2_snapshots").fetchone()[0], "l2_snapshots": l2_count,
"trades": self.conn.execute("SELECT COUNT(*) FROM trades").fetchone()[0], "trades": trade_count,
"funding": self.conn.execute("SELECT COUNT(*) FROM funding").fetchone()[0], "funding": funding_count,
"coins": [r[0] for r in self.conn.execute( "coins": [r[0] for r in coins_result],
"SELECT DISTINCT coin FROM l2_snapshots" "date_range": date_result,
).fetchall()],
"date_range": self.conn.execute(
"SELECT MIN(exchange_ts_ms), MAX(exchange_ts_ms) FROM l2_snapshots"
).fetchone(),
} }
def close(self): def close(self):