aljaz/hummingbot-dashboard
1
0
Fork 0
mirror of https://github.com/aljazceru/hummingbot-dashboard.git synced 2026-02-23 14:06:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

(feat) add db_path optional

Browse Source
This commit is contained in:
cardosofede
2023-12-07 17:18:00 -03:00
parent 6ddd74e4dd
commit 018aa5cbad
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
utils/optuna_database_manager.py
View File

@@ -1,5 +1,6 @@
import os
import json
from typing import Optional
import pandas as pd
from sqlalchemy import create_engine, text
@@ -9,9 +10,10 @@ from utils.data_manipulation import StrategyData
class OptunaDBManager:
def __init__(self, db_name):
def __init__(self, db_name, db_root_path: Optional[str]):
db_root_path = db_root_path or "data/backtesting"
self.db_name = db_name
self.db_path = f'sqlite:///{os.path.join("data/backtesting", db_name)}'
self.db_path = f'sqlite:///{os.path.join(db_root_path, db_name)}'
self.engine = create_engine(self.db_path, connect_args={'check_same_thread': False})
self.session_maker = sessionmaker(bind=self.engine)