1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
from Plugins.Plugin import PluginDescriptor
from Components.PluginComponent import plugins
from os import path as os_path, walk as os_walk
from string import lower
from mimetypes import guess_type
def getExtension(file):
p = file.rfind('.')
if p == -1:
ext = ""
else:
ext = file[p+1:]
return lower(ext)
def getType(file):
(type, _) = guess_type(file)
if type is None:
# Detect some mimetypes unknown to dm7025
# TODO: do mimetypes.add_type once should be better
ext = getExtension(file)
if ext == "ipk":
return "application/x-debian-package"
elif ext == "ogg":
return "application/ogg"
return type
class Scanner:
def __init__(self, name, mimetypes= [], paths_to_scan = [], description = "", openfnc = None):
self.mimetypes = mimetypes
self.name = name
self.paths_to_scan = paths_to_scan
self.description = description
self.openfnc = openfnc
def checkFile(self, file):
return True
def handleFile(self, res, file):
if (self.mimetypes is None or file.mimetype in self.mimetypes) and self.checkFile(file):
res.setdefault(self, []).append(file)
def __repr__(self):
return "<Scanner " + self.name + ">"
def open(self, list, *args, **kwargs):
if self.openfnc is not None:
self.openfnc(list, *args, **kwargs)
class ScanPath:
def __init__(self, path, with_subdirs = False):
self.path = path
self.with_subdirs = with_subdirs
def __repr__(self):
return self.path + "(" + str(self.with_subdirs) + ")"
# we will use this in a set(), so we need to implement __hash__ and __cmp__
def __hash__(self):
return self.path.__hash__() ^ self.with_subdirs.__hash__()
def __cmp__(self, other):
if self.path < other.path:
return -1
elif self.path > other.path:
return +1
else:
return self.with_subdirs.__cmp__(other.with_subdirs)
class ScanFile:
def __init__(self, path, mimetype = None, size = None, autodetect = True):
self.path = path
if mimetype is None and autodetect:
(self.mimetype, _) = guess_type(path)
else:
self.mimetype = mimetype
self.size = size
def __repr__(self):
return "<ScanFile " + self.path + " (" + str(self.mimetype) + ", " + str(self.size) + " MB)>"
def execute(option):
print "execute", option
if option is None:
return
(_, scanner, files, session) = option
scanner.open(files, session)
def scanDevice(mountpoint):
scanner = [ ]
for p in plugins.getPlugins(PluginDescriptor.WHERE_FILESCAN):
l = p()
if not isinstance(l, list):
l = [l]
scanner += l
print "scanner:", scanner
res = { }
# merge all to-be-scanned paths, with priority to
# with_subdirs.
paths_to_scan = set()
# first merge them all...
for s in scanner:
paths_to_scan.update(set(s.paths_to_scan))
# ...then remove with_subdir=False when same path exists
# with with_subdirs=True
for p in set(paths_to_scan):
if p.with_subdirs == True and ScanPath(path=p.path) in paths_to_scan:
paths_to_scan.remove(ScanPath(path=p.path))
# convert to list
paths_to_scan = list(paths_to_scan)
# now scan the paths
for p in paths_to_scan:
path = os_path.join(mountpoint, p.path)
for root, dirs, files in os_walk(path):
for f in files:
sfile = ScanFile(os_path.join(root, f))
for s in scanner:
s.handleFile(res, sfile)
# if we really don't want to scan subdirs, stop here.
if not p.with_subdirs:
del dirs[:]
# res is a dict with scanner -> [ScanFiles]
return res
def openList(session, files):
if not isinstance(files, list):
files = [ files ]
scanner = [ ]
for p in plugins.getPlugins(PluginDescriptor.WHERE_FILESCAN):
l = p()
if not isinstance(l, list):
l = [l]
scanner += l
print "scanner:", scanner
res = { }
for file in files:
for s in scanner:
s.handleFile(res, file)
choices = [ (r.description, r, res[r], session) for r in res ]
Len = len(choices)
if Len > 1:
from Screens.ChoiceBox import ChoiceBox
session.openWithCallback(
execute,
ChoiceBox,
title = "The following viewers were found...",
list = choices
)
return True
elif Len:
execute(choices[0])
return True
return False
def openFile(session, mimetype, file):
return openList(session, [ScanFile(file, mimetype)])
|
