Function removeRequiredAnyDefaultsForPython

scripts/codegen/python.ts:1142–1178

(
    code: string,
    definitions: Record<string, JSONSchema7>,
    definitionCollections: DefinitionCollections
)

}

function removeRequiredAnyDefaultsForPython(
    code: string,
    definitions: Record<string, JSONSchema7>,
    definitionCollections: DefinitionCollections
): string {
    const requiredFieldsByClass = new Map<string, Set<string>>();

    for (const [definitionName, schema] of Object.entries(definitions)) {
        const resolved = resolveObjectSchema(schema, definitionCollections) ?? resolveSchema(schema, definitionCollections);
        if (!resolved || !isObjectSchema(resolved) || !resolved.properties || !Array.isArray(resolved.required)) {
            continue;
        }

        const requiredFields = resolved.required.map(toSnakeCase);
        for (const className of new Set([definitionName, toPascalCase(definitionName)])) {
            const fields = requiredFieldsByClass.get(className) ?? new Set<string>();
            for (const field of requiredFields) {
                fields.add(field);
            }
            requiredFieldsByClass.set(className, fields);
        }
    }

    const classBlockRe = /(@dataclass\r?\nclass\s+(\w+):[\s\S]*?)(?=^@dataclass|^class\s+\w|^def\s+\w|\Z)/gm;
    return code.replace(classBlockRe, (block: string, _classPrefix: string, className: string) => {
        const requiredFields = requiredFieldsByClass.get(className);
        if (!requiredFields) {
            return block;
        }

        let updatedBlock = block;
        for (const field of requiredFields) {
            updatedBlock = updatedBlock.replace(new RegExp(`^(    ${escapeRegExp(field)}: Any) = None$`, "m"), "$1");
        }
        return updatedBlock;
    });
}

/**
 * Remove locally-emitted Enum class definitions whose name already comes from