diff --git a/pygrok/pygrok.py b/pygrok/pygrok.py index 142d122..12398b0 100644 --- a/pygrok/pygrok.py +++ b/pygrok/pygrok.py @@ -14,9 +14,17 @@ class Grok(object): def __init__(self, pattern, custom_patterns_dir=None, custom_patterns={}, fullmatch=True): self.pattern = pattern self.custom_patterns_dir = custom_patterns_dir - self.predefined_patterns = _reload_patterns(DEFAULT_PATTERNS_DIRS) + + #Adding PySpark support in case using library as zip file during spark-submit command. + try: + self.predefined_patterns = _reload_patterns(DEFAULT_PATTERNS_DIRS) + except NotADirectoryError: + DEFAULT_PATTERNS_DIRS = ['.' + '/patterns'] + self.fullmatch = fullmatch + + custom_pats = {} if custom_patterns_dir is not None: custom_pats = _reload_patterns([custom_patterns_dir])