1 files changed, 179 insertions, 0 deletions

diff --git a/dodai/config/db/__init__.py b/dodai/config/db/__init__.py
new file mode 100644
index 0000000..fa510ac
--- /dev/null
+++ b/dodai/config/db/__init__.py
@@ -0,0 +1,179 @@
+# Copyright (C) 2010  Leonard Thomas
+#
+# This file is part of Dodai.
+#
+# Dodai is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Dodai is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Dodai.  If not, see <http://www.gnu.org/licenses/>.
+
+class ConfigDb(object):
+
+    def __init__(self):
+        self.connections = {}
+        self._handlers = {}
+        from dodai.config.db.sa import Sa
+        self.register_handler('sa', Sa)
+
+    def register_handler(self, name, obj):
+        self._handlers[name] = [obj, None]
+
+    def add_config(self, config_parser=None):
+        if config_parser:
+            if hasattr(config_parser, 'sections') and \
+                    hasattr(config_parser, 'options'):
+                config_obj = ConfigDbFile(config_parser)
+                self._add_connections(config_obj)
+            else:
+                raise NotConfigParserObject()
+
+    def _add_connections(self, config_obj):
+        connections = config_obj()
+        for name, obj in connections.items():
+            self.connections[name] = obj
+
+    def load(self, name):
+        if name in self.connections:
+            connection = self.connections[name]
+            if connection.db_obj:
+                return connection.db_obj
+            else:
+                handler = self._load_handler(connection.handler)
+                db_obj = handler.load(connection)
+                self.connections[name].db_obj = db_obj
+                return db_obj
+
+    def _load_handler(self, name):
+        if name in self._handlers:
+            handler = self._handlers[name]
+            cls = handler[0]
+            obj = handler[1]
+            if not obj:
+                obj = cls()
+                self._handlers[name] = [cls, obj]
+            return obj
+        raise UnknownHandlerException(name)
+
+
+
+class ConfigDbFile(object):
+
+    OPTIONS_REQUIRED = [
+        ['protocol', 'hostname', 'port', 'username', 'password','database'],
+        ['protocol', 'filename']
+    ]
+    OPTIONS_EXTRA = ['protocol_extra', 'handler']
+    DEFAULT_HANDLER = 'sa'
+
+    def __init__(self, config_parser):
+        self.parser = config_parser
+        self._options = self._all_options()
+        self.connections = {}
+
+    def __call__(self):
+        if not self.connections:
+            for section in self.parser.sections():
+                if self._is_valid(section):
+                    obj = self._build_connection(section)
+                    self.connections[obj.name] = obj
+        return self.connections
+
+    def _all_options(self):
+        out = []
+        for option_group in self.OPTIONS_REQUIRED:
+            for option in option_group:
+                if option not in out:
+                    out.append(option)
+        for option in self.OPTIONS_EXTRA:
+            if option not in out:
+                out.append(option)
+        return out
+
+    def _is_valid(self, section):
+        for option_group in self.OPTIONS_REQUIRED:
+            total = len(option_group)
+            count = 0
+            for option in option_group:
+                if option in self.parser.options(section):
+                    value = self.parser.get(section, option)
+                    if value:
+                        count += 1
+            if count >= total:
+                return True
+        return False
+
+    def _build_connection(self, section):
+        obj = ConfigDbConnection()
+        for option in self._options:
+            obj.name = section
+            if self.parser.has_option(section, option):
+                value = self.parser.get(section, option)
+                setattr(obj, option, value)
+            if not hasattr(obj, 'handler') or not obj.handler:
+                obj.handler = self.DEFAULT_HANDLER
+        return obj
+
+
+class BaseConfigDb(object):
+
+    PROTOCOLS = ['postgresql', 'mysql', 'sqlite', 'mssql', 'oracle']
+
+    def _clean(self, obj):
+        obj.protocol = self._clean_protocol(obj.protocol)
+        if hasattr(obj, 'port'):
+            obj.port = self._clean_port(obj.port)
+
+    def _clean_protocol(self, data):
+        data = data.lower()
+        if data in ('postgres', 'postgre'):
+            data = 'postgresql'
+        if data not in self.PROTOCOLS:
+            raise InvalidProtocolException(data)
+        else:
+            return data
+
+    def _clean_port(self, data):
+        try:
+            data = int(data)
+        except ValueError:
+            data = None
+        except TypeError:
+            data = None
+        if data:
+            if data <1 or data > 65535:
+                raise InvalidPortException(data)
+        return data
+
+
+class ConfigDbConnection(object):
+
+    def __init__(self):
+        self.db_obj = None
+
+
+class NotConfigParserObject(Exception):
+    pass
+
+
+class InvalidProtocolException(Exception):
+    pass
+
+
+class InvalidPortException(Exception):
+    pass
+
+
+class UnknownHandlerException(Exception):
+    pass
+
+
+class UnknownConnectionException(Exception):
+    pass