ruff format .

Author: haacked

posthog/client.py

@@ -10,28 +10,54 @@
 from six import string_types
 from typing_extensions import Unpack
 
-from posthog.args import (ID_TYPES, ExceptionArg, OptionalCaptureArgs,
-                          OptionalSetArgs)
+from posthog.args import ID_TYPES, ExceptionArg, OptionalCaptureArgs, OptionalSetArgs
 from posthog.consumer import Consumer
-from posthog.contexts import (_get_current_context, get_context_distinct_id,
-                              get_context_session_id, new_context)
+from posthog.contexts import (
+    _get_current_context,
+    get_context_distinct_id,
+    get_context_session_id,
+    new_context,
+)
 from posthog.exception_capture import ExceptionCapture
-from posthog.exception_utils import (exc_info_from_error,
-                                     exception_is_already_captured,
-                                     exceptions_from_error_tuple,
-                                     handle_in_app, mark_exception_as_captured)
-from posthog.feature_flags import (InconclusiveMatchError,
-                                   match_feature_flag_properties)
+from posthog.exception_utils import (
+    exc_info_from_error,
+    exception_is_already_captured,
+    exceptions_from_error_tuple,
+    handle_in_app,
+    mark_exception_as_captured,
+)
+from posthog.feature_flags import InconclusiveMatchError, match_feature_flag_properties
 from posthog.poller import Poller
-from posthog.request import (DEFAULT_HOST, APIError, batch_post,
-                             determine_server_host, flags, get, remote_config)
-from posthog.types import (FeatureFlag, FeatureFlagResult, FlagMetadata,
-                           FlagsAndPayloads, FlagsResponse, FlagValue,
-                           normalize_flags_response, to_flags_and_payloads,
-                           to_payloads, to_values)
-from posthog.utils import (FlagCache, RedisFlagCache, SizeLimitedDict, clean,
-                           guess_timezone, remove_trailing_slash,
-                           system_context)
+from posthog.request import (
+    DEFAULT_HOST,
+    APIError,
+    batch_post,
+    determine_server_host,
+    flags,
+    get,
+    remote_config,
+)
+from posthog.types import (
+    FeatureFlag,
+    FeatureFlagResult,
+    FlagMetadata,
+    FlagsAndPayloads,
+    FlagsResponse,
+    FlagValue,
+    normalize_flags_response,
+    to_flags_and_payloads,
+    to_payloads,
+    to_values,
+)
+from posthog.utils import (
+    FlagCache,
+    RedisFlagCache,
+    SizeLimitedDict,
+    clean,
+    guess_timezone,
+    remove_trailing_slash,
+    system_context,
+)
 from posthog.version import VERSION
 
 try:
@@ -274,9 +300,9 @@ def feature_flags(self, flags):
             for flag in self._feature_flags
             if flag.get("key") is not None
         }
-        assert (
-            self.feature_flags_by_key is not None
-        ), "feature_flags_by_key should be initialized when feature_flags is set"
+        assert self.feature_flags_by_key is not None, (
+            "feature_flags_by_key should be initialized when feature_flags is set"
+        )
 
     def get_feature_variants(
         self,
@@ -1435,9 +1461,9 @@ def _locally_evaluate_flag(
         response = None
 
         if self.feature_flags:
-            assert (
-                self.feature_flags_by_key is not None
-            ), "feature_flags_by_key should be initialized when feature_flags is set"
+            assert self.feature_flags_by_key is not None, (
+                "feature_flags_by_key should be initialized when feature_flags is set"
+            )
             # Local evaluation
             flag = self.feature_flags_by_key.get(key)
             if flag:

posthog/request.py

@@ -139,23 +139,23 @@ def remote_config(
     timeout: int = 15,
 ) -> Any:
     """Get remote config flag value from remote_config API endpoint"""
-    url = remove_trailing_slash(host or DEFAULT_HOST) + f"/api/projects/@current/feature_flags/{key}/remote_config/"
-    
-    body = {
-        "api_key": project_api_key,
-        "sentAt": datetime.now(tz=tzutc()).isoformat()
-    }
-    
+    url = (
+        remove_trailing_slash(host or DEFAULT_HOST)
+        + f"/api/projects/@current/feature_flags/{key}/remote_config/"
+    )
+
+    body = {"api_key": project_api_key, "sentAt": datetime.now(tz=tzutc()).isoformat()}
+
     headers = {
         "Authorization": f"Bearer {personal_api_key}",
         "Content-Type": "application/json",
-        "User-Agent": USER_AGENT
+        "User-Agent": USER_AGENT,
     }
-    
+
     data = json.dumps(body, cls=DatetimeSerializer)
-    
+
     res = _session.post(url, data=data, headers=headers, timeout=timeout)
-    
+
     return _process_response(
         res,
         success_message=f"POST /api/projects/@current/feature_flags/{key}/remote_config/ completed successfully",

posthog/test/test_client.py

@@ -8,8 +8,7 @@
 from parameterized import parameterized
 
 from posthog.client import Client
-from posthog.contexts import (get_context_session_id, new_context,
-                              set_context_session)
+from posthog.contexts import get_context_session_id, new_context, set_context_session
 from posthog.request import APIError
 from posthog.test.test_utils import FAKE_TEST_API_KEY
 from posthog.types import FeatureFlag, LegacyFlagMetadata
@@ -1924,9 +1923,9 @@ def test_mock_system_context(
 
                 # Set up platform-specific mocks
                 if platform_method:
-                    getattr(mock_platform, platform_method).return_value = (
-                        platform_return
-                    )
+                    getattr(
+                        mock_platform, platform_method
+                    ).return_value = platform_return
 
                 # Special handling for Linux which uses distro module
                 if sys_platform == "linux":

remote_config_example.py

@@ -6,27 +6,29 @@
 import posthog
 
 # Initialize PostHog client
-#posthog.api_key = "phc_ml85lVi483SwbdDrqgb2LJOSXf050rPNpA6rrz7O5VY"
+# posthog.api_key = "phc_ml85lVi483SwbdDrqgb2LJOSXf050rPNpA6rrz7O5VY"
 posthog.api_key = "phc_8oSYhZJ7VpjQFNngdkUzEGctbY1u9P3SNJPcAKAhPvE"
 posthog.personal_api_key = "phs_QVofStqGt6UUMTlnxKoXb5cHAkQ7JesCZ1PUxfxNRUHKEUx"
 # posthog.personal_api_key = "phx_N8hy6HQcO0vplOc16fY2mYRBWVWl8mLDca38LgdmrwT3isA"
 posthog.host = "http://localhost:8000"
 posthog.debug = True
 
+
 def test_remote_config():
     """Test remote config payload retrieval."""
     print("Testing remote config endpoint...")
-    
+
     # Test feature flag key - replace with an actual flag key from your project
     flag_key = "unencrypted-remote-config-setting"
-    
+
     try:
         # Get remote config payload
         payload = posthog.get_remote_config_payload(flag_key)
         print(f"✅ Success! Remote config payload for '{flag_key}': {payload}")
-        
+
     except Exception as e:
         print(f"❌ Error getting remote config: {e}")
 
+
 if __name__ == "__main__":
-    test_remote_config()
+    test_remote_config()