Coverage for hookee/plugins/request_headers.py: 100.00%

16 statements  

« prev     ^ index     » next       coverage.py v7.2.3, created at 2023-04-23 03:26 +0000

1from hookee.pluginmanager import REQUEST_PLUGIN 

2from hookee.util import PrintUtil 

3 

4__author__ = "Alex Laird" 

5__copyright__ = "Copyright 2023, Alex Laird" 

6__version__ = "1.2.2" 

7 

8plugin_type = REQUEST_PLUGIN 

9description = "Print the `request`'s headers, if defined." 

10 

11print_util = None # type: PrintUtil 

12 

13 

14def setup(hookee_manager): 

15 global print_util 

16 

17 print_util = hookee_manager.print_util 

18 

19 

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) 

26 

27 return request