32 lines
919 B
Python
32 lines
919 B
Python
|
import logging
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
def setup_logger(name: str = 'dy-downloader', level: int = logging.INFO, log_file: str = None) -> logging.Logger:
|
||
|
|
logger = logging.getLogger(name)
|
||
|
|
logger.setLevel(level)
|
||
|
|
|
||
|
|
if logger.handlers:
|
||
|
|
return logger
|
||
|
|
|
||
|
|
formatter = logging.Formatter(
|
||
|
|
'%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
||
|
|
)
|
||
|
|
|
||
|
|
console_handler = logging.StreamHandler(sys.stdout)
|
||
|
|
console_handler.setLevel(level)
|
||
|
|
console_handler.setFormatter(formatter)
|
||
|
|
logger.addHandler(console_handler)
|
||
|
|
|
||
|
|
if log_file:
|
||
|
|
log_path = Path(log_file)
|
||
|
|
log_path.parent.mkdir(parents=True, exist_ok=True)
|
||
|
|
file_handler = logging.FileHandler(log_file, encoding='utf-8')
|
||
|
|
file_handler.setLevel(level)
|
||
|
|
file_handler.setFormatter(formatter)
|
||
|
|
logger.addHandler(file_handler)
|
||
|
|
|
||
|
|
return logger
|