Coverage for hookee/plugins/request_headers.py: 100.00%
16 statements
« prev ^ index » next coverage.py v7.4.1, created at 2024-02-02 23:09 +0000
« prev ^ index » next coverage.py v7.4.1, created at 2024-02-02 23:09 +0000
1from hookee.pluginmanager import REQUEST_PLUGIN
2from hookee.util import PrintUtil
4__author__ = "Alex Laird"
5__copyright__ = "Copyright 2023, Alex Laird"
6__version__ = "1.2.2"
8plugin_type = REQUEST_PLUGIN
9description = "Print the `request`'s headers, if defined."
11print_util = None # type: PrintUtil
14def setup(hookee_manager):
15 global print_util
17 print_util = hookee_manager.print_util
20def run(request):
21 if request.headers and "X-Forwarded-For" in request.headers:
22 print_util.print_basic("Client IP: {}".format(request.headers.get("X-Forwarded-For")),
23 color=print_util.request_color)
24 if request.headers:
25 print_util.print_dict("Headers", dict(request.headers), color=print_util.request_color)
27 return request