fix(revert): Revert upgrade flagsmith-flag-engine to v10 (#6653) (#6674)

Co-authored-by: Kim Gustyr <kim.gustyr@flagsmith.com>

Author: gagantrivedi

api/app/pagination.py

@@ -3,11 +3,10 @@
 from collections import OrderedDict
 from typing import Any
 
+from flag_engine.identities.models import IdentityModel
 from rest_framework.pagination import PageNumberPagination
 from rest_framework.response import Response
 
-from util.engine_models.identities.models import IdentityModel
-
 
 class CustomPagination(PageNumberPagination):
     page_size = 999

api/e2etests/e2e_seed_data.py

@@ -10,6 +10,7 @@
     VIEW_PROJECT,
 )
 from django.conf import settings
+from flag_engine.identities.models import IdentityModel as EngineIdentity
 
 from edge_api.identities.models import EdgeIdentity
 from environments.identities.models import Identity
@@ -24,7 +25,6 @@
 from organisations.subscriptions.constants import ENTERPRISE
 from projects.models import Project, UserProjectPermission
 from users.models import FFAdminUser, UserPermissionGroup
-from util.engine_models.identities.models import IdentityModel as EngineIdentity
 
 # Password used by all the test users
 PASSWORD = "Str0ngp4ssw0rd!"

api/edge_api/identities/export.py

@@ -4,12 +4,12 @@
 from decimal import Decimal
 
 from django.utils import timezone
+from flag_engine.identities.traits.types import map_any_value_to_trait_value
 
 from edge_api.identities.models import EdgeIdentity
 from environments.identities.traits.models import Trait
 from features.models import Feature, FeatureState
 from features.multivariate.models import MultivariateFeatureOption
-from util.engine_models.identities.traits.types import map_any_value_to_trait_value
 
 EXPORT_EDGE_IDENTITY_PAGINATION_LIMIT = 20000
 

api/edge_api/identities/models.py

@@ -3,6 +3,8 @@
 from contextlib import suppress
 
 from django.db.models import Prefetch, Q
+from flag_engine.features.models import FeatureStateModel
+from flag_engine.identities.models import IdentityFeaturesList, IdentityModel
 
 from api_keys.user import APIKeyUser
 from edge_api.identities.tasks import (
@@ -18,8 +20,6 @@
 from features.multivariate.models import MultivariateFeatureStateValue
 from features.versioning.versioning_service import get_environment_flags_dict
 from users.models import FFAdminUser
-from util.engine_models.features.models import FeatureStateModel
-from util.engine_models.identities.models import IdentityFeaturesList, IdentityModel
 from util.mappers import map_engine_identity_to_identity_document
 
 

api/edge_api/identities/serializers.py

@@ -3,6 +3,16 @@
 
 from django.utils import timezone
 from drf_spectacular.utils import extend_schema_field
+from flag_engine.features.models import FeatureModel as EngineFeatureModel
+from flag_engine.features.models import FeatureStateModel as EngineFeatureStateModel
+from flag_engine.features.models import (
+    MultivariateFeatureOptionModel as EngineMultivariateFeatureOptionModel,
+)
+from flag_engine.features.models import (
+    MultivariateFeatureStateValueModel as EngineMultivariateFeatureStateValueModel,
+)
+from flag_engine.identities.models import IdentityModel as EngineIdentity
+from flag_engine.utils.exceptions import DuplicateFeatureState
 from pydantic import ValidationError as PydanticValidationError
 from pyngo import drf_error_details
 from rest_framework import serializers
@@ -15,18 +25,6 @@
 from features.serializers import (  # type: ignore[attr-defined]
     FeatureStateValueSerializer,
 )
-from util.engine_models.features.models import FeatureModel as EngineFeatureModel
-from util.engine_models.features.models import (
-    FeatureStateModel as EngineFeatureStateModel,
-)
-from util.engine_models.features.models import (
-    MultivariateFeatureOptionModel as EngineMultivariateFeatureOptionModel,
-)
-from util.engine_models.features.models import (
-    MultivariateFeatureStateValueModel as EngineMultivariateFeatureStateValueModel,
-)
-from util.engine_models.identities.models import IdentityModel as EngineIdentity
-from util.engine_models.utils.exceptions import DuplicateFeatureState
 from util.mappers import (
     map_engine_identity_to_identity_document,
     map_feature_to_engine,

api/edge_api/identities/utils.py

@@ -1,6 +1,6 @@
 import typing
 
-from util.engine_models.features.models import FeatureStateModel
+from flag_engine.features.models import FeatureStateModel
 
 if typing.TYPE_CHECKING:
     from edge_api.identities.types import ChangeType, FeatureStateChangeDetails

api/edge_api/identities/views.py

@@ -9,6 +9,8 @@
 )
 from django.shortcuts import get_object_or_404
 from drf_spectacular.utils import extend_schema
+from flag_engine.identities.models import IdentityFeaturesList, IdentityModel
+from flag_engine.identities.traits.models import TraitModel
 from pyngo import drf_error_details
 from rest_framework import status, viewsets
 from rest_framework.decorators import action, api_view, permission_classes
@@ -54,8 +56,6 @@
 from features.models import FeatureState
 from features.permissions import IdentityFeatureStatePermissions
 from projects.exceptions import DynamoNotEnabledError
-from util.engine_models.identities.models import IdentityFeaturesList, IdentityModel
-from util.engine_models.identities.traits.models import TraitModel
 
 from . import edge_identity_service
 from .exceptions import TraitPersistenceError

api/environments/dynamodb/services.py

@@ -2,6 +2,8 @@
 from decimal import Decimal
 from typing import Generator, Iterable
 
+from flag_engine.identities.models import IdentityModel
+
 from environments.dynamodb import (
     CapacityBudgetExceeded,
     DynamoEnvironmentV2Wrapper,
@@ -14,7 +16,6 @@
 )
 from environments.models import Environment
 from projects.models import EdgeV2MigrationStatus
-from util.engine_models.identities.models import IdentityModel
 from util.mappers import map_engine_feature_state_to_identity_override
 
 logger = logging.getLogger(__name__)

api/environments/dynamodb/types.py

@@ -5,10 +5,9 @@
 
 import boto3
 from django.conf import settings
+from flag_engine.features.models import FeatureStateModel
 from pydantic import BaseModel
 
-from util.engine_models.features.models import FeatureStateModel
-
 if typing.TYPE_CHECKING:
     from projects.models import EdgeV2MigrationStatus
 

api/environments/dynamodb/wrappers/identity_wrapper.py

@@ -7,17 +7,15 @@
 from boto3.dynamodb.conditions import Attr, Key
 from django.conf import settings
 from django.core.exceptions import ObjectDoesNotExist
+from flag_engine.context.mappers import map_environment_identity_to_context
+from flag_engine.environments.models import EnvironmentModel
+from flag_engine.identities.models import IdentityModel
+from flag_engine.segments.evaluator import get_context_segments
 from rest_framework.exceptions import NotFound
 
 from edge_api.identities.search import EdgeIdentitySearchData
 from environments.dynamodb.constants import IDENTITIES_PAGINATION_LIMIT
 from environments.dynamodb.wrappers.exceptions import CapacityBudgetExceeded
-from util.engine_models.context.mappers import (
-    get_context_segments,
-    map_environment_identity_to_context,
-)
-from util.engine_models.environments.models import EnvironmentModel
-from util.engine_models.identities.models import IdentityModel
 from util.mappers import map_identity_to_identity_document
 
 from .base import BaseDynamoWrapper