From 2aff21b4ef20c4114ee02ba61d15a712c8bcd62f Mon Sep 17 00:00:00 2001
From: ooliver1 <oliverwilkes2006@icloud.com>
Date: Mon, 3 Oct 2022 19:36:56 +0100
Subject: [PATCH] feat: add console handlers

---
 botbase/botbase.py | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/botbase/botbase.py b/botbase/botbase.py
index 831d73d..cf75cf7 100644
--- a/botbase/botbase.py
+++ b/botbase/botbase.py
@@ -4,7 +4,7 @@
 from asyncio import sleep, wait_for
 from contextlib import suppress
 from importlib import import_module
-from logging import CRITICAL, INFO, Formatter, getLogger
+from logging import CRITICAL, INFO, Formatter, getLogger, StreamHandler
 from logging.handlers import RotatingFileHandler
 from pathlib import Path
 from random import choice
@@ -81,21 +81,23 @@
 """
 
 
-def get_handler():
+def get_handlers():
+    formatter = Formatter(
+        "%(levelname)-7s %(asctime)s %(filename)12s:%(funcName)-28s: %(message)s",
+        datefmt="%H:%M:%S %d/%m/%Y",
+    )
     h = RotatingFileHandler(
         "./logs/bot/io.log",
         maxBytes=1000000,
         backupCount=5,
         encoding="utf-8",
     )
-    h.setFormatter(
-        Formatter(
-            "%(levelname)-7s %(asctime)s %(filename)12s:%(funcName)-28s: %(message)s",
-            datefmt="%H:%M:%S %d/%m/%Y",
-        )
-    )
+    i = StreamHandler()
+
+    i.setFormatter(formatter)
+    h.setFormatter(formatter)
     h.namer = lambda name: name.replace(".log", "") + ".log"
-    return h
+    return h, i
 
 
 class BotBase(AutoShardedBot):
@@ -147,9 +149,10 @@ def __init__(self, *args, config_module: str = "config", **kwargs) -> None:
         log = getLogger()
         log.handlers = []
         log.setLevel(INFO)
-        h = get_handler()
+        h, i = get_handlers()
 
         log.addHandler(h)
+        log.addHandler(i)
         getLogger("asyncio").setLevel(CRITICAL)
 
         self.loop.set_exception_handler(self.asyncio_handler)