From da62e614d7018a7ec130d71a1923840bb5be6aa6 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 15 Feb 2026 18:10:42 +0100 Subject: [PATCH] chore: improve typesafety for migrations --- freqtrade/persistence/migrations.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/freqtrade/persistence/migrations.py b/freqtrade/persistence/migrations.py index c5f222122..1720b4d29 100644 --- a/freqtrade/persistence/migrations.py +++ b/freqtrade/persistence/migrations.py @@ -30,7 +30,9 @@ def get_backup_name(tabs: list[str], backup_prefix: str): return table_back_name -def get_last_sequence_ids(engine, trade_back_name: str, order_back_name: str): +def get_last_sequence_ids( + engine, trade_back_name: str, order_back_name: str +) -> tuple[int | None, int | None]: order_id: int | None = None trade_id: int | None = None @@ -48,14 +50,21 @@ def get_last_sequence_ids(engine, trade_back_name: str, order_back_name: str): return order_id, trade_id -def set_sequence_ids(engine, order_id, trade_id, pairlock_id=None, kv_id=None, custom_data_id=None): +def set_sequence_ids( + engine, + order_id: int | None = None, + trade_id: int | None = None, + pairlock_id: int | None = None, + kv_id: int | None = None, + custom_data_id: int | None = None, +): """ Set sequence ids to the given values. The id's given should be the next id to use, so the current max id + 1 - or current id if using nextval before migration. :param engine: SQLAlchemy engine - :param order_id: value to set for orders_id_seq - :param trade_id: value to set for trades_id_seq + :param order_id: value to set for orders_id_seq (optional) + :param trade_id: value to set for trades_id_seq (optional) :param pairlock_id: value to set for pairlocks_id_seq (optional) :param kv_id: value to set for KeyValueStore_id_seq (optional) :param custom_data_id: value to set for trade_custom_data_id_seq (optional)