258940b41f8da6b55ee77063d275364ecd6f76ae
[enigma2.git] / lib / python / Tools / Directories.py
1 import os
2
3 SCOPE_TRANSPONDERDATA = 0
4 SCOPE_SYSETC = 1
5 SCOPE_FONTS = 2
6 SCOPE_SKIN = 3
7 SCOPE_SKIN_IMAGE = 4
8 SCOPE_USERETC = 5
9 SCOPE_CONFIG = 6
10 SCOPE_LANGUAGE = 7
11 SCOPE_HDD = 8
12
13 PATH_CREATE = 0
14 PATH_DONTCREATE = 1
15 PATH_FALLBACK = 2
16 defaultPaths = {
17                 SCOPE_TRANSPONDERDATA: ("/etc/", PATH_DONTCREATE),
18                 SCOPE_SYSETC: ("/etc/", PATH_DONTCREATE),
19                 SCOPE_FONTS: ("/usr/share/fonts/", PATH_DONTCREATE),
20                 SCOPE_CONFIG: ("/etc/enigma2/", PATH_CREATE),
21                                             
22                 SCOPE_LANGUAGE: ("/usr/share/enigma2/po/", PATH_CREATE),
23
24                 SCOPE_SKIN: ("/usr/share/enigma2/", PATH_DONTCREATE),
25                 SCOPE_SKIN_IMAGE: ("/usr/share/enigma2/", PATH_DONTCREATE),
26                 SCOPE_HDD: ("/hdd/movie/", PATH_DONTCREATE),
27                 
28                 SCOPE_USERETC: ("", PATH_DONTCREATE) # user home directory
29         }
30         
31 FILE_COPY = 0 # copy files from fallback dir to the basedir
32 FILE_MOVE = 1 # move files
33 PATH_COPY = 2 # copy the complete fallback dir to the basedir
34 PATH_MOVE = 3 # move the fallback dir to the basedir (can be used for changes in paths)
35 fallbackPaths = {
36                 SCOPE_CONFIG: [("/home/root/", FILE_MOVE),
37                                            ("/usr/share/enigma2/defaults/", FILE_COPY)],
38                 SCOPE_HDD: [("/hdd/movies", PATH_MOVE)]
39         }
40
41 def resolveFilename(scope, base = ""):
42         print "getting scope", scope, "with base", base
43         path = defaultPaths[scope]
44         print "path:", path
45
46         if path[1] == PATH_CREATE:
47                 if (not pathExists(defaultPaths[scope][0])):
48                         os.mkdir(path[0])
49         
50         if not fileExists(path[0] + base):
51                 #try:
52                 if fallbackPaths.has_key(scope):
53                         for x in fallbackPaths[scope]:
54                                 if x[1] == FILE_COPY:
55                                         if fileExists(x[0] + base):
56                                                 os.system("cp " + x[0] + base + " " + path[0] + base)
57                                                 break
58                                 elif x[1] == FILE_MOVE:
59                                         if fileExists(x[0] + base):
60                                                 os.system("mv " + x[0] + base + " " + path[0] + base)
61                                                 break
62                                 elif x[1] == PATH_COPY:
63                                         if pathExists(x[0]):
64                                                 if not pathExists(defaultPaths[scope][0]):
65                                                         os.mkdir(path[0])
66                                                 os.system("cp -a " + x[0] + "* " + path[0])
67                                                 break
68                                 elif x[1] == PATH_MOVE:
69                                         if pathExists(x[0]):
70                                                 os.system("mv " + x[0] + " " + path[0])
71                                                 break
72
73         
74         # FIXME: we also have to handle DATADIR etc. here.
75         return path[0] + base
76
77         # this is only the BASE - an extension must be added later.
78         
79 def pathExists(path):
80         return os.path.exists(path)
81
82 def fileExists(f):
83         try:
84                 file = open(f)
85         except IOError:
86                 exists = 0
87         else:
88                 exists = 1
89         return exists
90
91 def getRecordingFilename(basename):
92         
93                 # filter out non-allowed characters
94         non_allowed_characters = "/.\\"
95         
96         filename = ""
97         for c in basename:
98                 if c in non_allowed_characters:
99                         c = "_"
100                 filename += c
101         
102         i = 0
103         while True:
104                 path = resolveFilename(SCOPE_HDD, filename)
105                 if i > 0:
106                         path += "_%03d" % i
107                 try:
108                         open(path + ".ts")
109                         i += 1
110                 except IOError:
111                         return path
112
113 # this is clearly a hack:
114 def InitFallbackFiles():
115         resolveFilename(SCOPE_CONFIG, "userbouquet.favourites.tv")
116         resolveFilename(SCOPE_CONFIG, "bouquets.tv")