Skip to content

Commit 2aff21b

Browse files
committed
feat: add console handlers
1 parent 515edf7 commit 2aff21b

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

botbase/botbase.py

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from asyncio import sleep, wait_for
55
from contextlib import suppress
66
from importlib import import_module
7-
from logging import CRITICAL, INFO, Formatter, getLogger
7+
from logging import CRITICAL, INFO, Formatter, getLogger, StreamHandler
88
from logging.handlers import RotatingFileHandler
99
from pathlib import Path
1010
from random import choice
@@ -81,21 +81,23 @@
8181
"""
8282

8383

84-
def get_handler():
84+
def get_handlers():
85+
formatter = Formatter(
86+
"%(levelname)-7s %(asctime)s %(filename)12s:%(funcName)-28s: %(message)s",
87+
datefmt="%H:%M:%S %d/%m/%Y",
88+
)
8589
h = RotatingFileHandler(
8690
"./logs/bot/io.log",
8791
maxBytes=1000000,
8892
backupCount=5,
8993
encoding="utf-8",
9094
)
91-
h.setFormatter(
92-
Formatter(
93-
"%(levelname)-7s %(asctime)s %(filename)12s:%(funcName)-28s: %(message)s",
94-
datefmt="%H:%M:%S %d/%m/%Y",
95-
)
96-
)
95+
i = StreamHandler()
96+
97+
i.setFormatter(formatter)
98+
h.setFormatter(formatter)
9799
h.namer = lambda name: name.replace(".log", "") + ".log"
98-
return h
100+
return h, i
99101

100102

101103
class BotBase(AutoShardedBot):
@@ -147,9 +149,10 @@ def __init__(self, *args, config_module: str = "config", **kwargs) -> None:
147149
log = getLogger()
148150
log.handlers = []
149151
log.setLevel(INFO)
150-
h = get_handler()
152+
h, i = get_handlers()
151153

152154
log.addHandler(h)
155+
log.addHandler(i)
153156
getLogger("asyncio").setLevel(CRITICAL)
154157

155158
self.loop.set_exception_handler(self.asyncio_handler)

0 commit comments

Comments
 (0)