growthbook 1.4.10__py2.py3-none-any.whl → 2.0.0__py2.py3-none-any.whl

growthbook/__init__.py

@@ -18,5 +18,5 @@ from .plugins import (
 )
 
 # x-release-please-start-version
-__version__ = "1.4.10"
+__version__ = "2.0.0"
 # x-release-please-end

growthbook/growthbook.py

@@ -782,7 +782,7 @@ class GrowthBook(object):
         )
 
         if features:
-            self.setFeatures(features)
+            self.set_features(features)
 
         # Register for automatic feature updates when cache expires
         if self._client_key:
@@ -813,7 +813,7 @@ class GrowthBook(object):
             self._api_host, self._client_key, self._decryption_key, self._cache_ttl
         )
         if response is not None and "features" in response.keys():
-            self.setFeatures(response["features"])
+            self.set_features(response["features"])
 
         if response is not None and "savedGroups" in response:
             self._saved_groups = response["savedGroups"]
@@ -828,7 +828,7 @@ class GrowthBook(object):
 
         if features is not None:
             if "features" in features:
-                self.setFeatures(features["features"])
+                self.set_features(features["features"])
             if "savedGroups" in features:
                 self._saved_groups = features["savedGroups"]
             feature_repo.save_in_cache(self._client_key, features, self._cache_ttl)
@@ -842,7 +842,7 @@ class GrowthBook(object):
 
         if data is not None:
             if "features" in data:
-                self.setFeatures(data["features"])
+                self.set_features(data["features"])
             if "savedGroups" in data:
                 self._saved_groups = data["savedGroups"]
             feature_repo.save_in_cache(self._client_key, features, self._cache_ttl)

growthbook/growthbook_client.py

@@ -169,40 +169,78 @@ class EnhancedFeatureRepository(FeatureRepository, metaclass=SingletonMeta):
             if callback in self._callbacks:
                 self._callbacks.remove(callback)
 
+    """
+    _start_sse_refresh flow mimics a bridge pattern to connect a blocking, synchronous background thread 
+    (the SSEClient) with your non-blocking, async main loop.
+
+    Bridge - _maintain_sse_connection - runs on the main async loop, calls `startAutoRefresh` (which in turn spawns a thread)
+    and waits indefinitely. (Awaiting a Future suspends the coroutine, costing zero CPU)
+
+    The SSEClient runs in a separate thread, makes a blocking HTTP request, and invokes `on_event` synchronously.
+
+    The Hand off - when the event arrives (we're still on the background thread), sse_handler uses `asyncio.run_coroutine_threadsafe` 
+    to schedule the async processing `_handle_sse_event` onto the main event loop.
+    """
     async def _start_sse_refresh(self) -> None:
         """Start SSE-based feature refresh"""
         with self._refresh_lock:
             if self._refresh_task is not None:  # Already running
                 return
 
-            async def sse_handler(event_data: Dict[str, Any]) -> None:
+            # SSEClient invokes `on_event` synchronously from a background thread.
+            async def _handle_sse_event(event_data: Dict[str, Any]) -> None:
                 try:
-                    if event_data['type'] == 'features-updated':
+                    event_type = event_data.get("type")
+                    if event_type == "features-updated":
                         response = await self.load_features_async(
                             self._api_host, self._client_key, self._decryption_key, self._cache_ttl
                         )
                         if response is not None:
                             await self._handle_feature_update(response)
-                    elif event_data['type'] == 'features':
-                        await self._handle_feature_update(event_data['data'])
+                    elif event_type == "features":
+                        await self._handle_feature_update(event_data.get("data", {}))
                 except Exception:
-                    traceback.print_exc()
+                    logger.exception("Error handling SSE event")
 
-            # Start the SSE connection task
-            self._refresh_task = asyncio.create_task(
-                self._maintain_sse_connection(sse_handler)
-            )
+            main_loop = asyncio.get_running_loop()
 
-    async def _maintain_sse_connection(self, handler: Callable) -> None:
-        """Maintain SSE connection with automatic reconnection"""
-        while not self._stop_event.is_set():
-            try:
-                await self.startAutoRefresh(self._api_host, self._client_key, handler)
-            except Exception as e:
-                if not self._stop_event.is_set():
-                    delay = self._backoff.next_delay()
-                    logger.error(f"SSE connection lost, reconnecting in {delay:.2f}s: {str(e)}")
-                    await asyncio.sleep(delay)
+            # We must not pass an `async def` callback here (it would never be awaited).
+            def sse_handler(event_data: Dict[str, Any]) -> None:
+                # Schedule async processing onto the main event loop.
+                try:
+                    asyncio.run_coroutine_threadsafe(_handle_sse_event(event_data), main_loop)
+                except Exception:
+                    logger.exception("Failed to schedule SSE event handler")
+
+            async def _maintain_sse_connection() -> None:
+                """
+                Start SSE streaming and keep the task alive until cancelled.
+                """
+                try:
+                    # NOTE: `startAutoRefresh` is synchronous and starts a background thread.
+                    self.startAutoRefresh(self._api_host, self._client_key, sse_handler)
+                    
+                    # Wait indefinitely until the task is cancelled - basically saying "Keep this service 'active' until someone cancels me."
+                    # reconnection logic is handled inside SSEClient's thread
+                    await asyncio.Future()
+                except asyncio.CancelledError:
+                    # Normal shutdown flow
+                    raise
+                except Exception:
+                    logger.exception("Unexpected error in SSE lifecycle task")
+                finally:
+                    try:
+                        # stopAutoRefresh blocks joining a thread, so it needs to be run in executor
+                        # to avoid blocking the async event loop
+                        await main_loop.run_in_executor(
+                            None, 
+                            lambda: self.stopAutoRefresh(timeout=10)
+                        )
+                    except Exception:
+                        logger.exception("Failed to stop SSE auto-refresh")
+
+            # Start a task that owns the SSE lifecycle and cleanup.
+            self._refresh_task = asyncio.create_task(_maintain_sse_connection())
 
     async def _start_http_refresh(self, interval: int = 60) -> None:
         """Enhanced HTTP polling with backoff"""
@@ -261,6 +299,12 @@ class EnhancedFeatureRepository(FeatureRepository, metaclass=SingletonMeta):
     async def stop_refresh(self) -> None:
         """Clean shutdown of refresh tasks"""
         self._stop_event.set()
+        # Ensure any SSE background thread is stopped as well.
+        try:
+            self.stopAutoRefresh(timeout=10)
+        except Exception:
+            # Best-effort cleanup; task cancellation below will proceed.
+            logger.exception("Error stopping SSE auto-refresh")
         if self._refresh_task:
             # Cancel the task
             self._refresh_task.cancel()

growthbook/plugins/growthbook_tracking.py

@@ -132,8 +132,8 @@ class GrowthBookTrackingPlugin(GrowthBookPlugin):
         """Setup feature evaluation tracking."""
         original_eval_feature = gb_instance.eval_feature
         
-        def eval_feature_wrapper(key: str):
-            result = original_eval_feature(key)
+        def eval_feature_wrapper(key: str, *args, **kwargs):
+            result = original_eval_feature(key, *args, **kwargs)
             self._track_feature_evaluated(key, result, gb_instance)
             return result
         

{growthbook-1.4.10.dist-info → growthbook-2.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: growthbook
-Version: 1.4.10
+Version: 2.0.0
 Summary: Powerful Feature flagging and A/B testing for Python apps
 Home-page: https://github.com/growthbook/growthbook-python
 Author: GrowthBook

growthbook-2.0.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+growthbook/__init__.py,sha256=tksQ8wdB4p_iIIPCa1e5NsUie_NM-b1173zp5wkCGLQ,444
+growthbook/common_types.py,sha256=YKUmmYfzgrzLQ7kp2IPLc8QBA-B0QbnbF5viekNiTpw,15703
+growthbook/core.py,sha256=C1Nes_AiEuu6ypPghKuIeM2F22XUsLK1KrLW0xDwLYU,35963
+growthbook/growthbook.py,sha256=Ee6-jWPtvlgRvxRbsX-ZhrMV-8_F2T78Q7P30DVMsQM,47833
+growthbook/growthbook_client.py,sha256=YnbKGbO2taWZXtlutuTf_ZSkL_WmhYLhr39fB6BFIcw,27578
+growthbook/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+growthbook/plugins/__init__.py,sha256=y2eAV1sA041XWcftBVTDH0t-ggy9r2C5oKRYRF6XR6s,602
+growthbook/plugins/base.py,sha256=PWBXUBj62hi25Y5Eif9WmEWagWdkwGXHi2dMtn44bo8,3637
+growthbook/plugins/growthbook_tracking.py,sha256=yN2xOHtRNsJuxkm16wY0YBQFxjEXDKnKcup7C9bQwe4,11351
+growthbook/plugins/request_context.py,sha256=WzoGxalxPfrsN3RzfkvVYaUGat1A3N4AErnaS9IZ48Y,13005
+growthbook-2.0.0.dist-info/licenses/LICENSE,sha256=D-TcBckB0dTPUlNJ8jBiTIJIj1ekHLB1CY7HJtJKhMY,1069
+growthbook-2.0.0.dist-info/METADATA,sha256=yCwwaIc8N5XnwmAMNVcTtqN0ZPaFsbCtcgmgG-dD3wg,22726
+growthbook-2.0.0.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
+growthbook-2.0.0.dist-info/top_level.txt,sha256=dzfRQFGYejCIUstRSrrRVTMlxf7pBqASTI5S8gGRlXw,11
+growthbook-2.0.0.dist-info/RECORD,,

growthbook-1.4.10.dist-info/RECORD

@@ -1,15 +0,0 @@
-growthbook/__init__.py,sha256=UisD9HLBgfbN9HN7bKSmt9-KIulTfmU1XojdFC-lOZw,445
-growthbook/common_types.py,sha256=YKUmmYfzgrzLQ7kp2IPLc8QBA-B0QbnbF5viekNiTpw,15703
-growthbook/core.py,sha256=C1Nes_AiEuu6ypPghKuIeM2F22XUsLK1KrLW0xDwLYU,35963
-growthbook/growthbook.py,sha256=xHrc_wgrfl9oTbAxIDSpFet7Wb8diW4Fetv02Cpc4aA,47829
-growthbook/growthbook_client.py,sha256=dN7BSWJ2RDNIWLnHh8tYKedOc6FzdecvkG88YzHESto,25082
-growthbook/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-growthbook/plugins/__init__.py,sha256=y2eAV1sA041XWcftBVTDH0t-ggy9r2C5oKRYRF6XR6s,602
-growthbook/plugins/base.py,sha256=PWBXUBj62hi25Y5Eif9WmEWagWdkwGXHi2dMtn44bo8,3637
-growthbook/plugins/growthbook_tracking.py,sha256=lWO9ErUSrnqhcpWLp03XIrh45-BdBssdmLDVvaGvulY,11317
-growthbook/plugins/request_context.py,sha256=WzoGxalxPfrsN3RzfkvVYaUGat1A3N4AErnaS9IZ48Y,13005
-growthbook-1.4.10.dist-info/licenses/LICENSE,sha256=D-TcBckB0dTPUlNJ8jBiTIJIj1ekHLB1CY7HJtJKhMY,1069
-growthbook-1.4.10.dist-info/METADATA,sha256=TzLKYNF9RfuJDh-iOFKOF-e95s9U9l3N5cSo4Otmghg,22727
-growthbook-1.4.10.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
-growthbook-1.4.10.dist-info/top_level.txt,sha256=dzfRQFGYejCIUstRSrrRVTMlxf7pBqASTI5S8gGRlXw,11
-growthbook-1.4.10.dist-info/RECORD,,