BigfootDev/flatbuffers
1
0
Fork 1
mirror of https://github.com/google/flatbuffers.git synced 2026-06-03 04:21:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Python] Render enums as Python IntEnum (#8145)

Browse Source 
This allows enums to be type check with mypy.
They will still behave like ints ->
> IntEnum is the same as Enum,
> but its members are also integers and can be used anywhere
> that an integer can be used.
> If any integer operation is performed with an IntEnum member,
> the resulting value loses its enumeration status.
https://docs.python.org/3/library/enum.html#enum.IntEnum

Only if the --python-typing flag is set.
This commit is contained in:
Felix
2024-06-03 17:39:14 +02:00
committed by GitHub
parent 6ede1ccc9e
commit dafd2f1f29
6 changed files with 31 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/idl_gen_python.cpp
View File

@@ -591,12 +591,21 @@ class PythonStubGenerator {
void GenerateEnumStub(std::stringstream &stub, const EnumDef *enum_def,
Imports *imports) const {
stub << "class " << namer_.Type(*enum_def);
if (version_.major != 3) stub << "(object)";
if (version_.major == 3){
imports->Import("enum", "IntEnum");
stub << "(IntEnum)";
}
else {
stub << "(object)";
}
stub << ":\n";
for (const EnumVal *val : enum_def->Vals()) {
stub << " " << namer_.Variant(*val) << ": "
<< ScalarType(enum_def->underlying_type.base_type) << "\n";
}
if (parser_.opts.generate_object_based_api & enum_def->is_union) {
imports->Import("flatbuffers", "table");
stub << "def " << namer_.Function(*enum_def)
@@ -2432,7 +2441,7 @@ class PythonGenerator : public BaseGenerator {
auto field_type = namer_.ObjectType(*ev.union_type.struct_def);
code +=
GenIndents(1) + "if unionType == " + union_type + "()." + variant + ":";
GenIndents(1) + "if unionType == " + union_type + "." + variant + ":";
if (parser_.opts.include_dependence_headers) {
auto package_reference = GenPackageReference(ev.union_type);
code += GenIndents(2) + "import " + package_reference;