Added parse_folder

cemsbr · Sep 6, 2016 · 845c022 · 845c022
1 parent fe8a0c0
commit 845c022
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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
@@ -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: