Merge pull request #7 from Robso-creator/issue-6

feat(issue-6): add logs to repo

requirements.txt

@@ -1,3 +1,4 @@
+colorlog==6.8.2
 discord==2.3.2
 loadotenv==1.0.1
 pre-commit==3.6.0

src/logger.py

@@ -0,0 +1,32 @@
+import logging
+
+import colorlog
+
+
+def setup_logger(logger_name, level=logging.INFO):
+    date_colored = '%(bold_black)s%(asctime)s'
+    level_colored = '%(log_color)s%(levelname)7s%(reset)s'
+    file_path_colored = '%(purple)s%(filename)s:%(name)s:%(funcName)s%(reset)s'
+    formatting = f'{date_colored}   {level_colored}   {file_path_colored} | %(message)s'
+
+    color_formatter = colorlog.ColoredFormatter(
+        formatting,
+        datefmt='%Y-%m-%d %H:%M:%S',
+        reset=True,
+        log_colors={
+            'DEBUG': 'cyan',
+            'INFO': 'green',
+            'WARNING': 'yellow',
+            'ERROR': 'red',
+            'CRITICAL': 'bold_red',
+        },
+    )
+
+    handler = colorlog.StreamHandler()
+    handler.setFormatter(color_formatter)
+
+    logger = logging.getLogger(logger_name)
+    logger.addHandler(handler)
+    logger.setLevel(level)
+
+    return logger

src/main.py

@@ -2,26 +2,31 @@
 
 import discord
 
+from src.logger import setup_logger
+
 intents = discord.Intents.default()
 intents.typing = False
 intents.presences = False
 intents.members = True
 intents.message_content = True
 
+_log = setup_logger('src.main')
+
 
 class MyClient(discord.Client):
     async def on_ready(self):
-        print(f'Logged on as {self.user}!')
+        _log.info(f'Logged on as {self.user}!')
 
     async def on_message(self, message):
-        print('message from {0.author}: {0.content}'.format(message))
-        if message.content == ('!roleta'):
+        _log.info('message from {0.author}: {0.content}'.format(message))
+        if message.content == '!roleta':
             await message.channel.send(
                 f'O sorteado foi: {self.users.copy()[random.randint(0, len(message.guild.members) - 1)]}',
             )
 
 
 if __name__ == '__main__':
-    import settings
+    from src import settings
+
     client = MyClient(intents=intents)
     client.run(settings.DISCORD_TOKEN)