Some pythonic suggestions for requests/hooks.py

This commit is contained in:
ueg1990
2015-09-12 20:43:25 -04:00
parent 7c74050d66
commit c5a41269d9
+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