rename parsers, better journald integration
[logparse.git] / logparse / interface.py
index dcb077794d764a33886defbace43b8bc3abaa876..46c5aeaac765da942b3cd0cc1b5b20198c724d7c 100644 (file)
@@ -1,5 +1,6 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+
 """
 This module is the entrypoint of the `logparse` shell command and also contains
 single-use functions which don't fit elsewhere. All user interaction with
@@ -11,15 +12,17 @@ This module provides the following methods:
     - `rotate_sim()`:   Simulate log rotation
 """
 
-import logging, logging.handlers
 import argparse
+from copy import copy
+import logging
+import logging.handlers
 import os
-from sys import stdin, version
+from sys import exit, stdin, version
 from subprocess import check_output
 from datetime import datetime
 
 import logparse
-from logparse import formatting, mail, config, load_parsers
+from logparse import formatting, mail, config, load_parsers, util
 
 
 def main():
@@ -35,35 +38,40 @@ def main():
     # Load config
 
     config.prefs = config.loadconf(argparser.parse_args().config)
+    if argparser.parse_args().time_period:
+        config.prefs.set("logparse", "period",
+                argparser.parse_args().time_period)
     
     # Set up logging
-
     logger = logging.getLogger(__name__)
-    loghandler = logging.handlers.SysLogHandler(address = '/dev/log')
-    loghandler.setFormatter(logging.Formatter(fmt='logparse[' + str(os.getpid()) + ']: %(message)s'))
-    loghandler.setLevel(logging.INFO)   # don't spam syslog with debug messages
-    if argparser.parse_args().quiet or config.prefs.getboolean("logparse", "quiet"):
-        logging.basicConfig(level=logging.CRITICAL)
-    elif argparser.parse_args().verbose or config.prefs.getboolean("logparse", "verbose"):
-        logging.basicConfig(level=logging.DEBUG)
+    if (argparser.parse_args().quiet
+            or config.prefs.getboolean("logparse", "quiet")):
+        logparse.logger.setLevel(logging.CRITICAL)
+    elif (argparser.parse_args().verbose 
+            or config.prefs.getboolean("logparse", "verbose")):
+        logparse.logger.setLevel(logging.DEBUG)
         logger.debug("Verbose mode turned on")
     else:
-        logging.basicConfig(level=logging.INFO)
-    logger.addHandler(loghandler)
+        logparse.logger.setLevel(logging.INFO)
+
+
 
     # Time analysis
 
     global start
     start = datetime.now()
-    logger.info("Beginning log analysis at {0} {1}".format(start.strftime(formatting.DATEFMT), start.strftime(formatting.TIMEFMT)))
-    logger.debug("This is {0} version {1}, running on Python {2}".format(logparse.__name__, logparse.__version__, version.replace('\n', '')))
+    logger.info("Beginning log analysis at {0} {1}".format(
+        start.strftime(formatting.DATEFMT), start.strftime(formatting.TIMEFMT)))
+    logger.debug("This is {0} version {1}, running on Python {2}".format(
+        logparse.__name__, logparse.__version__, version.replace('\n', '')))
      
     # Write header
 
     formatting.init_var()
 
     if argparser.parse_args().plain:
-        output = formatting.PlaintextOutput(linewidth=config.prefs.getint("plain", "linewidth"))
+        output = formatting.PlaintextOutput(
+                linewidth=config.prefs.getint("plain", "linewidth"))
         output.append_header()
     else:
         output = formatting.HtmlOutput()
@@ -86,6 +94,12 @@ def main():
     # Set up parsers
 
     loader = load_parsers.ParserLoader()
+
+    try:
+        loader.check_systemd()
+    except Exception as e:
+        logger.error("Failed to check systemd dependencies: ".format(e))
+
     if parser_names:
         for parser_name in parser_names:
             if parser_name not in ignore_logs:
@@ -97,44 +111,92 @@ def main():
 
     # Execute parsers
 
+    executed_parsers = []
+
     for parser in loader.parsers:
-        output.append_section(parser.parse_log())
+        if (argparser.parse_args().verbose 
+                or config.prefs.getboolean("logparse", "verbose")):
+            output.append_section(parser.parse_log())
 
-    # Write footer
+        else:
+            try:
+                output.append_section(parser.parse_log())
+            except Exception as e:
+                logger.error("Uncaught error executing logger {0}: {1}".format(
+                    parser.name, e))
+        executed_parsers.append(parser.name)
 
+    if len(executed_parsers) == 0:
+        exit()
+
+    # Write footer
     output.append_footer()
 
     # Write output
+    if ((argparser.parse_args().destination
+        or config.prefs.get("logparse", "output"))
+        and not argparser.parse_args().no_write):
 
-    if (argparser.parse_args().destination or config.prefs.get("logparse", "output")) and not argparser.parse_args().no_write:
+        # Determine destination path
         if argparser.parse_args().destination:
             dest_path = argparser.parse_args().destination
         else:
             dest_path = config.prefs.get("logparse", "output")
+
         logger.debug("Outputting to {0}".format(dest_path))
-        if (argparser.parse_args().embed_styles or config.prefs.getboolean("html", "embed-styles")) and not (argparser.parse_args().plain or config.prefs.getboolean("plain", "plain")):
-            output.embed_css(config.prefs.get("html", "css"))
-        if (not os.path.isfile(dest_path)) and not (argparser.parse_args().overwrite or config.prefs.getboolean("logparse", "overwrite")):
-            output.write(dest_path)
+
+        # Determine whether to clobber old file
+        if (not os.path.isfile(dest_path)) \
+        and not (argparser.parse_args().overwrite
+                or config.prefs.getboolean("logparse", "overwrite")):
+
+            if (argparser.parse_args().embed_styles
+                    or config.prefs.getboolean("html", "embed-styles")) \
+                and not (argparser.parse_args().plain
+                        or config.prefs.getboolean("plain", "plain")):
+                # Embed CSS stylesheet
+                output.embed_css(config.prefs.get("html", "css"))
+                output.write_embedded(dest_path)
+            else:
+                output.write(dest_path)
+
         elif logging.root.level == logging.CRITICAL:
+
+            # Don't write output if running in quiet mode (only stdout)
             pass
+
         else:
+
             logger.warning("Destination file already exists")
-            if input("Would you like to overwrite {0}? (y/n) [n] ".format(dest_path)) == 'y':
-                output.write(dest_path)
+            if input("Would you like to overwrite {0}? (y/n) [n] "
+                    .format(dest_path)) == 'y':
+                if (argparser.parse_args().embed_styles
+                        or config.prefs.getboolean("html", "embed-styles")) \
+                    and not (argparser.parse_args().plain
+                        or config.prefs.getboolean("plain", "plain")):
+
+                    output.embed_css(config.prefs.get("html", "css"))
+                    output.write_embedded(dest_path)
+
+                else:
+                    output.write(dest_path)
             else:
                 logger.warning("No output written")
 
     # Send email if requested
 
-    if (str(argparser.parse_args().to) or str(config.prefs.get("mail", "to"))) and not argparser.parse_args().no_mail:
+    if (str(argparser.parse_args().to) or str(config.prefs.get("mail", "to"))) \
+            and not argparser.parse_args().no_mail:
+
         if str(argparser.parse_args().to):
             to = argparser.parse_args().to
         else:
             to = config.prefs.get("mail", "to")
+
         mail.sendmail(
             mailbin=config.prefs.get("mail", "mailbin"),
-            body=(output.embed_css(config.prefs.get("html", "css")) if isinstance(output, formatting.HtmlOutput) else output.content),
+            body=(output.embed_css(config.prefs.get("html", "css"))
+                if isinstance(output, formatting.HtmlOutput) else output.content),
             recipient=to,
             subject=formatting.fsubject(config.prefs.get("mail", "subject")),
             html=isinstance(output, formatting.HtmlOutput),
@@ -143,9 +205,11 @@ def main():
     # Rotate logs if requested
 
     if not argparser.parse_args().no_rotate:
-        if argparser.parse_args().simulate or config.prefs.getboolean("logparse", "rotate"):
+        if (argparser.parse_args().simulate
+                or config.prefs.getboolean("logparse", "rotate")):
             rotate_sim()
-        elif config.prefs.getboolean("logparse", "rotate") or argparser.parse_args().rotate:
+        elif (config.prefs.getboolean("logparse", "rotate")
+                or argparser.parse_args().rotate):
             rotate()
         else:
             logger.debug("User doesn't want to rotate logs")
@@ -155,34 +219,75 @@ def main():
     # Finish up
 
     finish = datetime.now()
-    logger.info("Finished parsing logs at {0} {1} (total time: {2})".format(finish.strftime(formatting.DATEFMT), finish.strftime(formatting.TIMEFMT), finish - start))
+    logger.info("Finished parsing logs at {0} {1} (total time: {2})".format(
+        finish.strftime(formatting.DATEFMT),
+        finish.strftime(formatting.TIMEFMT),
+        finish - start))
 
     if argparser.parse_args().printout:
-        output.print_stdout()
+        if isinstance(output, formatting.HtmlOutput) \
+                and argparser.parse_args().embed_styles \
+                or config.prefs.getboolean("html", "embed-styles"):
+            output.print_stdout_embedded()
+        else:
+            output.print_stdout()
 
     return
 
+
 def get_argparser():
     """
     Initialise arguments (in a separate function for documentation purposes)
     """
-    argparser = argparse.ArgumentParser(description='grab logs of some common services and send them by email')
-    argparser.add_argument('-t','--to', help='mail recipient (\"to\" address)', required=False)
-    argparser.add_argument('-c', '--config', help='path to config file', required=False, default="/etc/logparse/logparse.conf")
-    argparser.add_argument('-p', '--print', help='print HTML to stdout', required=False, dest='printout', action='store_true', default=False)
-    argparser.add_argument('-d', '--destination', help='file to output HTML', required=False)
-    argparser.add_argument('-f', '--overwrite', help='force overwrite an existing output file', required=False, action='store_true', default=False)
-    argparser.add_argument('-v', '--verbose', help='verbose console/syslog output (for debugging)', required=False, default=False, action='store_true')
-    argparser.add_argument('-r', '--rotate', help='force rotate log files using systemd logrotate (overrides --rotate and "rotate" in logparse.conf)', required=False, default=False, action='store_true')
-    argparser.add_argument('-nr', '--no-rotate', help='do not rotate logfiles (overrides --rotate and logparse.conf)', required=False, default=False, action='store_true')
-    argparser.add_argument('-s', '--simulate', help="test run logrotate (do not actually change files)", required=False, default=False, action="store_true")
-    argparser.add_argument('-l', '--logs', help='services to analyse', required=False)
-    argparser.add_argument('-nl', '--ignore-logs', help='skip these services (takes precedence over -l)', required=False)
-    argparser.add_argument('-es', '--embed-styles', help='make CSS rules inline rather than linking the file', required=False, default=False, action='store_true')
-    argparser.add_argument('-nh', '--plain', help='write/send plain text rather than HTML', required=False, default=False, action='store_true')
-    argparser.add_argument('-q', '--quiet', help='no output to stdout', required=False, default=False, action='store_true')
-    argparser.add_argument('-nm', '--no-mail', help="do not send email (overrides config file)", required=False, default=False, action="store_true")
-    argparser.add_argument('-nw', '--no-write', help="do not write output file (overrides config file)", required=False, default=False, action="store_true")
+
+    argparser = argparse.ArgumentParser(description=
+            'Grab logs of some common services and send them by email')
+    argparser.add_argument('-t','--to', required=False,
+            help='mail recipient (\"to\" address)')
+    argparser.add_argument('-c', '--config', required=False,
+            default="/etc/logparse/logparse.conf",
+            help='path to config file')
+    argparser.add_argument('-p', '--print', required=False, dest='printout',
+            action='store_true', default=False,
+            help='print HTML to stdout')
+    argparser.add_argument('-d', '--destination', required=False, 
+            help='file to output HTML')
+    argparser.add_argument('-f', '--overwrite', required=False,
+            action='store_true', default=False, 
+            help='force overwrite an existing output file')
+    argparser.add_argument('-v', '--verbose', required=False, default=False,
+            action='store_true',
+            help='verbose console/syslog output (for debugging)')
+    argparser.add_argument('-r', '--rotate', required=False, default=False,
+            action='store_true',
+            help='force rotate log files using systemd logrotate (overrides \
+            --rotate and "rotate" in logparse.conf)')
+    argparser.add_argument('-nr', '--no-rotate', required=False, default=False,
+            action='store_true', 
+            help='do not rotate log files (overrides config)')
+    argparser.add_argument('-s', '--simulate', required=False, default=False,
+            action="store_true",
+            help="test run logrotate (do not actually change files)")
+    argparser.add_argument('-l', '--logs', required=False, 
+            help='services to analyse')
+    argparser.add_argument('-nl', '--ignore-logs', required=False,
+            help='skip these services (takes precedence over -l)')
+    argparser.add_argument('-es', '--embed-styles', required=False,
+            default=False, action='store_true',
+            help='make CSS rules inline rather than linking the file')
+    argparser.add_argument('-nh', '--plain', required=False, default=False,
+            action='store_true', help='write/send plain text rather than HTML')
+    argparser.add_argument('-q', '--quiet', required=False, default=False,
+            action='store_true', help='no output to stdout')
+    argparser.add_argument('-nm', '--no-mail', required=False, default=False,
+            action="store_true",
+            help="do not send email (overrides config file)")
+    argparser.add_argument('-nw', '--no-write', required=False, default=False,
+            action="store_true",
+            help="do not write output file (overrides config file)")
+    argparser.add_argument('-tp', '--time-period', required=False,
+            help="time period to analyse logs for (applies to all parsers)")
+
     return argparser
 
 
@@ -199,12 +304,16 @@ def rotate():
     try:
         if not os.geteuid() == 0:
             if stdin.isatty():
-                logger.warning("Not running as root, using sudo (may require password to be entered)")
-                rotate_shell = check_output("sudo logrotate /etc/logrotate.conf", shell=True)
+                logger.warning("Not running as root, using sudo \
+                        (may require password to be entered)")
+                rotate_shell = check_output(
+                        "sudo logrotate /etc/logrotate.conf", shell=True)
             else:
-                raise PermissionError("Root priviliges are required to run logrotate but are not provided")
+                raise PermissionError("Root priviliges are required to run \
+                        logrotate but were not provided")
         else:
-            rotate_shell = check_output("/usr/sbin/logrotate /etc/logrotate.conf", shell=True)
+            rotate_shell = check_output(
+                    "/usr/sbin/logrotate /etc/logrotate.conf", shell=True)
         logger.info("Rotated logfiles")
         logger.debug("logrotate output: " + rotate_shell)
     except Exception as e:
@@ -220,7 +329,8 @@ def rotate_sim():   # Simulate log rotation
     logger = logging.getLogger(__name__)
     try:
         if not os.geteuid() == 0:
-            logger.warning("Cannot run logrotate as root - you will see permission errors in the output below")
+            logger.warning("Cannot run logrotate as root - \
+                    you will see permission errors in the output below")
         sim_cmd = "logrotate -d /etc/logrotate.conf"
         logger.debug("Here is the output of `{0}` (simulated):".format(sim_cmd))
         sim = check_output(sim_cmd, shell=True)