| # Copyright 2021 The Chromium Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import re |
| from style_variable_generator.css_generator import CSSStyleGenerator |
| |
| class TSStyleGenerator(CSSStyleGenerator): |
| '''Generator for TS Variables''' |
| |
| @staticmethod |
| def GetName(): |
| return 'TS' |
| |
| def Render(self): |
| return self.ApplyTemplate(self, 'templates/ts_generator.tmpl', |
| self.GetParameters()) |
| |
| def GetParameters(self): |
| params = CSSStyleGenerator.GetParameters(self) |
| params['include_style_sheet'] = self.generator_options.get( |
| 'include_style_sheet', 'false') == 'true' |
| return params |
| |
| def GetFilters(self): |
| filters = CSSStyleGenerator.GetFilters(self) |
| filters['to_ts_var_name'] = self.ToTSVarName |
| return filters |
| |
| def ToTSVarName(self, model_name): |
| return re.sub(r'[\.\-]', '_', model_name.upper()) |
