git.maemo.org Git - findit/blob - src/files/search.py

   1 #!/usr/bin/env python
   2 # -*-coding: utf-8 -*-
   3 # vim: sw=4 ts=4 expandtab ai
   4
   5 from os import walk
   6 from os.path import join, abspath, normcase, basename, isdir, getsize
   7 from heapq import nlargest
   8
   9 from misc import size_hum_read, _
  10 from config import config
  11
  12 #==============================================================================
  13
  14 class Control(object):
  15
  16     def __init__(self, ui):
  17
  18         if ui == 'cli':
  19             self.present = Cli_Presentation()
  20         elif ui == 'gtk':
  21             self.present = Gtk_Presentation(self.start_search)
  22         elif ui == 'hildon':
  23             self.present = Hildon_Presentation(self.start_search)
  24
  25         # self.present - for updating windows in interactive presentations
  26         self.abstrac = Abstraction(self.present)
  27
  28         # Used only in non-interactive presentations
  29         self.present.start_search(self.start_search)
  30
  31     def start_search(self, get_data, get_stopit):
  32         filelist = []
  33         outtype, start_path, count = get_data()
  34         search_func = self.abstrac.filegetter(start_path, get_stopit)
  35         for fsize, fpath in nlargest(count, search_func):
  36             filelist.append([int(fsize), fpath, size_hum_read(fsize)])
  37         self.present.show_out_toplevel(None, outtype, filelist)
  38
  39     def run(self):
  40         return self.present.toplevel
  41
  42 #==============================================================================
  43
  44 class Abstraction(object):
  45
  46     def __init__(self, presentation):
  47         self.ignore_dirs = config['files']['ignore_dirs']
  48         self.presentation = presentation
  49
  50     def filegetter(self, startdir, get_stopit):
  51         """Generator of file sizes and paths based on os.walk."""
  52
  53         # Walk across directory tree
  54         for dirpath, dirnames, fnames in walk(startdir):
  55             # Eliminate unnecessary directories
  56             ignore_dirs = self.ignore_dirs
  57             for ign_dir in ignore_dirs[:]:
  58                 for dirname in dirnames[:]:
  59                     if ign_dir == normcase(join(abspath(dirpath), dirname)):
  60                         dirnames.remove(dirname)
  61                         ignore_dirs.remove(ign_dir)
  62
  63             for fname in fnames:
  64                 flpath = abspath(join(dirpath, fname))
  65                 self.presentation.show_current_status(flpath)
  66
  67                 # Stop search via 'stopit' signal
  68                 stopit = get_stopit()
  69                 if stopit:
  70                     stopit = False
  71                     print 'Stopped'
  72                     raise StopIteration
  73                 # Query only valid files
  74                 try:
  75                     # Return results (bytesize, path)
  76                     yield getsize(flpath), flpath
  77                 except OSError:
  78                     continue
  79
  80 #==============================================================================
  81
  82 class Cli_Presentation(object):
  83     def __init__(self):
  84         self.outtype = config['files']['outtype']
  85         self.start_path = config['files']['start_path']
  86         self.count = config['files']['count']
  87         self.stopit = False
  88
  89         self.toplevel = None
  90
  91     def get_data(self):
  92         return self.outtype, self.start_path, int(self.count)
  93
  94     def get_stopit(self):
  95         return False
  96
  97     def show_out_toplevel(self, _, outtype, results):
  98         out_submodule = __import__('files.' + outtype, None, None, outtype)
  99         out_submodule.Cli_Presentation(results).toplevel
 100
 101     def show_current_status(self, current_path):
 102         pass
 103         #print current_path
 104
 105     def start_search(self, start_func):
 106         start_func(self.get_data, self.get_stopit)
 107
 108 #==============================================================================
 109
 110 class Gtk_Presentation(object):
 111
 112     def __init__(self, start_func):
 113         import gtk
 114
 115         # "Start path" entry
 116         self.path_entry = gtk.Entry()
 117         self.path_entry.set_text(config['files']['start_path'])
 118
 119         # "Files quantity" label
 120         qty_label = gtk.Label(_('Files quantity'))
 121
 122         # "Files quantity" spin
 123         self.qty_spin = gtk.SpinButton()
 124         self.qty_spin.set_numeric(True)
 125         self.qty_spin.set_range(0, 65536)
 126         self.qty_spin.set_increments(1, 10)
 127         self.qty_spin.set_value(config['files']['count'])
 128
 129         # "Start" button
 130         self.start_btn = gtk.Button(_('Start'))
 131         self.start_btn.connect('released', self.start_btn_released, start_func)
 132
 133         # "Stop" button
 134         self.stop_btn = gtk.Button(_('Stop'))
 135         self.stop_btn.set_sensitive(False)
 136         self.stop_btn.connect('clicked', self.stop_btn_clicked)
 137
 138         # Output selection
 139         outtable_rbtn = gtk.RadioButton(None, _('Table'))
 140         outtable_rbtn.set_name('outtable')
 141         outdiagram_rbtn = gtk.RadioButton(outtable_rbtn, _('Diagram'))
 142         outdiagram_rbtn.set_name('outdiagram')
 143         out1_rbtn = gtk.RadioButton(outtable_rbtn, 'Another 1')
 144         out1_rbtn.set_name('outanother1')
 145         self.out_rbtns = [outtable_rbtn, outdiagram_rbtn, out1_rbtn]
 146
 147         hbox = gtk.HBox(False, 4)
 148         hbox.pack_start(qty_label, False, False, 0)
 149         hbox.pack_start(self.qty_spin, False, False, 0)
 150         hbox.pack_start(self.start_btn, False, False, 0)
 151         hbox.pack_start(self.stop_btn, False, False, 0)
 152         for btn in reversed(self.out_rbtns):
 153             hbox.pack_end(btn, False, False, 0)
 154             # Activate radio button
 155             if btn.get_name() == config['outtype']:
 156                 btn.set_active(True)
 157
 158         self.statusbar = gtk.Statusbar()
 159         self.context_id = self.statusbar.get_context_id('Current walked file')
 160
 161         self.vbox = gtk.VBox(False, 4)
 162         self.vbox.pack_start(self.path_entry, False, False, 0)
 163         self.vbox.pack_start(hbox, False, False, 0)
 164         self.vbox.pack_end(self.statusbar, False, False, 0)
 165
 166         self.toplevel = self.vbox
 167
 168         # For importing gtk only once (lambda not work)
 169         def show_current_status(current_path):
 170             self.statusbar.push(self.context_id, current_path)
 171             gtk.main_iteration()
 172         self.show_current_status = show_current_status
 173
 174         self.show_out_toplevel(None, config['outtype'], [(1, 'path', 'bytesize')])
 175
 176     #=== Functions ============================================================
 177     def start_btn_released(self, btn, start_func):
 178         self.stopit = False
 179         self.stop_btn.set_sensitive(True)
 180         self.start_btn.set_sensitive(False)
 181         start_func(self.get_data, self.get_stopit)
 182         self.stop_btn.set_sensitive(False)
 183         self.start_btn.set_sensitive(True)
 184
 185     def stop_btn_clicked(self, widget):
 186         self.stopit = True
 187         self.stop_btn.set_sensitive(False)
 188         self.start_btn.set_sensitive(True)
 189
 190     def get_data(self):
 191         for btn in self.out_rbtns:
 192             if btn.get_active():
 193                 out = btn.get_name()
 194         return out, self.path_entry.get_text(), int(self.qty_spin.get_value())
 195
 196     def get_stopit(self):
 197         return self.stopit
 198
 199     # Empty because search start by button
 200     def start_search(self, start_func):
 201         pass
 202
 203     #=== Output type selecting ================================================
 204     def show_out_toplevel(self, btn, outtype, results):
 205         print 'Entering <' + outtype + '> output mode...'
 206         out_submodule = __import__('files.' + outtype, None, None, outtype)
 207
 208         try:
 209             self.out_toplevel.destroy()
 210         except:
 211             pass
 212
 213         self.out_toplevel = out_submodule.Gtk_Presentation(results).toplevel
 214         self.vbox.add(self.out_toplevel)
 215         self.out_toplevel.show_all()
 216 ###        out_submodule.Gtk_Presentation().show_results(results)
 217
 218 #==============================================================================
 219
 220 class Hildon_Presentation(object):
 221
 222     def __init__(self, start_func):
 223         import gtk
 224         import hildon
 225
 226     def start_search(self, start_func):
 227         pass