Files
stat_pairs_backtest/src/tools/data_loader.py
T
2025-05-25 23:05:17 -04:00

25 lines
556 B
Python

import sys
import sqlite3
import pandas as pd
def load_sqlite_to_dataframe(db_path, query):
try:
conn = sqlite3.connect(db_path)
df = pd.read_sql_query(query, conn)
return df
except sqlite3.Error as excpt:
print(f"SQLite error: {excpt}")
raise
except Exception as e:
print(f"Error: {excpt}")
raise
finally:
if 'conn' in locals():
conn.close()
if __name__ == "__main__":
df1 = load_sqlite_to_dataframe(sys.argv[1], table_name='md_1min_bars')
print(df1)