3 # vim: sw=4 ts=4 expandtab ai
6 from os.path import join, abspath, normcase, basename, isdir, getsize
7 from heapq import nlargest
9 from misc import size_hum_read, _
11 #==============================================================================
13 class Control(object):
15 def __init__(self, ui, config):
18 self.present = eval(ui + '_Presentation(config, self.start_search)')
19 self.abstrac = Abstraction(self.config, self.present)
21 self.toplevel = self.present.toplevel
23 def start_search(self, get_data, get_stopit):
25 outtype, start_path, count = get_data()
26 search_func = self.abstrac.filegetter(start_path, get_stopit)
27 for fsize, fpath in nlargest(count, search_func):
28 filelist.append([int(fsize), fpath, size_hum_read(fsize)])
29 self.present.show_out_toplevel(None, outtype, filelist)
34 #==============================================================================
36 class Abstraction(object):
38 def __init__(self, config, presentation):
39 self.ignore_dirs = config['ignore_dirs']
40 self.presentation = presentation
42 def filegetter(self, startdir, get_stopit):
43 """Generator of file sizes and paths based on os.walk."""
45 # Walk across directory tree
46 for dirpath, dirnames, fnames in walk(startdir):
47 # Eliminate unnecessary directories
48 ignore_dirs = self.ignore_dirs
49 for ign_dir in ignore_dirs[:]:
50 for dirname in dirnames[:]:
51 if ign_dir == normcase(join(abspath(dirpath), dirname)):
52 dirnames.remove(dirname)
53 ignore_dirs.remove(ign_dir)
56 flpath = abspath(join(dirpath, fname))
57 self.presentation.show_current_status(flpath)
59 # Stop search via 'stopit' signal
65 # Query only valid files
67 # Return results (bytesize, path)
68 yield getsize(flpath), flpath
72 #==============================================================================
74 class Cli_Presentation(object):
75 def __init__(self, config, start_func):
76 self.start_func = start_func
78 self.outtype = config['outtype']
79 self.start_path = config['start_path']
80 self.count = config['count']
86 return self.outtype, self.start_path, int(self.count)
91 def show_out_toplevel(self, _, outtype, results):
92 out_submodule = __import__('files.' + outtype, None, None, outtype)
93 out_submodule.Cli_Presentation(results).toplevel
95 def show_current_status(self, current_path):
97 ### print current_path
100 self.start_func(self.get_data, self.get_stopit)
102 #==============================================================================
104 class Gtk_Presentation(object):
106 def __init__(self, config, start_func):
112 self.path_entry = gtk.Entry()
113 self.path_entry.set_text(self.config['start_path'])
115 # "Files quantity" label
116 qty_label = gtk.Label(_('Files quantity'))
118 # "Files quantity" spin
119 self.qty_spin = gtk.SpinButton()
120 self.qty_spin.set_numeric(True)
121 self.qty_spin.set_range(0, 65536)
122 self.qty_spin.set_increments(1, 10)
123 self.qty_spin.set_value(self.config['count'])
126 self.start_btn = gtk.Button(_('Start'))
127 self.start_btn.connect('released', self.start_btn_released, start_func)
130 self.stop_btn = gtk.Button(_('Stop'))
131 self.stop_btn.set_sensitive(False)
132 self.stop_btn.connect('clicked', self.stop_btn_clicked)
135 outtable_rbtn = gtk.RadioButton(None, _('Table'))
136 outtable_rbtn.set_name('outtable')
137 outdiagram_rbtn = gtk.RadioButton(outtable_rbtn, _('Diagram'))
138 outdiagram_rbtn.set_name('outdiagram')
139 out1_rbtn = gtk.RadioButton(outtable_rbtn, 'Another 1')
140 out1_rbtn.set_name('outanother1')
141 self.out_rbtns = [outtable_rbtn, outdiagram_rbtn, out1_rbtn]
143 hbox = gtk.HBox(False, 4)
144 hbox.pack_start(qty_label, False, False, 0)
145 hbox.pack_start(self.qty_spin, False, False, 0)
146 hbox.pack_start(self.start_btn, False, False, 0)
147 hbox.pack_start(self.stop_btn, False, False, 0)
148 for btn in reversed(self.out_rbtns):
149 hbox.pack_end(btn, False, False, 0)
150 # Activate radio button
151 if btn.get_name() == self.config['outtype']:
154 self.statusbar = gtk.Statusbar()
155 self.context_id = self.statusbar.get_context_id('Current walked file')
157 self.vbox = gtk.VBox(False, 4)
158 self.vbox.pack_start(self.path_entry, False, False, 0)
159 self.vbox.pack_start(hbox, False, False, 0)
160 self.vbox.pack_end(self.statusbar, False, False, 0)
162 self.toplevel = self.vbox
164 # For importing gtk only once (lambda not work)
165 def show_current_status(current_path):
166 self.statusbar.push(self.context_id, current_path)
168 self.show_current_status = show_current_status
170 self.show_out_toplevel(None, self.config['outtype'], [(1, 'path', 'bytesize')])
172 #=== Functions ============================================================
173 def start_btn_released(self, btn, start_func):
175 self.stop_btn.set_sensitive(True)
176 self.start_btn.set_sensitive(False)
177 start_func(self.get_data, self.get_stopit)
178 self.stop_btn.set_sensitive(False)
179 self.start_btn.set_sensitive(True)
181 def stop_btn_clicked(self, widget):
183 self.stop_btn.set_sensitive(False)
184 self.start_btn.set_sensitive(True)
187 for btn in self.out_rbtns:
190 return out, self.path_entry.get_text(), int(self.qty_spin.get_value())
192 def get_stopit(self):
198 #=== Output type selecting ================================================
199 def show_out_toplevel(self, btn, outtype, results):
200 print 'Entering <' + outtype + '> output mode...'
201 out_submodule = __import__('files.' + outtype, None, None, outtype)
204 self.out_toplevel.destroy()
208 self.out_toplevel = out_submodule.Gtk_Presentation(results).toplevel
209 self.vbox.add(self.out_toplevel)
210 self.out_toplevel.show_all()
211 ### out_submodule.Gtk_Presentation().show_results(results)
213 #==============================================================================
215 class Hildon_Presentation(object):
217 def __init__(self, config, start_func):