Index: mojo/public/tools/bindings/generators/mojom_python_generator.py |
diff --git a/mojo/public/tools/bindings/generators/mojom_python_generator.py b/mojo/public/tools/bindings/generators/mojom_python_generator.py |
index 37d39589ad929ff19540e089417ea21924e134ea..fe0b5117bb33358bbf4da1dfa9ce5bff2ecacc3a 100644 |
--- a/mojo/public/tools/bindings/generators/mojom_python_generator.py |
+++ b/mojo/public/tools/bindings/generators/mojom_python_generator.py |
@@ -34,6 +34,9 @@ def ConstantStyle(name): |
return '_'.join([x.upper() for x in components]) |
def GetNameForElement(element): |
+ if (mojom.IsEnumKind(element) or mojom.IsInterfaceKind(element) or |
+ mojom.IsStructKind(element)): |
+ return UpperCamelCase(element.name) |
if isinstance(element, mojom.EnumValue): |
return (UpperCamelCase(element.enum_name) + '.' + |
ConstantStyle(element.name)) |
@@ -82,6 +85,7 @@ class Generator(generator.Generator): |
'imports': self.GetImports(), |
'enums': self.module.enums, |
'module': self.module, |
+ 'structs': self.GetStructs(), |
} |
def GenerateFiles(self, args): |