hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

list_modules will now find directories.

Browse source
This commit is contained in:
Todd Gamblin 2013-05-13 10:35:38 -07:00
parent b2f78ef78e
commit eb0832e379
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
lib/spack/spack/utils.py
View file

@ -1,7 +1,6 @@
import os
import re
import errno
import glob
import shutil
import subprocess
import multiprocessing
@ -39,9 +38,17 @@ def install(src, dest):
def list_modules(directory):
"""Lists all of the modules, excluding __init__.py, in
a particular directory."""
os.chdir(directory)
for name in glob.glob("*.py"):
if name != '__init__.py':
for name in os.listdir(directory):
if name == '__init__.py':
continue
path = new_path(directory, name)
if os.path.isdir(path):
init_py = new_path(path, '__init__.py')
if os.path.isfile(init_py):
yield name
elif name.endswith('.py'):
yield re.sub('.py$', '', name)