protoc-event-support.patch 912 B

12345678910111213141516171819202122
  1. --- a/src/google/protobuf/compiler/python/helpers.cc (62b83ba8ff1149a180456030d46ca0881711b0e5)
  2. +++ b/src/google/protobuf/compiler/python/helpers.cc (94b865765aa38dc5e4ac99e0051d228ebc9d2a80)
  3. @@ -50,9 +50,18 @@ namespace protobuf {
  4. namespace compiler {
  5. namespace python {
  6. +TProtoStringType FixEv(TProtoStringType filename) {
  7. + constexpr auto kSuffixLen = 3;
  8. + if (filename.EndsWith(".ev")) {
  9. + return filename.substr(0, filename.length() - kSuffixLen) + "_ev.proto";
  10. + }
  11. + return filename;
  12. +}
  13. +
  14. // Returns the Python module name expected for a given .proto filename.
  15. TProtoStringType ModuleName(absl::string_view filename) {
  16. - TProtoStringType basename = StripProto(filename);
  17. + TProtoStringType str(std::string{filename});
  18. + TProtoStringType basename = StripProto(FixEv(str));
  19. absl::StrReplaceAll({{"-", "_"}, {"/", "."}}, &basename);
  20. return absl::StrCat(basename, "_pb2");
  21. }