Total Complexity | 7 |
Total Lines | 27 |
Duplicated Lines | 0 % |
Coverage | 18.75% |
1 | 1 | from logging import Filter |
|
6 | 1 | class RequestsLogFilter(Filter): |
|
7 | 1 | def filter(self, record): |
|
8 | if self.is_dropped_connection(record): |
||
9 | # Change record level to debug |
||
10 | record.levelno = logging.DEBUG |
||
11 | record.levelname = 'DEBUG' |
||
12 | |||
13 | # Retrieve logger for record |
||
14 | logger = logging.getLogger(record.name) |
||
15 | |||
16 | # Check if the logger has debug logging enabled |
||
17 | return logger.isEnabledFor(logging.DEBUG) |
||
18 | |||
19 | return True |
||
20 | |||
21 | 1 | @staticmethod |
|
22 | def is_dropped_connection(record): |
||
23 | if record.levelno != logging.INFO: |
||
24 | return False |
||
25 | |||
26 | if record.name != 'requests.packages.urllib3.connectionpool': |
||
27 | return False |
||
28 | |||
29 | if record.msg and not record.msg.startswith('Resetting dropped connection:'): |
||
30 | return False |
||
31 | |||
32 | return True |
||
33 | |||
56 |