Function map_identity_to_engine

api/util/mappers/engine.py:388–435

(
    identity: "Identity",
    *,
    with_overrides: bool = True,
    with_traits: bool = True,
)

def map_identity_to_engine(
    identity: "Identity",
    *,
    with_overrides: bool = True,
    with_traits: bool = True,
) -> IdentityModel:
    environment_api_key = identity.environment.api_key

    # Read relationships - grab all the data needed from the ORM here.
    if with_overrides:
        identity_feature_states: List["FeatureState"] = _get_prioritised_feature_states(
            identity.identity_features.all(),
        )
        multivariate_feature_state_values_by_feature_state_id = {
            feature_state.pk: feature_state.multivariate_feature_state_values.all()
            for feature_state in identity_feature_states
        }
    else:
        identity_feature_states = []
        multivariate_feature_state_values_by_feature_state_id = {}

    identity_traits: Iterable["Trait"] = (
        identity.identity_traits.all() if with_traits else []
    )

    # Prepare relationships.
    identity_feature_state_models = [
        map_feature_state_to_engine(
            feature_state,
            mv_fs_values=multivariate_feature_state_values_by_feature_state_id.pop(
                feature_state.pk,
            ),
        )
        for feature_state in identity_feature_states
    ]
    identity_trait_models = map_traits_to_engine(identity_traits)

    return IdentityModel(
        # Attributes:
        identifier=identity.identifier,
        environment_api_key=environment_api_key,
        created_date=identity.created_date,
        django_id=identity.pk,
        #
        # Relationships:
        identity_features=identity_feature_state_models,  # type: ignore[arg-type]
        identity_traits=identity_trait_models,
    )


_rule_type_adapter: TypeAdapter[RuleType] = TypeAdapter(RuleType)