utils.py 文件源码

def enrich_wrapper(func):
    """
    item_loader???pickle ?????????response???selector??, ???????
    ???enrich??????selector????????selector
    :param func:
    :return:
    """

    @wraps(func)
    def wrapper(*args, **kwargs):
        item_loader = args[1]
        response = args[2]
        selector = Selector(text=response.text)
        item_loader.selector = selector
        result = func(*args, **kwargs)
        item_loader.selector = None

        return result

    return wrapper