Create import_directory.py

import_directory.py

@@ -0,0 +1,43 @@
+import os
+import re
+import sys
+
+#---------------------------------------------------------------------------------------------------
+# Interface
+#---------------------------------------------------------------------------------------------------
+
+def do(path, env):
+    """ Imports all modules residing directly in directory "path" into the provided environment
+        (usually the callers environment). A typical call:
+        importdir.do("example_dir", globals())
+    """
+    __do(path, env)
+
+
+#---------------------------------------------------------------------------------------------------
+# Implementation
+#---------------------------------------------------------------------------------------------------
+
+# File name of a module:
+__module_file_regexp = "(.+)\.py(c?)$"
+
+def __get_module_names_in_dir(path):
+    """ Returns a set of all module names residing directly in directory "path".
+    """
+    result = set()
+
+    # Looks for all python files in the directory (not recursively) and add their name to result:
+    for entry in os.listdir(path):
+        if os.path.isfile(os.path.join(path, entry)):
+            regexp_result = re.search(__module_file_regexp, entry)
+            if regexp_result: # is a module file name
+                result.add(regexp_result.groups()[0])
+
+    return result
+
+def __do(path, env):
+    """ Implements do().
+    """
+    sys.path.append(path) # adds provided directory to list we can import from
+    for module_name in sorted(__get_module_names_in_dir(path)): # for each found module...
+        env[module_name] = __import__(module_name)              # ... import