bohr/heuristics/identities/same_if_same_name_ends.py

from typing import Optional, Tuple

from bohrapi.artifacts.identity import Identity
from bohrapi.core import Heuristic
from bohrlabels.core import OneOrManyLabels
from bohrlabels.labels import MatchLabel


@Heuristic(Identity, Identity)
def same_if_same_name_ends(
    identities: Tuple[Identity, Identity]
) -> Optional[OneOrManyLabels]:
    """
    >>> not_same_if_only_first_names_match((Identity({"names": ["Hlib Mr Babii"]}), Identity({"names": ["Hlib Babii"]})))
    MatchLabel.Match
    >>> not_same_if_only_first_names_match((Identity({"names": ["Hlib Babii Mr"]}), Identity({"names": ["Hlib Babii"]}))) is None
    True
    >>> not_same_if_only_first_names_match((Identity({}), Identity({}))) is None
    True
    """
    name1 = identities[0].name
    name2 = identities[1].name
    if name1 is not None and name2 is not None:
        if len(spl1 := name1.split(" ")) != len(spl2 := name2.split(" ")):
            if spl1[0] == spl2[0] and spl1[-1] == spl2[-1]:
                return MatchLabel.Match