initial version

svn: r2248

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.")
+    )
+