poyee-data-warehouse/dw_base/spark/udf/customs/indonesia_qymc_judge.py

prefix_list = [
    'PT.',
    'PT ',
    'CV.',
    'CV ']


def is_prefix_and_concat(qymc, qymc_org):
    if qymc is None or qymc_org is None:
        return qymc_org
    for prefix in prefix_list:
        if qymc.startswith(prefix) and not qymc_org.startswith(prefix.replace('.', ' ')):
            return prefix.replace('.', ' ') + qymc_org
    return qymc_org


def get_qymc_prefix(qymcs: list):
    if qymcs is None or len(qymcs) == 0:
        return None
    for prefix in prefix_list:
        for qymc in qymcs:
            if qymc.startswith(prefix):
                return prefix.replace('.', ' ')
    return None