From b821d848a58c40bfe26262352cd66aa6eeac5d4a Mon Sep 17 00:00:00 2001 From: "Jasper St. Pierre" Date: Tue, 28 Aug 2012 20:45:22 -0300 Subject: mallardwriter: Use a more generic system for language formatting --- giscanner/mallardwriter.py | 23 ++++++++++++++++------- 1 file changed, 16 insertions(+), 7 deletions(-) diff --git a/giscanner/mallardwriter.py b/giscanner/mallardwriter.py index 585713a2..876bd872 100644 --- a/giscanner/mallardwriter.py +++ b/giscanner/mallardwriter.py @@ -127,6 +127,8 @@ class MallardFormatter(object): return parent_chain class MallardFormatterC(MallardFormatter): + language = "C" + def format_type(self, type_): if isinstance(type_, ast.Array): try: @@ -139,6 +141,8 @@ class MallardFormatterC(MallardFormatter): return type_.target_fundamental class MallardFormatterPython(MallardFormatter): + language = "Python" + def format_type(self, type_): if isinstance(type_, ast.Array): return '[' + self.format_type(type_.element_type) + ']' @@ -157,17 +161,22 @@ class MallardFormatterPython(MallardFormatter): doc = doc.replace('%FALSE', 'False') return doc +LANGUAGES = { + "c": MallardFormatterC, + "python": MallardFormatterPython, +} + class MallardWriter(object): def __init__(self, transformer, language): self._transformer = transformer - self._language = language - if self._language == 'C': - self._formatter = MallardFormatterC(self._transformer) - elif self._language == 'Python': - self._formatter = MallardFormatterPython(self._transformer) - else: - raise SystemExit("Unsupported language: %s" % language) + try: + formatter_class = LANGUAGES[language.lower()] + except KeyError: + raise SystemExit("Unsupported language: %s" % (language,)) + + self._formatter = formatter_class(self._transformer) + self._language = self._formatter.language def write(self, output): nodes = [self._transformer.namespace] -- cgit v1.2.1