CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-20 11:21:44 +00:00
Code Issues 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

17
freqtrade/persistence/migrations.py
View File

@@ -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)