Libreria de logger

Author: GabiAndi

bin/pycamp_manager.py

@@ -1,10 +1,10 @@
 import argparse
 import datetime
-import logging
 from pycamp_bot.models import Pycamp
 from pycamp_bot.models import Pycampista
 from pycamp_bot.models import PycampistaAtPycamp
 from pycamp_bot.models import models_db_connection
+from pycamp_bot.logger import logger
 
 
 def parse_args():
@@ -59,20 +59,20 @@ def parse_args():
 
     if args.wich == 'pycampista':
         if args.add:
-            logging.info('Adding pycampista')
+            logger.info('Adding pycampista')
             pycampista = Pycampista.create(username=args.name)
 
         if args.arrive:
             arrive_time = datetime.datetime.fromisoformat(args.arrive)
-            logging.info('Changing arrive time to {}'.format(arrive_time))
+            logger.info('Changing arrive time to {}'.format(arrive_time))
             pycampista = Pycampista.select().where(
                 Pycampista.username == args.name)[0]
             pycampista.arrive = arrive_time
             pycampista.save()
 
         if args.departure:
             departure_time = datetime.datetime.fromisoformat(args.departure)
-            logging.info('Changing departure time to {}'.format(departure_time))
+            logger.info('Changing departure time to {}'.format(departure_time))
             pycampista = Pycampista.select().where(
                 Pycampista.username == args.name)[0]
             pycampista.leave = departure_time
@@ -83,20 +83,20 @@ def parse_args():
         print(pycampista)
     elif args.wich == 'pycamp':
         if args.add:
-            logging.info('Adding pycamp')
+            logger.info('Adding pycamp')
             pycamp = Pycamp.create(headquarters=args.name)
 
         if args.init:
             init_time = datetime.datetime.fromisoformat(args.init)
-            logging.info('Changing init time to {}'.format(init_time))
+            logger.info('Changing init time to {}'.format(init_time))
             pycamp = Pycamp.select().where(
                 Pycamp.headquarters == args.name)[0]
             pycamp.init = init_time
             pycamp.save()
 
         if args.end:
             end_time = datetime.datetime.fromisoformat(args.end)
-            logging.info('Changing end time to {}'.format(end_time))
+            logger.info('Changing end time to {}'.format(end_time))
             pycamp = Pycamp.select().where(
                 Pycamp.headquarters == args.name)[0]
             pycamp.end = end_time

bin/run_bot.py

@@ -1,4 +1,3 @@
-import logging
 import os
 from telegram.ext import Application
 from pycamp_bot.commands import auth
@@ -11,12 +10,7 @@
 from pycamp_bot.commands import schedule
 from pycamp_bot.commands import announcements
 from pycamp_bot.models import models_db_connection
-
-
-logging.basicConfig(
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    level=logging.INFO)
-logger = logging.getLogger(__name__)
+from pycamp_bot.logger import logger
 
 
 def set_handlers(application):

src/pycamp_bot/commands/announcements.py

@@ -1,12 +1,8 @@
-import logging
 from telegram.ext import CommandHandler
 from pycamp_bot.models import Project, Pycampista, Vote
 from pycamp_bot.commands.auth import get_admins_username
 
 
-logger = logging.getLogger(__name__)
-
-
 async def announce(update, context):
     username = update.message.from_user.username
     admins = get_admins_username()

src/pycamp_bot/commands/auth.py

@@ -1,10 +1,7 @@
-import logging
 import os
 from telegram.ext import CommandHandler
 from pycamp_bot.models import Pycampista
-
-
-logger = logging.getLogger(__name__)
+from pycamp_bot.logger import logger
 
 
 def get_admins_username():

src/pycamp_bot/commands/base.py

@@ -1,9 +1,6 @@
-import logging
 from telegram.ext import CommandHandler
 from pycamp_bot.commands.help_msg import get_help
-
-
-logger = logging.getLogger(__name__)
+from pycamp_bot.logger import logger
 
 
 async def msg_to_active_pycamp_chat(bot, text):

src/pycamp_bot/commands/manage_pycamp.py

@@ -1,12 +1,10 @@
 import datetime
-import logging
 from telegram.ext import CommandHandler
 from pycamp_bot.models import Pycamp
 from pycamp_bot.models import Pycampista
 from pycamp_bot.models import PycampistaAtPycamp
 from pycamp_bot.commands.auth import admin_needed
-
-logger = logging.getLogger(__name__)
+from pycamp_bot.logger import logger
 
 
 def get_pycamp_by_name(name):

src/pycamp_bot/commands/projects.py

@@ -1,16 +1,14 @@
-import logging
 from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters
 from pycamp_bot.models import Pycampista, Project, Vote
 from pycamp_bot.commands.base import msg_to_active_pycamp_chat
 from pycamp_bot.commands.manage_pycamp import active_needed, get_active_pycamp
 from pycamp_bot.commands.auth import admin_needed
+from pycamp_bot.logger import logger
 
 
 current_projects = {}
 NOMBRE, DIFICULTAD, TOPIC = ["nombre", "dificultad", "topic"]
 
-logger = logging.getLogger(__name__)
-
 
 def load_authorized(f):
     async def wrap(*args, **kargs):

src/pycamp_bot/commands/schedule.py

@@ -1,4 +1,3 @@
-import logging
 import string
 from telegram.ext import ConversationHandler, CommandHandler, MessageHandler, filters
 from pycamp_bot.models import Project, Slot, Pycampista, Vote
@@ -9,8 +8,6 @@
 
 DAY_LETTERS = []
 
-logger = logging.getLogger(__name__)
-
 
 def _dictToString(dicto):
     if dicto:

src/pycamp_bot/commands/voting.py

@@ -1,14 +1,11 @@
-import logging
 import peewee
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup
 from telegram.ext import CommandHandler, CallbackQueryHandler
 from pycamp_bot.commands.base import msg_to_active_pycamp_chat
 from pycamp_bot.commands.auth import admin_needed
 from pycamp_bot.commands.manage_pycamp import active_needed, get_active_pycamp
 from pycamp_bot.models import Pycampista, Project, Vote
-
-
-logger = logging.getLogger(__name__)
+from pycamp_bot.logger import logger
 
 
 def vote_authorized(f):

src/pycamp_bot/logger.py

@@ -0,0 +1,41 @@
+"""Módulo para hacer loggeos."""
+
+import logging
+
+
+LOG_FORMAT = '[%(asctime)s] - [%(filename)s %(lineno)d] %(levelname)s:'
+LOG_DATE_FORMAT = '%Y:%m:%d %H:%M:%S'
+
+
+class LogFormatter(logging.Formatter):
+    """Formateador del log."""
+    grey = "\x1b[38;20m"
+    blue = "\x1b[34;20m"
+    yellow = "\x1b[33;20m"
+    red = "\x1b[31;20m"
+    bold_red = "\x1b[31;1m"
+    reset = "\x1b[0m"
+
+    FORMATS = {
+        logging.DEBUG: LOG_FORMAT + ' ' + grey + '%(message)s' + reset,
+        logging.INFO: LOG_FORMAT + ' ' + blue + '%(message)s' + reset,
+        logging.WARNING: LOG_FORMAT + ' ' + yellow + '%(message)s' + reset,
+        logging.ERROR: LOG_FORMAT + ' ' + red + '%(message)s' + reset,
+        logging.CRITICAL: LOG_FORMAT + ' ' + bold_red + '%(message)s' + reset,
+    }
+
+    def format(self, record):
+        log_formatter = logging.Formatter(
+            fmt=self.FORMATS.get(record.levelno),
+            datefmt=LOG_DATE_FORMAT
+        )
+        return log_formatter.format(record)
+
+
+formatter = LogFormatter()
+ch = logging.StreamHandler()
+ch.setFormatter(formatter)
+logger = logging.getLogger(__name__)
+logger.addHandler(ch)
+logger.setLevel(logging.DEBUG)
+logger.propagate = False