Changed Fhir.resources to allow inclusion of custom resources

zensoup · Jan 5, 2019 · dbde90f · dbde90f
1 parent a33b512
commit dbde90f
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/fhirbug/Fhir/__init__.py b/fhirbug/Fhir/__init__.py
@@ -7,6 +7,7 @@
 import inspect
 
 from fhirbug.Fhir import resources
+from fhirbug.config import settings
 
 dir = os.path.dirname(__file__)
 for module_file in os.listdir(os.path.join(dir, 'Resources')):
@@ -18,8 +19,15 @@
         for cls_name, cls in clsmembers:
           setattr(resources, cls_name, cls)
 
-# Import extensions
-ext_module = importlib.import_module('fhirbug.Fhir.Resources.extensions')
-clsmembers = inspect.getmembers(ext_module, inspect.isclass)
+# Import internal extensions
+int_module = importlib.import_module('fhirbug.Fhir.Resources.extensions')
+clsmembers = inspect.getmembers(int_module, inspect.isclass)
 for cls_name, cls in clsmembers:
   setattr(resources, cls_name, cls)
+
+# Import external extensions
+if hasattr(settings, 'EXTENSIONS_PATH'):
+    ext_module = importlib.import_module(settings.EXTENSIONS_PATH)
+    clsmembers = inspect.getmembers(ext_module, inspect.isclass)
+    for cls_name, cls in clsmembers:
+        setattr(resources, cls_name, cls)