mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-10 12:47:25 -04:00
Compare commits
2 Commits
v0.3.0-alp
...
v0.3.0-alp
| Author | SHA1 | Date | |
|---|---|---|---|
| 90b6ad400d | |||
| 6e4b2a4cc5 |
@@ -32,15 +32,26 @@ def get_db_connection(db_path: str = "data/jobs.db") -> sqlite3.Connection:
|
|||||||
"""
|
"""
|
||||||
# Resolve path based on deployment mode
|
# Resolve path based on deployment mode
|
||||||
resolved_path = get_db_path(db_path)
|
resolved_path = get_db_path(db_path)
|
||||||
|
print(f"🔍 DIAGNOSTIC [get_db_connection]: Input path='{db_path}', Resolved path='{resolved_path}'")
|
||||||
|
|
||||||
# Ensure data directory exists
|
# Ensure data directory exists
|
||||||
db_path_obj = Path(resolved_path)
|
db_path_obj = Path(resolved_path)
|
||||||
db_path_obj.parent.mkdir(parents=True, exist_ok=True)
|
db_path_obj.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
# Check if database file exists
|
||||||
|
file_exists = db_path_obj.exists()
|
||||||
|
print(f"🔍 DIAGNOSTIC [get_db_connection]: Database file exists: {file_exists}")
|
||||||
|
|
||||||
conn = sqlite3.connect(resolved_path, check_same_thread=False)
|
conn = sqlite3.connect(resolved_path, check_same_thread=False)
|
||||||
conn.execute("PRAGMA foreign_keys = ON")
|
conn.execute("PRAGMA foreign_keys = ON")
|
||||||
conn.row_factory = sqlite3.Row
|
conn.row_factory = sqlite3.Row
|
||||||
|
|
||||||
|
# Verify tables exist
|
||||||
|
cursor = conn.cursor()
|
||||||
|
cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
|
||||||
|
tables = [row[0] for row in cursor.fetchall()]
|
||||||
|
print(f"🔍 DIAGNOSTIC [get_db_connection]: Tables in database: {tables}")
|
||||||
|
|
||||||
return conn
|
return conn
|
||||||
|
|
||||||
|
|
||||||
@@ -274,6 +285,15 @@ def initialize_dev_database(db_path: str = "data/trading_dev.db") -> None:
|
|||||||
initialize_database(db_path)
|
initialize_database(db_path)
|
||||||
print(f"🔍 DIAGNOSTIC: initialize_dev_database() COMPLETED successfully")
|
print(f"🔍 DIAGNOSTIC: initialize_dev_database() COMPLETED successfully")
|
||||||
|
|
||||||
|
# Verify tables were created
|
||||||
|
print(f"🔍 DIAGNOSTIC: Verifying tables exist in {db_path}")
|
||||||
|
verify_conn = sqlite3.connect(db_path)
|
||||||
|
verify_cursor = verify_conn.cursor()
|
||||||
|
verify_cursor.execute("SELECT name FROM sqlite_master WHERE type='table'")
|
||||||
|
tables = [row[0] for row in verify_cursor.fetchall()]
|
||||||
|
verify_conn.close()
|
||||||
|
print(f"🔍 DIAGNOSTIC: Tables found: {tables}")
|
||||||
|
|
||||||
|
|
||||||
def cleanup_dev_database(db_path: str = "data/trading_dev.db", data_path: str = "./data/dev_agent_data") -> None:
|
def cleanup_dev_database(db_path: str = "data/trading_dev.db", data_path: str = "./data/dev_agent_data") -> None:
|
||||||
"""
|
"""
|
||||||
|
|||||||
@@ -69,8 +69,16 @@ def get_db_path(base_db_path: str) -> str:
|
|||||||
Example:
|
Example:
|
||||||
PROD: "data/trading.db" -> "data/trading.db"
|
PROD: "data/trading.db" -> "data/trading.db"
|
||||||
DEV: "data/trading.db" -> "data/trading_dev.db"
|
DEV: "data/trading.db" -> "data/trading_dev.db"
|
||||||
|
|
||||||
|
Note:
|
||||||
|
This function is idempotent - calling it multiple times on the same
|
||||||
|
path will not add multiple _dev suffixes.
|
||||||
"""
|
"""
|
||||||
if is_dev_mode():
|
if is_dev_mode():
|
||||||
|
# Check if already has _dev suffix (idempotent)
|
||||||
|
if "_dev.db" in base_db_path:
|
||||||
|
return base_db_path
|
||||||
|
|
||||||
# Insert _dev before .db extension
|
# Insert _dev before .db extension
|
||||||
if base_db_path.endswith(".db"):
|
if base_db_path.endswith(".db"):
|
||||||
return base_db_path[:-3] + "_dev.db"
|
return base_db_path[:-3] + "_dev.db"
|
||||||
|
|||||||
Reference in New Issue
Block a user