| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """Generates Python source files from a mojom.Module.""" | 5 """Generates Python source files from a mojom.Module.""" |
| 6 | 6 |
| 7 import re | 7 import re |
| 8 from itertools import ifilter | 8 from itertools import ifilter |
| 9 | 9 |
| 10 import mojom.generate.generator as generator | 10 import mojom.generate.generator as generator |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 uccc = UpperCamelCase(name) | 28 uccc = UpperCamelCase(name) |
| 29 return uccc[0].lower() + uccc[1:] | 29 return uccc[0].lower() + uccc[1:] |
| 30 | 30 |
| 31 def ConstantStyle(name): | 31 def ConstantStyle(name): |
| 32 components = NameToComponent(name) | 32 components = NameToComponent(name) |
| 33 if components[0] == 'k': | 33 if components[0] == 'k': |
| 34 components = components[1:] | 34 components = components[1:] |
| 35 return '_'.join([x.upper() for x in components]) | 35 return '_'.join([x.upper() for x in components]) |
| 36 | 36 |
| 37 def GetNameForElement(element): | 37 def GetNameForElement(element): |
| 38 if (mojom.IsEnumKind(element) or mojom.IsInterfaceKind(element) or |
| 39 mojom.IsStructKind(element)): |
| 40 return UpperCamelCase(element.name) |
| 38 if isinstance(element, mojom.EnumValue): | 41 if isinstance(element, mojom.EnumValue): |
| 39 return (GetNameForElement(element.enum) + '.' + | 42 return (GetNameForElement(element.enum) + '.' + |
| 40 ConstantStyle(element.name)) | 43 ConstantStyle(element.name)) |
| 41 if isinstance(element, (mojom.NamedValue, | 44 if isinstance(element, (mojom.NamedValue, |
| 42 mojom.Constant)): | 45 mojom.Constant)): |
| 43 return ConstantStyle(element.name) | 46 return ConstantStyle(element.name) |
| 44 raise Exception('Unexpected element: ' % element) | 47 raise Exception('Unexpected element: ' % element) |
| 45 | 48 |
| 46 def ExpressionToText(token): | 49 def ExpressionToText(token): |
| 47 if isinstance(token, (mojom.EnumValue, mojom.NamedValue)): | 50 if isinstance(token, (mojom.EnumValue, mojom.NamedValue)): |
| (...skipping 74 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 122 'expression_to_text': ExpressionToText, | 125 'expression_to_text': ExpressionToText, |
| 123 'name': GetNameForElement, | 126 'name': GetNameForElement, |
| 124 } | 127 } |
| 125 | 128 |
| 126 @UseJinja('python_templates/module.py.tmpl', filters=python_filters) | 129 @UseJinja('python_templates/module.py.tmpl', filters=python_filters) |
| 127 def GeneratePythonModule(self): | 130 def GeneratePythonModule(self): |
| 128 return { | 131 return { |
| 129 'imports': self.GetImports(), | 132 'imports': self.GetImports(), |
| 130 'enums': self.module.enums, | 133 'enums': self.module.enums, |
| 131 'module': ComputeConstantValues(self.module), | 134 'module': ComputeConstantValues(self.module), |
| 135 'structs': self.GetStructs(), |
| 132 } | 136 } |
| 133 | 137 |
| 134 def GenerateFiles(self, args): | 138 def GenerateFiles(self, args): |
| 135 self.Write(self.GeneratePythonModule(), | 139 self.Write(self.GeneratePythonModule(), |
| 136 '%s.py' % self.module.name.replace('.mojom', '_mojom')) | 140 '%s.py' % self.module.name.replace('.mojom', '_mojom')) |
| 137 | 141 |
| 138 def GetImports(self): | 142 def GetImports(self): |
| 139 for each in self.module.imports: | 143 for each in self.module.imports: |
| 140 each['python_module'] = each['module_name'].replace('.mojom', '_mojom') | 144 each['python_module'] = each['module_name'].replace('.mojom', '_mojom') |
| 141 return self.module.imports | 145 return self.module.imports |
| 142 | 146 |
| 143 def GetJinjaParameters(self): | 147 def GetJinjaParameters(self): |
| 144 return { | 148 return { |
| 145 'lstrip_blocks': True, | 149 'lstrip_blocks': True, |
| 146 'trim_blocks': True, | 150 'trim_blocks': True, |
| 147 } | 151 } |
| OLD | NEW |