flo-python 0.1.0.dev3__tar.gz → 0.1.0.dev4__tar.gz

{flo_python-0.1.0.dev3 → flo_python-0.1.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flo-python
-Version: 0.1.0.dev3
+Version: 0.1.0.dev4
 Summary: Python SDK for the Flo distributed systems platform
 Project-URL: Homepage, https://github.com/floruntime/flo-python
 Project-URL: Documentation, https://github.com/floruntime/flo-python#readme

{flo_python-0.1.0.dev3 → flo_python-0.1.0.dev4}/examples/worker.py

@@ -158,56 +158,39 @@ async def generate_report(ctx: ActionContext) -> bytes:
 
 
 async def main():
-    # Create and connect client (shared configuration)
-    client = FloClient(
-        os.getenv("FLO_ENDPOINT", "localhost:3000"),
+    async with FloClient(
+        os.getenv("FLO_ENDPOINT", "localhost:4455"),
         namespace=os.getenv("FLO_NAMESPACE", "myapp"),
         debug=os.getenv("FLO_DEBUG", "").lower() in ("1", "true"),
-    )
-    await client.connect()
-
-    # Create a worker from the client
-    worker = client.new_action_worker(
-        concurrency=5,
-        action_timeout=300,  # 5 minutes
-    )
+    ) as client:
+        worker = client.new_action_worker(
+            concurrency=5,
+            action_timeout=300,  # 5 minutes
+        )
 
-    # Register action handlers using register_action()
-    worker.register_action("process-order", process_order)
-    worker.register_action("send-notification", send_notification)
-    worker.register_action("generate-report", generate_report)
-
-    # Register using decorator syntax
-    @worker.action("health-check")
-    async def health_check(ctx: ActionContext) -> bytes:
-        """Simple health check action."""
-        return ctx.to_bytes({
-            "status": "healthy",
-            "worker_id": ctx.task_id,
-            "timestamp": ctx.created_at,
-        })
-
-    # Handle shutdown signals
-    stop_event = asyncio.Event()
-
-    def signal_handler():
-        logger.info("Received shutdown signal")
-        worker.stop()
-        stop_event.set()
-
-    loop = asyncio.get_running_loop()
-    for sig in (signal.SIGINT, signal.SIGTERM):
-        loop.add_signal_handler(sig, signal_handler)
-
-    # Start worker (blocks until stopped)
-    logger.info("Starting worker...")
-    try:
-        await worker.start()
-    except KeyboardInterrupt:
-        logger.info("Interrupted")
-    finally:
-        await worker.close()
-        await client.close()
+        # Register action handlers
+        worker.register_action("process-order", process_order)
+        worker.register_action("send-notification", send_notification)
+        worker.register_action("generate-report", generate_report)
+
+        # Register using decorator syntax
+        @worker.action("health-check")
+        async def health_check(ctx: ActionContext) -> bytes:
+            return ctx.to_bytes({"status": "healthy", "worker_id": ctx.task_id})
+
+        # Handle shutdown signals
+        def signal_handler():
+            logger.info("Received shutdown signal")
+            worker.stop()
+
+        loop = asyncio.get_running_loop()
+        for sig in (signal.SIGINT, signal.SIGTERM):
+            loop.add_signal_handler(sig, signal_handler)
+
+        # Start worker (blocks until stopped)
+        logger.info("Starting worker...")
+        async with worker:
+            await worker.start()
         logger.info("Worker shutdown complete")
 
 

{flo_python-0.1.0.dev3 → flo_python-0.1.0.dev4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "flo-python"
-version = "0.1.0.dev3"
+version = "0.1.0.dev4"
 description = "Python SDK for the Flo distributed systems platform"
 readme = "README.md"
 license = "MIT"

{flo_python-0.1.0.dev3 → flo_python-0.1.0.dev4}/src/flo/client.py

@@ -372,11 +372,13 @@ class FloClient:
         Returns:
             A new ActionWorker instance ready to register actions and start.
 
-        Example:
+        Example::
+
             async with FloClient("localhost:3000", namespace="myapp") as client:
                 worker = client.new_action_worker(concurrency=5)
                 worker.register_action("process-order", process_order)
-                await worker.start()
+                async with worker:
+                    await worker.start()
         """
         from .worker import ActionWorker
 

{flo_python-0.1.0.dev3 → flo_python-0.1.0.dev4}/src/flo/worker.py

@@ -3,7 +3,8 @@
 Provides ActionWorker for executing actions and StreamWorker for
 processing stream records via consumer groups.
 
-Example:
+Example::
+
     from flo import FloClient, ActionContext
 
     async def process_order(ctx: ActionContext) -> bytes:
@@ -14,8 +15,9 @@ Example:
     async def main():
         async with FloClient("localhost:3000", namespace="myapp") as client:
             worker = client.new_action_worker(concurrency=5)
-            worker.action("process-order")(process_order)
-            await worker.start()
+            worker.register_action("process-order", process_order)
+            async with worker:
+                await worker.start()
 """
 
 import asyncio
@@ -626,6 +628,19 @@ class ActionWorker:
         if self._client:
             await self._client.close()
 
+    async def __aenter__(self) -> "ActionWorker":
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: object,
+    ) -> None:
+        """Async context manager exit — closes the worker."""
+        await self.close()
+
 
 # =============================================================================
 # Stream Worker
@@ -977,3 +992,16 @@ class StreamWorker:
         self.stop()
         if self._client:
             await self._client.close()
+
+    async def __aenter__(self) -> "StreamWorker":
+        """Async context manager entry."""
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: object,
+    ) -> None:
+        """Async context manager exit — closes the stream worker."""
+        await self.close()