Import from GRAMPS package

svn: r1492

gramps2/src/ReadXML.py

@@ -108,6 +108,20 @@ def importData(database, filename, callback):
         return 0
 
     xml_file.close()
+    
+    # Rename media files if they were conflicting with existing ones
+    #print parser.MediaFileMap
+    ObjectMap = parser.db.getObjectMap()
+    for OldMediaID in parser.MediaFileMap.keys():
+        NewMediaID = parser.MediaFileMap[OldMediaID]
+        oldfile = ObjectMap[NewMediaID].getPath()
+        oldpath = os.path.dirname(oldfile)
+        (junk,oldext) = os.path.splitext(os.path.basename(oldfile))
+        ObjectMap[NewMediaID].setPath(NewMediaID + oldext)
+	if NewMediaID != OldMediaID:
+            newfile = os.path.join( basefile, NewMediaID + oldext )
+            os.rename(oldfile,newfile)
+    
     return 1
 
 #-------------------------------------------------------------------------

gramps2/src/plugins/ReadPkg.py

@@ -0,0 +1,150 @@
+#
+# Gramps - a GTK+/GNOME based genealogy program
+#
+# Copyright (C) 2000  Donald N. Allingham
+#
+# 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
+#
+
+#
+# Written by Alex Roitman, largely based on ReadNative.py by Don Allingham 
+#
+
+"Import from Gramps package"
+
+from ReadXML import *
+import Utils
+from intl import gettext as _
+import gtk
+import const
+import os
+import shutil
+import time
+from QuestionDialog import ErrorDialog, WarningDialog
+import TarFile
+
+#-------------------------------------------------------------------------
+#
+#
+#
+#-------------------------------------------------------------------------
+def readData(database,active_person,cb):
+    ReadPkg(database,active_person,cb)
+
+#-------------------------------------------------------------------------
+#
+#
+#
+#-------------------------------------------------------------------------
+class ReadPkg:
+    def __init__(self,database,active_person,cb):
+        self.db = database
+        self.callback = cb
+        
+        self.top = gtk.FileSelection("%s - GRAMPS" % _("Import from GRAMPS package"))
+        self.top.hide_fileop_buttons()
+        self.top.ok_button.connect('clicked', self.on_ok_clicked)
+        self.top.cancel_button.connect('clicked', self.close_window)
+        self.top.show()
+
+    def close_window(self,obj):
+        self.top.destroy()
+        
+    def show_display(self):
+        self.window = gtk.Window()
+        self.window.set_title(_("Import from GRAMPS package"))
+        vbox = gtk.VBox()
+        self.window.add(vbox)
+        label = gtk.Label(_("Import from GRAMPS package"))
+        vbox.add(label)
+        adj = gtk.Adjustment(lower=0,upper=100)
+        self.progress_bar = gtk.ProgressBar(adj)
+        vbox.add(self.progress_bar)
+        self.window.show_all()
+        
+    def on_ok_clicked(self,obj):
+
+        name = self.top.get_filename()
+        if name == "":
+            return
+
+        Utils.destroy_passed_object(self.top)
+        self.show_display()
+
+        # Create tempdir, if it does not exist, then check for writability
+        tmpdir_path = os.path.expanduser("~/.gramps/tmp" )
+        if not os.path.isdir(tmpdir_path):
+            try:
+                os.mkdir(tmpdir_path,0700)
+            except:
+                ErrorDialog( _("Could not create temporary directory %s") % 
+                                tmpdir_path )
+                return
+        elif not os.access(tmpdir_path,os.W_OK):
+            ErrorDialog( _("Temporary directory %s is not writable") % tmpdir_path )
+            return
+
+        # Clean up tempdir, if it is not empty
+	files = os.listdir(tmpdir_path) ;
+        for filename in files:
+            os.remove( os.path.join(tmpdir_path,filename) )
+
+        try:
+            t = TarFile.ReadTarFile(name,tmpdir_path)
+	    t.extract()
+	    t.close()
+        except:
+            ErrorDialog(_("Error extracting into %s") % tmpdir_path )
+            return
+
+	dbname = os.path.join(tmpdir_path,const.xmlFile)  
+
+        try:
+            importData(self.db,dbname,self.progress)
+        except:
+            import DisplayTrace
+            DisplayTrace.DisplayTrace()
+
+        # Clean up tempdir after ourselves
+        files = os.listdir(tmpdir_path) 
+
+        dbdir_path = self.db.getSavePath() 
+        for filename in files:
+            oldfile = os.path.join(tmpdir_path,filename)
+            newfile = os.path.join(dbdir_path,filename)
+	    if filename != const.xmlFile:
+                shutil.copy2( oldfile, newfile )
+
+            os.remove( oldfile )
+
+        os.rmdir(tmpdir_path)
+
+        self.window.destroy()
+        self.callback(1)
+
+    def progress(self,val):
+        self.progress_bar.set_value(val*100.0)
+        while gtk.events_pending():
+            gtk.mainiteration()
+        
+#------------------------------------------------------------------------
+#
+# Register with the plugin system
+#
+#------------------------------------------------------------------------
+from Plugins import register_import
+
+register_import(readData,_("Import from GRAMPS package"))
+