new dirs

1 files changed, 84 insertions, 0 deletions

diff --git a/module/InitHomeDir.py b/module/InitHomeDir.py
new file mode 100644
index 000000000..0c66b5c32
--- /dev/null
+++ b/module/InitHomeDir.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+    This program 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.
+
+    This program 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 this program; if not, see <http://www.gnu.org/licenses/>.
+
+    @author: RaNaN
+
+	This modules inits working directories and global variables, pydir and homedir
+"""
+
+
+from os import makedirs
+from os import path
+from os import chdir
+from sys import platform
+from sys import argv
+
+import __builtin__
+__builtin__.pypath = path.abspath(path.join(__file__,"..",".."))
+
+
+homedir = ""
+
+if platform == 'nt':
+	homedir = path.expanduser("~")
+	if homedir == "~":
+		import ctypes
+		CSIDL_APPDATA = 26
+		_SHGetFolderPath = ctypes.windll.shell32.SHGetFolderPathW
+		_SHGetFolderPath.argtypes = [ctypes.wintypes.HWND,
+			                         ctypes.c_int,
+			                         ctypes.wintypes.HANDLE,
+			                         ctypes.wintypes.DWORD, ctypes.wintypes.LPCWSTR]
+	
+		path_buf = ctypes.wintypes.create_unicode_buffer(ctypes.wintypes.MAX_PATH)
+		result = _SHGetFolderPath(0, CSIDL_APPDATA, 0, 0, path_buf)
+		homedir = path_buf.value
+else:
+	homedir = path.expanduser("~")
+
+__builtin__.homedir = homedir
+
+
+args = " ".join(argv[1:])
+
+# dirty method to set configdir from commandline arguments
+
+if path.exists(path.join(pypath, "module", "config", "configdir")):
+	f = open(path.join(pypath, "module", "config", "configdir"), "rb")
+	c = f.read().strip()
+	configdir = path.join(pypath, c)
+               
+elif "--configdir=" in args:
+	pos = args.find("--configdir=")
+	end = args.find("-", pos+12)
+	
+	if end == -1:
+		configdir = args[pos+12:].strip()
+	else:
+		configdir = args[pos+12:end].strip()
+else:
+	if platform in ("posix","linux2"):
+		configdir = path.join(homedir, ".pyload")
+	else:
+		configdir = path.join(homedir, "pyload")
+		
+if not path.exists(configdir):
+	makedirs(configdir, 0700)
+
+__builtin__.configdir = configdir
+chdir(configdir)
+
+#print "Using %s as working directory." % configdir