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