nicer naming

Author: khvn26

flag_engine/context/types.py

@@ -27,19 +27,19 @@ class IdentityContext(TypedDict):
     traits: NotRequired[Dict[str, Optional[Union[str, float, bool]]]]
 
 
-class SegmentCondition1(TypedDict):
+class StrValueSegmentCondition(TypedDict):
     property: str
     operator: ConditionOperator
     value: str
 
 
-class SegmentCondition2(TypedDict):
+class InOperatorSegmentCondition(TypedDict):
     property: str
     operator: Literal["IN"]
     value: List[str]
 
 
-SegmentCondition = Union[SegmentCondition1, SegmentCondition2]
+SegmentCondition = Union[StrValueSegmentCondition, InOperatorSegmentCondition]
 
 
 class SegmentRule(TypedDict):

flag_engine/segments/evaluator.py

@@ -14,9 +14,10 @@
     EvaluationContext,
     FeatureContext,
     SegmentCondition,
+    SegmentContext,
+    SegmentRule,
+    StrValueSegmentCondition,
 )
-from flag_engine.context.types import SegmentCondition1 as StrValueSegmentCondition
-from flag_engine.context.types import SegmentContext, SegmentRule
 from flag_engine.environments.models import EnvironmentModel
 from flag_engine.identities.models import IdentityModel
 from flag_engine.identities.traits.types import ContextValue, is_trait_value

tests/unit/segments/test_segments_evaluator.py

@@ -10,9 +10,9 @@
     EvaluationContext,
     FeatureContext,
     SegmentCondition,
+    SegmentContext,
+    StrValueSegmentCondition,
 )
-from flag_engine.context.types import SegmentCondition1 as StrValueSegmentCondition
-from flag_engine.context.types import SegmentContext
 from flag_engine.environments.models import EnvironmentModel
 from flag_engine.features.models import FeatureModel, FeatureStateModel
 from flag_engine.identities.models import IdentityModel