summaryrefslogtreecommitdiff
path: root/cloudinit/log.py
diff options
context:
space:
mode:
authorJoshua Harlow <harlowja@yahoo-inc.com>2012-06-07 12:42:38 -0700
committerJoshua Harlow <harlowja@yahoo-inc.com>2012-06-07 12:42:38 -0700
commit8900f9cba622eeaf3810003c5a6ff7522312277b (patch)
tree96dd5295a29c69eabfede0630f697325149926f3 /cloudinit/log.py
parent87ebd91af8c230b8f1deed8e734297e40397eea0 (diff)
downloadvyos-cloud-init-8900f9cba622eeaf3810003c5a6ff7522312277b.tar.gz
vyos-cloud-init-8900f9cba622eeaf3810003c5a6ff7522312277b.zip
1. Adding some new helper files that split off file inclusion, templating, importing, constant usage.
1. Move all datasources to a new sources directory 1. Rename some files to be more consistent with python file/module naming.
Diffstat (limited to 'cloudinit/log.py')
-rw-r--r--cloudinit/log.py94
1 files changed, 94 insertions, 0 deletions
diff --git a/cloudinit/log.py b/cloudinit/log.py
new file mode 100644
index 00000000..4d270045
--- /dev/null
+++ b/cloudinit/log.py
@@ -0,0 +1,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())