add parsers for memory info and system info
[logparse.git] / logparse / parsers / load_parsers.py
index f2821106c6456d8a03d112adab1725f5624ec6e5..0dc291ad4db3800af2b35c385d99dce77315d9f6 100644 (file)
@@ -5,29 +5,50 @@
 #
 
 import imp
+import importlib
 import os
 import glob
+from pathlib import Path
+from sys import path
+from typing import NamedTuple
 
 parser_dir = "/usr/share/logparse/"
-main_module = "__init__.py"
+main_module = "__init__"
+default_parsers = ["cron", "httpd", "mem", "postfix", "smbd", "sshd", "sudo", "sysinfo", "temperature", "zfs"]
 
 import logging
 logger = logging.getLogger(__name__)
 
-def search():
+class Parser():
+    def __init__(self, name, path=None, info=None):
+        self.name = str(name)
+        self.path = Path(path) if path else None
+        self.info = dict(info) if info else None
+
+def findall():
     logger.debug("Searching for parsers in {0}".format(parser_dir))
+    path.append(os.path.abspath(parser_dir))
     parsers = []
-    parser_candidates = glob.glob(os.path.join(os.path.dirname(parser_dir), "*"))
-    logger.debug("Found parser candidates {0}".format(str(parser_candidates)))
-    for p in parser_candidates:
-        location = os.path.join(parser_dir, p)
-        if not os.path.isdir(parser_dir) or not main_module + '.py' in os.listdir(location):
+    parser_candidates = os.listdir(parser_dir)
+    for parser_name in parser_candidates:
+        location = os.path.join(parser_dir, parser_name)
+        if not os.path.isdir(location) or not main_module + '.py' in os.listdir(location):
+            logger.warning("Rejecting parser {0} due to invalid structure".format(location))
             continue
         info = imp.find_module(main_module, [location])
-        parsers.append({"name": p, "info": info})
+        parser_obj = Parser(parser_name, location, info)
+        parsers.append(parser_obj)
+        logger.debug("Added parser {0}".format(parser_obj.name))
     return parsers
 
-def load(parser):
-    logger.debug("Loading {0}".format(parser["name"]))
-    return imp.load_module(parser["name"], *parser["info"])
+def search(name):
+    logger.debug("Searching for parser {0}".format(name))
+    if name in default_parsers:
+        logger.debug("Found parser {0} in default modules".format(name))
+        return Parser('.'.join(__name__.split('.')[:-1] + [name]))
+    else:
+        return None
 
+def load(parser):
+    logger.debug("Loading parser {0} from {1}".format(parser.name, parser.path if parser.path != None else "defaults"))
+    return importlib.import_module(parser.name)