rework parser loading interface
[logparse.git] / logparse / interface.py
index 8d08c48487f418bba2a9b2746e75d918c12ef19a..90b3ea8e70881dafafc4a0737ff736871ae863c6 100644 (file)
@@ -1,54 +1,32 @@
-#
-#   __main__.py
-#
-#   This module is the entrypoint of the `logparse` shell command and also
-#   contains single-use functions which don't fit elsewhere.
-#
+"""
+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
+logparse should be through this module.
+
+This module provides the following methods:
+    main        Set up arguments, config, logging, and execute parsers
+    rotate      Rotate logs using systemd logrotate
+    rotate_sim  Simulate log rotation
+"""
 
 import logging, logging.handlers
 import argparse
 import os
-import glob
-import sys
+from sys import stdin, version
 from subprocess import check_output
 from datetime import datetime
 
 import logparse
-import logparse.config
-from logparse.config import prefs, loadconf
 from logparse import formatting, mail, config, load_parsers
 
-global argparser
-
-def rotate():       # Rotate logs using systemd logrotate
-    try:
-        if not os.geteuid() == 0:
-            if sys.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)
-            else:
-                raise PermissionError("Root priviliges are required to run logrotate but are not provided")
-        else:
-            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:
-        logger.warning("Failed to rotate log files: " + str(e))
-
-def rotate_sim():   # Simulate log rotation
-    try:
-        if not os.geteuid() == 0:
-            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)
-        logger.debug(sim)
-    except Exception as e:
-        logger.warning("Failed to get logrotate simulation: " + str(e))
-
 
 def main():
+    """
+    Initialisation and general management of logparse functionaliy.
+    """
+
     # Get arguments
+
     global argparser
     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)
@@ -69,9 +47,11 @@ def main():
     argparser.add_argument('-nw', '--no-write', help="do not write output file (overrides config file)", required=False, default=False, action="store_true")
 
     # Load config
-    config.prefs = loadconf(argparser.parse_args().config)
+
+    config.prefs = config.loadconf(argparser.parse_args().config)
     
     # 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'))
@@ -85,15 +65,12 @@ def main():
         logging.basicConfig(level=logging.INFO)
     logger.addHandler(loghandler)
 
-    logger.debug([x for x in config.prefs.sections()])
-    logger.debug(config.prefs.get("logparse", "output"))
-    logger.debug("Config test: " + config.prefs.get("logparse", "output"))
-
     # 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__, sys.version.replace('\n', '')))
+    logger.debug("This is {0} version {1}, running on Python {2}".format(logparse.__name__, logparse.__version__, version.replace('\n', '')))
      
     # Write header
 
@@ -106,32 +83,43 @@ def main():
         output = formatting.HtmlOutput()
         output.append_header(config.prefs.get("html", "header"))
 
-
     # Find parsers
     
-    loader = load_parsers.ParserLoader("logparse.parsers") 
-    parser_names = set([x.name for x in loader.parsers])
-
+    parser_names = []
+    ignore_logs = []
     if argparser.parse_args().logs:
-        parser_names = parser_names.intersection(set(argparser.parse_args().logs.split()))
+        parser_names = set(argparser.parse_args().logs.split())
     elif config.prefs.get("logparse", "parsers"):
-        parser_names = parser_names.intersection(set(config.prefs.get("logparse", "parsers").split()))
+        parser_names = set(config.prefs.get("logparse", "parsers").split())
 
     if argparser.parse_args().ignore_logs:
-        parser_names = parser_names.difference(set(argparser.parse_args().ignore_logs.split()))
+        ignore_logs = argparser.parse_args().ignore_logs.split()
     elif config.prefs.get("logparse", "ignore-parsers"):
-        parser_names = parser_names.difference(set(config.prefs.get("logparse", "ignore-parsers").split()))
+        ignore_logs = config.prefs.get("logparse", "ignore-parsers").split()
+
+    # Set up parsers
+
+    loader = load_parsers.ParserLoader()
+    if parser_names:
+        for parser_name in parser_names:
+            if parser_name not in ignore_logs:
+                loader.search(parser_name)
+    else:
+        loader.load_pkg()
+        if ignore_logs:
+            loader.ignore(ignore_logs)
 
     # Execute parsers
 
-    logger.debug("Queued the following parsers: " + str(loader.parsers))
     for parser in loader.parsers:
-        if parser.name in parser_names:
-            output.append_section(parser.parse_log())
+        output.append_section(parser.parse_log())
+
+    # Write footer
 
-    # Write HTML 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:
             dest_path = argparser.parse_args().destination
@@ -151,6 +139,8 @@ def main():
             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):
             to = argparser.parse_args().to
@@ -164,6 +154,8 @@ def main():
             html=isinstance(output, formatting.HtmlOutput),
             sender=config.prefs.get("mail", "from"))
 
+    # Rotate logs if requested
+
     if not argparser.parse_args().no_rotate:
         if argparser.parse_args().simulate or config.prefs.getboolean("logparse", "rotate"):
             rotate_sim()
@@ -174,7 +166,8 @@ def main():
     else:
         logger.debug("User doesn't want to rotate logs")
 
-    # Print end message
+    # 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))
 
@@ -182,3 +175,42 @@ def main():
         output.print_stdout()
 
     return
+
+
+def rotate():
+    """
+    Rotate logs using systemd logrotate. This requires root privileges, and a
+    basic check for this is attempted below. Root password will be prompted
+    for if permissions are not automatically granted.
+    """
+
+    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)
+            else:
+                raise PermissionError("Root priviliges are required to run logrotate but are not provided")
+        else:
+            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:
+        logger.warning("Failed to rotate log files: " + str(e))
+
+
+def rotate_sim():   # Simulate log rotation
+    """
+    Simulate log rotation using logrotate's -d flag. This does not require root
+    privileges, but permission errors will be shown in the output without it.
+    """
+
+    try:
+        if not os.geteuid() == 0:
+            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)
+        logger.debug(sim)
+    except Exception as e:
+        logger.warning("Failed to get logrotate simulation: " + str(e))