weeheavy/pelican-theme
1
0
Fork 0
forked from github/pelican
Code Pull requests Activity

Add --fatal=errors|warnings program option

Browse source
This commit is contained in:
Kernc 2015-11-15 20:34:35 +01:00
commit 9a04ce660c
2 changed files with 25 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

21
pelican/log.py
View file

@ -175,7 +175,22 @@ class LimitLogger(SafeLogger):
def enable_filter(self):
self.addFilter(LimitLogger.limit_filter)
logging.setLoggerClass(LimitLogger)
class FatalLogger(LimitLogger):
warnings_fatal = False
errors_fatal = False
def warning(self, *args, **kwargs):
super(FatalLogger, self).warning(*args, **kwargs)
if FatalLogger.warnings_fatal:
raise RuntimeError('Warning encountered')
def error(self, *args, **kwargs):
super(FatalLogger, self).error(*args, **kwargs)
if FatalLogger.errors_fatal:
raise RuntimeError('Error encountered')
logging.setLoggerClass(FatalLogger)
def supports_color():
@ -203,7 +218,9 @@ def get_formatter():
return TextFormatter()
def init(level=None, handler=logging.StreamHandler(), name=None):
def init(level=None, fatal='', handler=logging.StreamHandler(), name=None):
FatalLogger.warnings_fatal = fatal.startswith('warning')
FatalLogger.errors_fatal = bool(fatal)
logger = logging.getLogger(name)