From 8f55b21275e87beb19d961cb9c3fe932a1d262ca Mon Sep 17 00:00:00 2001
From: "WQY\\qiong" <qiongyu.wang@outlook.com>
Date: Thu, 16 Feb 2023 22:18:51 +0800
Subject: [PATCH] Support to set restore operation

---
 api/__init__.py | 2 +-
 api/database.py | 4 ++++
 api/parser.py   | 2 +-
 tjnetwork.py    | 3 +++
 4 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/api/__init__.py b/api/__init__.py
index bdd2505..f9e7c95 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -9,7 +9,7 @@ from .database import ChangeSet
 from .database import get_current_operation
 from .database import execute_undo, execute_redo
 from .database import have_snapshot, take_snapshot, pick_snapshot
-from .database import pick_operation, sync_with_server, get_restore_operation
+from .database import pick_operation, sync_with_server, get_restore_operation, set_restore_operation
 
 from .batch_cmd import execute_batch_command
 from .batch_cmds import execute_batch_commands
diff --git a/api/database.py b/api/database.py
index f2cba43..6a2d167 100644
--- a/api/database.py
+++ b/api/database.py
@@ -278,3 +278,7 @@ def sync_with_server(name: str, operation: int) -> ChangeSet:
 
 def get_restore_operation(name: str) -> int:
     return read(name, f'select * from restore_operation')['id']
+
+
+def set_restore_operation(name: str, operation: int) -> None:
+    write(name, f'update restore_operation set id = {operation}')
diff --git a/api/parser.py b/api/parser.py
index 8187882..2b3b009 100644
--- a/api/parser.py
+++ b/api/parser.py
@@ -229,7 +229,7 @@ def read_inp(project: str, inp: str):
 
     execute_batch_commands(project, _read_inp(inp))
     op = get_current_operation(project)
-    write(project, f'update restore_operation set id = {op}')
+    set_restore_operation(project, op)
 
 
 def dump_inp(project: str, inp: str):
diff --git a/tjnetwork.py b/tjnetwork.py
index ffbba3d..06c6b45 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -193,6 +193,9 @@ def execute_batch_commands(name: str, cs: ChangeSet) -> ChangeSet:
 def get_restore_operation(name: str) -> int:
     return api.get_restore_operation(name)
 
+def set_restore_operation(name: str, operation: int) -> None:
+    return api.set_restore_operation(name, operation)
+
 
 ############################################################
 # type