如何将参数传递给python grequests中的钩子

Question

根据Requests文档,可以将事件挂钩添加到.get()函数中.

requests.get('http://httpbin.org', hooks=dict(response=print_url))
def print_url(r, *args, **kwargs):
    print(r.url)

这很好但是如何使用自定义参数设置*args,例如,我想将一些自定义值传递给print_url(),如何在*args中设置它们？这样的事情失败了:

args = ("search_item", search_item)
rs = (grequests.get(u, hooks={'response': [parse_books],'args': [args]}) for u in urls)

Answer 1

您不能指定要传递给响应挂钩的额外参数.如果你想要指定额外的信息,你应该创建一个你调用的函数,然后返回一个函数作为钩子传递,例如,

def hook_factory(*factory_args, **factory_kwargs):
    def response_hook(response, *request_args, **request_kwargs):
        # use factory_kwargs
        # etc.
        return None # or the modified response
    return response_hook

grequests.get(u, hooks={'response': [hook_factory(search_item=search_item)]})