From e88478fc13c54d18ab37fccbb3f899454e70cbf8 Mon Sep 17 00:00:00 2001 From: "WQY\\qiong" Date: Fri, 17 Feb 2023 21:35:08 +0800 Subject: [PATCH] Improve update snapshot --- api/database.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/database.py b/api/database.py index 4b64d74..474bf18 100644 --- a/api/database.py +++ b/api/database.py @@ -192,7 +192,10 @@ def take_snapshot(name: str, tag: str) -> int | None: def update_snapshot(name: str, operation: int, tag: str) -> None: if tag == None or tag == '': return None - write(name, f"update snapshot_operation set tag = '{tag}' where id = {operation}") + if have_snapshot_for_operation(name, operation): + write(name, f"update snapshot_operation set tag = '{tag}' where id = {operation}") + else: + take_snapshot_for_operation(name, operation, tag) def update_snapshot_for_current_operation(name: str, tag: str) -> None: