feat(flags): add Unleash feature flagging integration

Author: aliu39

requirements-linting.txt

@@ -17,4 +17,5 @@ pre-commit # local linting
 httpcore
 openfeature-sdk
 launchdarkly-server-sdk
+UnleashClient
 typer

requirements-testing.txt

@@ -14,3 +14,4 @@ socksio
 httpcore[http2]
 setuptools
 Brotli
+UnleashClient

sentry_sdk/integrations/unleash.py

@@ -0,0 +1,41 @@
+from functools import wraps
+
+import sentry_sdk
+from sentry_sdk.integrations import Integration
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing import Optional
+    from UnleashClient import UnleashClient
+
+
+class UnleashIntegration(Integration):
+    identifier = "unleash"
+
+    def __init__(self, unleash_client):
+        # type: (Optional[UnleashClient]) -> None
+        self.unleash_client = unleash_client
+
+    @staticmethod
+    def setup_once():
+        # type: () -> None
+        integration = sentry_sdk.get_client().get_integration(UnleashIntegration)
+        if integration is None:
+            return
+
+        unleash_client = integration.unleash_client
+        old_is_enabled = unleash_client.is_enabled
+
+        @wraps(old_is_enabled)
+        def sentry_is_enabled(self, *a, **kw):
+            # TODO: # type:
+            key, value = a[0], a[1]  # TODO: this is a placeholder
+            if isinstance(value, bool):
+                flags = sentry_sdk.get_current_scope().flags
+                flags.set(key, value)
+            return old_is_enabled(self, *a, **kw)
+
+        unleash_client.is_enabled = sentry_is_enabled  # type: ignore
+
+        # TODO: get_variant

setup.py

@@ -80,6 +80,7 @@ def get_file_text(file_name):
         "starlette": ["starlette>=0.19.1"],
         "starlite": ["starlite>=1.48"],
         "tornado": ["tornado>=6"],
+        "unleash": ["UnleashClient>=6.0.1"],
     },
     entry_points={
         "opentelemetry_propagator": [

tox.ini

@@ -168,6 +168,10 @@ envlist =
     {py3.9,py3.11,py3.12}-langchain-latest
     {py3.9,py3.11,py3.12}-langchain-notiktoken
 
+    # LaunchDarkly
+    {py3.8,py3.12,py3.13}-launchdarkly-v9.8.0
+    {py3.8,py3.12,py3.13}-launchdarkly-latest
+
     # Litestar
     {py3.8,py3.11}-litestar-v{2.0}
     {py3.8,py3.11,py3.12}-litestar-v{2.6}
@@ -189,10 +193,6 @@ envlist =
     {py3.8,py3.12,py3.13}-openfeature-v0.7
     {py3.8,py3.12,py3.13}-openfeature-latest
 
-    # LaunchDarkly
-    {py3.8,py3.12,py3.13}-launchdarkly-v9.8.0
-    {py3.8,py3.12,py3.13}-launchdarkly-latest
-
     # OpenTelemetry (OTel)
     {py3.7,py3.9,py3.12,py3.13}-opentelemetry
 
@@ -291,6 +291,10 @@ envlist =
     {py3.7,py3.12,py3.13}-typer-v{0.15}
     {py3.7,py3.12,py3.13}-typer-latest
 
+    # Unleash
+    {py3.8,py3.12,py3.13}-unleash-v6.0.1
+    {py3.8,py3.12,py3.13}-unleash-latest
+
 [testenv]
 deps =
     # if you change requirements-testing.txt and your change is not being reflected
@@ -572,6 +576,10 @@ deps =
     launchdarkly-v9.8.0: launchdarkly-server-sdk~=9.8.0
     launchdarkly-latest: launchdarkly-server-sdk
 
+    # Unleash
+    unleash-v6.0.1: UnleashClient~=6.0.1
+    unleash-latest: UnleashClient
+
     # OpenTelemetry (OTel)
     opentelemetry: opentelemetry-distro
 
@@ -795,6 +803,7 @@ setenv =
     tornado: TESTPATH=tests/integrations/tornado
     trytond: TESTPATH=tests/integrations/trytond
     typer: TESTPATH=tests/integrations/typer
+    unleash: TESTPATH=tests/integrations/unleash
     socket: TESTPATH=tests/integrations/socket
 
 passenv =