Bootstrap

python调试时打印当所在位置信息

在写代码时 经常需要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")

;