aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/livestreamer/logger.py
blob: 34463b4c6a96c55bad154ec9b96b3ca6870b033f (plain)
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
import sys

class Logger(object):
    Levels = ["none", "error", "warning", "info", "debug"]
    Format = "[{module}][{level}] {msg}\n"

    output = sys.stdout
    level = 0

    @classmethod
    def set_level(cls, level):
        try:
            index = Logger.Levels.index(level)
        except ValueError:
            return

        cls.level = index

    @classmethod
    def set_output(cls, output):
        cls.output = output

    def __init__(self, module):
        self.module = module

    def msg(self, level, msg, *args):
        if Logger.level < level or level > len(Logger.Levels):
            return

        msg = msg.format(*args)

        self.output.write(Logger.Format.format(module=self.module,
                                               level=Logger.Levels[level],
                                               msg=msg))
        self.output.flush()

    def error(self, msg, *args):
        self.msg(1, msg, *args)

    def warning(self, msg, *args):
        self.msg(2, msg, *args)

    def info(self, msg, *args):
        self.msg(3, msg, *args)

    def debug(self, msg, *args):
        self.msg(4, msg, *args)


__all__ = ["Logger"]