Changed "*.*" to "*" in files.search so it possible find files w/o extension
[findit] / src / files / search.py
index 1bbccf6..0440610 100755 (executable)
@@ -3,8 +3,10 @@
 # vim: sw=4 ts=4 expandtab ai
 
 from os import walk
-from os.path import join, abspath, normcase, basename, isdir, getsize
+from os.path import join, abspath, normcase, isdir, getsize
 from heapq import nlargest
+from fnmatch import fnmatch
+from sys import platform
 
 from misc import size_hum_read, _
 from config import config
@@ -20,19 +22,25 @@ OUTTYPES = [
 
 class Control(object):
 
-    def __init__(self, ui, params):
-        self.present = eval(ui + '_Presentation(self.start_search, params)')
+    def __init__(self, ui, outtype, params):
+        self.present = eval(ui + '_Presentation(self.start_search, outtype, params)')
         self.abstrac = Abstraction(self.present)
 
         self.toplevel = self.present.toplevel
 
     def start_search(self, get_criteria, get_stopit):
         filelist = []
-        outtype, start_path, count = get_criteria()
-        search_func = self.abstrac.filegetter(start_path, get_stopit)
+        outtype, start_path, count, file_filter = get_criteria()
+        search_func = self.abstrac.filegetter(start_path, file_filter, get_stopit)
+
         for fsize, fpath in nlargest(count, search_func):
             filelist.append([int(fsize), fpath, size_hum_read(fsize)])
-        self.present.show_out_toplevel(outtype, filelist)
+
+        if not filelist:
+            self.present.nothing_founded()
+            return
+        results = [filelist, start_path]
+        self.present.show_out_toplevel(outtype, results)
 
     def run(self):
         self.present.run()
@@ -45,9 +53,8 @@ class Abstraction(object):
         self.ignore_dirs = config['files']['ignore_dirs']
         self.presentation = presentation
 
-    def filegetter(self, startdir, get_stopit):
+    def filegetter(self, startdir, file_filter, get_stopit):
         """Generator of file sizes and paths based on os.walk."""
-
         # Walk across directory tree
         for dirpath, dirnames, fnames in walk(startdir):
             # Eliminate unnecessary directories
@@ -59,37 +66,49 @@ class Abstraction(object):
                         ignore_dirs.remove(ign_dir)
 
             for fname in fnames:
-                flpath = abspath(join(dirpath, fname))
-                self.presentation.show_current_status(flpath)
-
-                # Stop search via 'stopit' signal
-                stopit = get_stopit()
-                if stopit:
-                    stopit = False
-                    print 'Stopped'
-                    raise StopIteration
-                # Query only valid files
-                try:
-                    # Return results (bytesize, path)
-                    yield getsize(flpath), flpath
-                except OSError:
-                    continue
+                # Store only necessary files
+                for mask in file_filter:
+                    if fnmatch(fname, mask):
+                        if platform == 'win32':
+                            # Crutch for non-unicode names
+                            flpath = unicode(join(dirpath, fname), '1251')
+                        else:
+                            flpath = join(dirpath, fname)
+                        # Show current path
+                        self.presentation.show_current_status(flpath)
+                        # Stop search via 'stopit' signal
+                        stopit = get_stopit()
+                        if stopit:
+                            stopit = False
+                            print 'Stopped'
+                            raise StopIteration
+                        # Query only valid files
+                        try:
+                            # Return results (bytesize, path)
+                            yield getsize(flpath), flpath
+                        except OSError:
+                            continue
 
 #==============================================================================
 
 class Cli_Presentation(object):
-    def __init__(self, start_func, params):
+    def __init__(self, start_func, outtype, params):
         self.start_func = start_func
 
-        self.outtype = params['outtype']
-        self.start_path = params['start_path']
-        self.count = params['count']
+        self.outtype = outtype
+        self.start_path = params[0]
+        self.count = params[1]
+        try:
+            self.file_filter = params[2].split(';')
+        except IndexError:
+            self.file_filter = '*'
+
         self.stopit = False
 
         self.toplevel = None
 
     def get_data(self):
-        return self.outtype, self.start_path, int(self.count)
+        return self.outtype, self.start_path, int(self.count), self.file_filter
 
     def get_stopit(self):
         return False
@@ -106,6 +125,9 @@ class Cli_Presentation(object):
         print '\\' + '\r',
         ### print current_path
 
+    def nothing_founded(self):
+        print _('Nothing founded!')
+
     def run(self):
         self.start_func(self.get_data, self.get_stopit)
 
@@ -113,24 +135,30 @@ class Cli_Presentation(object):
 
 class Gtk_Presentation(object):
 
-    def __init__(self, start_func, __):
+    def __init__(self, start_func, *unused):
         import gtk
         global gtk  # for show_current_status()
+        from misc import NotebookWCloseBtns
 
-        self.nb = gtk.Notebook()
-        self.nb.set_scrollable(True)
+        self.nb = NotebookWCloseBtns()
+        self.nb.notebook.set_scrollable(True)
+        self.nb.notebook.set_border_width(2)
 
         #====================
         # Notebook
         #====================
 
+        # "Start path" label
+        self.path_label = gtk.Label(_('Path'))
         # "Start path" entry
         self.path_entry = gtk.Entry()
         self.path_entry.set_text(config['files']['start_path'])
+        # "Browse" button
+        self.browse_btn = gtk.Button('Browse...')
+        self.browse_btn.connect('clicked', self.browse_btn_clicked)
 
         # "Files quantity" label
         qty_label = gtk.Label(_('Files quantity'))
-
         # "Files quantity" spin
         self.qty_spin = gtk.SpinButton()
         self.qty_spin.set_numeric(True)
@@ -138,15 +166,14 @@ class Gtk_Presentation(object):
         self.qty_spin.set_increments(1, 10)
         self.qty_spin.set_value(config['files']['count'])
 
-        # "Start" button
-        self.start_btn = gtk.Button(_('Start'))
-        self.start_btn.connect('released', self.start_btn_released, start_func)
-
-        # "Stop" button
-        self.stop_btn = gtk.Button(_('Stop'))
-        self.stop_btn.set_sensitive(False)
-        self.stop_btn.connect('clicked', self.stop_btn_clicked)
+        # "Filter" label
+        filter_label = gtk.Label(_('Filter (semicolon separated)'))
+        # "Filter" entry
+        self.filter_entry = gtk.Entry()
+        self.filter_entry.set_text(config['files']['filter'])
 
+        # "Output" label
+        out_label = gtk.Label(_('Output'))
         # Output selection
         btn = gtk.RadioButton(None, OUTTYPES[0][1])
         btn.set_name(OUTTYPES[0][0])
@@ -157,37 +184,59 @@ class Gtk_Presentation(object):
             btn.set_name(name)
             self.out_rbtns.append(btn)
 
-        hbox1 = gtk.HBox(False, 2)
-        hbox1.pack_start(qty_label, False, False, 0)
-        hbox1.pack_start(self.qty_spin, False, False, 0)
+        # "Start" button
+        self.start_btn = gtk.Button(_('Start'))
+        self.start_btn.connect('released', self.start_btn_released, start_func)
+        # "Stop" button
+        self.stop_btn = gtk.Button(_('Stop'))
+        self.stop_btn.set_sensitive(False)
+        self.stop_btn.connect('clicked', self.stop_btn_clicked)
+
+        path_hbox = gtk.HBox(False, 2)
+        path_hbox.pack_start(self.path_label, False, False, 4)
+        path_hbox.pack_start(self.path_entry, True, True, 0)
+        path_hbox.pack_start(self.browse_btn, False, False, 0)
+
+        qty_hbox = gtk.HBox(False, 2)
+        qty_hbox.pack_start(qty_label, False, False, 4)
+        qty_hbox.pack_start(self.qty_spin, False, False, 0)
 
-        hbox2 = gtk.HBox(False, 2)
+        filter_hbox = gtk.HBox(False, 2)
+        filter_hbox.pack_start(filter_label, False, False, 4)
+        filter_hbox.pack_start(self.filter_entry, True, True, 0)
+
+        out_hbox = gtk.HBox(False, 2)
+        out_hbox.pack_start(out_label, False, False, 4)
         for btn in self.out_rbtns:
-            hbox2.pack_start(btn, False, False, 0)
+            out_hbox.pack_start(btn, False, False, 0)
             # Activate radio button
             if btn.get_name() == config['outtype']:
                 btn.set_active(True)
 
-        hbox3 = gtk.HBox(True, 2)
-        hbox3.pack_start(self.start_btn, True, True, 0)
-        hbox3.pack_start(self.stop_btn, True, True, 0)
+        control_hbox = gtk.HBox(True, 2)
+        control_hbox.pack_start(self.start_btn, True, True, 0)
+        control_hbox.pack_start(self.stop_btn, True, True, 0)
 
         cr_vbox = gtk.VBox(False, 2)
-        cr_vbox.pack_start(self.path_entry, False, False, 0)
-        cr_vbox.pack_start(hbox1, False, False, 0)
-        cr_vbox.pack_start(hbox2, False, False, 0)
-        cr_vbox.pack_end(hbox3, False, False, 0)
-        self.nb.append_page(cr_vbox, gtk.Label(_('Criteria')))
+        cr_vbox.set_border_width(2)
+        cr_vbox.pack_start(path_hbox, False, False, 0)
+        cr_vbox.pack_start(qty_hbox, False, False, 0)
+        cr_vbox.pack_start(filter_hbox, False, False, 0)
+        cr_vbox.pack_start(out_hbox, False, False, 0)
+        cr_vbox.pack_end(control_hbox, False, False, 0)
+
+        self.nb.new_tab(cr_vbox, _('Criteria'), noclose=True)
 
         #====================
         # Others
         #====================
 
         self.statusbar = gtk.Statusbar()
+        self.statusbar.set_has_resize_grip(False)
         self.context_id = self.statusbar.get_context_id('Current walked file')
 
-        self.vbox = gtk.VBox(False, 2)
-        self.vbox.pack_start(self.nb, True, True, 0)
+        self.vbox = gtk.VBox()
+        self.vbox.pack_start(self.nb.notebook, True, True, 0)
         self.vbox.pack_end(self.statusbar, False, False, 0)
 
 #        self.show_out_toplevel(config['outtype'], [(1, 'path', 'bytesize')])
@@ -195,7 +244,22 @@ class Gtk_Presentation(object):
         self.toplevel = self.vbox
 
     #=== Functions ============================================================
+    def browse_btn_clicked(self, btn):
+        """Open directory browser. "Browse" button clicked callback."""
+        dialog = gtk.FileChooserDialog(title=_('Choose directory'),
+                                       action='select-folder',
+                                       buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
+                                                gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
+        path = abspath(self.path_entry.get_text())
+        dialog.set_current_folder(path)
+        dialog.show_all()
+        response = dialog.run()
+        if response == gtk.RESPONSE_OK:
+            self.path_entry.set_text(dialog.get_filename())
+        dialog.destroy()
+
     def start_btn_released(self, btn, start_func):
+        """Start file search. Button "Go" activate callback."""
         self.stopit = False
         self.stop_btn.set_sensitive(True)
         self.start_btn.set_sensitive(False)
@@ -204,45 +268,45 @@ class Gtk_Presentation(object):
         self.start_btn.set_sensitive(True)
 
     def stop_btn_clicked(self, widget):
+        """Stop search. "Stop" button clicked callback."""
         self.stopit = True
         self.stop_btn.set_sensitive(False)
         self.start_btn.set_sensitive(True)
 
     def get_criteria(self):
+        """Pick search criteria from window."""
         for btn in self.out_rbtns:
             if btn.get_active():
                 out = {}
                 out['name'] = btn.get_name()
                 out['label'] = btn.get_label()
-        return out, self.path_entry.get_text(), int(self.qty_spin.get_value())
+        file_filter = self.filter_entry.get_text().split(';')
+        # If no filter - show all files
+        if file_filter == ['']:
+            file_filter = ['*']
+        return out, \
+            self.path_entry.get_text(), int(self.qty_spin.get_value()), \
+            file_filter
 
     def get_stopit(self):
         return self.stopit
 
     def show_current_status(self, current_path):
+        """Show current walked path in statusbar and update window."""
         self.statusbar.push(self.context_id, current_path)
         gtk.main_iteration()
 
-    def _new_page(self, child, label):
-        self.nb.append_page(child, gtk.Label(label))
-        #self.nb.set_current_page(-1)
-        #child.grab_focus()
-
-    def _close_page(self):
-        pass
+    def nothing_founded(self):
+        self.statusbar.push(self.context_id, _('Nothing founded!'))
 
     def run(self):
         pass
 
     #=== Output type selecting ================================================
     def show_out_toplevel(self, outtype, results):
-        print 'Entering <' + outtype['name'] + '> output mode...'
         out_submodule = __import__('files.' + outtype['name'], None, None, outtype)
-
         self.out_toplevel = out_submodule.Gtk_Presentation(results).toplevel
-
-        self._new_page(self.out_toplevel, outtype['label'])
-        self.out_toplevel.show_all()
+        self.nb.new_tab(self.out_toplevel, outtype['label'])
 ###        out_submodule.Gtk_Presentation().show_results(results)
 
 #==============================================================================