Merge pull request #1489 from sbrunner/fix

Fix task done
sbrunner · Jan 26, 2025 · 0828694 · 0828694
2 parents fb8248e + 5203d23
commit 0828694
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/scan_to_paperless/process.py b/scan_to_paperless/process.py
@@ -2024,7 +2024,9 @@ async def _task(status: scan_to_paperless.status.Status) -> None:
         elif job_type == scan_to_paperless.status.JobType.DOWN:
             assert name is not None
             root_folder = os.path.join(os.environ.get("SCAN_SOURCE_FOLDER", "/source"), name)
-            shutil.rmtree(root_folder)
+            if os.path.exists(root_folder):
+                shutil.rmtree(root_folder)
+            status.remove_status(name)
         elif job_type == scan_to_paperless.status.JobType.CODE:
             assert name is not None
             print(f"Process code '{name}'")

diff --git a/scan_to_paperless/status.py b/scan_to_paperless/status.py
@@ -153,6 +153,12 @@ def set_status(
         if write:
             self.write()
 
+    def remove_status(self, name: str) -> None:
+        """Remove the status when a task is completely done."""
+        if name in self._status:
+            del self._status[name]
+            self.write()
+
     def _init(self) -> None:
         """Scan for changes for waiting documents."""
         self._update_scan_codes()