chg: avoid trying to load a tree that is currenly being built

Lookyloo · Oct 13, 2023 · 4143d5f · 4143d5f
1 parent 74bc8fc
commit 4143d5f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lookyloo/capturecache.py b/lookyloo/capturecache.py
@@ -95,6 +95,8 @@ def __init__(self, cache_entry: Dict[str, Any]):
     def tree(self) -> CrawledTree:
         if not self.capture_dir.exists():
             raise MissingCaptureDirectory(f'The capture {self.uuid} does not exists in {self.capture_dir}.')
+        while is_locked(self.capture_dir):
+            time.sleep(5)
         return load_pickle_tree(self.capture_dir, self.capture_dir.stat().st_mtime, self.logger)