Skip to content

Commit b48ba36

Browse files
markjmJens-G
authored andcommitted
python: add more precise typehints for immutable structs
1 parent 32776c0 commit b48ba36

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -838,6 +838,14 @@ void t_py_generator::generate_py_struct_definition(ostream& out,
838838
std::string thrift_spec_type = gen_type_hints_ ? ": typing.Any" : "";
839839
out << indent() << "thrift_spec" << thrift_spec_type << " = None" << '\n';
840840

841+
if (gen_type_hints_ && is_immutable(tstruct) && members.size() > 0) {
842+
out << '\n';
843+
for (m_iter = members.begin(); m_iter != members.end(); ++m_iter) {
844+
indent(out) << (*m_iter)->get_name()
845+
<< member_hint((*m_iter)->get_type(), (*m_iter)->get_req()) << '\n';
846+
}
847+
}
848+
841849
out << '\n';
842850

843851
/*

0 commit comments

Comments
 (0)