def get_files()

in utils/io.py [0:0]

• 19 lines of code
• 8 McCabe index (conditional complexity)

def get_files(path, formats=None):
    """Get files contained in path

    Arguments:
        path {str} -- path

    Keyword Arguments:
        formats {str/list} -- file formats; if None take all (default: {None})

    Returns:
        list -- file names
        list -- file paths
    """

    if formats:

        if isinstance(formats, str):
            formats = [formats]
        else:
            assert isinstance(formats, list)

        names = []
        paths = []

        for f_format in formats:
            files = [f for f in os.listdir(path)
                     if re.match(r'.*\.{}'.format(f_format), f)]

            files.sort()
            names.extend(files)
            paths.extend([os.path.join(path, f) for f in files])

        return names, paths

    names = os.listdir(path)
    names.sort()
    paths = [os.path.join(path, f) for f in names]

    return names, paths