bplaxco · September 25, 2015 15:17 · Sep 25, 2015
diff --git a/__init__.py b/__init__.py
@@ -0,0 +1,22 @@
+# Import classes from files in the sub directory
+# so files in models/ClassName.py with ClassName in it
+# can be imported as models.ClassName
+
+import pkgutil
+
+__all__ = []
+
+for _, name, ispkg in pkgutil.walk_packages(__path__):
+
+    if not name.startswith('__') and not ispkg:
+        # Ex: apps.assessment.models.<name>
+        full_name = '.'.join([__name__, name])
+
+        # Get the module that is holding the class
+        module = __import__(full_name, fromlist=[name])
+
+        # Get the class from the module and bring it into the global name space
+        globals()[name] = getattr(module, name)
+
+        # Add it to the __all__ array for * imports
+        __all__.append(name)