lacuscore 1.14.2__py3-none-any.whl → 1.14.3__py3-none-any.whl

lacuscore/lacuscore.py

@@ -19,7 +19,7 @@ from datetime import date, timedelta
 from ipaddress import ip_address, IPv4Address, IPv6Address
 from tempfile import NamedTemporaryFile
 from typing import Literal, Any, overload, cast, TYPE_CHECKING
-from collections.abc import Iterator
+from collections.abc import AsyncIterator
 from uuid import uuid4
 from urllib.parse import urlsplit
 
@@ -358,7 +358,7 @@ class LacusCore():
             return CaptureStatus.DONE
         return CaptureStatus.UNKNOWN
 
-    def consume_queue(self, max_consume: int) -> Iterator[Task]:  # type: ignore[type-arg]
+    async def consume_queue(self, max_consume: int) -> AsyncIterator[Task[None]]:
         """Trigger the capture for captures with the highest priority. Up to max_consume.
 
         :yield: Captures.
@@ -378,6 +378,8 @@ class LacusCore():
             yield task_logger.create_task(self._capture(uuid, priority), name=uuid,
                                           logger=logger,
                                           message='Capture raised an uncaught exception')
+            # Make sur the task starts.
+            await asyncio.sleep(0.5)
 
     async def _capture(self, uuid: str, priority: int) -> None:
         """Trigger a specific capture
@@ -643,8 +645,12 @@ class LacusCore():
             # If the capture fails to be stored in valkey, we must also remove the capture settings
             # so it is not dangling there.
 
-            if to_capture.document:
-                os.unlink(tmp_f.name)
+            try:
+                if to_capture.document:
+                    os.unlink(tmp_f.name)
+            except UnboundLocalError:
+                # Missing settings, the capture failed.
+                pass
 
             if retry:
                 if self.redis.zcard('lacus:to_capture') == 0:

{lacuscore-1.14.2.dist-info → lacuscore-1.14.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lacuscore
-Version: 1.14.2
+Version: 1.14.3
 Summary: Core of Lacus, usable as a module
 License: BSD-3-Clause
 Author: Raphaël Vinot

{lacuscore-1.14.2.dist-info → lacuscore-1.14.3.dist-info}/RECORD

@@ -1,10 +1,10 @@
 lacuscore/__init__.py,sha256=aLBshQPT9IBDKn5qWrX9A_exqtLFPyLsQiPWdfpAFjA,537
 lacuscore/helpers.py,sha256=d1rMHEA9A9NwfKksa2JvhLu-egaCBf1Vs2F7ALtAhmo,14264
 lacuscore/lacus_monitoring.py,sha256=r6IaYuh6sMq43eOWdZx0fU8p4PWVZlqSD6nr6yOaTUU,2713
-lacuscore/lacuscore.py,sha256=zUkfSSn9_tFE9vnFo309obHJdvBBfhVjwe7IDL7csoU,44492
+lacuscore/lacuscore.py,sha256=5JeM2XyMUZJWMIWul1j3GGKmzH8lUZubXkL-ec56rtk,44705
 lacuscore/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lacuscore/task_logger.py,sha256=2wDotU6r6vn-aKO8nZNdxSuisSj11LlcxuvW60qPL0Y,1909
-lacuscore-1.14.2.dist-info/LICENSE,sha256=4C4hLYrIkUD96Ggk-y_Go1Qf7PBZrEm9PSeTGe2nd4s,1516
-lacuscore-1.14.2.dist-info/METADATA,sha256=qkvp08WeFCrDYogV4NcoFFvGGlgvWZC3P6-RwkgtBJ0,2654
-lacuscore-1.14.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lacuscore-1.14.2.dist-info/RECORD,,
+lacuscore-1.14.3.dist-info/LICENSE,sha256=4C4hLYrIkUD96Ggk-y_Go1Qf7PBZrEm9PSeTGe2nd4s,1516
+lacuscore-1.14.3.dist-info/METADATA,sha256=BGqoy4gu2jancStlS3jBqfyjnVq9Z7APMmSRkKJzZV0,2654
+lacuscore-1.14.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lacuscore-1.14.3.dist-info/RECORD,,