#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000 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
#
"Web Site/Generate Web Site"
from RelLib import *
from TextDoc import *
from HtmlDoc import *
import const
import utils
import intl
_ = intl.gettext
import os
import re
import sort
import string
import time
from gtk import *
from gnome.ui import *
from libglade import *
from StyleEditor import *
active_person = None
db = None
topDialog = None
glade_file = os.path.dirname(__file__) + os.sep + "webpage.glade"
restrict = 1
restrict_photos = 0
no_photos = 0
styles = StyleSheet()
style_sheet_list = None
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
class HtmlLinkDoc(HtmlDoc):
def start_link(self,path):
self.f.write('' % path)
def end_link(self):
self.f.write('')
def newline(self):
self.f.write('
\n')
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
class IndividualPage:
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def __init__(self,person,photos,restrict,list,dir_name,doc):
self.person = person
self.doc = doc
self.list = list
self.alive = probably_alive(person) or restrict
self.photos = (photos == 2) or (photos == 1 and not self.alive)
self.dir = dir_name
tbl = TableStyle()
tbl.set_width(100)
tbl.set_column_widths([15,85])
self.doc.add_table_style("IndTable",tbl)
tbl = TableStyle()
tbl.set_width(100)
tbl.set_column_widths([15,15,70])
self.doc.add_table_style("FamTable",tbl)
cell = TableCellStyle()
cell.set_top_border(1)
cell.set_bottom_border(1)
self.doc.add_cell_style("TableHead",cell)
cell = TableCellStyle()
self.doc.add_cell_style("NormalCell",cell)
name = person.getPrimaryName().getRegularName()
self.doc.set_title(_("Summary of %s") % name)
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def write_normal_row(self,list):
self.doc.start_row()
for i in list:
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
self.doc.write_text(i)
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def write_link_row(self,title,person):
self.doc.start_row()
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
self.doc.write_text(title)
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
if person:
if person in self.list:
self.doc.start_link("i%s.html" % str(person.getId()))
self.doc.write_text(person.getPrimaryName().getRegularName())
if person in self.list:
self.doc.end_link()
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def create_page(self):
filebase = "i%d.html" % (self.person.getId())
self.doc.open("%s%s%s" % (self.dir,os.sep,filebase))
photo_list = self.person.getPhotoList()
name = self.person.getPrimaryName().getRegularName()
self.doc.start_paragraph("Title")
self.doc.write_text(_("Summary of %s") % name)
self.doc.end_paragraph()
self.doc.start_paragraph("Normal")
self.doc.end_paragraph()
if self.photos and len(photo_list) > 0:
file = photo_list[0].getPath()
self.doc.start_paragraph("Normal")
self.doc.add_photo(file,4.0,4.0)
self.doc.end_paragraph()
self.doc.start_table("one","IndTable")
self.write_normal_row([ "%s:" % _("Name"), name])
if self.person.getGender() == Person.male:
self.write_normal_row([ "%s:" % _("Gender"), _("Male")])
else:
self.write_normal_row([ "%s:" % _("Gender"), _("Female")])
family = self.person.getMainFamily()
if family:
self.write_link_row("%s:" % _("Father"), family.getFather())
self.write_link_row("%s:" % _("Mother"), family.getMother())
else:
self.write_link_row("%s:" % _("Father"), None)
self.write_link_row("%s:" % _("Mother"), None)
if self.person.getNote() != "":
note = string.replace(self.person.getNote(),"\n","
")
self.write_normal_row(["%s:" % _("Notes"), note])
self.doc.end_table()
self.doc.start_paragraph("Normal")
self.doc.end_paragraph()
self.write_facts()
self.write_families()
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def close(self):
self.doc.close()
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def write_facts(self):
self.doc.start_table("two","IndTable")
self.doc.start_row()
self.doc.start_cell("TableHead",2)
self.doc.start_paragraph("TableTitle")
self.doc.write_text(_("Individual Facts"))
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
event_list = [ self.person.getBirth(), self.person.getDeath() ]
event_list = event_list + self.person.getEventList()
for event in event_list:
name = _(event.getName())
date = event.getDate()
place = event.getPlace()
description = event.getDescription()
if date == "":
if place == "":
continue
else:
val = place + ". " + description
else:
if place == "":
val = date + ". " + description
else:
val = date + " in " + place + ". " + description
self.write_normal_row([name, val])
self.doc.end_table()
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def write_fam_fact(self,event):
name = event.getName()
date = event.getDate()
place = event.getPlace()
description = event.getDescription()
if date == "":
if place == "":
return
else:
val = place + ". " + description
else:
if place == "":
val = date + ". " + description
else:
val = date + " in " + place + ". " + description
self.write_normal_row(["",name, val])
#--------------------------------------------------------------------
#
#
#
#--------------------------------------------------------------------
def write_families(self):
self.doc.start_paragraph("Normal")
self.doc.end_paragraph()
self.doc.start_table("three","FamTable")
self.doc.start_row()
self.doc.start_cell("TableHead",3)
self.doc.start_paragraph("TableTitle")
self.doc.write_text(_("Marriages/Children"))
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
for family in self.person.getFamilyList():
if self.person == family.getFather():
spouse = family.getMother()
else:
spouse = family.getFather()
self.doc.start_row()
self.doc.start_cell("NormalCell",3)
self.doc.start_paragraph("Spouse")
if spouse:
self.doc.start_link("i%s.html" % str(spouse.getId()))
self.doc.write_text(spouse.getPrimaryName().getRegularName())
self.doc.end_link()
else:
self.doc.write_text(_("unknown"))
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
event_list = [ family.getMarriage(), family.getDivorce() ]
event_list = event_list + family.getEventList()
for event in event_list:
self.write_fam_fact(event)
child_list = family.getChildList()
if len(child_list) > 0:
self.doc.start_row()
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
self.doc.write_text("")
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
self.doc.write_text(_("Children"))
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.start_cell("NormalCell")
self.doc.start_paragraph("Normal")
first = 1
for child in family.getChildList():
if first == 1:
first = 0
else:
self.doc.write_text('\n')
self.doc.start_link("i%s.html" % str(child.getId()))
self.doc.write_text(child.getPrimaryName().getRegularName())
self.doc.end_link()
self.doc.end_paragraph()
self.doc.end_cell()
self.doc.end_row()
self.doc.end_table()
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def individual_filter(database,person,list):
list.append(person)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def ancestor_filter(database,person,list):
if person == None:
return
if person not in list:
list.append(person)
family = person.getMainFamily()
if family != None:
ancestor_filter(database,family.getFather(),list)
ancestor_filter(database,family.getMother(),list)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def descendant_filter(database,person,list):
if person == None or person in list:
return
if person not in list:
list.append(person)
for family in person.getFamilyList():
for child in family.getChildList():
descendant_filter(database,child,list)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def an_des_filter(database,person,list):
descendant_filter(database,person,list)
ancestor_filter(database,person,list)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def entire_db_filter(database,person,list):
for entry in database.getPersonMap().values():
list.append(entry)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def an_des_of_gparents_filter(database,person,list):
my_list = []
family = person.getMainFamily()
if family == None:
return
for p in [ family.getMother(), family.getFather() ]:
if p:
pf = p.getMainFamily()
if pf:
if pf.getFather():
my_list.append(pf.getFather())
if pf.getMother():
my_list.append(pf.getMother())
for person in my_list:
descendant_filter(database,person,list)
ancestor_filter(database,person,list)
#------------------------------------------------------------------------
#
# Mams menu items to filter functions
#
#------------------------------------------------------------------------
filter_map = {
_("Individual") : individual_filter,
_("Ancestors") : ancestor_filter,
_("Descendants") : descendant_filter,
_("Ancestors and descendants") : an_des_filter,
_("Grandparent's ancestors and descendants") : an_des_of_gparents_filter,
_("Entire database") : entire_db_filter
}
#-------------------------------------------------------------------------
#
#
#
#-------------------------------------------------------------------------
def probably_alive(person):
if person == None:
return 1
death = person.getDeath()
birth = person.getBirth()
if death.getDate() != "":
return 0
if birth.getDate() != "":
year = birth.getDateObj().get_start_date().getYear()
time_struct = time.localtime(time.time())
current_year = time_struct[0]
if year != -1 and current_year - year > 110:
return 0
return 1
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def report(database,person):
global active_person
global topDialog
global db
global styles
global style_sheet_list
active_person = person
db = database
font = FontStyle()
font.set(bold=1, face=FONT_SANS_SERIF, size=16)
p = ParagraphStyle()
p.set(align=PARA_ALIGN_CENTER,font=font)
styles.add_style("Title",p)
font = FontStyle()
font.set(bold=1,face=FONT_SANS_SERIF,size=12,italic=1)
p = ParagraphStyle()
p.set_font(font)
styles.add_style("TableTitle",p)
font = FontStyle()
font.set(bold=1,face=FONT_SANS_SERIF,size=12)
p = ParagraphStyle()
p.set_font(font)
styles.add_style("Spouse",p)
font = FontStyle()
font.set_size(12)
p = ParagraphStyle()
p.set_font(font)
styles.add_style("Normal",p)
style_sheet_list = StyleSheetList("webpage.xml",styles)
dic = {
"destroy_passed_object" : utils.destroy_passed_object,
"on_nophotos_toggled" : on_nophotos_toggled,
"on_style_edit_clicked" : on_style_edit_clicked,
"on_ok_clicked" : on_ok_clicked,
}
topDialog = GladeXML(glade_file,"top")
topDialog.signal_autoconnect(dic)
build_menu(None)
top = topDialog.get_widget("top")
topDialog.get_widget("targetDirectory").set_default_path(os.getcwd())
filterName = topDialog.get_widget("filterName")
popdown_strings = filter_map.keys()
popdown_strings.sort()
filterName.set_popdown_strings(popdown_strings)
name = person.getPrimaryName().getName()
topDialog.get_widget("personName").set_text(name)
top.show()
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def on_style_edit_clicked(obj):
StyleListDisplay(style_sheet_list,build_menu,None)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def build_menu(object):
menu = topDialog.get_widget("style_menu")
myMenu = GtkMenu()
for style in style_sheet_list.get_style_names():
menuitem = GtkMenuItem(style)
menuitem.set_data("d",style_sheet_list.get_style_sheet(style))
menuitem.show()
myMenu.append(menuitem)
menu.set_menu(myMenu)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def on_nophotos_toggled(obj):
if obj.get_active():
topDialog.get_widget("restrict_photos").set_sensitive(0)
else:
topDialog.get_widget("restrict_photos").set_sensitive(1)
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def on_ok_clicked(obj):
global active_person
global filter_map
global db
filterName = topDialog.get_widget("filter").get_text()
dir_name = topDialog.get_widget("targetDirectory").get_full_path(0)
templ_name = topDialog.get_widget("htmlTemplate").get_full_path(0)
restrict = topDialog.get_widget("restrict").get_active()
restrict_photos = topDialog.get_widget("restrict_photos").get_active()
no_photos = topDialog.get_widget("nophotos").get_active()
if dir_name == None:
dir_name = os.getcwd()
elif not os.path.isdir(dir_name):
parent_dir = os.path.dirname(dir_name)
if not os.path.isdir(parent_dir):
GnomeErrorDialog(_("Neither %s nor %s are directories") % \
(dir_name,parent_dir))
return
else:
try:
os.mkdir(dir_name)
except IOError, value:
GnomeErrorDialog(_("Could not create the directory : %s") % \
dir_name + "\n" + value[1])
return
except:
GnomeErrorDialog(_("Could not create the directory : %s") % \
dir_name)
return
filter = filter_map[filterName]
ind_list = []
filter(db,active_person,ind_list)
styles = topDialog.get_widget("style_menu").get_menu().get_active().get_data("d")
if no_photos == 1:
photos = 0
elif restrict_photos == 1:
photos = 1
else:
photos = 2
for person in ind_list:
doc = HtmlLinkDoc(styles,templ_name)
idoc = IndividualPage(person,photos,restrict,ind_list,dir_name,doc)
idoc.create_page()
idoc.close()
if len(ind_list) > 1:
dump_index(ind_list,styles,templ_name,dir_name)
utils.destroy_passed_object(obj)
#------------------------------------------------------------------------
#
# Writes a index file, listing all people in the person list.
#
#------------------------------------------------------------------------
def dump_index(person_list,styles,template,html_dir):
doc = HtmlLinkDoc(styles,template)
doc.set_title(_("Family Tree Index"))
doc.open(html_dir + os.sep + "index.html")
doc.start_paragraph("Title")
doc.write_text(_("Family Tree Index"))
doc.end_paragraph()
person_list.sort(sort.by_last_name)
for person in person_list:
name = person.getPrimaryName().getName()
id = person.getId()
doc.start_link("i%s.html" % str(person.getId()))
doc.write_text(name)
doc.end_link()
doc.newline()
doc.close()
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def get_description():
return _("Generates web (HTML) pages for individuals, or a set of individuals.")
#------------------------------------------------------------------------
#
#
#
#------------------------------------------------------------------------
def get_name():
return _("Web Site/Generate Web Site")