mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: improve typesafety for migrations

Browse Source
This commit is contained in:
Matthias
2026-02-15 18:10:42 +01:00
parent ca94909ccc
commit da62e614d7
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/persistence/migrations.py
+13 -4
View File
@@ -30,7 +30,9 @@ def get_backup_name(tabs: list[str], backup_prefix: str):
return table_back_name 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 order_id: int | None = None
trade_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 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. 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 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. if using nextval before migration.
:param engine: SQLAlchemy engine :param engine: SQLAlchemy engine
:param order_id: value to set for orders_id_seq :param order_id: value to set for orders_id_seq (optional)
:param trade_id: value to set for trades_id_seq :param trade_id: value to set for trades_id_seq (optional)
:param pairlock_id: value to set for pairlocks_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 kv_id: value to set for KeyValueStore_id_seq (optional)
:param custom_data_id: value to set for trade_custom_data_id_seq (optional) :param custom_data_id: value to set for trade_custom_data_id_seq (optional)