diff options
Diffstat (limited to 'ybd/utils.py')
| -rw-r--r-- | ybd/utils.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/ybd/utils.py b/ybd/utils.py index e6fd5a6..89648c8 100644 --- a/ybd/utils.py +++ b/ybd/utils.py @@ -375,12 +375,10 @@ def _find_extensions(paths): map(lambda x: efs.add_fs(x, OSFS(x)), paths) def get_extensions(kind): - return {str(os.path.splitext(x)[0][1:]): str(efs.getsyspath(x)) + return {os.path.splitext(x)[0][1:]: efs.getsyspath(x) for x in efs.walk.files(filter=['*.%s' % kind])} - a = {e: get_extensions(e) for e in extension_kinds} - print 'Extensions found: %s' % a - return a + return {e: get_extensions(e) for e in extension_kinds} def find_extensions(): |