如何在从自定义生成器写入文件时控制文本缩进

def generate_data(data, indent_level=0, indent_size=4):
    indent = ' ' * (indent_level * indent_size)
    if isinstance(data, dict):
        for key, value in data.items():
            yield f'{indent}{key}:'
            yield from generate_data(value, indent_level + 1, indent_size)
    elif isinstance(data, list):
        for item in data:
            yield f'{indent}-'
            yield from generate_data(item, indent_level + 1, indent_size)
    else:
        yield f'{indent}{data}'

def write_to_file(data, file_path):
    with open(file_path, 'w') as file:
        for line in generate_data(data):
            file.write(line + '\n')

# 示例数据
data = {
    'name': 'John',
    'age': 30,
    'languages': ['Python', 'JavaScript', 'Java'],
    'address': {
        'street': '123 Main St',
        'city': 'New York',
        'country': 'USA'
    }
}

# 写入文件
write_to_file(data, 'output.txt')

name: John
age: 30
languages:
    - Python
    - JavaScript
    - Java
address:
    street: 123 Main St
    city: New York
    country: USA