OpenBackdoor

1
# -*- coding: utf-8 -*-
2
import logging
3
import os
4
import datetime
5
from typing import *
6

7
def init_logger(
8
    log_file: Optional[str] = None,
9
    log_file_level=logging.NOTSET,
10
    log_level=logging.INFO,
11
):  
12
    if isinstance(log_file_level, str):
13
        log_file_level = getattr(logging, log_file_level)
14
    if isinstance(log_level, str):
15
        log_level = getattr(logging, log_level)
16
    log_format = logging.Formatter("[\033[032m%(asctime)s\033[0m %(levelname)s] %(module)s %(message)s")
17
    logger = logging.getLogger()
18
    logger.setLevel(log_level)
19

20
    console_handler = logging.StreamHandler()
21
    console_handler.setFormatter(log_format)
22
    logger.handlers = [console_handler]
23

24
    if log_file and log_file != '':
25
        file_handler = logging.FileHandler(log_file)
26
        file_handler.setLevel(log_file_level)
27
        file_handler.setFormatter(log_format)
28
        logger.addHandler(file_handler)
29
    return logger
30

31
logger = init_logger()
32