Converter class

1 files changed, 22 insertions, 0 deletions

diff --git a/dealconvert/__init__.py b/dealconvert/__init__.py
new file mode 100644
index 0000000..f613fac
--- /dev/null
+++ b/dealconvert/__init__.py
@@ -0,0 +1,22 @@
+from .formats import *
+
+class DealConverter(object):
+    def __init__(self, input_file):
+        self.input = input_file
+        self.formats = {}
+        self.parser = self._detect_format(self.input)
+
+    def output(self, output_files):
+        deal_set = self.parser.parse(self.input)
+        for output in output_files:
+            self._detect_format(output).output(output, deal_set)
+
+    def _detect_format(self, filename):
+        for deal_format in globals()['formats'].__all__:
+            if deal_format not in self.formats:
+                self.formats[deal_format] = getattr(
+                    globals()[deal_format],
+                    deal_format.upper() + 'Format')()
+            if self.formats[deal_format].match_file(filename):
+                return self.formats[deal_format]
+        raise ValueError('Unrecognized file extension: %s' % filename)