From eb0832e3795e751b64f813f11b3a3ac03631b0bc Mon Sep 17 00:00:00 2001 From: Todd Gamblin Date: Mon, 13 May 2013 10:35:38 -0700 Subject: list_modules will now find directories. --- lib/spack/spack/utils.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) (limited to 'lib') diff --git a/lib/spack/spack/utils.py b/lib/spack/spack/utils.py index 465c83caf9..2781262b5a 100644 --- a/lib/spack/spack/utils.py +++ b/lib/spack/spack/utils.py @@ -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) -- cgit v1.2.3-70-g09d2