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
|
import os, os.path, stat, sys
import gobject, gtk
from threading import Thread
import time
import fringtools
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_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,path):
self.stop()
self.thread = WalkThread(self,path,self.showhidden)
self.thread.start()
def stop(self):
if self.thread:
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()
class WalkThread( Thread ):
""" A separate class for the thread. """
def __init__(self, master, path, showhidden):
""" Parameters: A FringWalker instance, a string with the path and a bool """
Thread.__init__(self)
self.stopsignal = False
self.master = master
self.path = path
self.showhidden = showhidden
def run(self):
""" Parse the root directory """
print "start walking",self.path
l = []
i = 0
subdirectories = []
total = 0
# make a first run for the root directory
try:
for fn in os.listdir(self.path):
if self.stopsignal: return None
if not self.showhidden and fn[0] == '.': continue
try: p = os.path.join(self.path, fn)
except: continue
s = os.lstat(p)
if stat.S_ISDIR(s.st_mode):
subdirectories.append( (fn,p) );
elif stat.S_ISREG(s.st_mode):
l.append((fn, None, s.st_size))
total += s.st_size
i += 1
except OSError:
pass
# emit an intermediate version to fill up the screen while waiting
self.master._progress_fn(self,
0, len(subdirectories),
(os.path.split(self.path)[-1], l, total))
# now walk the subdirectories with the faster extension function
c = 0
for fn, p in subdirectories:
c += 1
sub = fringtools.build_tree(p, self.showhidden)
if self.stopsignal: return None
total += sub[2]
l.append( (fn,sub[1],sub[2]) );
l.sort(treearray_cmp_fn)
# emit an intermediate version after each directory
self.master._progress_fn(self,
c, len(subdirectories),
(os.path.split(self.path)[-1], l, total))
l.sort(treearray_cmp_fn)
# emit final signal
self.master._finished_fn(self,(os.path.split(self.path)[-1], l, total))
print "finished walking",self.path
|