Function diff_between

sdk/python/feast/diff/registry_diff.py:253–311

Returns the difference between the current and desired repo states. Args: registry: The registry storing the current repo state. current_project: The Feast project for which the diff is being computed. desired_repo_contents: The desired repo state.

(
    registry: BaseRegistry,
    current_project: str,
    desired_repo_contents: RepoContents,
)

def diff_between(
    registry: BaseRegistry,
    current_project: str,
    desired_repo_contents: RepoContents,
) -> RegistryDiff:
    """
    Returns the difference between the current and desired repo states.

    Args:
        registry: The registry storing the current repo state.
        current_project: The Feast project for which the diff is being computed.
        desired_repo_contents: The desired repo state.
    """
    diff = RegistryDiff()

    (
        objs_to_keep,
        objs_to_delete,
        objs_to_update,
        objs_to_add,
    ) = extract_objects_for_keep_delete_update_add(
        registry, current_project, desired_repo_contents
    )

    for object_type in FEAST_OBJECT_TYPES:
        objects_to_keep = objs_to_keep[object_type]
        objects_to_delete = objs_to_delete[object_type]
        objects_to_update = objs_to_update[object_type]
        objects_to_add = objs_to_add[object_type]

        for e in objects_to_add:
            diff.add_feast_object_diff(
                FeastObjectDiff(
                    name=e.name,
                    feast_object_type=object_type,
                    current_feast_object=None,
                    new_feast_object=e,
                    feast_object_property_diffs=[],
                    transition_type=TransitionType.CREATE,
                )
            )
        for e in objects_to_delete:
            diff.add_feast_object_diff(
                FeastObjectDiff(
                    name=e.name,
                    feast_object_type=object_type,
                    current_feast_object=e,
                    new_feast_object=None,
                    feast_object_property_diffs=[],
                    transition_type=TransitionType.DELETE,
                )
            )
        for e in objects_to_update:
            current_obj = [_e for _e in objects_to_keep if _e.name == e.name][0]
            diff.add_feast_object_diff(
                diff_registry_objects(current_obj, e, object_type)
            )