__init__.py 文件源码

def feed_response(sources, modifier=None):
    def decorator(func):
        @wraps(func)
        @responses.activate
        def func_wrapper(obj, *args, **kwargs):
            source_list = sources
            feeds = []
            if type(sources) is not list:
                source_list = [sources]

            for source in source_list:
                if source == 'twitter':
                    feeds.append(_twitter(modifier))
                elif source == 'instagram':
                    feeds.append(_instagram(modifier))
                elif source == 'facebook':
                    feeds.append(_facebook(modifier))
            feeds.extend(args)
            return func(obj, *feeds, **kwargs)
        return func_wrapper
    return decorator