xyzshop/plot/refill/helper.py

import re
import logging

logger = logging.getLogger('plot')

def filter_chname(chnames):
    if chnames is not None:
        chnames = chnames.strip()

    if chnames is None or len(chnames) == 0:
        chnames = set()
    else:
        items = re.split(r',', chnames)
        names = [name.strip() for name in items]
        namelist = []
        for name in names:
            if len(name) > 0:
                namelist.append(name)
        chnames = set(namelist)
    return chnames

def filter_mchids(mchids):
    if mchids is not None:
        mchids = mchids.strip()

    if mchids is None or len(mchids) == 0:
        mchids = set()
    else:
        items = re.split(r',', mchids)
        names = [name.strip() for name in items]
        namelist = []
        for name in names:
            if len(name) > 0:
                namelist.append(int(name))
        mchids = set(namelist)
    return mchids

def filter_cardtype(card_types):
    if card_types is not None:
        card_types = card_types.strip()

    if card_types is None or len(card_types) == 0:
        result = {4, 5, 6} #默认话费类型
    else:
        items = re.split(r',', card_types)
        stypes = [type.strip() for type in items]
        types = []
        for stype in stypes:
            try:
                type = int(stype)
                types.append(type)
            except Exception as ex:
                logger.error(ex)

        result = set(types)
    return result