File size: 1,398 Bytes
26225c5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import os
import os.path as osp
import pyrootutils
__all__ = ['get_config_structure']
def get_config_structure(start_directory=None, indent=0, verbose=False):
"""Parse a config file structure in search for .yaml files
"""
# If not provided, search the project configs directory
if start_directory is None:
root = str(pyrootutils.setup_root(
search_from='',
indicator=[".git", "README.md"],
pythonpath=True,
dotenv=True))
start_directory = osp.join(root, 'configs')
# Structure to store the file hierarchy:
# - first value is a dictionary of directories
# - second value is a list of yaml files
struct = ({}, [])
# Recursively gather files and directories in the current directory
for item in os.listdir(start_directory):
item_path = os.path.join(start_directory, item)
if os.path.isdir(item_path):
if verbose:
print(f"{' ' * indent}Directory: {item}")
struct[0][item] = get_config_structure(
start_directory=item_path, indent=indent + 1)
elif os.path.isfile(item_path):
filename, extension = osp.splitext(item)
if extension == '.yaml':
struct[1].append(filename)
if verbose:
print(f"{' ' * indent}File: {item}")
return struct
|