zenx 0.7.9__py3-none-any.whl → 0.8.1__py3-none-any.whl

zenx/engine.py

@@ -61,6 +61,7 @@ class Engine:
             if self.shutdown_event.is_set():
                 logger.info("shutdown", spider=spider_name)
             if spider.background_tasks:
+                logger.debug("waiting", background_tasks=len(spider.background_tasks), belong_to="spider")
                 await asyncio.gather(*spider.background_tasks)
             await client.close()
             await db.close()

zenx/pipelines/manager.py

@@ -16,7 +16,8 @@ class PipelineManager:
         self.pipelines = {name:Pipeline.get_pipeline(name)(logger, db, settings) for name in pipeline_names}
         self.settings = settings
         self._fire_and_forget_pipelines = [p for p in self.pipelines.values() if p.name != "preprocess"]
-    
+        self._background_tasks = set()
+
     
     async def start_pipelines(self) -> None:
         """ connect and monitor """
@@ -36,9 +37,14 @@ class PipelineManager:
                 self.logger.exception("process_item", item=item, pipeline=preprocess_pipeline.name)
                 raise
         for pipeline in self._fire_and_forget_pipelines:
-            asyncio.create_task(pipeline.process_item(item, spider))
+            t = asyncio.create_task(pipeline.process_item(item, spider))
+            self._background_tasks.add(t)
+            t.add_done_callback(self._background_tasks.discard)
 
     
     async def close_pipelines(self) -> None:
+        if self._background_tasks:
+            self.logger.debug("waiting", background_tasks=len(self._background_tasks), belong_to="pipeline_manager")
+            await asyncio.gather(*self._background_tasks)
         for pipeline in self.pipelines.values():
             await pipeline.close()

{zenx-0.7.9.dist-info → zenx-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zenx
-Version: 0.7.9
+Version: 0.8.1
 Summary: mini-framework
 Requires-Python: >=3.12
 Requires-Dist: curl-cffi>=0.12.0

{zenx-0.7.9.dist-info → zenx-0.8.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 zenx/cli.py,sha256=pHKhOTdqI6NQQoYK91waRIMpxCXLYtXEryzVaTbmvqc,2810
 zenx/debug_runner.py,sha256=B2Jd9A4_EHDa-ohLcwnFIxOV73FagTWXX2fl3qgwlpY,809
 zenx/discovery.py,sha256=YANVGzy2IG1fYruUud-11Y-ynyO6iEp3EjlHnhIQJQI,1014
-zenx/engine.py,sha256=XuPzQuDASkaEuxGFXU5ctntGMOyHitfxq0RgqT0sRmM,2444
+zenx/engine.py,sha256=CYwvmPSaI1lC-y_DfuOzX_1Ii14fxU8IdfMM7AIrUUU,2551
 zenx/exceptions.py,sha256=BJXxzwwX2CU6inhppfblx8c8Z6Mhvsk7MAhQ1LAnhBg,37
 zenx/logger.py,sha256=UmEk0vV1mSCozV7z_DDgCCXdAManDr5wgkrhKiRQtyU,1651
 zenx/settings.py,sha256=0RLeKcqHJdD1vuYIQtGeXkp2YIvv3tNyz9O8jw620VQ,1008
@@ -12,7 +12,7 @@ zenx/clients/http.py,sha256=fb6COYot6vidNFRBWgoU6CYEfnYWJP0JuVkydvxsHb4,5700
 zenx/pipelines/__init__.py,sha256=IxkZ0UpEJdYjLdd-PMcC9PzzzArTBNNcpgKc7NiOe5Y,131
 zenx/pipelines/base.py,sha256=N_388z5DFMeaU6wMwcClZAbQFWKh4kpAF7eUJhpQevs,1863
 zenx/pipelines/google_rpc.py,sha256=8DGcUlE6Th39XgD7WMMxqBMhUkH0kr3l9-cN_1x4CY4,4610
-zenx/pipelines/manager.py,sha256=bP2WIMblbWYHOJPU44l_IF3xLD0eHBYV4CCTFZc2FHI,1662
+zenx/pipelines/manager.py,sha256=lm2A_4h8NNdurFfwrrLwx5Z1tqJ9yZp2qgYvwGWd1lc,2017
 zenx/pipelines/preprocess.py,sha256=zXuhaLdH5KfBi_Bf-K5BgzFEujdmPWWpwYKkpOifT5U,816
 zenx/pipelines/websocket.py,sha256=jkRp5a6ro3q2JPAm8qtZWnf0EzFFUFOQgzgSxURXTvk,4359
 zenx/resources/proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -20,8 +20,8 @@ zenx/resources/proto/feed_pb2.py,sha256=ZyICOLnyuXekkvV4bAHZ1nE1-wwzcYYRRrmRJCMr
 zenx/resources/proto/feed_pb2_grpc.py,sha256=Mim6FfBgIMj0PmTqHk036nVUMJH3A6I3ts6r1j3bQF8,7441
 zenx/spiders/__init__.py,sha256=rs5LuqdM2MQlUYiTGJrzkYhzN8_SSLTrR7wGjSRrrSo,25
 zenx/spiders/base.py,sha256=BGxOvNnjaz6gmZm7NIgYc0RddWDmpIp53ukxprUUd-Y,1932
-zenx-0.7.9.dist-info/METADATA,sha256=orKA7TZTu6I7LK1o0txmKI4XDzckpy-JOhJ7HY4UFhg,1273
-zenx-0.7.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-zenx-0.7.9.dist-info/entry_points.txt,sha256=8JXob2f1VtvzGFris-e9Usqywg7oca-cChDlH9moOZU,38
-zenx-0.7.9.dist-info/top_level.txt,sha256=JeXwvK86d7sB-2x-avugFnZIZa33zaHWKI8RHWJR6KY,5
-zenx-0.7.9.dist-info/RECORD,,
+zenx-0.8.1.dist-info/METADATA,sha256=LOXt2oFxxR1Q66CIzM0qh1nV_zy4JwhOXnzNizUGZnA,1273
+zenx-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+zenx-0.8.1.dist-info/entry_points.txt,sha256=8JXob2f1VtvzGFris-e9Usqywg7oca-cChDlH9moOZU,38
+zenx-0.8.1.dist-info/top_level.txt,sha256=JeXwvK86d7sB-2x-avugFnZIZa33zaHWKI8RHWJR6KY,5
+zenx-0.8.1.dist-info/RECORD,,