Fix typing

Author: cmanallen

mypy.ini

@@ -74,6 +74,8 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 [mypy-openai.*]
 ignore_missing_imports = True
+[mypy-openfeature.*]
+ignore_missing_imports = True
 [mypy-huggingface_hub.*]
 ignore_missing_imports = True
 [mypy-arq.*]

sentry_sdk/flag_utils.py

@@ -1,7 +1,13 @@
 from copy import copy
+from typing import TYPE_CHECKING
 
 import itertools
 
+if TYPE_CHECKING:
+    from typing import TypedDict
+
+    FlagData = TypedDict("FlagData", {"flag": str, "result": bool})
+
 
 class FlagBuffer:
 
@@ -13,9 +19,11 @@ def __init__(self, capacity):
 
     @property
     def index(self):
+        # type: () -> int
         return self.ip % self.capacity
 
     def clear(self):
+        # type: () -> None
         self.buffer = []
         self.ip = 0
 
@@ -27,7 +35,7 @@ def copy(self):
         return buffer
 
     def get(self):
-        # type: () -> list[dict]
+        # type: () -> list[FlagData]
         if self.ip >= self.capacity:
             iterator = itertools.chain(
                 range(self.index, self.capacity), range(0, self.index)
@@ -58,5 +66,5 @@ def __init__(self, flag, result):
 
     @property
     def asdict(self):
-        # type: () -> dict
+        # type: () -> FlagData
         return {"flag": self.flag, "result": self.result}

sentry_sdk/scope.py

@@ -689,7 +689,7 @@ def clear(self):
 
         # self._last_event_id is only applicable to isolation scopes
         self._last_event_id = None  # type: Optional[str]
-        self._flags = None
+        self._flags = None  # type: Optional[FlagBuffer]
 
     @_attr_setter
     def level(self, value):