filetime check

dmapp/dmcore/state.py

@@ -1,7 +1,13 @@
 from config import logger, states, tasks
 
 
-def save(doc_id: str, *, task: str | None = None, workspace: str | None = None) -> None:
+def save(
+    doc_id: str,
+    *,
+    task: str | None = None,
+    workspace: str | None = None,
+    filetime: str | None = None,
+) -> None:
     """
     Upsert a document with _id=doc_id, setting any of the provided fields.
     Leave fields you don’t pass unchanged.
@@ -11,6 +17,9 @@ def save(doc_id: str, *, task: str | None = None, workspace: str | None = None)
         updates["task"] = task
     if workspace is not None:
         updates["workspace"] = workspace
+    if filetime is not None:
+        updates["filetime"] = filetime
+
     if updates:
         states.update_one(
             {"_id": doc_id},
@@ -28,9 +37,18 @@ def retrieve(doc_id: str) -> dict[str, str | None]:
     return {
         "task": doc.get("task") if doc else None,
         "workspace": doc.get("workspace") if doc else None,
+        "filetime": doc.get("filetime") if doc else None,
     }
 
 
+# just
+def update_work_state(work: str, task_id: str):
+    """
+    update work state
+    """
+    states.update_one({"_id": "work"}, {"$set": {work: task_id}})
+
+
 def init_work_state(wd: dict):
     """
     init work states with default values