From 647000c15adc52e6afa718eb4622c12db1b24a1e Mon Sep 17 00:00:00 2001 From: Jason Zedlitz Date: Mon, 13 Oct 2003 21:06:29 +0000 Subject: [PATCH] initial version svn: r2248 --- gramps2/src/plugins/Partition.py | 228 +++++++++++++++++++++++++++++++ 1 file changed, 228 insertions(+) create mode 100644 gramps2/src/plugins/Partition.py diff --git a/gramps2/src/plugins/Partition.py b/gramps2/src/plugins/Partition.py new file mode 100644 index 000000000..d297be39c --- /dev/null +++ b/gramps2/src/plugins/Partition.py @@ -0,0 +1,228 @@ +# Copyright (C) 2003 Jesper Zedlitz +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +# + +"Export/Partition" + +#------------------------------------------------------------------------ +# +# standard python modules +# +#------------------------------------------------------------------------ +import os +import posixpath + +#------------------------------------------------------------------------ +# +# GRAMPS modules +# +#------------------------------------------------------------------------ +import Utils +import RelLib +import GrampsCfg + +from QuestionDialog import ErrorDialog +from gettext import gettext as _ + +#------------------------------------------------------------------------ +# +# GNOME/GTK modules +# +#------------------------------------------------------------------------ +import gobject +import gtk +import gtk.glade +from gnome.ui import * +import WriteXML + +personSeen = [] +familySeen = [] +database_for_unlinked_persons = RelLib.GrampsDB() +prefix = "/tmp/test" + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def work_on_person( db, person ): + global personSeen + + if (len(person.getFamilyList()) + len(person.getParentList())) > 0: + database = db + else: + database = database_for_unlinked_persons + + if( database.getPersonMap().has_key( person.getId() ) ): + return + + database.addPersonNoMap( person, person.getId() ) + personSeen.append(person) + + for source_ref in person.getPrimaryName().getSourceRefList(): + work_on_sourceref( database, source_ref) + + for name in person.getAlternateNames(): + for source_ref in name.getSourceRefList(): + work_on_sourceref( database, source_ref) + + work_on_event( database, person.getBirth() ) + work_on_event( database, person.getDeath() ) + for event in person.getEventList(): + work_on_event(database, event) + + # recursion + for fam in person.getFamilyList(): + work_on_family( database, fam ) + + for fam in person.getParentList(): + work_on_family( database, fam[0] ) + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def work_on_family( database, family ): + global familySeen + if database.getFamilyMap().has_key( family.getId() ): + return + + database.getFamilyMap()[family.getId()] = family + familySeen.append(family) + + work_on_event( database, family.getMarriage() ) + for event in family.getEventList(): + work_on_event(database, event) + + + # recursion + father = family.getFather() + if( father != None ): + work_on_person( database, father ) + + mother = family.getMother() + if( mother != None ): + work_on_person( database, mother ) + + for person in family.getChildList(): + work_on_person( database, person ) + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def work_on_sourceref( database, source_ref ): + if source_ref == None: + return + + source = source_ref.getBase() + if source == None: + return + + if database.getSourceMap().has_key(source.getId()): + return + + database.addSourceNoMap( source, source.getId() ); + + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def work_on_event( database, event ): + if event == None: + return + + for source_ref in event.getSourceRefList(): + work_on_sourceref( database, source_ref) + + work_on_place( database, event.getPlace() ) + + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def work_on_place( database, place ): + if place == None: + return + + if place in database.getPlaces(): + return + database.addPlaceNoMap(place, place.getId()) + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +def report(db, person): + global personSeen + global prefix + global database_for_unlinked_persons + + text = "=== Partitions ===\n" + count = 0 + for p in db.getPersonMap().values(): + if not p in personSeen: + database = RelLib.GrampsDB() + work_on_person( database, p ) + + person_len = len(database.getPersonKeys()) + if person_len > 0: + g = WriteXML.XmlWriter(database,None,0,0) + g.write(prefix+str(count)+".xml") + text += "partition "+prefix+str(count)+".xml written ( "+str(person_len)+" persons)\n" + count += 1 + + g = WriteXML.XmlWriter(database_for_unlinked_persons,None,0,0) + g.write(prefix+".xml") + text += "partition "+prefix+".xml written ( "+str(len(database_for_unlinked_persons.getPersonKeys()))+" persons)\n" + + base = os.path.dirname(__file__) + glade_file = "%s/summary.glade" % base + + topDialog = gtk.glade.XML(glade_file,"summary","gramps") + topDialog.signal_autoconnect({ + "destroy_passed_object" : Utils.destroy_passed_object, + }) + + top = topDialog.get_widget("summary") + textwindow = topDialog.get_widget("textwindow") + textwindow.get_buffer().set_text(text) + top.show() + +#------------------------------------------------------------------------- +# +# +# +#------------------------------------------------------------------------- +from Plugins import register_report + +register_report( + report, + _("Partitions"), + status=(_("Alpha")), + category=_("Export"), + description=_("This program partitions individuals in a database into disjoint partitions.\n"+ + "A partition is composed of people related by one or more multiple relations.\n"+ + "There should be no known relationship between people in different partitions.") + ) +