Function main

scripts/generate_sre_constants.py:86–133

(
    infile="Lib/re/_constants.py",
    outfile_constants="crates/sre_engine/src/constants.rs",
)

def main(
    infile="Lib/re/_constants.py",
    outfile_constants="crates/sre_engine/src/constants.rs",
):
    ns = {}
    with open(infile) as fp:
        code = fp.read()
    exec(code, ns)

    content = [sre_constants_header]
    content.append("use bitflags::bitflags;\n\n")
    content.append(f"pub const SRE_MAGIC: usize = {ns['MAGIC']};\n")
    content.extend(
        dump_enum(
            ns["OPCODES"],
            "SreOpcode",
            "#[derive(num_enum::TryFromPrimitive, Debug, PartialEq, Eq)]",
        )
    )
    content.extend(
        dump_enum(
            ns["ATCODES"],
            "SreAtCode",
            "#[derive(num_enum::TryFromPrimitive, Debug, PartialEq, Eq)]",
            "AT_",
        )
    )
    content.extend(
        dump_enum(
            ns["CHCODES"],
            "SreCatCode",
            "#[derive(num_enum::TryFromPrimitive, Debug)]",
            "CATEGORY_",
        )
    )

    content.extend(
        dump_bitflags(
            ns,
            "SRE_FLAG_",
            "#[derive(Debug, PartialEq, Eq, Clone, Copy)]",
            "SreFlag",
            "u16",
        )
    )
    content.extend(dump_bitflags(ns, "SRE_INFO_", "", "SreInfo", "u32"))

    update_file(outfile_constants, "".join(content))


if __name__ == "__main__":