Merge pull request #5 from baikonur-oss/empty-whitelist-bug

Fix empty whitelist case
baikonur-oss · Aug 5, 2019 · be20b5b · be20b5b
2 parents 8743801 + 20f9df3
commit be20b5b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lambda/main.py b/lambda/main.py
@@ -45,9 +45,14 @@
 
 LOG_TYPE_FIELD: str = os.environ['LOG_TYPE_FIELD']
 LOG_TIMESTAMP_FIELD: str = os.environ['LOG_TIMESTAMP_FIELD']
-LOG_TYPE_FIELD_WHITELIST: set = set(str(os.environ['LOG_TYPE_WHITELIST']).split(','))
 LOG_TYPE_UNKNOWN_PREFIX: str = os.environ['LOG_TYPE_UNKNOWN_PREFIX']
 
+WHITELIST: str = str(os.environ['LOG_TYPE_WHITELIST'])
+if WHITELIST == "":
+    LOG_TYPE_FIELD_WHITELIST: set = set()
+else:
+    LOG_TYPE_FIELD_WHITELIST: set = set(WHITELIST.split(','))
+
 ELASTICSEARCH_HOST = os.environ['ES_HOST']
 INDEX_NAME_PREFIX = os.environ['INDEX_NAME_PREFIX']