Merge pull request #2774 from ueg1990/pythonic_hooks

Some pythonic suggestions for requests/hooks.py
This commit is contained in:
2015-09-13 01:05:34 -04:00
+5 -13
View File
@@ -12,34 +12,26 @@ Available hooks:
The response generated from a Request.
"""
from collections import defaultdict
HOOKS = ['response']
def default_hooks():
hooks = {}
for event in HOOKS:
hooks[event] = []
return hooks
return dict((event, []) for event in HOOKS)
# TODO: response is the only one
def dispatch_hook(key, hooks, hook_data, **kwargs):
"""Dispatches a hook dictionary on a given piece of data."""
hooks = hooks or dict()
if key in hooks:
hooks = hooks.get(key)
hooks = hooks.get(key)
if hooks:
if hasattr(hooks, '__call__'):
hooks = [hooks]
for hook in hooks:
_hook_data = hook(hook_data, **kwargs)
if _hook_data is not None:
hook_data = _hook_data
return hook_data
return hook_data