Add param for Dry run to use a DB file instead of memory (#182)

2017-12-14 06:10:11 -08:00
parent 4b38100ae2
commit 2ac8b685d6
4 changed files with 40 additions and 15 deletions
--- a/freqtrade/persistence.py
+++ b/freqtrade/persistence.py
@@ -29,10 +29,15 @@ def init(config: dict, engine: Optional[Engine] = None) -> None:
    _CONF.update(config)
    if not engine:
        if _CONF.get('dry_run', False):
-            engine = create_engine('sqlite://',
-                                   connect_args={'check_same_thread': False},
-                                   poolclass=StaticPool,
-                                   echo=False)
+            # the user wants dry run to use a DB
+            if _CONF.get('dry_run_db', False):
+                engine = create_engine('sqlite:///tradesv3.dry_run.sqlite')
+            # Otherwise dry run will store in memory
+            else:
+                engine = create_engine('sqlite://',
+                                       connect_args={'check_same_thread': False},
+                                       poolclass=StaticPool,
+                                       echo=False)
        else:
            engine = create_engine('sqlite:///tradesv3.sqlite')