aurora

1
import sys
2
import logging
3

4

5
class LoggerHandler(logging.Handler):
6
    r"""
7
    Logger handler used in Web UI.
8
    """
9

10
    def __init__(self):
11
        super().__init__()
12
        self.log = ""
13

14
    def reset(self):
15
        self.log = ""
16

17
    def emit(self, record):
18
        if record.name == "httpx":
19
            return
20
        log_entry = self.format(record)
21
        self.log += log_entry
22
        self.log += "\n\n"
23

24

25
def get_logger(name: str) -> logging.Logger:
26
    r"""
27
    Gets a standard logger with a stream hander to stdout.
28
    """
29
    formatter = logging.Formatter(
30
        fmt="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
31
        datefmt="%m/%d/%Y %H:%M:%S"
32
    )
33
    handler = logging.StreamHandler(sys.stdout)
34
    handler.setFormatter(formatter)
35

36
    logger = logging.getLogger(name)
37
    logger.setLevel(logging.INFO)
38
    logger.addHandler(handler)
39

40
    return logger
41

42

43
def reset_logging() -> None:
44
    r"""
45
    Removes basic config of root logger. (unused in script)
46
    """
47
    root = logging.getLogger()
48
    list(map(root.removeHandler, root.handlers))
49
    list(map(root.removeFilter, root.filters))
50