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
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
|
import os, os.path, stat, sys
import threading
import posixpath # instead of os.path for gnomevfs operations
import time
import gobject, gtk
from gnomevfs import *
try: import fringtools
except: print 'Error: Could not find "fringtools" extension module'
WALKER_CLASSIC = 0
WALKER_GNOMEVFS = 1
WALKER_CPP = 2
def print_tree( treearray, tab=0 ):
""" An utility function to print out a tree array """
fn, data, size = treearray
print " "*tab,"%s (%i)"%(fn,size)
if not data: return
if tab > 1: return
for e in data:
print_tree(e,tab+1)
def treearray_cmp_by_size( a, b ):
return cmp(a[2], b[2]) # compare sizes
def treearray_cmp_fn( a, b ):
""" a and b are tuples describing a directory tree. Compare first by directory
status, then by size, and finally by name. """
a_dir = a[1] != None
b_dir = b[1] != None
if a_dir and not b_dir:
return 1
elif b_dir and not a_dir:
return -1
elif a_dir and b_dir:
return cmp(a[2], b[2]) # compare sizes
else:
return cmp(a[0], b[0]) # compare names
class FringWalker( gobject.GObject ):
""" Manages requests for walking directories
Directory entries are represented by tuples of the following form:
(name, contents, size)
name: the file name of the directory
contents: a list of entries within the entry. If None, the entry is not a directory
size: the size of the entry.
"""
__gsignals__ = {
'list-changed': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gobject.TYPE_PYOBJECT,)),
'finished': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gobject.TYPE_PYOBJECT,)),
'manually-stopped': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, () ),
'progress': (gobject.SIGNAL_RUN_LAST, gobject.TYPE_NONE, (gobject.TYPE_INT,gobject.TYPE_INT)),
}
def __init__(self):
gobject.GObject.__init__(self)
self.thread = None
self.showhidden = False
def walk(self,uri,method=WALKER_GNOMEVFS):
self.stop()
self.thread = WalkThread(self,uri,self.showhidden,method)
self.thread.start()
def stop(self):
if self.thread:
self.emit("manually-stopped")
self.thread.stopsignal = True
self.thread = None
def _progress_fn(self, walkthread, c, l, r):
# only emit if called from the current request
if walkthread == self.thread:
gtk.gdk.threads_enter()
self.emit("progress", c, l)
self.emit("list-changed", r)
gtk.gdk.threads_leave()
def _finished_fn(self, walkthread, r):
if walkthread == self.thread:
gtk.gdk.threads_enter()
self.emit("finished", r)
gtk.gdk.threads_leave()
self.thread = None
class WalkThread( threading.Thread ):
""" A separate class for the thread. """
def __init__(self, master, uri, showhidden, method=WALKER_CLASSIC):
""" Parameters: A FringWalker instance, a string with the path and a bool """
threading.Thread.__init__(self)
self.stopsignal = False
self.master = master
self.uri = URI(uri)
self.showhidden = showhidden
self.method = method
self.max_recursion_sort = 3
def __uri_tail(self, uri):
""" Return the tail (the filename) of a gnomevfs uri """
f = format_uri_for_display(str(uri))
f = posixpath.split( f )[-1]
return f
def build_tree_python(self, path):
l = []
total = 0
for fn in os.listdir(path):
if self.stopsignal: return (None,None,0)
if not self.showhidden and fn[0] == ".": continue
try: p = os.path.join(path, fn)
except: continue
s = os.lstat(p)
if stat.S_ISDIR(s.st_mode):
sub = self.build_tree_python(p)
l.append( sub )
total += sub[2]
elif stat.S_ISREG(s.st_mode):
l.append( (fn, None, s.st_size) )
total += s.st_size
return (os.path.split(path)[-1], l, total)
def build_tree_gnomevfs(self, uri, recursionlvl=0):
try: h = DirectoryHandle(uri)
except InvalidURIError:
print uri,"is not a valid uri, skipped"
return (str(uri), None, 0)
except NotFoundError:
print uri,"not found, skipped"
return (str(uri), None, 0)
l = []
total = 0
try:
while True:
if self.stopsignal: return (None,None,0)
d = h.next()
if not self.showhidden and d.name[0] == ".": continue
if d.type == 2: # directory
sub = self.build_tree_gnomevfs(uri.append_path(d.name),recursionlvl+1)
l.append( sub )
total += sub[2]
else:
l.append( (d.name, None, d.size) )
total += d.size
except StopIteration: pass
if recursionlvl <= self.max_recursion_sort:
l.sort(treearray_cmp_by_size)
return (self.__uri_tail(uri), l, total)
def run(self):
""" Parse the root directory """
# write some debug information
starttime = time.time()
print "start walking",self.uri,
if self.method == WALKER_CPP: print "(using c++ extension)"
elif self.method == WALKER_GNOMEVFS: print "(using python and gnomevfs)"
else: print "(using classic python)"
# scan root directory first (using gnomevfs)
try: h = DirectoryHandle(self.uri)
except InvalidURIError:
print uri,"is not a valid uri, skipped"
return (str(self.uri), None, 0)
except NotFoundError:
print uri,"not found, skipped"
return (str(self.uri), None, 0)
subdirectories = []
l = []
total = 0
d = h.next()
try:
while True:
if self.stopsignal: return
d = h.next()
if not self.showhidden and d.name[0] == ".": continue
if d.type == 2: # directory
subdirectories.append( d.name );
else:
l.append( (d.name, None, d.size) )
total += d.size
except StopIteration: pass
# emit an intermediate version to fill up the screen while waiting
self.master._progress_fn(self,
0, len(subdirectories),
(self.__uri_tail(self.uri), l, total))
# now walk the subdirectories with the faster extension function
c = 0
for directory in subdirectories:
c += 1
if self.method == WALKER_CPP:
path = get_local_path_from_uri(str(self.uri))+os.sep+directory
sub = fringtools.build_tree(path, self.showhidden)
elif self.method == WALKER_GNOMEVFS:
uri = self.uri
sub = self.build_tree_gnomevfs(uri.append_path(directory))
else:
path = get_local_path_from_uri(str(self.uri))+os.sep+directory
sub = self.build_tree_python(path)
if self.stopsignal: return
total += sub[2]
l.append( (directory,sub[1],sub[2]) );
l.sort(treearray_cmp_fn)
# emit an intermediate version after each directory
self.master._progress_fn(self,
c, len(subdirectories),
(self.__uri_tail(self.uri), l, total))
l.sort(treearray_cmp_fn)
# emit final signal
self.master._finished_fn(self,(self.__uri_tail(self.uri), l, total))
print "finished walking",self.uri,"(time=%s)"%round(time.time()-starttime,2)
|