Skip to content

Commit 167c6e9

Browse files
committed
THRIFT-5885 Fix enum generation
- Also an issue for latest builds on later windows not only the appveyor builds like originally thoughts - Used the code as suggested from the ticket as-is without too much understanding
1 parent 1e4e3b3 commit 167c6e9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

compiler/cpp/src/thrift/generate/t_py_generator.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -984,8 +984,8 @@ void t_py_generator::generate_py_struct_definition(ostream& out,
984984
t_type* type = (*m_iter)->get_type();
985985
if (type->is_enum()) {
986986
out << indent() << "if name == \"" << (*m_iter)->get_name() << "\":" << '\n'
987-
<< indent() << indent_str() << "super().__setattr__(name, value if hasattr(value, 'value') else "
988-
<< type_name(type) << ".__members__.get(value))" << '\n'
987+
<< indent() << indent_str() << "super().__setattr__(name, value if hasattr(value, 'value') or value is None else "
988+
<< type_name(type) << "(value))" << '\n'
989989
<< indent() << indent_str() << "return" << '\n';
990990
}
991991
}

0 commit comments

Comments
 (0)