-
Notifications
You must be signed in to change notification settings - Fork 0
/
importdir.py
44 lines (34 loc) · 1.4 KB
/
importdir.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import os
import re
import sys
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 each found module...
for module_name in sorted(__get_module_names_in_dir(path)):
env[module_name] = __import__(module_name) # ... import
env[module_name] = __import__(module_name) # ... import