yaml_to_dir.py

import os
import yaml
import json
import argparse


def create_folder_structure(base_dir, classification_dict, level=0, max_levels=10):
    if level > max_levels:
        return

    # Process the current level of the dictionary
    for key, value in classification_dict.items():
        # print(key, value)
        if value is None:
            print(key, "Value is None")
            continue  # Skip if value is None

        # Check if the value is a dictionary, which means we have a class with subclasses
        if isinstance(value, dict):
            # Retrieve the 'name' key if present, otherwise use an empty string
            dir_name = f"{key} {value.get('name', '')}".strip()
            current_dir = os.path.join(base_dir, dir_name)
            os.makedirs(current_dir, exist_ok=True)
            print(f"Created directory DICT: {current_dir}")

            # If there are subclasses, recursively process them
            subclasses = value.get('subclasses', [])
            # print(subclasses)
            if isinstance(subclasses, list):
                # print("LIST")
                for subclass in subclasses:
                    # Each subclass is a dictionary; get its key and value
                    if subclass:
                        # print(subclass)
                        subclass_key, subclass_value = list(subclass.items())[0]
                        # Call the function recursively with the subclass dictionary
                        create_folder_structure(current_dir, {subclass_key: subclass_value}, level + 1, max_levels)

        # If the value is a string, we have a subclass entry
        elif isinstance(value, str):
            dir_name = f"{key} {value}".strip()
            current_dir = os.path.join(base_dir, dir_name)
            os.makedirs(current_dir, exist_ok=True)
            print(f"Created directory STRING: {current_dir}")
            

def main():
    # Set up argument parsing
    parser = argparse.ArgumentParser(
        description="Create folder structure from a YAML file."
    )
    parser.add_argument("file", type=str, help="Path to the YAML file")
    parser.add_argument(
        "output_dir",
        type=str,
        help="Base output directory where folders will be created",
    )

    # Parse arguments
    args = parser.parse_args()
    
    
    if args.file.endswith("yaml"):
        with open(args.file, "r", encoding="utf8") as file:
            data = yaml.safe_load(file)
            # print(json.dumps(data, indent=4))
        create_folder_structure(args.output_dir, data)
        
    
    elif args.file.endswith("json"):
        with open(args.file, "r", encoding="utf-8") as file:
            data = json.load(file)
            # print(data)
        create_folder_structure(args.output_dir, data)
        
    
    else:
        print("No valid file was provided")

    # Create the folder structure


if __name__ == "__main__":
    main()