#-*- coding:utf8 -*-


import logging
from logging import CRITICAL, ERROR, WARNING, INFO, DEBUG, NOTSET
import os


class AutoLogger:


    def __init__(self, path, level=WARNING, name="Auto logger"):
        # 初始化日志记录器, 默认为警告级别
        self.logger = logging.getLogger(name)
        self.logger.setLevel(level)


        # 初始化文件日志处理器
        self.fh = logging.FileHandler(path)
        self.fh.setLevel(level)


        # 初始化控制台日志处理器
        self.ch = logging.StreamHandler()
        self.ch.setLevel(level)


        # 初始化日志输出格式
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        self.fh.setFormatter(formatter)
        self.ch.setFormatter(formatter)


        # 绑定处理器至日志记录器
        self.logger.addHandler(self.fh)
        self.logger.addHandler(self.ch)


    # 设置日志级别
    def setLevel(self, level):
        self.fh.setLevel(level)
        self.ch.setLevel(level)


    def setfileLevel(self, level):
        self.fh.setLevel(level)


    def setConsoleLevel(self, level):
        self.ch.setLevel(level)


    # 写日志
    def write_msg(self, typ, msg=None):
        if typ in (CRITICAL, ERROR, WARNING, INFO, DEBUG, NOTSET):
            self.logger.log(typ, msg)
        else:
            print("日志写入失败")
      
if __name__ == "__main__":
    logger = AutoLogger(path=os.getcwd() + "/log.txt")
    logger.write_msg(DEBUG, "debug ...")
    logger.write_msg(WARNING, "waring ...")
    logger.write_msg(ERROR, "error ...")