Set restore op to current

api/__init__.py

@@ -16,7 +16,7 @@ from .database import update_snapshot, update_snapshot_for_current_operation
 from .database import delete_snapshot, delete_snapshot_by_operation
 from .database import get_operation_by_snapshot, get_snapshot_by_operation
 from .database import pick_snapshot
-from .database import pick_operation, sync_with_server, get_restore_operation, set_restore_operation
+from .database import pick_operation, sync_with_server, get_restore_operation, set_restore_operation, set_restore_operation_to_current
 
 from .batch_cmd import execute_batch_command
 from .batch_cmds import execute_batch_commands

api/database.py

@@ -335,3 +335,7 @@ def get_restore_operation(name: str) -> int:
 
 def set_restore_operation(name: str, operation: int) -> None:
     write(name, f'update restore_operation set id = {operation}')
+
+
+def set_restore_operation_to_current(name: str) -> None:
+    return set_restore_operation(name, get_current_operation(name))

tjnetwork.py

@@ -288,6 +288,9 @@ def get_restore_operation(name: str) -> int:
 def set_restore_operation(name: str, operation: int) -> None:
     return api.set_restore_operation(name, operation)
 
+def set_restore_operation_to_current(name: str) -> None:
+    return api.set_restore_operation_to_current(name)
+
 
 ############################################################
 # type