#!/usr/bin/env python
# coding: utf-8
import logging
import time
import os
def get_log(log_model_name, style='console'):
"""
:param log_model_name: 要打印日誌的模塊名
:param style: 日誌打印方式,console:打印到控制檯;file:打印到文件中
:return:
"""
# 創建一個logger
logger = logging.getLogger(log_model_name)
logger.setLevel(logging.INFO)
# 設置日誌存放路徑,日誌文件名
# 獲取本地實際,轉換爲設置的格式
rq = time.strftime('%Y%m%d%H%M',time.localtime(time.time()))
# 設置所有日誌和錯誤日誌的存放路徑
path = os.path.dirname(os.path.abspath('log_print.py'))
all_log_path = os.path.join(path, 'All_Logs/')
error_log_path = os.path.join(path, 'Error_Logs/')
# 判斷要保存的日誌文件夾是否存在,不存在就創建
if not os.path.exists(all_log_path):
os.mkdir(all_log_path)
if not os.path.exists(error_log_path):
os.mkdir(error_log_path)
# 設置日誌文件名
all_log_name = all_log_path + rq + '.log'
error_log_name = error_log_path + rq + '.log'
if style == 'console':
# 創建一個handler輸出到控制檯
# 創建一個handler寫入所有日誌
all_log = logging.StreamHandler()
all_log.setLevel(logging.INFO)
# 創建一個handler寫入錯誤日誌
error_log = logging.StreamHandler(error_log_name)
error_log.setLevel(logging.ERROR)
# 定義日誌輸出格式
# 以時間-日誌名稱-日誌級別-日誌內容的形式展示
all_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(message)s')
# 以時間-日誌名稱-日誌級別-文件名-函數行數-錯誤內容
error_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(module)s-%(message)s')
# 將定義好的輸出形式添加到handler
all_log.setFormatter(all_log_formatter)
error_log.setFormatter(error_log_formatter)
# 給logger添加handler
logger.addHandler(all_log)
logger.addHandler(error_log)
return logger
else:
# 日誌寫入到文件
# 創建一個handler寫入所有日誌
all_log = logging.FileHandler(all_log_name)
all_log.setLevel(logging.INFO)
# 創建一個handler寫入錯誤日誌
error_log = logging.FileHandler(error_log_name)
error_log.setLevel(logging.ERROR)
# 定義日誌輸出格式
# 以時間-日誌名稱-日誌級別-日誌內容的形式展示
all_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(message)s')
# 以時間-日誌名稱-日誌級別-文件名-函數行數-錯誤內容
error_log_formatter = logging.Formatter('%(asctime)s-%(name)s-%(levelname)s-%(module)s-%(message)s')
# 將定義好的輸出形式添加到handler
all_log.setFormatter(all_log_formatter)
error_log.setFormatter(error_log_formatter)
# 給logger添加handler
logger.addHandler(all_log)
logger.addHandler(error_log)
return logger
if __name__ == '__main__':
logger0 = get_log('test', 'console') # test需要打印日誌的項目名稱
logger0.info('aaa 1213')
logger0.info('aaa 7890')
style=='console':
style=='file':