summaryrefslogtreecommitdiff
path: root/cloudinit/log.py
blob: 4d270045e20214fbded854cefbcf73c319856c34 (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
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
# vim: tabstop=4 shiftwidth=4 softtabstop=4

import logging
import logging.handlers
import sys

# Logging levels for easy access
CRITICAL = logging.CRITICAL
FATAL = logging.FATAL
ERROR = logging.ERROR
WARNING = logging.WARNING
WARN = logging.WARN
INFO = logging.INFO
DEBUG = logging.DEBUG
NOTSET = logging.NOTSET

# File log rotation settings
ROTATE_AMOUNT = 10  # Only keep the past 9 + 1 active
ROTATE_SIZE = 10 * 1024 * 1024  # 10 MB


class ConsoleFormatter(logging.Formatter):

    def _get_mini_level(self, record):
        if record.levelno in [INFO, NOTSET] or not record.levelname:
            return ''
        lvlname = record.levelname
        return lvlname[0].upper() + ": "

    def format(self, record):
        record.message = record.getMessage()
        rdict = dict(record.__dict__)
        rdict['minilevelname'] = self._get_mini_level(record)
        # Skipping exception info for the console...
        return self._fmt % (rdict)


def setupLogging(level, filename=None, filelevel=logging.DEBUG):
    root = getLogger()
    consolelg = logging.StreamHandler(sys.stdout)
    consolelg.setFormatter(ConsoleFormatter('%(minilevelname)s%(message)s'))
    consolelg.setLevel(level)
    root.addHandler(consolelg)
    if filename:
        filelg = logging.handlers.RotatingFileHandler(filename, maxBytes=ROTATE_SIZE, backupCount=ROTATE_AMOUNT)
        filelg.setFormatter(logging.Formatter('%(levelname)s: @%(name)s : %(message)s'))
        filelg.setLevel(filelevel)
        root.addHandler(filelg)
    root.setLevel(level)


def logging_set_from_cfg(cfg):
    log_cfgs = []
    logcfg = util.get_cfg_option_str(cfg, "log_cfg", False)
    if logcfg:
        # if there is a 'logcfg' entry in the config, respect
        # it, it is the old keyname
        log_cfgs = [logcfg]
    elif "log_cfgs" in cfg:
        for cfg in cfg['log_cfgs']:
            if isinstance(cfg, list):
                log_cfgs.append('\n'.join(cfg))
            else:
                log_cfgs.append()

    if not len(log_cfgs):
        sys.stderr.write("Warning, no logging configured\n")
        return

    for logcfg in log_cfgs:
        try:
            logging.config.fileConfig(StringIO.StringIO(logcfg))
            return
        except:
            pass

    raise Exception("no valid logging found\n")


def getLogger(name='cloudinit'):
    return logging.getLogger(name)


# Fixes this annoyance...
# No handlers could be found for logger XXX annoying output...
try:
    from logging import NullHandler
except ImportError:
    class NullHandler(logging.Handler):
        def emit(self, record):
            pass

logger = getLogger()
logger.addHandler(NullHandler())