134 lines
5.2 KiB
Python
134 lines
5.2 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2006 Donald N. Allingham
|
|
# Copyright (C) 2008 Brian G. Matherly
|
|
#
|
|
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
#
|
|
|
|
# Written by Alex Roitman, largely based on ReadNative.py by Don Allingham
|
|
|
|
"Import from Gramps package"
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import os
|
|
import tarfile
|
|
from gramps.gen.const import GRAMPS_LOCALE as glocale
|
|
_ = glocale.translation.gettext
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Set up logging
|
|
#
|
|
#------------------------------------------------------------------------
|
|
import logging
|
|
log = logging.getLogger(".ReadPkg")
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Gramps modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from gramps.gen.const import XMLFILE
|
|
from gramps.gen.utils.file import media_path
|
|
## we need absolute import as this is dynamically loaded:
|
|
from gramps.plugins.importer.importxml import importData
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
#
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
def impData(database, name, user):
|
|
# 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 mediapath dir of the family tree we import to
|
|
oldmediapath = database.get_mediapath()
|
|
#use home dir if no media path
|
|
my_media_path = media_path(database)
|
|
media_dir = "%s.media" % os.path.basename(name)
|
|
tmpdir_path = os.path.join(my_media_path, media_dir)
|
|
if not os.path.isdir(tmpdir_path):
|
|
try:
|
|
os.mkdir(tmpdir_path, 0o700)
|
|
except:
|
|
user.notify_error( _("Could not create media directory %s") %
|
|
tmpdir_path )
|
|
return
|
|
elif not os.access(tmpdir_path, os.W_OK):
|
|
user.notify_error(_("Media directory %s is not writable") % tmpdir_path)
|
|
return
|
|
else:
|
|
# mediadir exists and writable -- User could have valuable stuff in
|
|
# it, have him remove it!
|
|
user.notify_error(_("Media directory %s exists. Delete it first, then"
|
|
" restart the import process") % tmpdir_path)
|
|
return
|
|
try:
|
|
archive = tarfile.open(name)
|
|
for tarinfo in archive:
|
|
archive.extract(tarinfo, tmpdir_path)
|
|
archive.close()
|
|
except:
|
|
user.notify_error(_("Error extracting into %s") % tmpdir_path)
|
|
return
|
|
|
|
imp_db_name = os.path.join(tmpdir_path, XMLFILE)
|
|
|
|
importer = importData
|
|
info = importer(database, imp_db_name, user)
|
|
|
|
newmediapath = database.get_mediapath()
|
|
#import of gpkg should not change media path as all media has new paths!
|
|
if not oldmediapath == newmediapath :
|
|
database.set_mediapath(oldmediapath)
|
|
|
|
# Set correct media dir if possible, complain if problems
|
|
if oldmediapath is None:
|
|
database.set_mediapath(tmpdir_path)
|
|
user.warn(
|
|
_("Base path for relative media set"),
|
|
_("The base media path of this Family Tree has been set to "
|
|
"%s. Consider taking a simpler path. You can change this "
|
|
"in the Preferences, while moving your media files to the "
|
|
"new position, and using the media manager tool, option "
|
|
"'Replace substring in the path' to set"
|
|
" correct paths in your media objects."
|
|
) % tmpdir_path)
|
|
else:
|
|
user.warn(
|
|
_("Cannot set base media path"),
|
|
_("The Family Tree you imported into already has a base media "
|
|
"path: %(orig_path)s. The imported media objects however "
|
|
"are relative from the path %(path)s. You can change the "
|
|
"media path in the Preferences or you can convert the "
|
|
"imported files to the existing base media path. You can "
|
|
"do that by moving your media files to the "
|
|
"new position, and using the media manager tool, option "
|
|
"'Replace substring in the path' to set"
|
|
" correct paths in your media objects."
|
|
) % {'orig_path': oldmediapath, 'path': tmpdir_path}
|
|
)
|
|
|
|
# Remove xml file extracted to media dir we imported from
|
|
os.remove(imp_db_name)
|
|
|
|
return info
|