gramps/src/plugins/WritePkg.py

216 lines
7.6 KiB
Python
Raw Normal View History

2002-10-20 19:55:16 +05:30
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2004 Donald N. Allingham
2002-10-20 19:55:16 +05:30
#
# 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$
2002-10-20 19:55:16 +05:30
"Export to GRAMPS package"
#-------------------------------------------------------------------------
#
# standard python modules
#
#-------------------------------------------------------------------------
import time
import os
from cStringIO import StringIO
#-------------------------------------------------------------------------
#
# GNOME/GTK modules
#
#-------------------------------------------------------------------------
import gtk
import gtk.glade
import gnome
2002-10-20 19:55:16 +05:30
#-------------------------------------------------------------------------
#
# GRAMPS modules
#
#-------------------------------------------------------------------------
import WriteXML
import TarFile
import Utils
2003-05-15 00:51:18 +05:30
from QuestionDialog import MissingMediaDialog
2002-10-20 19:55:16 +05:30
2003-08-17 07:44:33 +05:30
from gettext import gettext as _
2002-10-20 19:55:16 +05:30
_title_string = _("Export to GRAMPS package")
2002-10-20 19:55:16 +05:30
#-------------------------------------------------------------------------
#
# writeData
#
#-------------------------------------------------------------------------
def writeData(database,person):
try:
PackageWriter(database)
except:
import DisplayTrace
DisplayTrace.DisplayTrace()
#-------------------------------------------------------------------------
#
# PackageWriter
#
#-------------------------------------------------------------------------
class PackageWriter:
def __init__(self,database):
self.db = database
2003-03-19 09:57:34 +05:30
choose = gtk.FileChooserDialog(_('Save GRAMPS Package'),
None,
gtk.FILE_CHOOSER_ACTION_SAVE,
(gtk.STOCK_CANCEL,
gtk.RESPONSE_CANCEL,
gtk.STOCK_OPEN,
gtk.RESPONSE_OK))
2003-03-19 09:57:34 +05:30
choose.set_local_only(gtk.FALSE)
filter = gtk.FileFilter()
filter.set_name(_('GRAMPS package'))
filter.add_pattern('*.gpkg')
choose.add_filter(filter)
2002-10-20 19:55:16 +05:30
response = choose.run()
2002-10-20 19:55:16 +05:30
if response == gtk.RESPONSE_OK:
name = choose.get_filename()
if os.path.splitext(name)[1] != ".gpkg":
name = name + ".gpkg"
choose.destroy()
self.export(name)
else:
choose.destroy()
def on_help_clicked(self,obj):
"""Display the relevant portion of GRAMPS manual"""
gnome.help_display('gramps-manual','export-data')
2002-10-20 19:55:16 +05:30
def export(self, filename):
missmedia_action = 0
2003-05-15 00:51:18 +05:30
#--------------------------------------------------------------
def remove_clicked():
# File is lost => remove all references and the object itself
2004-04-09 09:35:48 +05:30
mobj = self.db.find_family_from_id(ObjectId)
for p_id in self.db.get_family_keys():
p = self.db.find_family_from_id(p_id)
2004-02-21 11:41:59 +05:30
nl = p.get_media_list()
2003-05-15 00:51:18 +05:30
for o in nl:
if o.get_reference() == mobj:
2003-05-15 00:51:18 +05:30
nl.remove(o)
2004-02-21 11:41:59 +05:30
p.set_media_list(nl)
for key in self.db.get_person_keys():
p = self.db.try_to_find_person_from_id(key)
2004-02-21 11:41:59 +05:30
nl = p.get_media_list()
2003-05-15 00:51:18 +05:30
for o in nl:
if o.get_reference() == mobj:
2003-05-15 00:51:18 +05:30
nl.remove(o)
2004-02-21 11:41:59 +05:30
p.set_media_list(nl)
for key in self.db.get_source_keys():
p = self.db.try_to_find_source_from_id(key)
2004-02-21 11:41:59 +05:30
nl = p.get_media_list()
2003-05-15 00:51:18 +05:30
for o in nl:
if o.get_reference() == mobj:
2003-05-15 00:51:18 +05:30
nl.remove(o)
2004-02-21 11:41:59 +05:30
p.set_media_list(nl)
for key in self.db.get_place_id_keys():
p = self.db.try_to_find_place_from_id(key)
2004-02-21 11:41:59 +05:30
nl = p.get_media_list()
2003-05-15 00:51:18 +05:30
for o in nl:
if o.get_reference() == mobj:
2003-05-15 00:51:18 +05:30
nl.remove(o)
2004-02-21 11:41:59 +05:30
p.set_media_list(nl)
self.db.remove_object(ObjectId)
2003-05-15 00:51:18 +05:30
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
2003-05-15 00:51:18 +05:30
def fs_ok_clicked(obj):
name = fs_top.get_filename()
if os.path.isfile(name):
g = open(name,"rb")
t.add_file(base,mtime,g)
g.close()
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()
2003-05-15 00:51:18 +05:30
#---------------------------------------------------------------
2002-10-20 19:55:16 +05:30
t = TarFile.TarFile(filename)
2003-05-15 00:51:18 +05:30
mtime = time.time()
# Write media files first, since the database may be modified
# during the process (i.e. when removing object)
2004-04-09 09:35:48 +05:30
for ObjectId in self.db.get_object_keys():
oldfile = self.db.try_to_find_object_from_id(ObjectId).get_path()
2003-05-15 00:51:18 +05:30
base = os.path.basename(oldfile)
if os.path.isfile(oldfile):
g = open(oldfile,"rb")
t.add_file(base,mtime,g)
g.close()
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' : oldfile },
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()
2002-10-20 19:55:16 +05:30
2003-05-15 00:51:18 +05:30
# Write XML now
g = StringIO()
2002-10-20 19:55:16 +05:30
gfile = WriteXML.XmlWriter(self.db,None,1)
gfile.write_handle(g)
mtime = time.time()
t.add_file("data.gramps",mtime,g)
g.close()
t.close()
#-------------------------------------------------------------------------
#
# Register the plugin
#
#-------------------------------------------------------------------------
from Plugins import register_export
register_export(writeData,_title_string)