From 845c0221aea4902d7d65fb24cd0bf13280291930 Mon Sep 17 00:00:00 2001 From: Carlos Eduardo Moreira dos Santos Date: Tue, 6 Sep 2016 01:34:12 -0300 Subject: [PATCH] Added parse_folder --- sparklogstats/__init__.py | 5 +++++ sparklogstats/logparser.py | 5 +++++ 2 files changed, 10 insertions(+) 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: