From ed87abd93abe991f127fa9e0f1c8e03972e871d4 Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 13 Aug 2020 09:41:12 +0200 Subject: [PATCH] Allow selecting only a certain table range in migration --- freqtrade/persistence/migrations.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/freqtrade/persistence/migrations.py b/freqtrade/persistence/migrations.py index bd7dd103f..1bce8fef2 100644 --- a/freqtrade/persistence/migrations.py +++ b/freqtrade/persistence/migrations.py @@ -5,6 +5,9 @@ from sqlalchemy import inspect logger = logging.getLogger(__name__) +def get_table_names_for_table(inspector, tabletype): + return [t for t in inspector.get_table_names() if t.startswith(tabletype)] + def has_column(columns: List, searchname: str) -> bool: return len(list(filter(lambda x: x["name"] == searchname, columns))) == 1 @@ -21,7 +24,7 @@ def check_migrate(engine, decl_base) -> None: inspector = inspect(engine) cols = inspector.get_columns('trades') - tabs = inspector.get_table_names() + tabs = get_table_names_for_table(inspector, 'trades') table_back_name = 'trades_bak' for i, table_back_name in enumerate(tabs): table_back_name = f'trades_bak{i}'