在写代码时 经常需要print 打印要调试的信息,但使用print 太多就会显得非常混乱 不知在哪一行执行了调试,为此写了info打印功能,可以将所在的文件和处于的行数直接打印出来。只要导入这个程序并且调用 info功能info() 就可以了,如:
2024-03-02 12:05:24 [main.py:7]: 打印调试
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
"""
自定义 log
"""
import os
import time
import datetime
import traceback
def get_invoker_info():
invoker_file=traceback.extract_stack()[-3].filename # 调用本logger的代码的所在文件
#invoker_func=traceback.extract_stack()[-3].name # 调用本logger的代码(其实就是下面的log方法和error方法)所在的函数
invoker_func=traceback.extract_stack()[-3][1] # 代码所在行号
# run_name = '[%s] ' % os.path.basename(__file__) # 文件所在.py
return invoker_file,invoker_func
def info(*args):
msg = ""
for tmp_msg in args:
msg += str(tmp_msg)
t = time.strftime('%Y-%m-%d %H:%M:%S')
fd = datetime.datetime.strptime(t, '%Y-%m-%d %H:%M:%S')
#eta = (fd + datetime.timedelta(hours=8)).strftime("%Y-%m-%d %H:%M:%S")
eta = (fd + datetime.timedelta()).strftime("%Y-%m-%d %H:%M:%S")
#formatOr = '{0},[{1}].{2}():{3}' # 2022-05-10 09:40:04,[sz_tftp.py].sz_tftp_server():
#formatOr = 'INFO '+'{0} [{1}:{2}]: {3}'
formatOr = '{0} [{1}:{2}]: {3}'
invoker_file,invoker_func = get_invoker_info()
content = formatOr.format(eta,os.path.basename(invoker_file),invoker_func,str(msg))
print(content)
t_path = os.path.split(os.path.realpath(__file__))[0]
if not os.path.exists(t_path + r'\logs/'):
os.mkdir(t_path + r'\logs/')
log_name_time = time.strftime('%Y%m%d', time.localtime(time.time())) # 创建已天为单位的文件
#log_name_time_time = time.strftime('%Y%m%d_%H%m%S', time.localtime(time.time())) # 创建以秒为单位的文件
log_name = '%slog_%s.log' % (t_path+'\logs/', log_name_time) # 固件文件名称
with open(log_name, 'a+') as f:
f.write(content+'\n')
"""
def info(*args):
msg = ""
for tmp_msg in args:
msg = msg + str(tmp_msg)
print(time.strftime('%Y-%m-%d %X', time.localtime()) + ': ' , str(msg))
"""
if __name__ == '__main__':
info(" hello "," world")