|
9 | 9 | import logging
|
10 | 10 | from config import DEBUG # pylint: disable=import-error # This works fine?
|
11 | 11 |
|
12 |
| -def get_logger(name): |
13 |
| - """Function to get a logger |
14 |
| - Useful for modules that have already initialized a logger, such as discord.py |
15 |
| - """ |
16 |
| - __logger = logging.getLogger(name) |
17 |
| - __logger.setLevel(logging.DEBUG if DEBUG else logging.INFO) |
18 |
| - __ch = logging.StreamHandler() |
19 |
| - __ch.setFormatter(CustomFormatter()) |
20 |
| - __logger.addHandler(__ch) |
21 |
| - return __logger |
22 |
| - |
23 |
| -def init_logger(name="root"): |
24 |
| - """Function to create a designated logger for separate modules""" |
25 |
| - __logger = logging.Logger(name) |
26 |
| - __ch = logging.StreamHandler() |
27 |
| - __ch.setLevel(logging.DEBUG if DEBUG else logging.INFO) |
28 |
| - __ch.setFormatter(CustomFormatter()) |
29 |
| - __logger.addHandler(__ch) |
30 |
| - return __logger |
31 |
| - |
32 | 12 |
|
33 | 13 | class CustomFormatter(logging.Formatter):
|
34 | 14 | """Custom formatter class"""
|
@@ -59,9 +39,45 @@ class CustomFormatter(logging.Formatter):
|
59 | 39 | logging.CRITICAL: bold_red +
|
60 | 40 | "[%(asctime)s][%(levelname)7s] %(message)s" + reset
|
61 | 41 | }
|
62 |
| - # Documenting my dwindling sanity here |
63 | 42 |
|
64 | 43 | def format(self, record):
|
65 | 44 | log_fmt = self.FORMATS.get(record.levelno)
|
66 | 45 | formatter = logging.Formatter(log_fmt, datefmt="%I:%M.%S%p")
|
67 | 46 | return formatter.format(record)
|
| 47 | + |
| 48 | + |
| 49 | +def overwrite_ipy_loggers(): |
| 50 | + for k, v in logging.Logger.manager.loggerDict.items(): |
| 51 | + print(k, v) |
| 52 | + if k in [ |
| 53 | + "mixin", |
| 54 | + "dispatch", |
| 55 | + "http", |
| 56 | + "gateway", |
| 57 | + "client", |
| 58 | + "context" |
| 59 | + ]: |
| 60 | + for h in v.handlers: |
| 61 | + h.setFormatter(CustomFormatter) |
| 62 | + |
| 63 | + |
| 64 | +def get_logger(name): |
| 65 | + """Function to get a logger |
| 66 | + Useful for modules that have already initialized a logger, such as discord.py |
| 67 | + """ |
| 68 | + __logger = logging.getLogger(name) |
| 69 | + __logger.setLevel(logging.DEBUG if DEBUG else logging.INFO) |
| 70 | + __ch = logging.StreamHandler() |
| 71 | + __ch.setFormatter(CustomFormatter()) |
| 72 | + __logger.addHandler(__ch) |
| 73 | + return __logger |
| 74 | + |
| 75 | + |
| 76 | +def init_logger(name="root"): |
| 77 | + """Function to create a designated logger for separate modules""" |
| 78 | + __logger = logging.Logger(name) |
| 79 | + __ch = logging.StreamHandler() |
| 80 | + __ch.setLevel(logging.DEBUG if DEBUG else logging.INFO) |
| 81 | + __ch.setFormatter(CustomFormatter()) |
| 82 | + __logger.addHandler(__ch) |
| 83 | + return __logger |
0 commit comments