From 0569b99f300405a8cb53fb26dbeab4e9c9c86f76 Mon Sep 17 00:00:00 2001
From: daniel_sagi <daniel.sagi@aquasec.com>
Date: Thu, 19 Jul 2018 14:52:58 +0300
Subject: [PATCH] added automatic imports for report package

---
 kube-hunter.py     | 2 +-
 report/__init__.py | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/kube-hunter.py b/kube-hunter.py
index 5c786d0..a7703a6 100644
--- a/kube-hunter.py
+++ b/kube-hunter.py
@@ -22,7 +22,7 @@ except:
 if config.log.lower() != "none":
     logging.basicConfig(level=loglevel, format='%(message)s', datefmt='%H:%M:%S')
 
-from report import default
+import report
 
 from src.core.events import handler
 from src.core.events.types import HuntFinished, HuntStarted
diff --git a/report/__init__.py b/report/__init__.py
index e69de29..e1e1462 100644
--- a/report/__init__.py
+++ b/report/__init__.py
@@ -0,0 +1,7 @@
+from os.path import dirname, basename, isfile
+import glob
+
+# dynamically importing all modules in folder
+files = glob.glob(dirname(__file__)+"/*.py")
+for module_name in (basename(f)[:-3] for f in files if isfile(f) and not f.endswith('__init__.py')):
+    exec('from {} import *'.format(module_name))
\ No newline at end of file