From a270b514e62d0764862c154a94552fd08aa86756 Mon Sep 17 00:00:00 2001 From: Don Allingham Date: Fri, 12 May 2006 04:51:44 +0000 Subject: [PATCH] 2006-05-11 Don Allingham * src/DataViews/_PersonView.py: add additional fields, and build filter * src/PeopleModel.py: handle both search and filters * src/Filters/_GenericFilter.py: import RelLib svn: r6625 --- ChangeLog | 6 +++ src/DataViews/_PersonView.py | 87 ++++++++++++++++++++++++++++++----- src/Filters/_GenericFilter.py | 2 + src/PeopleModel.py | 56 +++++++++++++++------- 4 files changed, 123 insertions(+), 28 deletions(-) diff --git a/ChangeLog b/ChangeLog index f8ddd1cec..b024c318e 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,9 @@ +2006-05-11 Don Allingham + * src/DataViews/_PersonView.py: add additional fields, and build + filter + * src/PeopleModel.py: handle both search and filters + * src/Filters/_GenericFilter.py: import RelLib + 2006-05-11 Alex Roitman * src/UndoHistory.py (gtk_color_to_str): Correct byte color. (UndoHistory._selection_changed): Enable/disable buttons. diff --git a/src/DataViews/_PersonView.py b/src/DataViews/_PersonView.py index 920215af3..0bbeeb0ee 100644 --- a/src/DataViews/_PersonView.py +++ b/src/DataViews/_PersonView.py @@ -62,7 +62,9 @@ import Config import const from Editors import EditPerson -from Filters import SearchBar +from Filters import SearchBar, GenericFilter +from Filters.Rules.Person import * + from DdTargets import DdTargets column_names = [ @@ -89,6 +91,7 @@ class PersonView(PageView.PersonNavView): dbstate.connect('active-changed',self.goto_active_person) self.handle_col = PeopleModel.COLUMN_INT_ID self.model = None + self.generic_filter = None self.func_list = { 'F2' : self.key_goto_home_person, @@ -213,44 +216,99 @@ class PersonView(PageView.PersonNavView): return hpaned def build_filter_sidebar(self): - table = gtk.Table(3,8) + table = gtk.Table(3,10) table.set_border_width(6) - table.set_col_spacings(6) table.set_row_spacings(6) self.filter_name = gtk.Entry() self.filter_id = gtk.Entry() self.filter_birth = gtk.Entry() self.filter_death = gtk.Entry() + self.filter_event = gtk.Entry() + self.filter_source = gtk.Entry() + self.filter_note = gtk.Entry() + self.filter_gender = gtk.Entry() + self.filter_regex = gtk.CheckButton(_('Use regular expressions')) + self.apply_btn = gtk.Button(stock=gtk.STOCK_FIND) + self.apply_btn.connect('clicked', self.filter_clicked) + + table.set_col_spacing(0,6) + table.set_col_spacing(1,6) table.attach(GrampsWidgets.MarkupLabel(_('Filter')), 0, 3, 0, 1, xoptions=gtk.FILL, yoptions=0) table.attach(GrampsWidgets.BasicLabel(_('Name')), 1, 2, 1, 2, xoptions=gtk.FILL, yoptions=0) + table.attach(self.filter_name, 2, 3, 1, 2, xoptions=gtk.FILL, yoptions=0) table.attach(GrampsWidgets.BasicLabel(_('ID')), 1, 2, 2, 3, xoptions=gtk.FILL, yoptions=0) + table.attach(self.filter_id, 2, 3, 2, 3, xoptions=gtk.FILL, yoptions=0) table.attach(GrampsWidgets.BasicLabel(_('Birth')), - 1,2,3,4, xoptions=gtk.FILL, yoptions=0) + 1, 2, 3, 4, xoptions=gtk.FILL, yoptions=0) + table.attach(self.filter_birth, 2, 3, 3, 4, xoptions=gtk.FILL, yoptions=0) table.attach(GrampsWidgets.BasicLabel(_('Death')), 1, 2, 4, 5, xoptions=gtk.FILL, yoptions=0) + table.attach(self.filter_death, 2, 3, 4, 5, xoptions=gtk.FILL, yoptions=0) - table.attach(self.apply_btn, 2, 3, 5, 6, xoptions=gtk.FILL, + table.attach(GrampsWidgets.BasicLabel(_('Event')), + 1, 2, 5, 6, xoptions=gtk.FILL, yoptions=0) + + table.attach(self.filter_event, 2, 3, 5, 6, + xoptions=gtk.FILL, yoptions=0) + + table.attach(GrampsWidgets.BasicLabel(_('Source')), + 1, 2, 6, 7, xoptions=gtk.FILL, yoptions=0) + + table.attach(self.filter_source, 2, 3, 6, 7, + xoptions=gtk.FILL, yoptions=0) + + table.attach(GrampsWidgets.BasicLabel(_('Note')), + 1, 2, 7, 8, xoptions=gtk.FILL, yoptions=0) + + table.attach(self.filter_note, 2, 3, 7, 8, + xoptions=gtk.FILL, yoptions=0) + + table.attach(self.filter_regex, 2, 3, 8, 9, xoptions=gtk.FILL, + yoptions=0) + table.attach(self.apply_btn, 2, 3, 9, 10, xoptions=0, yoptions=0) return table + + def filter_clicked(self, obj): + name = self.filter_name.get_text().strip() + gid = self.filter_id.get_text().strip() + birth = self.filter_birth.get_text().strip() + death = self.filter_death.get_text().strip() + event = self.filter_event.get_text().strip() + source = self.filter_source.get_text().strip() + note = self.filter_note.get_text().strip() + gender = self.filter_gender.get_text().strip() + regex = self.filter_regex.get_active() + + if not name and not gid and not death and not event and \ + not source and not note and not gender: + self.generic_filter = None + else: + self.generic_filter = GenericFilter() + if name: + rule = SearchName([name]) + self.generic_filter.add_rule(rule) + self.build_tree() + print self.generic_filter def drag_begin(self, widget, *data): widget.drag_source_set_icon_stock(self.get_stock()) @@ -417,9 +475,10 @@ class PersonView(PageView.PersonNavView): search = (0, '', False) else: search = self.search_bar.get_value() + self.generic_filter = None self.model = PeopleModel.PeopleModel( - self.dbstate.db, None, search) + self.dbstate.db, self.generic_filter, search) self.tree.set_model(self.model) @@ -565,10 +624,13 @@ class PersonView(PageView.PersonNavView): continue name = column_names[pair[1]] try: - column = gtk.TreeViewColumn(name, self.renderer, markup=pair[1], - background=self.model.marker_color_column) + column = gtk.TreeViewColumn( + name, self.renderer, markup=pair[1], + background=self.model.marker_color_column) except AttributeError: - column = gtk.TreeViewColumn(name, self.renderer, markup=pair[1]) + column = gtk.TreeViewColumn( + name, self.renderer, markup=pair[1]) + column.set_resizable(True) column.set_fixed_width(pair[2]) column.set_sizing(gtk.TREE_VIEW_COLUMN_FIXED) @@ -584,11 +646,11 @@ class PersonView(PageView.PersonNavView): selected_ids = self.get_selected_objects() if not self.inactive: try: - person = self.dbstate.db.get_person_from_handle(selected_ids[0]) + handle = selected_ids[0] + person = self.dbstate.db.get_person_from_handle(handle) self.dbstate.change_active_person(person) except: pass - #self.dbstate.change_active_person(None) if len(selected_ids) == 1: self.tree.drag_source_set(BUTTON1_MASK, @@ -604,7 +666,8 @@ class PersonView(PageView.PersonNavView): selected_ids = self.get_selected_objects() nonempty_ids = [h for h in selected_ids if h] if nonempty_ids: - data = (DdTargets.PERSON_LINK.drag_type, id(self), nonempty_ids[0], 0) + data = (DdTargets.PERSON_LINK.drag_type, id(self), + nonempty_ids[0], 0) sel_data.set(sel_data.target, 8 ,pickle.dumps(data)) def person_added(self,handle_list): diff --git a/src/Filters/_GenericFilter.py b/src/Filters/_GenericFilter.py index e47ca38aa..3842beec1 100644 --- a/src/Filters/_GenericFilter.py +++ b/src/Filters/_GenericFilter.py @@ -24,6 +24,8 @@ Package providing filtering framework for GRAMPS. """ +import RelLib + #------------------------------------------------------------------------- # # GenericFilter diff --git a/src/PeopleModel.py b/src/PeopleModel.py index 9c75717c2..d7f28f5ee 100644 --- a/src/PeopleModel.py +++ b/src/PeopleModel.py @@ -148,12 +148,15 @@ class PeopleModel(gtk.GenericTreeModel): self.iter2path = {} self.path2iter = {} self.sname_sub = {} - if search: + if not data_filter: col = search[0] text = search[1] inv = search[2] func = lambda x: self.on_get_value(x, col) or u"" data_filter = SearchFilter(func, text, inv) + self._build_data = self._build_search_sub + else: + self._build_data = self._build_filter_sub self.rebuild_data(data_filter, skip) def rebuild_data(self, data_filter=None, skip=[]): @@ -163,25 +166,12 @@ class PeopleModel(gtk.GenericTreeModel): self.calculate_data(data_filter, skip) self.assign_data() - def calculate_data(self, dfilter=None, skip=[]): - """ - Calculates the new path to node values for the model. - """ - - if dfilter: - self.dfilter = dfilter - self.temp_iter2path = {} - self.temp_path2iter = {} - self.temp_sname_sub = {} - - if not self.db.is_open(): - return + def _build_search_sub(self,dfilter, skip): + self.sortnames = {} ngn = NameDisplay.displayer.name_grouping_name nsn = NameDisplay.displayer.raw_sorted_name - self.sortnames = {} - cursor = self.db.get_person_cursor() node = cursor.first() @@ -197,6 +187,40 @@ class PeopleModel(gtk.GenericTreeModel): node = cursor.next() cursor.close() + def _build_filter_sub(self,dfilter, skip): + self.sortnames = {} + + ngn = NameDisplay.displayer.name_grouping_name + nsn = NameDisplay.displayer.raw_sorted_name + + handle_list = dfilter.apply(self.db, self.db.get_person_handles()) + + for handle in handle_list: + d = self.db.get_raw_person_data(handle) + if not (handle in skip or (dfilter and not dfilter.match(handle))): + name_data = d[_NAME_COL] + self.sortnames[handle] = nsn(name_data) + try: + self.temp_sname_sub[name_data[5]].append(handle) + except: + self.temp_sname_sub[name_data[5]] = [handle] + + def calculate_data(self, dfilter=None, skip=[]): + """ + Calculates the new path to node values for the model. + """ + + if dfilter: + self.dfilter = dfilter + self.temp_iter2path = {} + self.temp_path2iter = {} + self.temp_sname_sub = {} + + if not self.db.is_open(): + return + + self._build_data(dfilter, skip) + self.temp_top_path2iter = locale_sort(self.temp_sname_sub.keys()) for name in self.temp_top_path2iter: self.build_sub_entry(name)