import os
import random
import sys
import multiprocessing
import time
from shutil import copyfile
from colorlog import ColoredFormatter
log directory
LOG_DIR = “./logs”
level fot stdout
STD_LEVEL = logging.WARN
level for file
FILE_LEVEL = logging.DEBUG
CRITICAL = 50
FATAL = CRITICAL
ERROR = 40
WARNING = 30
WARN = WARNING
INFO = 20
DEBUG = 10
NOTSET = 0
class Log:
def init(self, log_dir=LOG_DIR):
“”"
function:init create log file, judge if log file is too bigger
:param log_dir: directory of logs
“”"
log_path = log_dir + “/log”
if not os.path.exists(log_dir):
os.mkdir(log_dir)
os.chmod(log_dir, 0o777)
if not os.path.exists(log_path):
f = open(log_path, mode=‘w’, encoding=‘utf-8’)
f.close()
os.chmod(log_path, 0o777)
# if log file is more than 1MB, copy to a file and clear log file
if os.path.getsize(log_path) / 1048576 > 1:
# print(os.path.getsize(log_path))
copyfile(log_path, log_dir + “/log” + str(time.time()).replace(“.”, “”))
with open(log_path, ‘w’) as f:
f.truncate()
f.close()
self.logger_format = logging.Formatter(‘%(asctime)s - %(levelname)s: %(message)s’)
self.c_logger_format = ColoredFormatter(fmt=‘%(log_color)s%(asctime)s’
’ - %(log_color)s%(levelname)s: %(log_color)s%(message)s’,
reset=True,
secondary_log_colors={},
style=‘%’
)
self.logger = logging.getLogger(str(random.random()))
self.logger.handlers.clear()
self.logger.setLevel(logging.DEBUG)
self.filehandler = logging.FileHandler(log_path, mode='a')
self.filehandler.setLevel(FILE_LEVEL)
self.filehandler.setFormatter(self.logger_format)
self.stdouthandler = logging.StreamHandler(sys.stdout)
self.stdouthandler.setLevel(STD_LEVEL)
self.stdouthandler.setFormatter(self.c_logger_format)
self.logger.addHandler(self.stdouthandler)
self.logger.addHandler(self.filehandler)
self.__lock = multiprocessing.Lock()
def __log(self, msg=None, level=logging.INFO):
if level == logging.DEBUG:
self.logger.debug(msg)
elif level == logging.INFO:
self.logger.info(msg)
elif level == logging.WARNING:
self.logger.warning(msg)
elif level == logging.ERROR:
self.logger.error(msg)
elif level == logging.CRITICAL:
self.logger.critical(msg)
def log_show_store(self, msg, level):
"""
function: show logs on screen and store logs into log file
:param msg:message of log
:param level:level of log
return: None
"""
self.__lock.acquire()
self.logger.addHandler(self.stdouthandler)
self.logger.addHandler(self.filehandler)
self.__log(msg=msg, level=level)
self.__lock.release()
def log_show(self, msg, level):
"""
function:show msg on the screen
:param msg:message of log
:param level: level of log
"""
self.__lock.acquire()
self.logger.removeHandler(self.filehandler)
self.logger.addHandler(self.stdouthandler)
self.__log(msg, level=level)
self.logger.removeHandler(self.stdouthandler)
self.logger.addHandler(self.filehandler)
self.__lock.release()
def log_store(self, msg, level):
"""
function:store msg into log file
:param msg:message of log
:param level: level of log
"""
self.__lock.acquire()
self.logger.removeHandler(self.stdouthandler)
self.logger.addHandler(self.filehandler)
self.__log(msg, level=level)
self.logger.removeHandler(self.filehandler)
self.logger.addHandler(self.stdouthandler)
self.__lock.release()
def log_wrapper(self, *args):
"""
function: record arg and result of functions
:param args:
:return: wrapper
"""
self.log_store(' '.join(args), FILE_LEVEL)
def wrapper(func):
def inner(*args, **kwargs):
self.log_store(getattr(func, "__name__") + " call " + str(args) + str(kwargs), 10)
ret = func(*args, **kwargs)
self.log_store(getattr(func, "__name__") + " return " + str(ret), 10)
return ret
return inner
return wrapper
### 导入及全局变量
colorlog需要安装一下,可以控制台输出彩色的
multiprocessing用于进程安全,加锁
设置了几个全局变量,方便修改
LOG\_DIR是日志保存的目录
STD\_LEVEL是控制台输出的级别,只输出WARN及以上的
FILE\_LEVEL是保存日志文件的级别,一般调试时使用DEBUG,生产环境时使用INFO
### 函数
\_\_init\_\_
1. 创建目录和日志文件,如果不存在。若大于1M,复制为当前日期时间的日志文件
2. 设置了控制台和文件的格式器,控制台是彩色的
3. 设置了控制台和文件的处理器
log\_show\_store
保存到日志文件并打印到控制台
log\_show
仅打印到控制台
log\_store
仅保存到日志文件
log\_wrapper
日志装饰器,用于添加到重要函数上,记录函数调用参数及返回值
## 实验及结果
“”"
-- coding:utf-8 --
@File: main.py
@Author:frank yu
@DateTime: 2021.10.12 20:56
@Contact: frankyu112058@gmail.com
@Description:
“”"
from utils.log import Log, DEBUG, WARN, INFO
from concurrent.futures import ThreadPoolExecutor, ProcessPoolExecutor
log = Log()
@log.log_wrapper(“wrapper for test”)
def test(a, b, c=5):
log.log_show('main test 1 ’ + str(a), level=DEBUG)
log.log_show('main test 2 ’ + str(b), level=WARN)
log.log_store('main test 3 ’ + str(a), level=DEBUG)
log.log_store('main test 4 ’ + str(b), level=WARN)
log.log_show_store('main test 5 ’ + str(a + b), level=INFO)
return a + b + c
def even(num):
log.log_store('main even 1 ’ + str(num), level=INFO)
def odd(num):
log.log_store('main odd 1 ’ + str(num), level=INFO)
multi thread test
def multi_thread():
exe = ThreadPoolExecutor(4)
for i in range(10):
if i & 1 == 0:
exe.submit(even, i)
else:
exe.submit(odd, i)
multi process test
def multi_process():
exe = ProcessPoolExecutor(4)
for i in range(10):
if i & 1 == 0:
exe.submit(even, i)
else:
exe.submit(odd, i)
if name == ‘main’:
test(3, 4, 6)
multi_thread()
multi_process()
msg这里传的是 **模块 函数 函数行数 消息** ,程序出错方便定位。
test测试了日志级别和日志装饰器
multi\_thread测试了多线程
multi\_process测试了多进程
结果如下:
![](https://img-blog.csdnimg.cn/b9f2e453c3204bdf863afefb76441237.PNG?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBAbGFkeV9raWxsZXI5,size_20,color_FFFFFF,t_70,g_se,x_16)
测试结果
可以看到控制台仅输出了WARNING级别,当然,如果有更高级别也会输出。
可以看到test函数的调用及参数和返回结果
可以看到多线程和多进程状态下结果没有丢失
## 更新
用了段时间,彩色的花里胡哨,很少用到,formatter有点少,而且封装后不能定位打印日志的文件,还是直接使用logger吧。
log.py
“”"
-- coding:utf-8 --
@File: log.py
@Author:frank yu
@DateTime: 2023.08.17 15:07
@Contact: frankyu112058@gmail.com
@Description:
日志封装
- 可以多线程、多进程使用
- 默认只有控制台handler,可以通过set_std、set_file、set_std_file设置,尤其是想保存到文件时
- 其他地方使用只import log对象即可
“”"
import functools
import logging
import os
import random
import sys
import multiprocessing
import time
from shutil import copyfile
log directory
LOG_DIR = “./logs”
level for stdout
STD_LEVEL = logging.DEBUG
level for file
FILE_LEVEL = logging.DEBUG
class Log:
def init(self, log_dir=LOG_DIR):
“”"
function:init create log file, judge if log file is too bigger
:param log_dir: directory of logs
“”"
log_path = log_dir + “/log”
if not os.path.exists(log_dir):
os.mkdir(log_dir)
os.chmod(log_dir, 0o777)
if not os.path.exists(log_path):
f = open(log_path, mode=‘w’, encoding=‘utf-8’)
f.close()
os.chmod(log_path, 0o777)
# if log file is more than 1MB, copy to a file and clear log file
if os.path.getsize(log_path) / 1048576 > 1:
# print(os.path.getsize(log_path))
copyfile(log_path, log_dir + “/log” + str(time.time()).replace(“.”, “”))
with open(log_path, ‘w’) as f:
f.truncate()
f.close()
self.logger_format = logging.Formatter(‘[%(asctime)s %(filename)s:%(funcName)s:%(lineno)d %(levelname)-8s] %(’
‘message)s’)
self.logger = logging.getLogger(str(random.random()))
self.logger.handlers.clear()
self.logger.setLevel(logging.DEBUG)
self.filehandler = logging.FileHandler(log_path, mode='a')
self.filehandler.setLevel(FILE_LEVEL)
self.filehandler.setFormatter(self.logger_format)
self.stdouthandler = logging.StreamHandler(sys.stdout)
self.stdouthandler.setLevel(STD_LEVEL)
self.stdouthandler.setFormatter(self.logger_format)
self.logger.addHandler(self.stdouthandler)
self.__lock = multiprocessing.Lock()
def set_std_file(self):
"""
function: set std and file handler
return: None
"""
self.__lock.acquire()
self.logger.addHandler(self.stdouthandler)
self.logger.addHandler(self.filehandler)
self.__lock.release()
def set_std(self):
"""
function: set std handler
"""
网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!