X-Git-Url: http://code.vuplus.com/gitweb/?a=blobdiff_plain;f=lib%2Fpython%2FTools%2FDirectories.py;h=1858152b1247073549a1aee59c328388f73d4298;hb=ae59e9cff74f8b5c36203b43ab5be0ead370b096;hp=8e7e4d7e1f5e47e2ba7a53bd81bc0462c15777fa;hpb=3d48300d74e5f41d3db939b088ad41238a90fa88;p=vuplus_dvbapp

diff --git a/lib/python/Tools/Directories.py b/lib/python/Tools/Directories.py
old mode 100644
new mode 100755
index 8e7e4d7..1858152
--- a/lib/python/Tools/Directories.py
+++ b/lib/python/Tools/Directories.py
@@ -1,5 +1,20 @@
 # -*- coding: utf-8 -*-
-from os import path as os_path, mkdir, system
+
+from os import path as os_path, mkdir, rmdir, system, walk, stat as os_stat, listdir, readlink, makedirs, error as os_error, symlink, access, F_OK, R_OK, W_OK
+from stat import S_IMODE
+from re import compile
+
+try:
+	from os import chmod
+	have_chmod = True
+except:
+	have_chmod = False
+
+try:
+	from os import utime
+	have_utime = True
+except:
+	have_utime = False
 
 SCOPE_TRANSPONDERDATA = 0
 SCOPE_SYSETC = 1
@@ -14,6 +29,10 @@ SCOPE_PLUGINS = 9
 SCOPE_MEDIA = 10
 SCOPE_PLAYLIST = 11
 SCOPE_CURRENT_SKIN = 12
+SCOPE_DEFAULTDIR = 13
+SCOPE_DEFAULTPARTITION = 14
+SCOPE_DEFAULTPARTITIONMOUNTDIR = 15
+SCOPE_METADIR = 16
 
 PATH_CREATE = 0
 PATH_DONTCREATE = 1
@@ -24,7 +43,7 @@ defaultPaths = {
 		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),
@@ -33,7 +52,12 @@ defaultPaths = {
 		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/6", PATH_DONTCREATE),
+		SCOPE_DEFAULTPARTITIONMOUNTDIR: ("/usr/share/enigma2/dealer", PATH_CREATE),
+		SCOPE_METADIR: ("/usr/lib/enigma2/python/Plugins/meta", PATH_CREATE),
 	}
 
 FILE_COPY = 0 # copy files from fallback dir to the basedir
@@ -104,23 +128,38 @@ def resolveFilename(scope, base = "", path_prefix = None):
 def pathExists(path):
 	return os_path.exists(path)
 
-def createDir(path):
+def isMount(path):
+	return os_path.ismount(path)
+
+def createDir(path, makeParents = False):
 	try:
-		mkdir(path)
+		if makeParents:
+			makedirs(path)
+		else:
+			mkdir(path)
 	except:
 		ret = 0
 	else:
 		ret = 1
 	return ret
 
-def fileExists(f, mode='r'):
+def removeDir(path):
 	try:
-		file = open(f, mode)
-	except IOError:
-		exists = 0
+		rmdir(path)
+	except:
+		ret = 0
 	else:
-		exists = 1
-	return exists
+		ret = 1
+	return ret
+
+def fileExists(f, mode='r'):
+	if mode == 'r':
+		acc_mode = R_OK
+	elif mode == 'w':
+		acc_mode = W_OK
+	else:
+		acc_mode = F_OK
+	return access(f, acc_mode)
 
 def getRecordingFilename(basename, dirname = None):
 	# filter out non-allowed characters
@@ -130,12 +169,12 @@ def getRecordingFilename(basename, dirname = None):
 	basename = basename.replace('\xc2\x86', '').replace('\xc2\x87', '')
 	
 	for c in basename:
-		if c in non_allowed_characters:
+		if c in non_allowed_characters or ord(c) < 32:
 			c = "_"
 		filename += c
 
 	if dirname is not None:
-		filename = ''.join([dirname, filename])
+		filename = ''.join((dirname, filename))
 
 	i = 0
 	while True:
@@ -154,3 +193,67 @@ def InitFallbackFiles():
 	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
+
+def copyfile(src, dst):
+	try:
+		f1 = open(src, "rb")
+		if os_path.isdir(dst):
+			dst = os_path.join(dst, os_path.basename(src))
+		f2 = open(dst, "w+b")
+		while True:
+			buf = f1.read(16*1024)
+			if not buf:
+				break
+			f2.write(buf)
+		st = os_stat(src)
+		mode = S_IMODE(st.st_mode)
+		if have_chmod:
+			chmod(dst, mode)
+		if have_utime:
+			utime(dst, (st.st_atime, st.st_mtime))
+	except:
+		print "copy", src, "to", dst, "failed!"
+		return -1
+	return 0
+
+def copytree(src, dst, symlinks=False):
+	names = listdir(src)
+	if os_path.isdir(dst):
+		dst = os_path.join(dst, os_path.basename(src))
+		if not os_path.isdir(dst):
+			mkdir(dst)
+	else:
+		makedirs(dst)
+	for name in names:
+		srcname = os_path.join(src, name)
+		dstname = os_path.join(dst, name)
+		try:
+			if symlinks and os_path.islink(srcname):
+				linkto = readlink(srcname)
+				symlink(linkto, dstname)
+			elif os_path.isdir(srcname):
+				copytree(srcname, dstname, symlinks)
+			else:
+				copyfile(srcname, dstname)
+		except:
+			print "dont copy srcname (no file or link or folder)"
+	try:
+		st = os_stat(src)
+		mode = S_IMODE(st.st_mode)
+		if have_chmod:
+			chmod(dst, mode)
+		if have_utime:
+			utime(dst, (st.st_atime, st.st_mtime))
+	except:
+		print "copy stats for", src, "failed!"