From bc8f00009da5ed1217d1341018a9d00da76ea7f3 Mon Sep 17 00:00:00 2001 From: gcarq Date: Sun, 21 May 2017 17:58:54 +0200 Subject: [PATCH] use a different db handle for dry_run --- main.py | 2 +- persistence.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index c21f56434..f4f3a1aeb 100755 --- a/main.py +++ b/main.py @@ -24,7 +24,7 @@ from utils import get_conf __author__ = "gcarq" __copyright__ = "gcarq 2017" __license__ = "GPLv3" -__version__ = "0.6.1" +__version__ = "0.7.0" conf = get_conf() diff --git a/persistence.py b/persistence.py index 5b38aeefa..12779d94f 100644 --- a/persistence.py +++ b/persistence.py @@ -6,11 +6,16 @@ from sqlalchemy.orm import scoped_session, sessionmaker from sqlalchemy.types import Enum from exchange import Exchange +from utils import get_conf +if get_conf().get('dry_run', False): + db_handle = 'sqlite:///tradesv2.dry_run.sqlite' +else: + db_handle = 'sqlite:///tradesv2.sqlite' -Base = declarative_base() -engine = create_engine('sqlite:///tradesv2.sqlite', echo=False) +engine = create_engine(db_handle, echo=False) Session = scoped_session(sessionmaker(bind=engine, autoflush=True, autocommit=True)) +Base = declarative_base() class Trade(Base):