logging_exceptions/logging_exceptions_example.py at master · Bernhard10/logging_exceptions · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import logging
import argparse
import logging_exceptions

parser = argparse.ArgumentParser()
logging_exceptions.update_parser(parser)


def foo():
    log = logging.getLogger("main.foo")
    log.info("Logging is enabled for logger 2.")
    for level in [logging.DEBUG, logging.INFO, logging.WARNING, logging.ERROR, logging.CRITICAL]:
        log.log(level, "Logging at level %s.", level)


def raise_error_contextmngr(value):
    e = ValueError("Another ValueError")
    log = logging.getLogger("main.inside_ctxt")
    log.info("Before with-context. This is logged directly")
    with logging_exceptions.log_to_exception(log, e):
        log.debug("This is DEBUG ... %s", value)
        log.info("This is an INFO ... %s", value)
        log.warning("This is a WARNING ... %s", value)
        log.error("This is an ERROR ... %s", value)
        log.critical("This is CRITICAL ... %s", value)
    log.info("After with-context. This is logged directly")
    raise e


def raise_error_contextmngr2(value):
    e = ValueError("Another ValueError")
    log = logging.getLogger("main.inside_ctxt2")
    log.info("Before with-context. This is logged directly")
    with logging_exceptions.log_to_exception(log, e):
        log.debug("This is DEBUG ... %s", value)
        log.info("This is an INFO ... %s", value)
        log.warning("This is a WARNING ... %s", value)
        log.error("This is an ERROR ... %s", value)
        log.critical("This is CRITICAL ... %s", value)
        log.info("Raising inside with context")
        raise e

def helper_function(log):
    with logging_exceptions.log_at_caller(log):
        log.error("Logged from within helper function")

class Filter1(logging.Filter):
    def __init__(self, name):
        super(Filter1, self).__init__()
        self.name = name

    def filter(self, record):
        record.msg = record.msg + " Filtered by {}".format(self.name)
        return True


if __name__ == "__main__":
    args = parser.parse_args()
    logging.basicConfig(format = "%(levelname)s:%(name)s.%(funcName)s[%(lineno)d]: %(message)s")
    logging_exceptions.use_colored_output(True)
    logging_exceptions.config_from_args(args)
    log = logging.getLogger("main1")
    log.info("Logging is enabled for logger 1.")
    fltr = Filter1("Main1 logger")
    log.addFilter(fltr)
    log.info("Now with filter")
    for level in [logging.DEBUG, logging.INFO, logging.WARNING, logging.ERROR, logging.CRITICAL]:
        log.log(level, "Logging at level %s.", level)

    foo()

    try:
        raise_error_contextmngr(120)
    except Exception as e:
        logging_exceptions.log_exception(e, logging.WARNING)
    try:
        raise_error_contextmngr(555)
    except Exception as e:
        logging_exceptions.log_exception(e)

    log = logging.getLogger("another.logger.name")
    log.info("The following should log from the main module level (doesn't work with root logger)")
    helper_function(log)

    fltr = Filter1("CTMNGR2")
    log = logging.getLogger("main.inside_ctxt2")
    log.addFilter(fltr)

    log = logging.getLogger()
    log.handlers[0].addFilter(Filter1("RootHandler"))
    try:
        raise_error_contextmngr2(12345)
    except Exception as e:
        logging_exceptions.log_exception(e, with_stacktrace=False)


    log.info("Almost there")
    try:
        raise_error_contextmngr2(-1)
    except Exception as e:
        with logging_exceptions.log_to_exception(log, e):
            log.critical("using a BARE RAISE works as intended.")
        raise