diff --git a/tldp/sources.py b/tldp/sources.py index 5016268..cfca999 100644 --- a/tldp/sources.py +++ b/tldp/sources.py @@ -59,13 +59,9 @@ def scansourcedirs(dirnames): for sdir in sorted(dirs): for fname in sorted(os.listdir(sdir)): candidates = list() - possible = os.path.join(sdir, fname) - if os.path.isfile(possible): + possible = arg_issourcedoc(os.path.join(sdir, fname)) + if possible: candidates.append(SourceDocument(possible)) - elif os.path.isdir(possible): - possible = sourcedoc_fromdir(possible) - if possible: - candidates.append(SourceDocument(possible)) else: logger.warning("Skipping non-directory, non-plain file %s", possible) @@ -81,6 +77,14 @@ def scansourcedirs(dirnames): return found +def arg_issourcedoc(filename): + if os.path.isfile(filename): + return filename + elif os.path.isdir(filename): + return sourcedoc_fromdir(filename) + return None + + def sourcedoc_fromdir(dirname): candidates = list() stem, _ = stem_and_ext(dirname)