1 # -*- coding: utf-8 -*-
3 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
4 from stat import S_IMODE
19 SCOPE_TRANSPONDERDATA = 0
31 SCOPE_CURRENT_SKIN = 12
33 SCOPE_DEFAULTPARTITION = 14
34 SCOPE_DEFAULTPARTITIONMOUNTDIR = 15
36 SCOPE_CURRENT_PLUGIN = 17
42 SCOPE_TRANSPONDERDATA: ("/etc/", PATH_DONTCREATE),
43 SCOPE_SYSETC: ("/etc/", PATH_DONTCREATE),
44 SCOPE_FONTS: ("/usr/share/fonts/", PATH_DONTCREATE),
45 SCOPE_CONFIG: ("/etc/enigma2/", PATH_CREATE),
46 SCOPE_PLUGINS: ("/usr/lib/enigma2/python/Plugins/", PATH_CREATE),
48 SCOPE_LANGUAGE: ("/usr/share/enigma2/po/", PATH_DONTCREATE),
50 SCOPE_SKIN: ("/usr/share/enigma2/", PATH_DONTCREATE),
51 SCOPE_SKIN_IMAGE: ("/usr/share/enigma2/", PATH_DONTCREATE),
52 SCOPE_HDD: ("/hdd/movie/", PATH_DONTCREATE),
53 SCOPE_MEDIA: ("/media/", PATH_DONTCREATE),
54 SCOPE_PLAYLIST: ("/etc/enigma2/playlist/", PATH_CREATE),
56 SCOPE_USERETC: ("", PATH_DONTCREATE), # user home directory
58 SCOPE_DEFAULTDIR: ("/usr/share/enigma2/defaults/", PATH_CREATE),
59 SCOPE_DEFAULTPARTITION: ("/dev/mtdblock/6", PATH_DONTCREATE),
60 SCOPE_DEFAULTPARTITIONMOUNTDIR: ("/usr/share/enigma2/dealer", PATH_CREATE),
61 SCOPE_METADIR: ("/usr/share/meta", PATH_CREATE),
64 FILE_COPY = 0 # copy files from fallback dir to the basedir
65 FILE_MOVE = 1 # move files
66 PATH_COPY = 2 # copy the complete fallback dir to the basedir
67 PATH_MOVE = 3 # move the fallback dir to the basedir (can be used for changes in paths)
69 SCOPE_CONFIG: [("/home/root/", FILE_MOVE),
70 ("/usr/share/enigma2/defaults/", FILE_COPY)],
71 SCOPE_HDD: [("/hdd/movies", PATH_MOVE)]
74 def resolveFilename(scope, base = "", path_prefix = None):
76 # you can only use the ~/ if we have a prefix directory
77 assert path_prefix is not None
78 base = os_path.join(path_prefix, base[2:])
80 # don't resolve absolute paths
84 if scope == SCOPE_CURRENT_SKIN:
85 from Components.config import config
86 tmp = defaultPaths[SCOPE_SKIN]
87 pos = config.skin.primary_skin.value.rfind('/')
89 #if basefile is not available use default skin path as fallback
90 tmpfile = tmp[0]+config.skin.primary_skin.value[:pos+1] + base
91 if fileExists(tmpfile):
92 path = tmp[0]+config.skin.primary_skin.value[:pos+1]
98 elif scope == SCOPE_CURRENT_PLUGIN:
99 tmp = defaultPaths[SCOPE_PLUGINS]
100 from Components.config import config
101 skintmp = defaultPaths[SCOPE_SKIN]
102 pos = config.skin.primary_skin.value.rfind('/')
104 #if basefile is not available inside current skin path, use the original provided file as fallback
105 skintmpfile = skintmp[0]+config.skin.primary_skin.value[:pos+1] + base
106 if fileExists(skintmpfile):
107 path = skintmp[0]+config.skin.primary_skin.value[:pos+1]
113 tmp = defaultPaths[scope]
118 if flags == PATH_CREATE:
119 if not pathExists(path):
122 fallbackPath = fallbackPaths.get(scope)
124 if fallbackPath and not fileExists(path + base):
125 for x in fallbackPath:
126 if x[1] == FILE_COPY:
127 if fileExists(x[0] + base):
128 system("cp " + x[0] + base + " " + path + base)
130 elif x[1] == FILE_MOVE:
131 if fileExists(x[0] + base):
132 system("mv " + x[0] + base + " " + path + base)
134 elif x[1] == PATH_COPY:
136 if not pathExists(defaultPaths[scope][0]):
138 system("cp -a " + x[0] + "* " + path)
140 elif x[1] == PATH_MOVE:
142 system("mv " + x[0] + " " + path)
145 # FIXME: we also have to handle DATADIR etc. here.
147 # this is only the BASE - an extension must be added later.
149 def pathExists(path):
150 return os_path.exists(path)
153 return os_path.ismount(path)
155 def createDir(path, makeParents = False):
176 def fileExists(f, mode='r'):
183 return access(f, acc_mode)
185 def getRecordingFilename(basename, dirname = None):
186 # filter out non-allowed characters
187 non_allowed_characters = "/.\\:*?<>|\""
190 basename = basename.replace('\xc2\x86', '').replace('\xc2\x87', '')
193 if c in non_allowed_characters or ord(c) < 32:
197 if dirname is not None:
198 filename = ''.join((dirname, filename))
200 while len(filename) > 240:
201 filename = filename.decode('UTF-8')
202 filename = filename[:-1]
203 filename = filename.encode('UTF-8')
207 path = resolveFilename(SCOPE_HDD, filename)
216 # this is clearly a hack:
217 def InitFallbackFiles():
218 resolveFilename(SCOPE_CONFIG, "userbouquet.favourites.tv")
219 resolveFilename(SCOPE_CONFIG, "bouquets.tv")
220 resolveFilename(SCOPE_CONFIG, "userbouquet.favourites.radio")
221 resolveFilename(SCOPE_CONFIG, "bouquets.radio")
223 # returns a list of tuples containing pathname and filename matching the given pattern
224 # example-pattern: match all txt-files: ".*\.txt$"
225 def crawlDirectory(directory, pattern):
226 expression = compile(pattern)
228 for root, dirs, files in walk(directory):
230 if expression.match(file) is not None:
231 list.append((root, file))
234 def copyfile(src, dst):
237 if os_path.isdir(dst):
238 dst = os_path.join(dst, os_path.basename(src))
239 f2 = open(dst, "w+b")
241 buf = f1.read(16*1024)
246 mode = S_IMODE(st.st_mode)
250 utime(dst, (st.st_atime, st.st_mtime))
252 print "copy", src, "to", dst, "failed!"
256 def copytree(src, dst, symlinks=False):
258 if os_path.isdir(dst):
259 dst = os_path.join(dst, os_path.basename(src))
260 if not os_path.isdir(dst):
265 srcname = os_path.join(src, name)
266 dstname = os_path.join(dst, name)
268 if symlinks and os_path.islink(srcname):
269 linkto = readlink(srcname)
270 symlink(linkto, dstname)
271 elif os_path.isdir(srcname):
272 copytree(srcname, dstname, symlinks)
274 copyfile(srcname, dstname)
276 print "dont copy srcname (no file or link or folder)"
279 mode = S_IMODE(st.st_mode)
283 utime(dst, (st.st_atime, st.st_mtime))
285 print "copy stats for", src, "failed!"
287 def getSize(path, pattern=".*"):
289 if os_path.isdir(path):
290 files = crawlDirectory(path, pattern)
292 filepath = os_path.join(file[0], file[1])
293 path_size += os_path.getsize(filepath)
294 elif os_path.isfile(path):
295 path_size = os_path.getsize(path)