diff options
Diffstat (limited to 'src/pyssg/discovery.py')
| -rw-r--r-- | src/pyssg/discovery.py | 40 |
1 files changed, 31 insertions, 9 deletions
diff --git a/src/pyssg/discovery.py b/src/pyssg/discovery.py index a99f8bb..041ad64 100644 --- a/src/pyssg/discovery.py +++ b/src/pyssg/discovery.py @@ -1,31 +1,53 @@ import os +import logging +from logging import Logger +log: Logger = logging.getLogger(__name__) -def get_file_list(directory: str, - extensions: list[str], + +def get_file_list(path: str, + exts: list[str], exclude: list[str]=None) -> list[str]: + log.debug('retrieving file list in path "%s" that contain file' + ' extensions (%s) except (%s)', + path, ', '.join(exts), + ', '.join(exclude if exclude is not None else [])) out: list[str] = [] - for root, dirs, files in os.walk(directory): + for root, dirs, files in os.walk(path): if exclude is not None: + log.debug('removing excludes from list') dirs[:] = [d for d in dirs if d not in exclude] for f in files: - if f.endswith(tuple(extensions)): - out.append(os.path.join(root, f).replace(directory, '')[1:]) + if f.endswith(tuple(exts)): + stripped_f: str = os.path.join(root, f).replace(path, '')[1:] + out.append(stripped_f) + log.debug('added file "%s" without "%s" part: "%s"', + f, path, stripped_f) + else: + log.debug('ignoring file "%s" as it doesn\'t contain' + ' any of the extensions (%s)', f, ', '.join(exts)) return out -def get_dir_structure(directory: str, +def get_dir_structure(path: str, exclude: list[str]=None) -> list[str]: + log.debug('retrieving dir structure in path "%s" except (%s)', + path, ', '.join(exclude if exclude is not None else [])) out: list[str] = [] - for root, dirs, files in os.walk(directory): + for root, dirs, files in os.walk(path): if exclude is not None: + log.debug('removing excludes from list') dirs[:] = [d for d in dirs if d not in exclude] for d in dirs: if root in out: out.remove(root) - out.append(os.path.join(root, d)) + log.debug('removed dir "%s" as it already is in the list', root) + joined_dir: str = os.path.join(root, d) + out.append(joined_dir) + log.debug('added dir "%s" to the list', joined_dir) - return [o.replace(directory, '')[1:] for o in out] + log.debug('removing "%s" from all dirs in list', path) + return [o.replace(path, '')[1:] for o in out] |