fe6840666c
Import of gpkg exports media in home dir, sets base media path to that dir, so user can easily change position with move in filemanager, and change of base media path 0001966: import of gpkg copies media files to hidden dir & not accessible svn: r10366
230 lines
8.4 KiB
Python
230 lines
8.4 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2008 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$
|
|
|
|
"Export to GRAMPS package"
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# standard python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import time
|
|
import os
|
|
import sys
|
|
import tarfile
|
|
from cStringIO import StringIO
|
|
from gettext import gettext as _
|
|
import ExportOptions
|
|
#from BasicUtils import UpdateCallback
|
|
import gen.proxy
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Set up logging
|
|
#
|
|
#------------------------------------------------------------------------
|
|
import logging
|
|
log = logging.getLogger(".WritePkg")
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GNOME/GTK modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GRAMPS modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from GrampsDbUtils import XmlWriter
|
|
from PluginUtils import register_export
|
|
import Utils
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# writeData
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
def writeData(database, filename, person, option_box, callback=None):
|
|
option_box.parse_options()
|
|
|
|
restrict = option_box.restrict
|
|
private = option_box.private
|
|
|
|
if private:
|
|
database = gen.proxy.PrivateProxyDb(database)
|
|
|
|
if restrict:
|
|
database = gen.proxy.LivingProxyDb(
|
|
database, gen.proxy.LivingProxyDb.MODE_RESTRICT)
|
|
|
|
if not option_box.cfilter.is_empty():
|
|
database = gen.proxy.FilterProxyDb(database, option_box.cfilter)
|
|
|
|
writer = PackageWriter(database, filename, callback)
|
|
return writer.export()
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# PackageWriter
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
class PackageWriter:
|
|
|
|
def __init__(self, database, filename, callback=None):
|
|
self.db = database
|
|
self.callback = callback
|
|
self.filename = filename
|
|
|
|
def export(self):
|
|
# missmedia_action = 0
|
|
#--------------------------------------------------------------
|
|
def remove_clicked():
|
|
# File is lost => remove all references and the object itself
|
|
for p_id in self.db.get_family_handles():
|
|
p = self.db.get_family_from_handle(p_id)
|
|
nl = p.get_media_list()
|
|
for o in nl:
|
|
if o.get_reference_handle() == m_id:
|
|
nl.remove(o)
|
|
p.set_media_list(nl)
|
|
self.db.commit_family(p,None)
|
|
for key in self.db.get_person_handles(sort_handles=False):
|
|
p = self.db.get_person_from_handle(key)
|
|
nl = p.get_media_list()
|
|
for o in nl:
|
|
if o.get_reference_handle() == m_id:
|
|
nl.remove(o)
|
|
p.set_media_list(nl)
|
|
self.db.commit_person(p,None)
|
|
for key in self.db.get_source_handles():
|
|
p = self.db.get_source_from_handle(key)
|
|
nl = p.get_media_list()
|
|
for o in nl:
|
|
if o.get_reference_handle() == m_id:
|
|
nl.remove(o)
|
|
p.set_media_list(nl)
|
|
self.db.commit_source(p,None)
|
|
for key in self.db.get_place_handles():
|
|
p = self.db.get_place_from_handle(key)
|
|
nl = p.get_media_list()
|
|
for o in nl:
|
|
if o.get_reference_handle() == m_id:
|
|
nl.remove(o)
|
|
p.set_media_list(nl)
|
|
self.db.commit_place(p,None)
|
|
for key in self.db.get_event_handles():
|
|
p = self.db.get_event_from_handle(key)
|
|
nl = p.get_media_list()
|
|
for o in nl:
|
|
if o.get_reference_handle() == m_id:
|
|
nl.remove(o)
|
|
p.set_media_list(nl)
|
|
self.db.commit_event(p,None)
|
|
self.db.remove_object(m_id,None)
|
|
|
|
def leave_clicked():
|
|
# File is lost => do nothing, leave as is
|
|
pass
|
|
|
|
def select_clicked():
|
|
# File is lost => select a file to replace the lost one
|
|
def fs_close_window(obj):
|
|
pass
|
|
|
|
def fs_ok_clicked(obj):
|
|
name = unicode(fs_top.get_filename(),
|
|
sys.getfilesystemencoding())
|
|
if os.path.isfile(name):
|
|
archive.add(name)
|
|
|
|
fs_top = gtk.FileSelection("%s - GRAMPS" % _("Select file"))
|
|
fs_top.hide_fileop_buttons()
|
|
fs_top.ok_button.connect('clicked',fs_ok_clicked)
|
|
fs_top.cancel_button.connect('clicked',fs_close_window)
|
|
fs_top.run()
|
|
fs_top.destroy()
|
|
#---------------------------------------------------------------
|
|
|
|
archive = tarfile.open(self.filename,'w:gz')
|
|
|
|
# Write media files first, since the database may be modified
|
|
# during the process (i.e. when removing object)
|
|
for m_id in self.db.get_media_object_handles():
|
|
mobject = self.db.get_object_from_handle(m_id)
|
|
filename = str(Utils.media_path_full(self.db, mobject.get_path()))
|
|
archname = str(mobject.get_path())
|
|
if os.path.isfile(filename):
|
|
archive.add(filename, archname)
|
|
# else:
|
|
# # File is lost => ask what to do
|
|
# if missmedia_action == 0:
|
|
# mmd = MissingMediaDialog(
|
|
# _("Media object could not be found"),
|
|
# _("%(file_name)s is referenced in the database, "
|
|
# "but no longer exists. The file may have been "
|
|
# "deleted or moved to a different location. "
|
|
# "You may choose to either remove the reference "
|
|
# "from the database, keep the reference to the "
|
|
# "missing file, or select a new file."
|
|
# ) % { 'file_name' : filename },
|
|
# remove_clicked, leave_clicked, select_clicked)
|
|
# missmedia_action = mmd.default_action
|
|
# elif missmedia_action == 1:
|
|
# remove_clicked()
|
|
# elif missmedia_action == 2:
|
|
# leave_clicked()
|
|
# elif missmedia_action == 3:
|
|
# select_clicked()
|
|
|
|
# Write XML now
|
|
g = StringIO()
|
|
gfile = XmlWriter(self.db, self.callback, 2)
|
|
gfile.write_handle(g)
|
|
tarinfo = tarfile.TarInfo('data.gramps')
|
|
tarinfo.size = len(g.getvalue())
|
|
tarinfo.mtime = time.time()
|
|
if os.sys.platform != "win32":
|
|
tarinfo.uid = os.getuid()
|
|
tarinfo.gid = os.getgid()
|
|
g.seek(0)
|
|
archive.addfile(tarinfo, g)
|
|
archive.close()
|
|
g.close()
|
|
|
|
return True
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Register the plugin
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
_title = _('GRAM_PS package (portable XML)')
|
|
_description = _('GRAMPS package is an archived XML database together '
|
|
'with the media object files.')
|
|
_config = (_('GRAMPS package export options'), ExportOptions.WriterOptionBox)
|
|
_filename = 'gpkg'
|
|
|
|
register_export(writeData, _title, _description, _config, _filename)
|