Add exception handling for module discovery

2019-12-15 07:32:28 +01:00 · 2019-12-15 07:32:28 +01:00 · 1fbca4a652
commit 1fbca4a652
parent 440d72896f
1 changed files with 5 additions and 2 deletions
--- a/main.py
+++ b/main.py
@ -5,8 +5,11 @@ import os
 print("Loading modules")
 modules = {}
 for folder in os.listdir("./modules"):
-    curr_module = importlib.import_module('.' + folder, package="modules")
-    modules[folder] = curr_module.get_module()
+    try:
+        curr_module = importlib.import_module('.' + folder, package="modules")
+        modules[folder] = curr_module.get_module()
+    except AttributeError:
+        pass
 print("Modules loaded successfully.\n")

 print("Available Modules:")