98 lines
3.0 KiB
Python
98 lines
3.0 KiB
Python
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
|
|
class bcolors:
|
|
"""
|
|
Define constant color value for different level
|
|
"""
|
|
DEBUG = '\033[94m'
|
|
INFO = '\033[95m'
|
|
SUCCESS = '\033[92m'
|
|
WARNING = '\033[93m'
|
|
FAIL = '\033[91m'
|
|
ENDC = '\033[0m'
|
|
|
|
class lvl:
|
|
"""
|
|
Define constant value for level log
|
|
"""
|
|
NOTSET = 0
|
|
DEBUG = 10
|
|
INFO = 20
|
|
SUCCESS = 25
|
|
WARNING = 30
|
|
FAIL = 40
|
|
|
|
class SingleLevelFilter(logging.Filter):
|
|
"""Filter for one level"""
|
|
def __init__(self, passlevel, reject):
|
|
"""
|
|
Constructor
|
|
passlevel : level to filter
|
|
reject : true on reject state
|
|
"""
|
|
self.passlevel = passlevel
|
|
self.reject = reject
|
|
def filter(self, record):
|
|
if self.reject:
|
|
return (record.levelno != self.passlevel)
|
|
else:
|
|
return (record.levelno == self.passlevel)
|
|
|
|
class Log(object):
|
|
"""
|
|
Log Manager
|
|
"""
|
|
def __init__(self) :
|
|
"""
|
|
Define 3 differents log :
|
|
activity.log -> all activity server
|
|
warning.log -> only warning server (including illegal acess)
|
|
error.log' -> error server
|
|
Write all message on terminal too
|
|
"""
|
|
logging.addLevelName(lvl.SUCCESS, "SUCCESS")
|
|
|
|
self.logger = logging.getLogger()
|
|
self.logger.setLevel(logging.DEBUG)
|
|
formatter = logging.Formatter('%(asctime)-15s :: %(levelname)s :: %(message)s')
|
|
|
|
file_handler = RotatingFileHandler('log/activity.log', 'a', 1000000, 1)
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler)
|
|
|
|
file_handler_warning = RotatingFileHandler('log/warning.log', 'a', 1000000, 1)
|
|
f1 = SingleLevelFilter(logging.WARNING, False)
|
|
file_handler_warning.addFilter(f1)
|
|
file_handler_warning.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler_warning)
|
|
|
|
file_handler_error = RotatingFileHandler('log/error.log', 'a', 1000000, 1)
|
|
file_handler_error.setLevel(logging.ERROR)
|
|
file_handler_error.setFormatter(formatter)
|
|
self.logger.addHandler(file_handler_error)
|
|
|
|
steam_handler = logging.StreamHandler()
|
|
steam_handler.setLevel(logging.NOTSET)
|
|
self.logger.addHandler(steam_handler)
|
|
|
|
|
|
def printL(self,pMsg,pLvl):
|
|
"""
|
|
Add color and write in log with an define level
|
|
pMsg : message to write in log
|
|
pLvl : level of log message
|
|
"""
|
|
if pLvl == lvl.DEBUG :
|
|
pMsg = bcolors.DEBUG + str(pMsg) + bcolors.ENDC
|
|
elif pLvl == lvl.INFO :
|
|
pMsg = bcolors.INFO + str(pMsg) + bcolors.ENDC
|
|
elif pLvl == lvl.SUCCESS :
|
|
pMsg = bcolors.SUCCESS + str(pMsg) + bcolors.ENDC
|
|
elif pLvl == lvl.WARNING :
|
|
pMsg = bcolors.WARNING + str(pMsg) + bcolors.ENDC
|
|
elif pLvl == lvl.FAIL :
|
|
pMsg = bcolors.FAIL + str(pMsg) + bcolors.ENDC
|
|
self.logger.log(pLvl,pMsg)
|