import logging
import logging.handlers
import os

'''
os.getcwd()代表py执行目录
日志模块
'''

#  创建log文件夹
new_path = os.path.join(os.getcwd(), 'log')
if not os.path.exists(new_path):
    os.makedirs(new_path)

LOG_FILENAME = 'log/jd.log'
logger = logging.getLogger()

def set_logger():
    if not logger.handlers:
        logger.setLevel(logging.INFO)
        formatter = logging.Formatter('%(asctime)s - %(levelname)s: %(message)s')
        console_handler = logging.StreamHandler()
        console_handler.setFormatter(formatter)
        logger.addHandler(console_handler)
        file_handler = logging.handlers.RotatingFileHandler(
            LOG_FILENAME, maxBytes=5*1024*1024, backupCount=5, encoding="utf-8")
        file_handler.setFormatter(formatter)
        logger.addHandler(file_handler)


set_logger()