1 files changed, 100 insertions, 0 deletions

diff --git a/dodai/__init__.py b/dodai/__init__.py
new file mode 100644
index 0000000..f1e90bc
--- /dev/null
+++ b/dodai/__init__.py
@@ -0,0 +1,100 @@
+# 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/>.
+
+
+
+import os
+import sys
+from dodai.config import Config
+from dodai.config.db import ConfigDbFile
+
+
+class Configure(Config):
+
+    CONFIG_FILES = ['config', 'configs', 'configure', 'connect', 'connections',
+                    'connection']
+
+    def __init__(self, project_name, config_files=None):
+        self.config_files = []
+        self.project_name = project_name
+        self.home_directory = self._home_directory()
+        self._default_files()
+        self._add_files(config_files)
+        self._add_config_files()
+
+    def _add_files(self, filenames):
+        if filenames:
+            if isinstance(filenames, list) or isinstance(filenames, tuple):
+                for filename in filenames:
+                    self._add_file(filename)
+            else:
+                self._add_file(filenames)
+
+    def _add_file(self, filename):
+        if os.path.isfile(filename):
+            if not filename in self.config_files:
+                self.config_files.append(filename)
+
+    def _default_files(self):
+        dirs = []
+        dirs.append(self._construct_system_config_directory())
+        dirs.append(self._construct_user_config_directory())
+        dirs.append(self._construct_project_config_directory())
+        files = []
+        for dir in dirs:
+            for name in self.CONFIG_FILES:
+                files.append(os.path.join(dir, name))
+        for filename in files:
+            self._add_file(filename)
+
+    def _home_directory(self):
+        out = None
+        try:
+            from win32com.shell import shellcon, shell
+            out = shell.SHGetFolderPath(0, shellcon.CSIDL_APPDATA, 0, 0)
+        except ImportError:
+            out = os.path.expanduser("~")
+        return out
+
+    def _construct_system_config_directory(self):
+        out = os.path.join('etc', self.project_name.lower())
+        return out
+
+    def _construct_user_config_directory(self):
+        project_directory = ".{dir}".format(dir=self.project_name.lower())
+        out = os.path.join(self.home_directory, project_directory)
+        return out
+
+    def _construct_project_config_directory(self):
+        dir = os.path.dirname(os.path.abspath(sys.argv[0]))
+        out = os.path.join(dir, 'config')
+        return out
+
+    def _add_config_files(self):
+        if self.config_files:
+            self.files().add_file(self.config_files)
+            config = self.files().parser()
+            if self._has_connections(config):
+                self.dbs().add_config(config)
+
+    def _has_connections(self, config):
+        obj = ConfigDbFile(config)
+        connections = obj()
+        if connections:
+            return True
+        else:
+            return False