growthbook 1.4.4__tar.gz → 1.4.5__tar.gz

{growthbook-1.4.4/growthbook.egg-info → growthbook-1.4.5}/PKG-INFO

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

{growthbook-1.4.4 → growthbook-1.4.5}/growthbook/__init__.py

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

{growthbook-1.4.4 → growthbook-1.4.5}/growthbook/common_types.py

@@ -426,6 +426,7 @@ class Options:
     sticky_bucket_service: Optional[AbstractStickyBucketService] = None
     sticky_bucket_identifier_attributes: Optional[List[str]] = None
     on_experiment_viewed: Optional[Callable[[Experiment, Result, Optional[UserContext]], None]] = None
+    on_feature_usage: Optional[Callable[[str, 'FeatureResult'], None]] = None
     tracking_plugins: Optional[List[Any]] = None
 
 

{growthbook-1.4.4 → growthbook-1.4.5}/growthbook/growthbook.py

@@ -565,6 +565,7 @@ class GrowthBook(object):
         features: dict = {},
         qa_mode: bool = False,
         on_experiment_viewed=None,
+        on_feature_usage=None,
         api_host: str = "",
         client_key: str = "",
         decryption_key: str = "",
@@ -603,6 +604,7 @@ class GrowthBook(object):
 
         self._qaMode = qa_mode or qaMode
         self._trackingCallback = on_experiment_viewed or trackingCallback
+        self._featureUsageCallback = on_feature_usage
 
         self._streaming = streaming
         self._streaming_timeout = streaming_connection_timeout
@@ -823,6 +825,7 @@ class GrowthBook(object):
             self._tracked.clear()
             self._assigned.clear()
             self._trackingCallback = None
+            self._featureUsageCallback = None
             self._forcedVariations.clear()
             self._overrides.clear()
             self._groups.clear()
@@ -886,11 +889,18 @@ class GrowthBook(object):
         )
 
     def eval_feature(self, key: str) -> FeatureResult:
-        return core_eval_feature(key=key, 
-                                 evalContext=self._get_eval_context(), 
-                                 callback_subscription=self._fireSubscriptions,
-                                 tracking_cb=self._track
-                                 )
+        result = core_eval_feature(key=key, 
+                                   evalContext=self._get_eval_context(), 
+                                   callback_subscription=self._fireSubscriptions,
+                                   tracking_cb=self._track
+                                   )
+        # Call feature usage callback if provided
+        if self._featureUsageCallback:
+            try:
+                self._featureUsageCallback(key, result)
+            except Exception:
+                pass
+        return result
 
     # @deprecated, use get_all_results
     def getAllResults(self):

{growthbook-1.4.4 → growthbook-1.4.5}/growthbook/growthbook_client.py

@@ -500,24 +500,50 @@ class GrowthBookClient:
         async with self._context_lock:
             context = await self.create_evaluation_context(user_context)
             result = core_eval_feature(key=key, evalContext=context, tracking_cb=self._track)
+            # Call feature usage callback if provided
+            if self.options.on_feature_usage:
+                try:
+                    self.options.on_feature_usage(key, result)
+                except Exception:
+                    logger.exception("Error in feature usage callback")
             return result
 
     async def is_on(self, key: str, user_context: UserContext) -> bool:
         """Check if a feature is enabled with proper async context management"""
         async with self._context_lock:
             context = await self.create_evaluation_context(user_context)
-            return core_eval_feature(key=key, evalContext=context, tracking_cb=self._track).on
+            result = core_eval_feature(key=key, evalContext=context, tracking_cb=self._track)
+            # Call feature usage callback if provided
+            if self.options.on_feature_usage:
+                try:
+                    self.options.on_feature_usage(key, result)
+                except Exception:
+                    logger.exception("Error in feature usage callback")
+            return result.on
     
     async def is_off(self, key: str, user_context: UserContext) -> bool:
         """Check if a feature is set to off with proper async context management"""
         async with self._context_lock:
             context = await self.create_evaluation_context(user_context)
-            return core_eval_feature(key=key, evalContext=context, tracking_cb=self._track).off
+            result = core_eval_feature(key=key, evalContext=context, tracking_cb=self._track)
+            # Call feature usage callback if provided
+            if self.options.on_feature_usage:
+                try:
+                    self.options.on_feature_usage(key, result)
+                except Exception:
+                    logger.exception("Error in feature usage callback")
+            return result.off
     
     async def get_feature_value(self, key: str, fallback: Any, user_context: UserContext) -> Any:
         async with self._context_lock:
             context = await self.create_evaluation_context(user_context)
             result = core_eval_feature(key=key, evalContext=context, tracking_cb=self._track)
+            # Call feature usage callback if provided
+            if self.options.on_feature_usage:
+                try:
+                    self.options.on_feature_usage(key, result)
+                except Exception:
+                    logger.exception("Error in feature usage callback")
             return result.value if result.value is not None else fallback
 
     async def run(self, experiment: Experiment, user_context: UserContext) -> Result:

{growthbook-1.4.4 → growthbook-1.4.5/growthbook.egg-info}/PKG-INFO

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

{growthbook-1.4.4 → growthbook-1.4.5}/setup.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 1.4.4
+current_version = 1.4.5
 commit = True
 tag = True
 

{growthbook-1.4.4 → growthbook-1.4.5}/tests/conftest.py

@@ -1,4 +1,5 @@
 import pytest
+import pytest_asyncio
 import asyncio
 import os
 import sys
@@ -22,7 +23,7 @@ def reset_singleton():
                 instance._stop_event.set()
         SingletonMeta._instances.clear()
 
-@pytest.fixture(autouse=True)
+@pytest_asyncio.fixture(autouse=True)
 async def cleanup_tasks():
     """Cleanup any pending tasks after each test."""
     yield

{growthbook-1.4.4 → growthbook-1.4.5}/tests/test_growthbook.py

@@ -215,6 +215,85 @@ def test_tracking():
     gb.destroy()
 
 
+def test_feature_usage_callback():
+    """Test that feature usage callback is called correctly"""
+    calls = []
+    
+    def feature_usage_cb(key, result):
+        calls.append([key, result])
+    
+    gb = GrowthBook(
+        attributes={"id": "1"},
+        on_feature_usage=feature_usage_cb,
+        features={
+            "feature-1": Feature(defaultValue=True),
+            "feature-2": Feature(defaultValue=False),
+            "feature-3": Feature(
+                defaultValue="blue",
+                rules=[
+                    FeatureRule(force="red", condition={"id": "1"})
+                ]
+            ),
+        }
+    )
+    
+    # Test eval_feature
+    result1 = gb.eval_feature("feature-1")
+    assert len(calls) == 1
+    assert calls[0][0] == "feature-1"
+    assert calls[0][1].value is True
+    assert calls[0][1].source == "defaultValue"
+    
+    # Test is_on
+    gb.is_on("feature-2")
+    assert len(calls) == 2
+    assert calls[1][0] == "feature-2"
+    assert calls[1][1].value is False
+    
+    # Test get_feature_value
+    value = gb.get_feature_value("feature-3", "blue")
+    assert len(calls) == 3
+    assert calls[2][0] == "feature-3"
+    assert calls[2][1].value == "red"
+    assert value == "red"
+    
+    # Test is_off
+    gb.is_off("feature-1")
+    assert len(calls) == 4
+    assert calls[3][0] == "feature-1"
+    
+    # Calling same feature multiple times should trigger callback each time
+    gb.eval_feature("feature-1")
+    gb.eval_feature("feature-1")
+    assert len(calls) == 6
+    
+    gb.destroy()
+
+
+def test_feature_usage_callback_error_handling():
+    """Test that feature usage callback errors are handled gracefully"""
+    
+    def failing_callback(key, result):
+        raise Exception("Callback error")
+    
+    gb = GrowthBook(
+        attributes={"id": "1"},
+        on_feature_usage=failing_callback,
+        features={
+            "feature-1": Feature(defaultValue=True),
+        }
+    )
+    
+    # Should not raise an error even if callback fails
+    result = gb.eval_feature("feature-1")
+    assert result.value is True
+    
+    # Should work with is_on as well
+    assert gb.is_on("feature-1") is True
+    
+    gb.destroy()
+
+
 def test_handles_weird_experiment_values():
     gb = GrowthBook(attributes={"id": "1"})
 

{growthbook-1.4.4 → growthbook-1.4.5}/tests/test_growthbook_client.py

@@ -767,5 +767,129 @@ async def test_handles_tracking_errors():
             calls = getMockedTrackingCalls()
             assert len(calls) == 1, "Expected exactly 1 tracking call"
 
+    finally:
+        await client.close()
+
+
+@pytest.mark.asyncio
+async def test_feature_usage_callback():
+    """Test that feature usage callback is called correctly"""
+    calls = []
+    
+    def feature_usage_cb(key, result):
+        calls.append([key, result])
+    
+    client = GrowthBookClient(Options(
+        api_host="https://localhost.growthbook.io",
+        client_key="test-key",
+        enabled=True,
+        on_feature_usage=feature_usage_cb
+    ))
+    
+    user_context = UserContext(attributes={"id": "1"})
+    
+    try:
+        # Set up mocks for feature repository
+        mock_features = {
+            "features": {
+                "feature-1": {"defaultValue": True},
+                "feature-2": {"defaultValue": False},
+                "feature-3": {
+                    "defaultValue": "blue",
+                    "rules": [
+                        {"force": "red", "condition": {"id": "1"}}
+                    ]
+                },
+            },
+            "savedGroups": {}
+        }
+        
+        with patch('growthbook.FeatureRepository.load_features_async', 
+                  new_callable=AsyncMock, return_value=mock_features), \
+             patch('growthbook.growthbook_client.EnhancedFeatureRepository.start_feature_refresh',
+                  new_callable=AsyncMock), \
+             patch('growthbook.growthbook_client.EnhancedFeatureRepository.stop_refresh',
+                  new_callable=AsyncMock):
+            
+            # Initialize client
+            await client.initialize()
+            
+            # Test eval_feature
+            result1 = await client.eval_feature("feature-1", user_context)
+            assert len(calls) == 1
+            assert calls[0][0] == "feature-1"
+            assert calls[0][1].value is True
+            assert calls[0][1].source == "defaultValue"
+            
+            # Test is_on
+            await client.is_on("feature-2", user_context)
+            assert len(calls) == 2
+            assert calls[1][0] == "feature-2"
+            assert calls[1][1].value is False
+            
+            # Test get_feature_value
+            value = await client.get_feature_value("feature-3", "blue", user_context)
+            assert len(calls) == 3
+            assert calls[2][0] == "feature-3"
+            assert calls[2][1].value == "red"
+            assert value == "red"
+            
+            # Test is_off
+            await client.is_off("feature-1", user_context)
+            assert len(calls) == 4
+            assert calls[3][0] == "feature-1"
+            
+            # Calling same feature multiple times should trigger callback each time
+            await client.eval_feature("feature-1", user_context)
+            await client.eval_feature("feature-1", user_context)
+            assert len(calls) == 6
+            
+    finally:
+        await client.close()
+
+
+@pytest.mark.asyncio
+async def test_feature_usage_callback_error_handling():
+    """Test that feature usage callback errors are handled gracefully"""
+    
+    def failing_callback(key, result):
+        raise Exception("Callback error")
+    
+    client = GrowthBookClient(Options(
+        api_host="https://localhost.growthbook.io",
+        client_key="test-key",
+        enabled=True,
+        on_feature_usage=failing_callback
+    ))
+    
+    user_context = UserContext(attributes={"id": "1"})
+    
+    try:
+        # Set up mocks for feature repository
+        mock_features = {
+            "features": {
+                "feature-1": {"defaultValue": True},
+            },
+            "savedGroups": {}
+        }
+        
+        with patch('growthbook.FeatureRepository.load_features_async', 
+                  new_callable=AsyncMock, return_value=mock_features), \
+             patch('growthbook.growthbook_client.EnhancedFeatureRepository.start_feature_refresh',
+                  new_callable=AsyncMock), \
+             patch('growthbook.growthbook_client.EnhancedFeatureRepository.stop_refresh',
+                  new_callable=AsyncMock):
+            
+            # Initialize client
+            await client.initialize()
+            
+            # Should not raise an error even if callback fails
+            result = await client.eval_feature("feature-1", user_context)
+            assert result.value is True
+            
+            # Should work with is_on as well
+            is_on = await client.is_on("feature-1", user_context)
+            assert is_on is True
+            
     finally:
         await client.close()