zenx 0.8.1__tar.gz → 0.8.2__tar.gz

{zenx-0.8.1 → zenx-0.8.2}/PKG-INFO

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

{zenx-0.8.1 → zenx-0.8.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zenx"
-version = "0.8.1"
+version = "0.8.2"
 description = "mini-framework"
 requires-python = ">=3.12"
 dependencies = [

{zenx-0.8.1 → zenx-0.8.2}/zenx/pipelines/google_rpc.py

@@ -83,8 +83,6 @@ try:
                 await self._stub.SubmitFeedMessage(feed_message)
             except grpc.RpcError as e:
                 self.logger.error("processing", exception=str(e), id=item['_id'], feed=self._feed_id, pipeline=self.name)
-            else:
-                self.logger.info("processed", id=item['_id'], feed=self._feed_id, pipeline=self.name)
 
 
         async def close(self) -> None:

{zenx-0.8.1 → zenx-0.8.2}/zenx/pipelines/websocket.py

@@ -24,6 +24,7 @@ try:
             self._connected = asyncio.Event()
             self._ws_client: Optional[websockets.ClientConnection] = None
             self._monitor_state_task: Optional[asyncio.Task] = None
+            self._listening_task: Optional[asyncio.Task] = None
 
 
         async def start(self) -> None:
@@ -37,6 +38,7 @@ try:
                 raise
             else:
                 self._monitor_state_task = asyncio.create_task(self._monitor_state())
+                self._listening_task = asyncio.create_task(self._listen())
 
         
         async def _monitor_state(self) -> None:
@@ -48,6 +50,15 @@ try:
                     await asyncio.sleep(0.5)
                 
 
+        async def _listen(self) -> None:
+            try:
+                await self._connected.wait()
+                async for msg in self._ws_client:
+                    self.logger.debug("received", msg=msg.decode(), pipeline=self.name)
+            except Exception:
+                await asyncio.sleep(1)
+
+
         async def _connect(self) -> None:
             self._connected.clear()
             self.logger.debug("connecting", pipeline=self.name)
@@ -84,12 +95,13 @@ try:
             
 
         async def close(self) -> None:
-            if self._monitor_state_task and not self._monitor_state_task.done():
-                self._monitor_state_task.cancel()
-                try:
-                    await self._monitor_state_task
-                except asyncio.CancelledError:
-                    pass
+            for t in [self._monitor_state_task, self._listening_task]:
+                if t and not t.done():
+                    t.cancel()
+                    try:
+                        await t
+                    except asyncio.CancelledError:
+                        pass
             await self._ws_client.close()
             self.logger.debug("closed", pipeline=self.name)
 

{zenx-0.8.1 → zenx-0.8.2}/zenx.egg-info/PKG-INFO

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