X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/700f532b8d7fb96ec6af2ac54526f66a834623c3..04686c6d059d527c3b6e98d4fa8a82a226e2ce65:/lib/python/Tools/Directories.py

diff --git a/lib/python/Tools/Directories.py b/lib/python/Tools/Directories.py
index 187d3edf..31e72641 100644
--- a/lib/python/Tools/Directories.py
+++ b/lib/python/Tools/Directories.py
@@ -1,4 +1,6 @@
-import os
+# -*- coding: utf-8 -*-
+from os import path as os_path, mkdir, system, walk
+from re import compile
 
 SCOPE_TRANSPONDERDATA = 0
 SCOPE_SYSETC = 1
@@ -7,49 +9,143 @@ 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
+SCOPE_DEFAULTDIR = 13
+SCOPE_DEFAULTPARTITION = 14
 
 PATH_CREATE = 0
 PATH_DONTCREATE = 1
-
+PATH_FALLBACK = 2
 defaultPaths = {
 		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/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: ("", PATH_DONTCREATE), # user home directory
 		
-		SCOPE_USERETC: ("", PATH_DONTCREATE) # user home directory
+		SCOPE_DEFAULTDIR: ("/usr/share/enigma2/defaults/", PATH_CREATE),
+		SCOPE_DEFAULTPARTITION: ("/dev/mtdblock/4", PATH_DONTCREATE),
 	}
 
-def resolveFilename(scope, base):
-	# in future, we would check for file existence here,
-	# so we can provide default/fallbacks.
-	
-	path = defaultPaths[scope]
-	if path[1] == PATH_CREATE:
-		if (not os.path.exists(path[0])):
-			os.mkdir(path[0])
-	
-	# FIXME: we also have to handle DATADIR etc. here.
-	return path[0] + base
+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)
+
+	fallbackPath = fallbackPaths.get(scope)
+
+	if fallbackPath and not fileExists(path + base):
+		for x in fallbackPath:
+			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 path + base
 	# this is only the BASE - an extension must be added later.
-def getRecordingFilename(basename):
+
+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 = ""
 	
-		# filter out non-allowed characters
-	non_allowed_characters = "/.\\"
+	basename = basename.replace('\xc2\x86', '').replace('\xc2\x87', '')
 	
-	filename = ""
 	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 += "_%03d" % i
 		try:
@@ -57,3 +153,21 @@ def getRecordingFilename(basename):
 			i += 1
 		except IOError:
 			return path
+
+# 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")
+
+# returns a list of tuples containing pathname and filename matching the given pattern
+# example-pattern: match all txt-files: ".*\.txt$"
+def crawlDirectory(directory, pattern):
+	expression = compile(pattern)
+	list = []
+	for root, dirs, files in walk(directory):
+		for file in files:
+			if expression.match(file) is not None:
+				list.append((root, file))
+	return list