Function ensureClassVarImport

scripts/codegen/python.ts:667–679

(code: string)

}

function ensureClassVarImport(code: string): string {
    // Already imported?
    if (/\bfrom typing import [^\n]*\bClassVar\b/.test(code)) return code;
    return code.replace(
        /^from typing import (.+)$/m,
        (_match, names) => {
            const list = names.split(",").map((n: string) => n.trim()).filter(Boolean);
            list.push("ClassVar");
            list.sort();
            return `from typing import ${[...new Set(list)].join(", ")}`;
        }
    );
}

function pushPyExperimentalComment(lines: string[], subject: PyExperimentalSubject, indent = ""): void {
    lines.push(pyExperimentalComment(subject, indent));