Skip to content

Commit da62e61

Browse files
committed
chore: improve typesafety for migrations
1 parent ca94909 commit da62e61

1 file changed

Lines changed: 13 additions & 4 deletions

File tree

freqtrade/persistence/migrations.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,9 @@ def get_backup_name(tabs: list[str], backup_prefix: str):
3030
return table_back_name
3131

3232

33-
def get_last_sequence_ids(engine, trade_back_name: str, order_back_name: str):
33+
def get_last_sequence_ids(
34+
engine, trade_back_name: str, order_back_name: str
35+
) -> tuple[int | None, int | None]:
3436
order_id: int | None = None
3537
trade_id: int | None = None
3638

@@ -48,14 +50,21 @@ def get_last_sequence_ids(engine, trade_back_name: str, order_back_name: str):
4850
return order_id, trade_id
4951

5052

51-
def set_sequence_ids(engine, order_id, trade_id, pairlock_id=None, kv_id=None, custom_data_id=None):
53+
def set_sequence_ids(
54+
engine,
55+
order_id: int | None = None,
56+
trade_id: int | None = None,
57+
pairlock_id: int | None = None,
58+
kv_id: int | None = None,
59+
custom_data_id: int | None = None,
60+
):
5261
"""
5362
Set sequence ids to the given values.
5463
The id's given should be the next id to use, so the current max id + 1 - or current id
5564
if using nextval before migration.
5665
:param engine: SQLAlchemy engine
57-
:param order_id: value to set for orders_id_seq
58-
:param trade_id: value to set for trades_id_seq
66+
:param order_id: value to set for orders_id_seq (optional)
67+
:param trade_id: value to set for trades_id_seq (optional)
5968
:param pairlock_id: value to set for pairlocks_id_seq (optional)
6069
:param kv_id: value to set for KeyValueStore_id_seq (optional)
6170
:param custom_data_id: value to set for trade_custom_data_id_seq (optional)

0 commit comments

Comments
 (0)