diff --git a/sparklogstats/__init__.py b/sparklogstats/__init__.py index e69de29..39cd29e 100644 --- a/sparklogstats/__init__.py +++ b/sparklogstats/__init__.py @@ -0,0 +1,5 @@ +"""Main classes for the user.""" +from .logparser import LogParser + + +__all__ = ('LogParser',) diff --git a/sparklogstats/logparser.py b/sparklogstats/logparser.py index 5743b78..8900bc0 100644 --- a/sparklogstats/logparser.py +++ b/sparklogstats/logparser.py @@ -1,3 +1,4 @@ +import glob from json import loads @@ -9,6 +10,10 @@ def __init__(self): def _reset(self): self.app = Application() + def parse_folder(self, folder): + logs = sorted(glob.glob(folder + '/app-*')) + return (self.parse_file(log) for log in logs) + def parse_file(self, filename): self._reset() with open(filename) as file: