X-Git-Url: http://code.vuplus.com/gitweb/?a=blobdiff_plain;f=lib%2Fpython%2FTools%2FDirectories.py;h=8ed2c8a46017951c8702db09bd1047d3a6094c99;hb=3c919c91a7778a049bad0317f6f7df86ff509415;hp=1e35722d06e758dfed114ef1ddc34ac38b0526c8;hpb=04f4636324ab06c5a3a4f618100048e159eaf6f9;p=vuplus_dvbapp diff --git a/lib/python/Tools/Directories.py b/lib/python/Tools/Directories.py old mode 100644 new mode 100755 index 1e35722..f0ef0de --- a/lib/python/Tools/Directories.py +++ b/lib/python/Tools/Directories.py @@ -1,6 +1,21 @@ # -*- coding: utf-8 -*- -from os import path as os_path, mkdir, rmdir, system, walk + +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 +from enigma import eEnv + +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 @@ -17,31 +32,34 @@ SCOPE_PLAYLIST = 11 SCOPE_CURRENT_SKIN = 12 SCOPE_DEFAULTDIR = 13 SCOPE_DEFAULTPARTITION = 14 -SCOPE_DEFAULTPARTITIONMOUNTDIR = 14 +SCOPE_DEFAULTPARTITIONMOUNTDIR = 15 +SCOPE_METADIR = 16 +SCOPE_CURRENT_PLUGIN = 17 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_TRANSPONDERDATA: (eEnv.resolve("${sysconfdir}/"), PATH_DONTCREATE), + SCOPE_SYSETC: (eEnv.resolve("${sysconfdir}/"), PATH_DONTCREATE), + SCOPE_FONTS: (eEnv.resolve("${datadir}/fonts/"), PATH_DONTCREATE), + SCOPE_CONFIG: (eEnv.resolve("${sysconfdir}/enigma2/"), PATH_CREATE), + SCOPE_PLUGINS: (eEnv.resolve("${libdir}/enigma2/python/Plugins/"), PATH_CREATE), + + SCOPE_LANGUAGE: (eEnv.resolve("${datadir}/enigma2/po/"), PATH_DONTCREATE), + + SCOPE_SKIN: (eEnv.resolve("${datadir}/enigma2/"), PATH_DONTCREATE), + SCOPE_SKIN_IMAGE: (eEnv.resolve("${datadir}/enigma2/"), PATH_DONTCREATE), SCOPE_HDD: ("/hdd/movie/", PATH_DONTCREATE), SCOPE_MEDIA: ("/media/", PATH_DONTCREATE), - SCOPE_PLAYLIST: ("/etc/enigma2/playlist/", PATH_CREATE), + SCOPE_PLAYLIST: (eEnv.resolve("${sysconfdir}/enigma2/playlist/"), PATH_CREATE), SCOPE_USERETC: ("", PATH_DONTCREATE), # user home directory - SCOPE_DEFAULTDIR: ("/usr/share/enigma2/defaults/", PATH_CREATE), - SCOPE_DEFAULTPARTITION: ("/dev/mtdblock/4", PATH_DONTCREATE), - SCOPE_DEFAULTPARTITIONMOUNTDIR: ("/usr/share/enigma2/dealer", PATH_CREATE), + SCOPE_DEFAULTDIR: (eEnv.resolve("${datadir}/enigma2/defaults/"), PATH_CREATE), + SCOPE_DEFAULTPARTITION: ("/dev/mtdblock6", PATH_DONTCREATE), + SCOPE_DEFAULTPARTITIONMOUNTDIR: (eEnv.resolve("${datadir}/enigma2/dealer"), PATH_CREATE), + SCOPE_METADIR: (eEnv.resolve("${datadir}/meta"), PATH_CREATE), } FILE_COPY = 0 # copy files from fallback dir to the basedir @@ -50,7 +68,7 @@ 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)], + (eEnv.resolve("${datadir}/enigma2/defaults/"), FILE_COPY)], SCOPE_HDD: [("/hdd/movies", PATH_MOVE)] } @@ -69,7 +87,27 @@ def resolveFilename(scope, base = "", path_prefix = None): tmp = defaultPaths[SCOPE_SKIN] pos = config.skin.primary_skin.value.rfind('/') if pos != -1: - path = tmp[0]+config.skin.primary_skin.value[:pos+1] + #if basefile is not available use default skin path as fallback + tmpfile = tmp[0]+config.skin.primary_skin.value[:pos+1] + base + if fileExists(tmpfile): + path = tmp[0]+config.skin.primary_skin.value[:pos+1] + else: + path = tmp[0] + else: + path = tmp[0] + + elif scope == SCOPE_CURRENT_PLUGIN: + tmp = defaultPaths[SCOPE_PLUGINS] + from Components.config import config + skintmp = defaultPaths[SCOPE_SKIN] + pos = config.skin.primary_skin.value.rfind('/') + if pos != -1: + #if basefile is not available inside current skin path, use the original provided file as fallback + skintmpfile = skintmp[0]+config.skin.primary_skin.value[:pos+1] + base + if fileExists(skintmpfile): + path = skintmp[0]+config.skin.primary_skin.value[:pos+1] + else: + path = tmp[0] else: path = tmp[0] else: @@ -80,7 +118,11 @@ def resolveFilename(scope, base = "", path_prefix = None): if flags == PATH_CREATE: if not pathExists(path): - mkdir(path) + try: + mkdir(path) + except OSError: + print "resolveFilename: Couldn't create %s" % path + return None fallbackPath = fallbackPaths.get(scope) @@ -112,9 +154,15 @@ 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: @@ -131,13 +179,13 @@ def removeDir(path): return ret def fileExists(f, mode='r'): - try: - file = open(f, mode) - except IOError: - exists = 0 + if mode == 'r': + acc_mode = R_OK + elif mode == 'w': + acc_mode = W_OK else: - exists = 1 - return exists + acc_mode = F_OK + return access(f, acc_mode) def getRecordingFilename(basename, dirname = None): # filter out non-allowed characters @@ -147,12 +195,17 @@ 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)) + + while len(filename) > 240: + filename = filename.decode('UTF-8') + filename = filename[:-1] + filename = filename.encode('UTF-8') i = 0 while True: @@ -175,10 +228,75 @@ def InitFallbackFiles(): # 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)) + if directory: + expression = compile(pattern) + 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!" + +def getSize(path, pattern=".*"): + path_size = 0 + if os_path.isdir(path): + files = crawlDirectory(path, pattern) + for file in files: + filepath = os_path.join(file[0], file[1]) + path_size += os_path.getsize(filepath) + elif os_path.isfile(path): + path_size = os_path.getsize(path) + return path_size