sima/lib/logger.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright (c) 2009, 2010, 2013 Jack Kaliko <efrim@azylum.org>
   4 #
   5 #  This file is part of sima
   6 #
   7 #  sima is free software: you can redistribute it and/or modify
   8 #  it under the terms of the GNU General Public License as published by
   9 #  the Free Software Foundation, either version 3 of the License, or
  10 #  (at your option) any later version.
  11 #
  12 #  sima is distributed in the hope that it will be useful,
  13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #  GNU General Public License for more details.
  16 #
  17 #  You should have received a copy of the GNU General Public License
  18 #  along with sima.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 #
  21
  22 """
  23 Logging facility for sima.
  24 """
  25
  26 import logging
  27 import sys
  28
  29 LOG_FORMATS = {
  30         logging.DEBUG: '{asctime} {filename}:{lineno}({funcName}) {levelname}: {message}',
  31         logging.INFO:  '{asctime} {levelname}: {message}'
  32         }
  33 DATE_FMT = "%Y-%m-%d %H:%M:%S"
  34
  35
  36 class LevelFilter(logging.Filter):# Logging facility
  37     """
  38     Enable logging between two log level by filtering everything < level.
  39     """
  40
  41     def __init__(self, filt_level):
  42         logging.Filter.__init__(self)
  43         self.level = filt_level
  44
  45     def filter(self, record):
  46         """Defines loglevel"""
  47         return record.levelno <= self.level
  48
  49
  50 def set_logger(level='info', logfile=None, name='sima'):
  51     """
  52     logger:
  53         level: in debug, info, warning,…
  54         file: provides to log to file
  55
  56     """
  57     user_log_level = getattr(logging, level.upper())
  58     if user_log_level > logging.DEBUG:
  59         log_format = LOG_FORMATS.get(logging.INFO)
  60     else:
  61         log_format = LOG_FORMATS.get(logging.DEBUG)
  62     logg = logging.getLogger(name)
  63     formatter = logging.Formatter(log_format, DATE_FMT, '{')
  64     logg.setLevel(user_log_level)
  65     if logfile:
  66         # create file handler
  67         fileh = logging.FileHandler(logfile)
  68         #fileh.setLevel(user_log_level)
  69         fileh.setFormatter(formatter)
  70         logg.addHandler(fileh)
  71     else:
  72         # create console handler with a specified log level (STDOUT)
  73         couth = logging.StreamHandler(sys.stdout)
  74         #couth.setLevel(user_log_level)
  75         couth.addFilter(LevelFilter(logging.WARNING))
  76
  77         # create console handler with warning log level (STDERR)
  78         cerrh = logging.StreamHandler(sys.stderr)
  79         #cerrh.setLevel(logging.WARNING)
  80         cerrh.setLevel(logging.ERROR)
  81
  82         # add formatter to the handlers
  83         cerrh.setFormatter(formatter)
  84         couth.setFormatter(formatter)
  85
  86         # add the handlers to SIMA_LOGGER
  87         logg.addHandler(couth)
  88         logg.addHandler(cerrh)
  89
  90 # VIM MODLINE
  91 # vim: ai ts=4 sw=4 sts=4 expandtab