X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/74853a81c5f017675176455eeae4738fa36809e4..db2fbdae678d0f024e4dab8b4b87446f43d3c80b:/lib/python/Tools/Directories.py

diff --git a/lib/python/Tools/Directories.py b/lib/python/Tools/Directories.py
index 9cd29511..395402ee 100644
--- a/lib/python/Tools/Directories.py
+++ b/lib/python/Tools/Directories.py
@@ -1,4 +1,5 @@
-import os
+# -*- coding: utf-8 -*-
+from os import path as os_path, mkdir, system
 
 SCOPE_TRANSPONDERDATA = 0
 SCOPE_SYSETC = 1
@@ -6,42 +7,153 @@ SCOPE_FONTS = 2
 SCOPE_SKIN = 3
 SCOPE_SKIN_IMAGE = 4
 SCOPE_USERETC = 5
+SCOPE_CONFIG = 6
+SCOPE_LANGUAGE = 7
+SCOPE_HDD = 8
+SCOPE_PLUGINS = 9
+SCOPE_MEDIA = 10
+SCOPE_PLAYLIST = 11
+SCOPE_CURRENT_SKIN = 12
 
+PATH_CREATE = 0
+PATH_DONTCREATE = 1
+PATH_FALLBACK = 2
 defaultPaths = {
-		SCOPE_TRANSPONDERDATA: "/etc/",
-		SCOPE_SYSETC: "/etc/",
-		SCOPE_FONTS: "/usr/share/fonts/",
+		SCOPE_TRANSPONDERDATA: ("/etc/", PATH_DONTCREATE),
+		SCOPE_SYSETC: ("/etc/", PATH_DONTCREATE),
+		SCOPE_FONTS: ("/usr/share/fonts/", PATH_DONTCREATE),
+		SCOPE_CONFIG: ("/etc/enigma2/", PATH_CREATE),
+		SCOPE_PLUGINS: ("/usr/lib/enigma2/python/Plugins/", PATH_CREATE),
+    
+		SCOPE_LANGUAGE: ("/usr/share/enigma2/po/", PATH_DONTCREATE),
 
-		SCOPE_SKIN: "/usr/share/tuxbox/enigma2/",
-		SCOPE_SKIN_IMAGE: "/usr/share/tuxbox/enigma2/",
+		SCOPE_SKIN: ("/usr/share/enigma2/", PATH_DONTCREATE),
+		SCOPE_SKIN_IMAGE: ("/usr/share/enigma2/", PATH_DONTCREATE),
+		SCOPE_HDD: ("/hdd/movie/", PATH_DONTCREATE),
+		SCOPE_MEDIA: ("/media/", PATH_DONTCREATE),
+		SCOPE_PLAYLIST: ("/etc/enigma2/playlist/", PATH_CREATE),
 		
-		SCOPE_USERETC: "" # user home directory
+		SCOPE_USERETC: ("", PATH_DONTCREATE) # user home directory
 	}
 
-def resolveFilename(scope, base):
-	# in future, we would check for file existence here,
-	# so we can provide default/fallbacks.
-	
+FILE_COPY = 0 # copy files from fallback dir to the basedir
+FILE_MOVE = 1 # move files
+PATH_COPY = 2 # copy the complete fallback dir to the basedir
+PATH_MOVE = 3 # move the fallback dir to the basedir (can be used for changes in paths)
+fallbackPaths = {
+		SCOPE_CONFIG: [("/home/root/", FILE_MOVE),
+					   ("/usr/share/enigma2/defaults/", FILE_COPY)],
+		SCOPE_HDD: [("/hdd/movies", PATH_MOVE)]
+	}
+
+def resolveFilename(scope, base = "", path_prefix = None):
+	if base[0:2] == "~/":
+		# you can only use the ~/ if we have a prefix directory
+		assert path_prefix is not None
+		base = os_path.join(path_prefix, base[2:])
+
+	# don't resolve absolute paths
+	if base[0:1] == '/':
+		return base
+
+	if scope == SCOPE_CURRENT_SKIN:
+		from Components.config import config
+		tmp = defaultPaths[SCOPE_SKIN]
+		pos = config.skin.primary_skin.value.rfind('/')
+		if pos != -1:
+			path = tmp[0]+config.skin.primary_skin.value[:pos+1]
+		else:
+			path = tmp[0]
+	else:
+		tmp = defaultPaths[scope]
+		path = tmp[0]
+
+	flags = tmp[1]
+
+	if flags == PATH_CREATE:
+		if (not pathExists(path)):
+			mkdir(path)
+
+	#if len(base) > 0 and base[0] == '/':
+		#path = ("", None)
+
+	if not fileExists(path + base):
+		#try:
+		if fallbackPaths.has_key(scope):
+			for x in fallbackPaths[scope]:
+				if x[1] == FILE_COPY:
+					if fileExists(x[0] + base):
+						system("cp " + x[0] + base + " " + path + base)
+						break
+				elif x[1] == FILE_MOVE:
+					if fileExists(x[0] + base):
+						system("mv " + x[0] + base + " " + path + base)
+						break
+				elif x[1] == PATH_COPY:
+					if pathExists(x[0]):
+						if not pathExists(defaultPaths[scope][0]):
+							mkdir(path)
+						system("cp -a " + x[0] + "* " + path)
+						break
+				elif x[1] == PATH_MOVE:
+					if pathExists(x[0]):
+						system("mv " + x[0] + " " + path)
+						break
 	# FIXME: we also have to handle DATADIR etc. here.
-	return defaultPaths[scope] + base
+	return path + base
 
 	# this is only the BASE - an extension must be added later.
-def getRecordingFilename(basename):
-	
-		# filter out non-allowed characters
-	non_allowed_characters = "/.\\"
 	
+def pathExists(path):
+	return os_path.exists(path)
+
+def createDir(path):
+	try:
+		mkdir(path)
+	except:
+		ret = 0
+	else:
+		ret = 1
+	return ret
+
+def fileExists(f, mode='r'):
+	try:
+		file = open(f, mode)
+	except IOError:
+		exists = 0
+	else:
+		exists = 1
+	return exists
+
+def getRecordingFilename(basename, dirname = None):
+	# filter out non-allowed characters
+	non_allowed_characters = "/.\\:*?<>|\""
 	filename = ""
+	
+	basename = basename.replace('\xc2\x86', '').replace('\xc2\x87', '')
+	
 	for c in basename:
 		if c in non_allowed_characters:
 			c = "_"
 		filename += c
-	
+
+	if dirname is not None:
+		filename = ''.join([dirname, filename])
+
 	i = 0
 	while True:
-		path = "/hdd/movies/" + filename
+		path = resolveFilename(SCOPE_HDD, filename)
 		if i > 0:
-			path += str(i)
-		if os.glob(path + "*").size() == 0:
+			path += "_%03d" % i
+		try:
+			open(path + ".ts")
+			i += 1
+		except IOError:
 			return path
-		i+=1
+
+# this is clearly a hack:
+def InitFallbackFiles():
+	resolveFilename(SCOPE_CONFIG, "userbouquet.favourites.tv")
+	resolveFilename(SCOPE_CONFIG, "bouquets.tv")
+	resolveFilename(SCOPE_CONFIG, "userbouquet.favourites.radio")
+	resolveFilename(SCOPE_CONFIG, "bouquets.radio")