Command line imports: gedcom and iso

svn: r1552
This commit is contained in:
Alex Roitman
2003-05-20 20:31:16 +00:00
parent 7eded566a4
commit f5c9c8f2cd
3 changed files with 207 additions and 105 deletions

View File

@@ -71,26 +71,30 @@ def writeData(database,person):
#-------------------------------------------------------------------------
class PackageWriter:
def __init__(self,database):
def __init__(self,database,cl=0,name=""):
self.db = database
self.cl = cl
self.name = name
if self.cl:
self.cl_run()
else:
base = os.path.dirname(__file__)
glade_file = "%s/%s" % (base,"cdexport.glade")
base = os.path.dirname(__file__)
glade_file = "%s/%s" % (base,"cdexport.glade")
dic = {
"destroy_passed_object" : Utils.destroy_passed_object,
"on_ok_clicked" : self.on_ok_clicked
}
dic = {
"destroy_passed_object" : Utils.destroy_passed_object,
"on_ok_clicked" : self.on_ok_clicked
}
self.top = gtk.glade.XML(glade_file,"packageExport")
self.top = gtk.glade.XML(glade_file,"packageExport")
Utils.set_titles(self.top.get_widget('packageExport'),
Utils.set_titles(self.top.get_widget('packageExport'),
self.top.get_widget('title'),
_('Export to CD'))
self.top.signal_autoconnect(dic)
self.top.get_widget("packageExport").show()
self.top.signal_autoconnect(dic)
self.top.get_widget("packageExport").show()
def copy_file(self,src,dest):
original = open(src,"r")
@@ -115,6 +119,36 @@ class PackageWriter:
th.write(data)
th.close()
def cl_run(self):
base = os.path.basename(self.name)
try:
uri = gnome.vfs.URI('burn:///%s' % base)
gnome.vfs.make_directory(uri,gnome.vfs.OPEN_WRITE)
except gnome.vfs.error, msg:
print msg
os._exit(1)
try:
uri = gnome.vfs.URI('burn:///%s/.thumb' % base)
gnome.vfs.make_directory(uri,gnome.vfs.OPEN_WRITE)
except gnome.vfs.error, msg:
print msg
os._exit(1)
for obj in self.db.getObjectMap().values():
oldfile = obj.getPath()
root = os.path.basename(oldfile)
if os.path.isfile(oldfile):
self.copy_file(oldfile,'burn:///%s/%s' % (base,root))
if obj.getMimeType()[0:5] == "image":
self.make_thumbnail(base,root,obj.getPath())
# Write XML now
g = gnome.vfs.create('burn:///%s/data.gramps' % base,gnome.vfs.OPEN_WRITE )
gfile = WriteXML.XmlWriter(self.db,None,1)
gfile.write_handle(g)
g.close()
def on_ok_clicked(self,obj):
Utils.destroy_passed_object(obj)

View File

@@ -1,7 +1,7 @@
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000 Donald N. Allingham
# Copyright (C) 2000-2003 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
@@ -334,7 +334,7 @@ def writeData(database,person):
class GedcomWriter:
"""Writes a GEDCOM file from the passed database"""
def __init__(self,db,person):
def __init__(self,db,person,cl=0,name=""):
self.db = db
self.person = person
self.restrict = 1
@@ -350,66 +350,71 @@ class GedcomWriter:
self.pidmap = {}
self.sidval = 0
self.sidmap = {}
self.cl = cl
self.name = name
glade_file = "%s/gedcomexport.glade" % os.path.dirname(__file__)
if self.cl:
self.cl_setup()
else:
glade_file = "%s/gedcomexport.glade" % os.path.dirname(__file__)
self.topDialog = gtk.glade.XML(glade_file,"gedcomExport")
self.topDialog.signal_autoconnect({
"destroy_passed_object" : Utils.destroy_passed_object,
"gnu_free" : self.gnu_free,
"standard_copyright" : self.standard_copyright,
"no_copyright" : self.no_copyright,
"on_ok_clicked" : self.on_ok_clicked
})
Utils.set_titles(self.topDialog.get_widget('gedcomExport'),
self.topDialog.get_widget('title'),
_('GEDCOM export'))
self.topDialog = gtk.glade.XML(glade_file,"gedcomExport")
self.topDialog.signal_autoconnect({
"destroy_passed_object" : Utils.destroy_passed_object,
"gnu_free" : self.gnu_free,
"standard_copyright" : self.standard_copyright,
"no_copyright" : self.no_copyright,
"on_ok_clicked" : self.on_ok_clicked
})
filter_obj = self.topDialog.get_widget("filter")
self.copy = 0
Utils.set_titles(self.topDialog.get_widget('gedcomExport'),
self.topDialog.get_widget('title'),
_('GEDCOM export'))
all = GenericFilter.GenericFilter()
all.set_name(_("Entire Database"))
all.add_rule(GenericFilter.Everyone([]))
des = GenericFilter.GenericFilter()
des.set_name(_("Descendants of %s") % person.getPrimaryName().getName())
des.add_rule(GenericFilter.IsDescendantOf([person.getId()]))
ans = GenericFilter.GenericFilter()
ans.set_name(_("Ancestors of %s") % person.getPrimaryName().getName())
ans.add_rule(GenericFilter.IsAncestorOf([person.getId()]))
com = GenericFilter.GenericFilter()
com.set_name(_("People with common ancestor with %s") %
person.getPrimaryName().getName())
com.add_rule(GenericFilter.HasCommonAncestorWith([person.getId()]))
self.filter_menu = GenericFilter.build_filter_menu([all,des,ans,com])
filter_obj.set_menu(self.filter_menu)
gedmap = GedcomInfo.GedcomInfoDB()
target_obj = self.topDialog.get_widget("target")
myMenu = gtk.Menu()
for name in gedmap.get_name_list():
menuitem = gtk.MenuItem(name)
myMenu.append(menuitem)
data = gedmap.get_description(name)
menuitem.set_data("data",data)
menuitem.show()
target_obj.set_menu(myMenu)
self.target_menu = myMenu
pathname = "%s.ged" % os.path.dirname(db.getSavePath())
filter_obj = self.topDialog.get_widget("filter")
self.copy = 0
all = GenericFilter.GenericFilter()
all.set_name(_("Entire Database"))
all.add_rule(GenericFilter.Everyone([]))
des = GenericFilter.GenericFilter()
des.set_name(_("Descendants of %s") % person.getPrimaryName().getName())
des.add_rule(GenericFilter.IsDescendantOf([person.getId()]))
ans = GenericFilter.GenericFilter()
ans.set_name(_("Ancestors of %s") % person.getPrimaryName().getName())
ans.add_rule(GenericFilter.IsAncestorOf([person.getId()]))
com = GenericFilter.GenericFilter()
com.set_name(_("People with common ancestor with %s") %
person.getPrimaryName().getName())
com.add_rule(GenericFilter.HasCommonAncestorWith([person.getId()]))
self.filter_menu = GenericFilter.build_filter_menu([all,des,ans,com])
filter_obj.set_menu(self.filter_menu)
gedmap = GedcomInfo.GedcomInfoDB()
target_obj = self.topDialog.get_widget("target")
myMenu = gtk.Menu()
for name in gedmap.get_name_list():
menuitem = gtk.MenuItem(name)
myMenu.append(menuitem)
data = gedmap.get_description(name)
menuitem.set_data("data",data)
menuitem.show()
target_obj.set_menu(myMenu)
self.target_menu = myMenu
pathname = "%s.ged" % os.path.dirname(db.getSavePath())
filetgt = self.topDialog.get_widget('fileentry1')
filetgt.set_filename(pathname)
self.topDialog.get_widget("gedcomExport").show()
filetgt = self.topDialog.get_widget('fileentry1')
filetgt.set_filename(pathname)
self.topDialog.get_widget("gedcomExport").show()
def gnu_free(self,obj):
self.copy = 1
@@ -487,6 +492,41 @@ class GedcomWriter:
self.export_data(name)
closebtn.set_sensitive(1)
def cl_setup(self):
self.restrict = 0
self.private = 0
self.copy = 0
for p in self.db.getPersonKeys():
self.plist[p] = 1
gedmap = GedcomInfo.GedcomInfoDB()
self.target_ged = gedmap.standard
self.dest = self.target_ged.get_dest()
self.adopt = self.target_ged.get_adopt()
self.conc = self.target_ged.get_conc()
self.altname = self.target_ged.get_alt_name()
self.cal = self.target_ged.get_alt_calendar()
self.obje = self.target_ged.get_obje()
self.resi = self.target_ged.get_resi()
self.prefix = self.target_ged.get_prefix()
self.source_refs = self.target_ged.get_source_refs()
self.cnvtxt = keep_utf8
self.flist = {}
self.slist = {}
for key in self.plist.keys():
p = self.db.getPerson(key)
add_persons_sources(p,self.slist,self.private)
for family in p.getFamilyList():
add_familys_sources(family,self.slist,self.private)
self.flist[family.getId()] = 1
self.export_data(self.name)
def export_data(self,filename):
try:
self.g = open(filename,"w")
@@ -558,17 +598,19 @@ class GedcomWriter:
for key in pkeys:
self.write_person(self.db.getPerson(key))
index = index + 1
if index%100 == 0:
if index%100 == 0 and not self.cl:
self.pbar.set_fraction(index/nump)
while(gtk.events_pending()):
gtk.mainiteration()
self.pbar.set_fraction(1.0)
if not self.cl:
self.pbar.set_fraction(1.0)
self.write_families()
if self.source_refs:
self.write_sources()
else:
self.sbar.set_fraction(1.0)
if not self.cl:
self.sbar.set_fraction(1.0)
self.g.write("0 TRLR\n")
self.g.close()
@@ -666,11 +708,12 @@ class GedcomWriter:
break
index = index + 1
if index % 100 == 0:
if index % 100 == 0 and not self.cl:
self.fbar.set_fraction(index/nump)
while(gtk.events_pending()):
gtk.mainiteration()
self.fbar.set_fraction(1.0)
if not self.cl:
self.fbar.set_fraction(1.0)
def write_sources(self):
nump = float(len(self.slist))
@@ -691,11 +734,12 @@ class GedcomWriter:
if source.getNote():
self.write_long_text("NOTE",1,self.cnvtxt(source.getNote()))
index = index + 1
if index % 100 == 0:
if index % 100 == 0 and not self.cl:
self.sbar.set_fraction(index/nump)
while(gtk.events_pending()):
gtk.mainiteration()
self.sbar.set_fraction(1.0)
if not self.cl:
self.sbar.set_fraction(1.0)
def write_person(self,person):
self.g.write("0 @%s@ INDI\n" % self.pid(person.getId()))