fix: TypeError: 'NoneType' object is not subscriptable (Python 3.11) (#170)

Author: khvn26

flagsmith/api/types.py

@@ -49,7 +49,7 @@ class MultivariateFeatureStateValueModel(typing.TypedDict):
 
 class FeatureStateModel(typing.TypedDict):
     enabled: bool
-    feature_segment: NotRequired[FeatureSegmentModel]
+    feature_segment: NotRequired[typing.Optional[FeatureSegmentModel]]
     feature_state_value: object
     feature: FeatureModel
     featurestate_uuid: str

flagsmith/mappers.py

@@ -266,7 +266,7 @@ def _map_environment_document_feature_states_to_feature_contexts(
                 for multivariate_feature_state_value in multivariate_feature_state_values
             ]
 
-        if "feature_segment" in feature_state:
-            feature_context["priority"] = feature_state["feature_segment"]["priority"]
+        if feature_segment := feature_state.get("feature_segment"):
+            feature_context["priority"] = feature_segment["priority"]
 
         yield feature_context

tests/data/environment.json

@@ -51,7 +51,8 @@
                 "id": 1
             },
             "segment_id": null,
-            "enabled": true
+            "enabled": true,
+            "feature_segment": null
         }
     ],
     "updated_at": "2023-07-14T16:12:00.000000Z",