📦 p0n1 / epub_to_audiobook

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43import logging
import datetime
from pathlib import Path

def get_formatter(is_worker):
    if is_worker:
        return logging.Formatter(
            "%(asctime)s - [Worker-%(process)d] - %(filename)s:%(lineno)d - %(funcName)s - %(levelname)s - %(message)s"
        )
    else:
        return logging.Formatter(
            "%(asctime)s - %(filename)s:%(lineno)d - %(funcName)s - %(levelname)s - %(message)s"
        )

def setup_logging(log_level, log_file=None, is_worker=False):
    formatter = get_formatter(is_worker)

    root_logger = logging.getLogger()
    for handler in root_logger.handlers[:]:
        root_logger.removeHandler(handler)
    root_logger.setLevel(log_level)

    console_handler = logging.StreamHandler()
    console_handler.setFormatter(formatter)
    root_logger.addHandler(console_handler)

    if not log_file:
        log_file = generate_unique_log_path("app") # Default prefix "app"

    # Ensure the directory for the log file exists
    Path(log_file).parent.mkdir(parents=True, exist_ok=True)

    file_handler = logging.FileHandler(log_file)
    file_handler.setFormatter(formatter)
    root_logger.addHandler(file_handler)

def generate_unique_log_path(prefix: str) -> Path:
    """Generates a unique log file path with a timestamp."""
    timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
    log_dir = Path("logs")
    log_dir.mkdir(parents=True, exist_ok=True)
    return log_dir / f"{prefix}_{timestamp}.log"