Check if a is contained in b

check_if_contained.py

from typing import TypeVar, List, Callable

Ta = TypeVar('Ta')
Tb = TypeVar('Tb')


def check_if_contained(
        a: List[Ta],
        b: List[Tb],
        compare: Callable[[Ta, Tb], bool] = lambda x, y: x == y,
        get_b_key: Callable[[Tb], str] = lambda x: x
) -> bool:
    """
    Check if a is contained in b

    :param a:
    :param b:
    :param compare:
    :param get_b_key:
    :return:
    """

    alter_register: dict[str] = {}
    y = 0

    for i in range(len(a)):
        found = False
        # If a[i] is already in the register, skip it
        if a[i] in alter_register:
            continue

        # Search for a[i] in b
        for j in range(y, len(b)):
            b_key = get_b_key(b[j])
            alter_register[b_key] = b[j]
            if compare(a[i], b[j]):
                found = True
                y = j + 1
                continue

        if not found:
            return False
    return True


if __name__ == '__main__':
    _a = [1, 2, 7, 8, 9]
    _b = [{'id': 1}, {'id': 2}, {'id': 3}, {'id': 4}, {'id': 5}, {'id': 6}, {'id': 7}, {'id': 8}, {'id': 9}]
    print(check_if_contained(_a, _b, lambda x, y: x == y['id'], lambda x: x['id']))