mypy

Author: dmarticus

posthog/client.py

@@ -1,4 +1,5 @@
 import atexit
+import hashlib
 import logging
 import numbers
 import os
@@ -11,20 +12,28 @@
 
 import distro  # For Linux OS detection
 from dateutil.tz import tzutc
-import hashlib
 from six import string_types
 
 from posthog.consumer import Consumer
 from posthog.exception_capture import ExceptionCapture
 from posthog.exception_utils import exc_info_from_error, exceptions_from_error_tuple, handle_in_app
 from posthog.feature_flags import InconclusiveMatchError, match_feature_flag_properties
 from posthog.poller import Poller
-from posthog.request import DEFAULT_HOST, APIError, batch_post, decide, determine_server_host, get, flags, remote_config
+from posthog.request import (
+    DEFAULT_HOST,
+    APIError,
+    batch_post,
+    decide,
+    determine_server_host,
+    flags,
+    get,
+    remote_config,
+)
 from posthog.types import (
-    FlagsResponse,
     FeatureFlag,
     FlagMetadata,
     FlagsAndPayloads,
+    FlagsResponse,
     FlagValue,
     normalize_flags_response,
     to_flags_and_payloads,
@@ -170,7 +179,7 @@ def system_context() -> dict[str, Any]:
 
 
 def is_token_in_rollout(
-    token: str, percentage: float = 0, included_hashes: set[str] = None, excluded_hashes: set[str] = None
+    token: str, percentage: float = 0, included_hashes: Optional[set[str]] = None, excluded_hashes: Optional[set[str]] = None
 ) -> bool:
     """
     Determines if a token should be included in a rollout based on:

posthog/test/test_client.py

@@ -1,14 +1,14 @@
+import hashlib
 import time
 import unittest
 from datetime import datetime
 from uuid import uuid4
-import hashlib
 
 import mock
 import six
 from parameterized import parameterized
 
-from posthog.client import Client, is_token_in_rollout, INCLUDED_HASHES, EXCLUDED_HASHES
+from posthog.client import EXCLUDED_HASHES, INCLUDED_HASHES, Client, is_token_in_rollout
 from posthog.request import APIError
 from posthog.test.test_utils import FAKE_TEST_API_KEY
 from posthog.types import FeatureFlag, LegacyFlagMetadata