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

def get_tjsldwdm(sldwdm, sldwdm_1, sldwdm_2, qkzl):
    if sldwdm == sldwdm_1 or sldwdm == sldwdm_2:
        return sldwdm
    if sldwdm_1 != 'KGS' and sldwdm_2 != 'KGS':
        return sldwdm
    if qkzl is None or qkzl == '':
        return sldwdm
    return 'KGS'


def get_tjsldwdm_en(sldwdm, sldwdm_1, sldwdm_2, qkzl, sldw_en):
    if sldwdm == sldwdm_1 or sldwdm == sldwdm_2:
        return sldw_en
    if sldwdm_1 != 'KGS' and sldwdm_2 != 'KGS':
        return sldw_en
    if qkzl is None or qkzl == '':
        return sldw_en
    return 'KILOGRAM'


def get_tjsldwdm_cn(sldwdm, sldwdm_1, sldwdm_2, qkzl, sldw_cn):
    if sldwdm == sldwdm_1 or sldwdm == sldwdm_2:
        return sldw_cn
    if sldwdm_1 != 'KGS' and sldwdm_2 != 'KGS':
        return sldw_cn
    if qkzl is None or qkzl == '':
        return sldw_cn
    return '千克'


def get_tjsldwdm_sl(sldwdm, sldwdm_1, sldwdm_2, qkzl, sl):
    if sldwdm == sldwdm_1 or sldwdm == sldwdm_2:
        return sl
    if sldwdm_1 != 'KGS' and sldwdm_2 != 'KGS':
        return sl
    if qkzl is None or qkzl == '':
        return sl
    return qkzl


def get_tjsldwdm_sldj(sldwdm, sldwdm_1, sldwdm_2, qkzl, sldj, zldj):
    if sldwdm == sldwdm_1 or sldwdm == sldwdm_2:
        return sldj
    if sldwdm_1 != 'KGS' and sldwdm_2 != 'KGS':
        return sldj
    if qkzl is None or qkzl == '':
        return sldj
    return zldj


if __name__ == '__main__':
    print(get_tjsldwdm('XXX', 'KGS', None, ''))