Accept Merge Request #188: (script -> master)

Merge Request: Add some python scripts

Created By: @王琼钰
Accepted By: @王琼钰
URL: https://tjwater.coding.net/p/tjwatercloud/d/TJWaterServer/git/merge/188?initial=true

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

create_project.py

@@ -0,0 +1,13 @@
+import sys
+from tjnetwork import *
+
+def main():
+    if  len(sys.argv) != 2:
+        print("create_project which_inp")
+        return
+
+    inp = sys.argv[1]
+    read_inp(inp, f'./inp/{inp}.inp')
+
+if __name__ == '__main__':
+    main()

dev_script.py

@@ -0,0 +1,11 @@
+from tjnetwork import *
+
+if __name__ == '__main__':
+    p = 'net4'
+    copy_project('net3', p)
+    open_project(p)
+    set_vertex(p, ChangeSet({'link' : '20', 'coords': [{'x': 20.0, 'y': 40.0}, {'x': 40.0, 'y': 20.0}]}))
+    set_restore_operation_to_current(p)
+    close_project(p)
+    for i in range(1, 11):
+        copy_project(p, f'{p}_{i}')

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