Function apply_diff_to_registry

sdk/python/feast/diff/registry_diff.py:314–425

Applies the given diff to the given Feast project in the registry. Args: registry: The registry to be updated. registry_diff: The diff to apply. project: Feast project to be updated. commit: Whether the change should be persisted immediately

(
    registry: BaseRegistry,
    registry_diff: RegistryDiff,
    project: str,
    commit: bool = True,
    no_promote: bool = False,
)

def apply_diff_to_registry(
    registry: BaseRegistry,
    registry_diff: RegistryDiff,
    project: str,
    commit: bool = True,
    no_promote: bool = False,
):
    """
    Applies the given diff to the given Feast project in the registry.

    Args:
        registry: The registry to be updated.
        registry_diff: The diff to apply.
        project: Feast project to be updated.
        commit: Whether the change should be persisted immediately
        no_promote: If True, save new feature view version snapshots without
            promoting them to the active definition. New versions are accessible
            only via explicit @v<N> reads.
    """
    for feast_object_diff in registry_diff.feast_object_diffs:
        # There is no need to delete the object on an update, since applying the new object
        # will automatically delete the existing object.
        if feast_object_diff.transition_type == TransitionType.DELETE:
            if feast_object_diff.feast_object_type == FeastObjectType.ENTITY:
                entity_obj = cast(Entity, feast_object_diff.current_feast_object)
                registry.delete_entity(entity_obj.name, project, commit=False)
            elif feast_object_diff.feast_object_type == FeastObjectType.FEATURE_SERVICE:
                feature_service_obj = cast(
                    FeatureService, feast_object_diff.current_feast_object
                )
                registry.delete_feature_service(
                    feature_service_obj.name, project, commit=False
                )
            elif feast_object_diff.feast_object_type in [
                FeastObjectType.FEATURE_VIEW,
                FeastObjectType.ON_DEMAND_FEATURE_VIEW,
                FeastObjectType.STREAM_FEATURE_VIEW,
                FeastObjectType.LABEL_VIEW,
            ]:
                feature_view_obj = cast(
                    BaseFeatureView, feast_object_diff.current_feast_object
                )
                registry.delete_feature_view(
                    feature_view_obj.name,
                    project,
                    commit=False,
                )
            elif feast_object_diff.feast_object_type == FeastObjectType.DATA_SOURCE:
                ds_obj = cast(DataSource, feast_object_diff.current_feast_object)
                registry.delete_data_source(
                    ds_obj.name,
                    project,
                    commit=False,
                )
            elif feast_object_diff.feast_object_type == FeastObjectType.PERMISSION:
                permission_obj = cast(
                    Permission, feast_object_diff.current_feast_object
                )