bazarr/libs/requests/hooks.py

34 lines
733 B
Python
Raw Normal View History

"""
requests.hooks
~~~~~~~~~~~~~~
This module provides the capabilities for the Requests hooks system.
Available hooks:
``response``:
The response generated from a Request.
"""
2022-11-07 18:06:49 +00:00
HOOKS = ["response"]
def default_hooks():
2018-10-31 16:08:29 +00:00
return {event: [] for event in HOOKS}
2022-11-07 18:06:49 +00:00
# TODO: response is the only one
def dispatch_hook(key, hooks, hook_data, **kwargs):
"""Dispatches a hook dictionary on a given piece of data."""
2018-10-31 16:08:29 +00:00
hooks = hooks or {}
hooks = hooks.get(key)
if hooks:
2022-11-07 18:06:49 +00:00
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