diff --git a/api/__init__.py b/api/__init__.py index 00c1638..c24d109 100644 --- a/api/__init__.py +++ b/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 diff --git a/api/database.py b/api/database.py index e82a702..6d9089f 100644 --- a/api/database.py +++ b/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)) diff --git a/create_project.py b/create_project.py new file mode 100644 index 0000000..a3e0262 --- /dev/null +++ b/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() diff --git a/dev_script.py b/dev_script.py new file mode 100644 index 0000000..a34e68b --- /dev/null +++ b/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}') diff --git a/tjnetwork.py b/tjnetwork.py index 12b9247..12bfeb4 100644 --- a/tjnetwork.py +++ b/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