commit 5d3b5a039bf94a2c1744b70e3f0d9dd249ce393e
author arkuhn <contact@adamrkuhn.com> Sat Jun 22 08:42:44 2024 -0400
committer Jens Geyer <Jens-G@users.noreply.github.com> Wed Aug 21 22:52:29 2024 +0200
tree 5c6164d162f75a0e061006a4ce207132fde1d24a
parent 7de26c45443bc1642f90d5365bf8eef15e239ce5

Generate thrift_spec class attribute (with type hints)

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

diff --git a/compiler/cpp/src/thrift/generate/t_py_generator.cc b/compiler/cpp/src/thrift/generate/t_py_generator.cc
index 7b5c54d..893d635 100644
--- a/compiler/cpp/src/thrift/generate/t_py_generator.cc
+++ b/compiler/cpp/src/thrift/generate/t_py_generator.cc
@@ -831,6 +831,8 @@
   out << ":" << '\n';
   indent_up();
   generate_python_docstring(out, tstruct);
+  std::string thrift_spec_type = gen_type_hints_ ? ": typing.Any" : "";
+  out << indent() << "thrift_spec" << thrift_spec_type << " = None" << '\n';
 
   out << '\n';