15b2843d93
* src/plugins/ReadPkg.py: attempt to os.remove(directory) failed * src/GrampsDb/_GrampsDbFactories.py: import from gen.db svn: r9520
128 lines
4.4 KiB
Python
128 lines
4.4 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2006 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
|
|
#
|
|
# $Id$
|
|
|
|
# Written by Alex Roitman, largely based on ReadNative.py by Don Allingham
|
|
|
|
"Import from Gramps package"
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import os
|
|
import tarfile
|
|
from gettext import gettext as _
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Set up logging
|
|
#
|
|
#------------------------------------------------------------------------
|
|
import logging
|
|
log = logging.getLogger(".ReadPkg")
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GNOME/GTK+ modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GRAMPS modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import const
|
|
from GrampsDbUtils import gramps_db_reader_factory
|
|
from QuestionDialog import ErrorDialog
|
|
from PluginUtils import register_import
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
def impData(database, name, cb=None, cl=0):
|
|
# Create tempdir, if it does not exist, then check for writability
|
|
# THE TEMP DIR is named as the filname.gpkg.media and is created
|
|
# in the same dir as the database that we are importing into.
|
|
db_path = os.path.dirname(database.get_save_path())
|
|
media_dir = "%s.media" % os.path.basename(name)
|
|
tmpdir_path = os.path.join(db_path, media_dir)
|
|
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
|
|
else: # tempdir exists and writable -- clean it up if not empty
|
|
files = os.listdir(tmpdir_path) ;
|
|
for filename in files:
|
|
try:
|
|
os.remove(os.path.join(tmpdir_path, filename))
|
|
except OSError:
|
|
try:
|
|
os.removedirs(os.path.join(tmpdir_path, filename))
|
|
except:
|
|
print "could not remove: '%s'" % os.path.join(tmpdir_path, filename)
|
|
try:
|
|
archive = tarfile.open(name)
|
|
for tarinfo in archive:
|
|
archive.extract(tarinfo, tmpdir_path)
|
|
archive.close()
|
|
except:
|
|
ErrorDialog(_("Error extracting into %s") % tmpdir_path)
|
|
return
|
|
|
|
imp_db_name = os.path.join(tmpdir_path, const.XMLFILE)
|
|
|
|
importer = gramps_db_reader_factory(const.APP_GRAMPS_XML)
|
|
importer(database, imp_db_name, cb)
|
|
|
|
# Clean up tempdir after ourselves
|
|
# THIS HAS BEEN CHANGED, because now we want to keep images
|
|
# stay after the import is over. Just delete the XML file.
|
|
os.remove(imp_db_name)
|
|
|
|
## files = os.listdir(tmpdir_path)
|
|
## for filename in files:
|
|
## os.remove(os.path.join(tmpdir_path, filename))
|
|
## os.rmdir(tmpdir_path)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Register with the plugin system
|
|
#
|
|
#------------------------------------------------------------------------
|
|
_mime_type = 'application/x-gramps-package'
|
|
_filter = gtk.FileFilter()
|
|
_filter.set_name(_('GRAMPS packages'))
|
|
_filter.add_mime_type(_mime_type)
|
|
_format_name = _('GRAMPS package')
|
|
|
|
register_import(impData, _filter, _mime_type, 0, _format_name)
|