1 files changed, 68 insertions, 0 deletions

diff --git a/dodai/tools/__init__.py b/dodai/tools/__init__.py
index 9d2fad7..ab90cfe 100644
--- a/dodai/tools/__init__.py
+++ b/dodai/tools/__init__.py
@@ -15,3 +15,71 @@
 # You should have received a copy of the GNU General Public License
 # along with Dodai.  If not, see <http://www.gnu.org/licenses/>.
 
+import sys
+import os
+import platform
+
+def home_directory():
+    """Returns the full real path to the home directory of the user who
+    is executing this script.
+
+    """
+    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 os.path.realpath(out)
+
+def config_directory_system(project_name=None):
+    """Returns the system config directory with the passed in project
+    appended to the path.  Returns None for windows and java systems.
+
+    """
+    path = None
+    if platform.system and platform.system() not in ['Windows', 'Java']:
+        if project_name:
+            path = os.path.join('/etc', project_name.strip())
+        else:
+            path = '/etc'
+    return path
+
+def config_directory_user(project_name):
+    """Returns the config direcotry of the project which is located in
+    the user's home directory.  Returns None for Java and unknown systems
+
+    """
+    path = None
+    project = '.{0}'.format(project_name.strip())
+    if platform.system and platform.system() not in ['Java']:
+        home_dir = home_directory()
+        path = os.path.join(home_dir, project)
+    return path
+
+def config_directory_project():
+    """Returns the config directory that is located in the same directory
+    of the executable that ran this script
+
+    """
+    path = os.path.dirname(os.path.abspath(sys.argv[0]))
+    return os.path.join(path, 'config')
+
+
+def config_directories(project_name):
+    """Returns a list of possible project directories
+
+    """
+    dirs = []
+    dir = config_directory_system(project_name)
+    if dir:
+        dirs.append(dir)
+    dir = config_directory_user(project_name)
+    if dir:
+        dirs.append(dir)
+    dir = config_directory_project()
+    if dir:
+        dirs.append(dir)
+    return dirs
+
+