# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2007 Thom Sturgill # Copyright (C) 2007-2008 Brian G. Matherly # Copyright (C) 2008 Rob G. Healey # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Pubilc 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$ """ Web Calendar generator. Created 4/22/07 by Thom Sturgill based on Calendar.py (with patches) by Doug Blank with input dialog based on NarrativeWeb.py by Don Allingham. 2008-05-11 Jason Simanek Improving markup for optimal separation of content and presentation. 2008-June-22 Rob G. Healey *** Remove StyleEditor, make it css based as is NarrativeWeb, move title to first tab, re-word note tabs, complete re-write of calendar build, added year glance, and blank year, added easter and dst start/stop from Calendar.py, etc. Reports/Web Page/Web Calendar """ #------------------------------------------------------------------------ # # python modules # #------------------------------------------------------------------------ import os import time import datetime import calendar import math import codecs import shutil from gettext import gettext as _ from xml.parsers import expat try: set() except: from sets import Set as set #------------------------------------------------------------------------ # # Set up logging # #------------------------------------------------------------------------ import logging log = logging.getLogger(".WebPage") #------------------------------------------------------------------------ # # GRAMPS module # #------------------------------------------------------------------------ import gen.lib import const from GrampsCfg import get_researcher from PluginUtils import PluginManager from ReportBase import (Report, ReportUtils, MenuReportOptions, CATEGORY_WEB, MODE_GUI) from PluginUtils import FilterOption, EnumeratedListOption, PersonOption, \ BooleanOption, NumberOption, StringOption, DestinationOption import Utils import GrampsLocale from QuestionDialog import ErrorDialog, WarningDialog from Utils import probably_alive from DateHandler import displayer as _dd from DateHandler import parser as _dp #------------------------------------------------------------------------ # # constants # #------------------------------------------------------------------------ _CALENDARSCREEN = "calendar-screen.css" _CALENDARPRINT = "calendar-print.css" # This information defines the list of styles in the Web calendar # options dialog as well as the location of the corresponding SCREEN # stylesheets. _CSS_FILES = [ # First is used as default selection. [_("Evergreen"), 'Web_Evergreen.css'], [_("Nebraska"), 'Web_Nebraska.css'], [_("Simply Red"), 'Web_Simply-Red.css'], [_("No style sheet"), ''], ] _CHARACTER_SETS = [ # First is used as default selection. [_('Unicode (recommended)'), 'utf-8'], ['ISO-8859-1', 'iso-8859-1' ], ['ISO-8859-2', 'iso-8859-2' ], ['ISO-8859-3', 'iso-8859-3' ], ['ISO-8859-4', 'iso-8859-4' ], ['ISO-8859-5', 'iso-8859-5' ], ['ISO-8859-6', 'iso-8859-6' ], ['ISO-8859-7', 'iso-8859-7' ], ['ISO-8859-8', 'iso-8859-8' ], ['ISO-8859-9', 'iso-8859-9' ], ['ISO-8859-10', 'iso-8859-10' ], ['ISO-8859-13', 'iso-8859-13' ], ['ISO-8859-14', 'iso-8859-14' ], ['ISO-8859-15', 'iso-8859-15' ], ['koi8_r', 'koi8_r', ], ] _CC = [ '', '' 'Creative Commons License - By attribution', '' 'Creative Commons License - By attribution, No derivations', '' 'Creative Commons License - By attribution, Share-alike', '' 'Creative Commons License - By attribution, Non-commercial', '' 'Creative Commons License - By attribution, Non-commercial, No derivations', '' 'Creative Commons License - By attribution, Non-commerical, Share-alike' ] _COPY_OPTIONS = [ _('Standard copyright'), _('Creative Commons - By attribution'), _('Creative Commons - By attribution, No derivations'), _('Creative Commons - By attribution, Share-alike'), _('Creative Commons - By attribution, Non-commercial'), _('Creative Commons - By attribution, Non-commercial, No derivations'), _('Creative Commons - By attribution, Non-commercial, Share-alike'), _('No copyright notice'), ] def make_date(year, month, day): """ Return a Date object of the particular year/month/day. """ retval = gen.lib.Date() retval.set_yr_mon_day(year, month, day) return retval #------------------------------------------------------------------------ # # WebCalReport # #------------------------------------------------------------------------ class WebCalReport(Report): """ Create WebCalReport object that produces the report. """ def __init__(self, database, options): Report.__init__(self, database, options) menu = options.menu self.database = database self.options = options self.html_dir = menu.get_option_by_name('target').get_value() filter_option = menu.get_option_by_name('filter') self.filter = filter_option.get_filter() self.ext = menu.get_option_by_name('ext').get_value() self.copy = menu.get_option_by_name('cright').get_value() self.encoding = menu.get_option_by_name('encoding').get_value() self.css = menu.get_option_by_name('css').get_value() self.country = menu.get_option_by_name('country').get_value() self.year = menu.get_option_by_name('year').get_value() self.fullyear = menu.get_option_by_name('fullyear').get_value() self.blankyear = menu.get_option_by_name('blankyear').get_value() self.surname = menu.get_option_by_name('surname').get_value() self.alive = menu.get_option_by_name('alive').get_value() self.birthday = menu.get_option_by_name('birthdays').get_value() self.anniv = menu.get_option_by_name('anniversaries').get_value() self.title_text = menu.get_option_by_name('title').get_value() self.home_link = menu.get_option_by_name('home_link').get_value() self.month_notes = [menu.get_option_by_name('note_jan').get_value(), menu.get_option_by_name('note_feb').get_value(), menu.get_option_by_name('note_mar').get_value(), menu.get_option_by_name('note_apr').get_value(), menu.get_option_by_name('note_may').get_value(), menu.get_option_by_name('note_jun').get_value(), menu.get_option_by_name('note_jul').get_value(), menu.get_option_by_name('note_aug').get_value(), menu.get_option_by_name('note_sep').get_value(), menu.get_option_by_name('note_oct').get_value(), menu.get_option_by_name('note_nov').get_value(), menu.get_option_by_name('note_dec').get_value()] self.warn_dir = True # Only give warning once. # Set first weekday according to Locale xmllang = Utils.xml_lang() if xmllang == "en-US": # USA calendar starts on a Sunday calendar.setfirstweekday(calendar.SUNDAY) else: # European calendar starts on Monday, default calendar.setfirstweekday(calendar.MONDAY) def copy_file(self, from_fname, to_fname, to_dir=''): """ Copy a file from a source to a (report) destination. If to_dir is not present and if the target is not an archive, then the destination directory will be created. Normally 'to_fname' will be just a filename, without directory path. 'to_dir' is the relative path name in the destination root. It will be prepended before 'to_fname'. """ dest = os.path.join(self.html_dir, to_dir, to_fname) destdir = os.path.dirname(dest) if not os.path.isdir(destdir): os.makedirs(destdir) if from_fname != dest: shutil.copyfile(from_fname, dest) elif self.warn_dir: WarningDialog( _("Possible destination error") + "\n" + _("You appear to have set your target directory " "to a directory used for data storage. This " "could create problems with file management. " "It is recommended that you consider using " "a different directory to store your generated " "web pages.")) self.warn_dir = False def dst(self, area="us"): """ Return Daylight Saving Time start/stop in a given area ("us", "eu"). US calculation valid 1976-2099; EU 1996-2099 """ year = self.year if area == "us": if year > 2006: start = (year, 3, 14 - (math.floor(1 + year * 5 / 4) % 7)) # March stop = (year, 11, 7 - (math.floor(1 + year * 5 / 4) % 7)) # November else: start = (year, 4, (2 + 6 * year - math.floor(year / 4)) % 7 + 1) # April stop = (year, 10, (31 - (math.floor(year * 5 / 4) + 1) % 7)) # October elif area == "eu": start = (year, 3, (31 - (math.floor(year * 5 / 4) + 4) % 7)) # March stop = (year, 10, (31 - (math.floor(year * 5 / 4) + 1) % 7)) # Oct return (start, stop) def add_day_item(self, text, year, month, day): if day == 0: # This may happen for certain "about" dates. day = 1 # Use first day of the month month_dict = self.calendar.get(month, {}) day_list = month_dict.get(day, []) day_list.append(text) month_dict[day] = day_list self.calendar[month] = month_dict def get_holidays(self, country = "United States"): """ Looks in multiple places for holidays.xml files the holidays file will be used first if it exists in user's plugins, else the GRAMPS plugins will be checked. No more of having duel holidays files being read. User directory is first choice if it exists, and does not use both holiday files any longer """ holiday_file = 'holidays.xml' holiday_full_path = "" fname1 = os.path.join(const.USER_PLUGINS, holiday_file) fname2 = os.path.join(const.PLUGINS_DIR, holiday_file) if os.path.exists(fname1): holiday_full_path = fname1 elif os.path.exists(fname2): holiday_full_path = fname2 if holiday_full_path != "": self.process_holiday_file(holiday_full_path, country) def process_holiday_file(self, filename, country): """ This will process a holiday file """ year = self.year parser = Xml2Obj() element = parser.Parse(filename) mycalendar = Holidays(element, country) date = datetime.date(year, 1, 1) while date.year == year: holidays = mycalendar.check_date( date ) for text in holidays: if text == "Easter": date1 = _easter(year) self.add_day_item(text, date1[0], date1[1], date1[2]) elif text == "Daylight Saving begins": if Utils.xml_lang() == "en-US": date2 = self.dst("us") else: date2 = self.dst("eu") dst_start = date2[0] dst_stop = date2[1] self.add_day_item(text, dst_start[0], dst_start[1], dst_start[2]) self.add_day_item("Daylight Saving ends", dst_stop[0], dst_stop[1], dst_stop[2]) elif text == "Daylight Saving ends": pass self.add_day_item(text, date.year, date.month, date.day) date = date.fromordinal( date.toordinal() + 1) def copy_css(self): """ Copies all the necessary files... """ # Copy the _CALENDARSCREEN css if self.css != "": from_file = os.path.join(const.DATA_DIR, self.css) self.copy_file(from_file, _CALENDARSCREEN, "styles") # copy calendar-print stylesheet from_file = os.path.join(const.DATA_DIR, "Web_Print-Default.css") self.copy_file(from_file, _CALENDARPRINT, "styles") # Copy GRAMPS favicon to target from_file = os.path.join(const.IMAGE_DIR, "favicon.ico") self.copy_file(from_file, "favicon.ico", "images") # Copy arrow image if "Year At A Glance" is requested, # and if the file exists if self.fullyear: from_file = os.path.join(const.IMAGE_DIR, "arrow102.gif") if os.path.exists(from_file): self.copy_file(from_file, "arrow102.gif", "images") def display_nav_links(self, of, currentsection, cal): """ 'cal' - one of "yg", "by", "wc", "ip" """ # Check to see if home_link will be used??? navs = [ (self.home_link, _('Home'), self.home_link), (1, _('Jan'), True), (2, _('Feb'), True), (3, _('Mar'), True), (4, _('Apr'), True), (5, _('May'), True), (6, _('Jun'), True), (7, _('Jul'), True), (8, _('Aug'), True), (9, _('Sep'), True), (10, _('Oct'), True), (11, _('Nov'), True), (12, _('Dec'), True), ('fullyear', _('Year Glance'), self.fullyear), ('blankyear', _('Blank Calendar'), self.blankyear) ] for url_fname, nav_text, cond in navs: if cond: # Figure out if we need
  • or just plain
  • cs = '' if url_fname == currentsection: cs = ' id="CurrentSection"' if type(url_fname) == int: url_fname = GrampsLocale.long_months[url_fname] new_dir = str(self.year) url = _('unknown') if ((cal == "yg") or (cal == "by")): url = _subdirs("yg", new_dir, url_fname) elif cal == "ip": url = _subdirs("ip", new_dir, url_fname) else: url = _subdirs("wc", new_dir, url_fname) if not _has_webpage_extension(url): url += self.ext of.write(' %s
  • \n' % (cs, url, nav_text)) def calendar_build(self, of, cal, month): """ This does the work of building the calendar 'cal' - one of "yg", "by", "wc" 'month' - month number 1, 2, .., 12 """ year = self.year # Begin calendar head title = GrampsLocale.long_months[month] of.write('\n' % title) of.write(' \n' % title) of.write(' \n') of.write(' \n') of.write(' \n' % title) of.write(' \n') # This calendar has first column sunday. Do not use locale! calendar.setfirstweekday(calendar.SUNDAY) # Calendar weekday names header of.write(' \n') of.write(' \n') for day_col in range(5): of.write(' \n') of.write(' \n') of.write(' \n') of.write(' \n') of.write(' \n') # Compute the first day to display for this month. # It can also be a day in the previous month. current_date = datetime.date(year, month, 1) # first day of the month # isoweekday: 1=monday, 2=tuesday, etc if current_date.isoweekday() != 7: # start dow here is 7, sunday # Compute the sunday before this date. current_ord = current_date.toordinal() - current_date.isoweekday() else: # First day of the month is sunday, that's OK current_ord = current_date.toordinal() # get last month's last week for previous days in the month if month == 1: prevmonth = calendar.monthcalendar(year-1, 12) else: prevmonth = calendar.monthcalendar(year, month-1) num_weeks = len(prevmonth) lastweek_prevmonth = prevmonth[num_weeks - 1] # get next month's first week for next days in the month if month == 12: nextmonth = calendar.monthcalendar(year+1, 1) else: nextmonth = calendar.monthcalendar(year, month + 1) firstweek_nextmonth = nextmonth[0] # Begin calendar monthinfo = calendar.monthcalendar(year, month) nweeks = len(monthinfo) for week_row in range(0, nweeks): week = monthinfo[week_row] of.write(' \n' % week_row) for days_row in range(0, 7): if days_row == 0: dayclass = "weekend sunday" elif days_row == 6: dayclass = "weekend saturday" else: dayclass = "weekday" day = week[days_row] if day == 0: # a day in the previous or in the next month if week_row == 0: # day in the previous month specday = lastweek_prevmonth[days_row] specclass = "previous " + dayclass elif week_row == nweeks-1: # day in the next month specday = firstweek_nextmonth[days_row] specclass = "next " + dayclass of.write(' \n') else: # normal day number in current month if cal == "by": of.write(' \n') else: of.write(' \n') current_ord += 1 of.write(' \n') def write_header(self, of, title, cal, mystyle): """ This creates the header for the Calendars including style embedded for special purpose """ of.write('\n ') of.write('\n' % (xmllang, xmllang)) of.write('\n') of.write(' %s\n' % title) of.write(' \n' % self.encoding) of.write(' \n') of.write(' \n') author = get_researcher().get_name() of.write(' \n' % author) if ((cal == "yg") or (cal == "by")): # year glance and blank_year # have same directory levels fname1 = _subdirs("yg", "styles", _CALENDARSCREEN) fname2 = _subdirs("yg", "styles", _CALENDARPRINT) fname3 = _subdirs("yg", "images", "favicon.ico") elif cal == "ip": fname1 = _subdirs("ip", "styles", _CALENDARSCREEN) fname2 = _subdirs("ip", "styles", _CALENDARPRINT) fname3 = _subdirs("ip", "images", "favicon.ico") else: fname1 = _subdirs("wc", "styles", _CALENDARSCREEN) fname2 = _subdirs("wc", "styles", _CALENDARPRINT) fname3 = _subdirs("wc", "images", "favicon.ico") # link to calendar-screen css of.write(' \n' % fname1) # link to calendar-print css if not cal == "yg": of.write(' \n' % fname2) # create a link to GRAMPS favicon of.write(' \n' % fname3) # Add calendar specific embedded style of.write(mystyle) of.write('\n') return author def write_footer(self, of, cal): """ Writes the footer section of the pages """ # Display date as user set in preferences value = _dp.parse(time.strftime('%b %d %Y')) value = _dd.display(value) msg = _('Generated by ' 'GRAMPS on %(date)s') % {'date' : value} author = get_researcher().get_name() if author: author = author.replace(',,,', '') of.write(' \n') of.write('\n') of.write('\n') def create_file(self, fname, subdir=None): """ Create a file in the html_dir tree. If the directory does not exist, create it. """ if subdir: subdir = os.path.join(self.html_dir, subdir) else: subdir = self.html_dir if not os.path.isdir(subdir): os.makedirs(subdir) fname = os.path.join(subdir, fname) of = codecs.EncodedFile(open(fname, "w"), 'utf-8', self.encoding, 'xmlcharrefreplace') return of def close_file(self, of): of.close() def indiv_date(self, month, day_num, day_list): """ This method creates the indiv pages for "Year At A Glance" 'dat_list' - lines of text to display at this day """ year = self.year # Create names for long and short month names, in LOcale lng_month = GrampsLocale.long_months[month] shrt_month = GrampsLocale.short_months[month] # Name the file, and create it cal_fname = '%s%d%s' % (shrt_month, day_num, self.ext) of = self.create_file(cal_fname, os.path.join(str(year), lng_month)) arrow = os.path.join(self.html_dir, "images", "arrow102.gif") mystyle = """ """ # TODO. Merge this with code from blank_year(), year_glance(), print_page() # Add Header to calendar title = "%d %s %d" % (day_num, lng_month, year) author = self.write_header(of, title, "ip", mystyle) of.write('\n' % (shrt_month, day_num)) of.write(' \n') # Create navigation menu of.write(' \n') of.write('

    %s %d, %d

    \n' % (lng_month, day_num, year)) # if arrow file exists in IMAGE_DIR, use it arrow = os.path.join(const.IMAGE_DIR, "arrow102.gif") if os.path.isfile(arrow): of.write(' \n') of.write('
    ') of.write('%s
    ') if cal == "yg": of.write(GrampsLocale.short_days[1]) else: of.write(GrampsLocale.long_days[1]) of.write('') if cal == "yg": of.write(GrampsLocale.short_days[day_col+2]) else: of.write(GrampsLocale.long_days[day_col+2]) of.write('') if cal == "yg": of.write(GrampsLocale.short_days[7]) else: of.write(GrampsLocale.long_days[7]) of.write('
    \n' % (specday, specclass)) of.write('
    %d
    \n' % specday) of.write('
    \n' % (day, dayclass)) of.write('
    %d
    \n' % day) of.write('
    []: specclass = "highlight " + dayclass of.write('class="%s">\n' % specclass) if cal == "yg": # Year at a Glance lng_month = GrampsLocale.long_months[month] shrt_month = GrampsLocale.short_months[month] of.write(' \n' % (lng_month, shrt_month, day, self.ext)) of.write('
    %d' '
    \n' % day) self.indiv_date(month, day, day_list) else: # WebCal of.write('
    %d
    \n' % day) of.write('
      \n') for p in day_list: for line in p.splitlines(): of.write('
    • ') of.write(line) of.write('
    • \n') of.write('
    \n') else: of.write('class="%s">\n' % dayclass) of.write('
    %d
    \n' % day) else: # Either a day of previous month, or a day of next month of.write('class="%s">\n' % dayclass) of.write('
    %d
    \n' % day) of.write('
    \n') # Write footer and close file self.write_footer(of, "by") self.close_file(of) def year_glance(self): """ This method will create the Full Year At A Glance Page... """ year = self.year # Name the file, and create it cal_fname = 'fullyear%s' % (self.ext) of = self.create_file(cal_fname, "%d" % year) # Add specific styles for "Year At A Glance" page mystyle = """ """ # TODO. See note in indiv_date() # Add header to page title = "%d, At A Glance" % year author = self.write_header(of, title, "yg", mystyle) of.write('\n') # body will terminate in write_footer # Header Title of.write(' \n') # Create navigation menu of.write(' \n') # End Navigation Menu of.write('

    \n') of.write(_(' This calendar is meant to give you access to all your data at a glance ' 'compressed into one page. Clicking on a red square will take you to a ' 'page that shows all the events for that date!\n')) of.write('

    \n\n') # Create progress bar for it self.progress.set_pass(_('Creating Year At A Glance page'), 12) for month in range(1, 13): # Create full year self.progress.step() # build the calendar self.calendar_build(of, "yg", month) # TODO. Add week padding to make them all 6 weeks long. # See six_weeks nweeks = len(calendar.monthcalendar(year, month)) for i in range(nweeks+1, 7): of.write(' \n' % i) of.write(' \n') of.write(' \n') of.write(' \n') # close table body before writing note of.write(' \n') # create note section for each calendar month note = self.month_notes[month-1].strip() note = note or " " of.write(' \n') of.write(' \n') of.write(' \n') of.write(' %s\n' % note) of.write(' \n') of.write(' \n') of.write(' \n') of.write(' \n\n') # write footer section, and close file self.write_footer(of, "yg") self.close_file(of) def write_report(self): """ The short method that runs through each month and creates a page. """ if not os.path.isdir(self.html_dir): parent_dir = os.path.dirname(self.html_dir) if not os.path.isdir(parent_dir): ErrorDialog(_("Neither %s nor %s are directories") % \ (self.html_dir, parent_dir)) return else: try: os.mkdir(self.html_dir) except IOError, value: ErrorDialog(_("Could not create the directory: %s") % \ self.html_dir + "\n" + value[1]) return except: ErrorDialog(_("Could not create the directory: %s") % \ self.html_dir) return # initialize the dict to fill: self.calendar = {} self.progress = Utils.ProgressMeter(_("Generate XHTML Calendars"), '') # Generate the CSS file self.copy_css() # get the information, first from holidays: if self.country != 0: # Don't include holidays self.get_holidays(_COUNTRIES[self.country]) # _country is currently global self.progress.set_pass(_("Getting information from holidays file"), '') # get data from database: self.collect_data() # generate the report: self.progress.set_pass(_("Creating Calendar pages"), 12) for month in range(1, 13): self.progress.step() self.print_page(month) if self.fullyear: self.year_glance() if self.blankyear: self.blank_year() # Close the progress meter self.progress.close() def print_page(self, month): """ This method provides information and header/ footer to the calendar month """ year = self.year # Name the file, and create it # TODO. Do we want locale month name here? cal_fname = "%s%s" % (GrampsLocale.long_months[month], self.ext) of = self.create_file(cal_fname, "%d" % year) # Add specific styles to calendar head mystyle = """ """ # TODO. See note in indiv_date() # Add Header to calendar author = self.write_header(of, self.title_text, "wc", mystyle) of.write('\n') # terminated in write_footer # Header Title of.write(' \n') # end header # Create Navigation Menu of.write(' \n') # End Navigation Menu # build the calendar self.calendar_build(of, "wc", month) # close table body before note section of.write(' \n') # create note section for "WebCal" note = self.month_notes[month-1].strip() note = note or " " of.write(' \n') of.write(' \n') of.write(' %s\n' % note) of.write(' \n') of.write(' \n') of.write(' \n\n') # write footer section, and close file self.write_footer(of, "wc") self.close_file(of) def collect_data(self): """ This method runs through the data, and collects the relevant dates and text. """ self.progress.set_pass(_("Filtering"), '') people = self.filter.apply(self.database, self.database.get_person_handles(sort_handles=False)) self.progress.set_pass(_("Reading database"), len(people)) for person_handle in people: self.progress.step() person = self.database.get_person_from_handle(person_handle) birth_ref = person.get_birth_ref() birth_date = None if birth_ref: birth_event = self.database.get_event_from_handle(birth_ref.ref) birth_date = birth_event.get_date_object() living = probably_alive(person, self.database, make_date(self.year, 1, 1), 0) if self.birthday and birth_date != None and ((self.alive and living) or not self.alive): year = birth_date.get_year() month = birth_date.get_month() day = birth_date.get_day() age = self.year - year # add some things to handle maiden name: father_lastname = None # husband, actually if self.surname == 0: # get husband's last name: if person.get_gender() == gen.lib.Person.FEMALE: family_list = person.get_family_handle_list() if len(family_list) > 0: fhandle = family_list[0] # first is primary fam = self.database.get_family_from_handle(fhandle) father_handle = fam.get_father_handle() mother_handle = fam.get_mother_handle() if mother_handle == person_handle: if father_handle: father = self.database.get_person_from_handle(father_handle) if father != None: father_lastname = father.get_primary_name().get_surname() short_name = _get_short_name(person, father_lastname) if age == 0: # person is 0 years old, display nothing text = "" elif age == 1: # person is 1, and therefore display it correctly # TODO. Make this translatable text = _('%(short_name)s, %(age)d year old') % {'short_name' : short_name, 'age' : age} else: text = _('%(short_name)s, %(age)d years old') % {'short_name' : short_name, 'age' : age} self.add_day_item(text, year, month, day) if self.anniv and ((self.alive and living) or not self.alive): family_list = person.get_family_handle_list() for fhandle in family_list: fam = self.database.get_family_from_handle(fhandle) father_handle = fam.get_father_handle() mother_handle = fam.get_mother_handle() if father_handle == person.get_handle(): spouse_handle = mother_handle else: continue # with next person if this was the marriage event if spouse_handle: spouse = self.database.get_person_from_handle(spouse_handle) if spouse: spouse_name = _get_short_name(spouse) short_name = _get_short_name(person) if self.alive: if not probably_alive(spouse, self.database, make_date(self.year, 1, 1), 0): continue married = True for event_ref in fam.get_event_ref_list(): event = self.database.get_event_from_handle(event_ref.ref) if event and event.type in [gen.lib.EventType.DIVORCE, gen.lib.EventType.ANNULMENT, gen.lib.EventType.DIV_FILING]: married = False if married: for event_ref in fam.get_event_ref_list(): event = self.database.get_event_from_handle(event_ref.ref) event_obj = event.get_date_object() year = event_obj.get_year() month = event_obj.get_month() day = event_obj.get_day() years = self.year - year if years == 0: text = "" # zero year anniversary # display nothing else: text = _('%(spouse)s and %(person)s, %(nyears)d year anniversary') % { 'spouse' : spouse_name, 'person' : short_name, 'nyears' : years, } self.add_day_item(text, year, month, day) #------------------------------------------------------------------------ # # WebCalOptions # #------------------------------------------------------------------------ class WebCalOptions(MenuReportOptions): """ Defines options and provides handling interface. """ def __init__(self, name, dbase): self.__db = dbase self.__pid = None self.__filter = None MenuReportOptions.__init__(self, name, dbase) def add_menu_options(self, menu): """ Add options to the menu for the web calendar. """ self.__add_report_options(menu) self.__add_content_options(menu) self.__add_notes_options(menu) def __add_report_options(self, menu): """ Options on the "Report Options" tab. """ category_name = _("Report Options") target = DestinationOption( _("Destination"), os.path.join(const.USER_HOME, "WEBCAL")) target.set_help( _("The destination directory for the web files")) target.set_directory_entry(True) menu.add_option(category_name, "target", target) title = StringOption(_('Calendar Title'), _('My Family Calendar')) title.set_help(_("The title of the calendar")) menu.add_option(category_name, "title", title) self.__filter = FilterOption(_("Filter"), 0) self.__filter.set_help( _("Select filter to restrict people that appear on calendar")) menu.add_option(category_name, "filter", self.__filter) self.__filter.connect('value-changed', self.__filter_changed) self.__pid = PersonOption(_("Filter Person")) self.__pid.set_help(_("The center person for the filter")) menu.add_option(category_name, "pid", self.__pid) self.__pid.connect('value-changed', self.__update_filters) self.__update_filters() ext = EnumeratedListOption(_("File extension"), ".html" ) for etype in ['.html', '.htm', '.shtml', '.php', '.php3', '.cgi']: ext.add_item(etype, etype) ext.set_help( _("The extension to be used for the web files")) menu.add_option(category_name, "ext", ext) cright = EnumeratedListOption(_('Copyright'), 0 ) index = 0 for copt in _COPY_OPTIONS: cright.add_item(index, copt) index += 1 cright.set_help( _("The copyright to be used for the web files")) menu.add_option(category_name, "cright", cright) encoding = EnumeratedListOption(_('Character set encoding'), _CHARACTER_SETS[0][1]) for eopt in _CHARACTER_SETS: encoding.add_item(eopt[1], eopt[0]) encoding.set_help( _("The encoding to be used for the web files")) menu.add_option(category_name, "encoding", encoding) css = EnumeratedListOption(_('StyleSheet'), _CSS_FILES[0][1]) for style in _CSS_FILES: css.add_item(style[1], style[0]) css.set_help( _("The Style Sheet to be used for the web page")) menu.add_option(category_name, "css", css) def __add_content_options(self, menu): """ Options on the "Content Options" tab. """ category_name = _("Content Options") year = NumberOption(_("Year of calendar"), time.localtime()[0], 1000, 3000) year.set_help(_("Year of calendar")) menu.add_option(category_name, "year", year) fullyear = BooleanOption(_('Create Year At A Glance'), False) fullyear.set_help(_('Wheather to create A Full Year High-lighted calendar')) menu.add_option(category_name, 'fullyear', fullyear) blankyear = BooleanOption(_('Create A Printable Blank Full- Year Calendar'), False) blankyear.set_help(_('Wheather to create A Full Year Printable calendar')) menu.add_option(category_name, 'blankyear', blankyear) country = EnumeratedListOption(_('Country for holidays'), 0 ) index = 0 for item in _COUNTRIES: country.add_item(index, item) index += 1 country.set_help( _("Holidays will be included for the selected " "country")) menu.add_option(category_name, "country", country) home_link = StringOption(_('Home link'), '../index.html') home_link.set_help(_("The link to be included to direct the user to " "the main page of the web site")) menu.add_option(category_name, "home_link", home_link) alive = BooleanOption(_("Include only living people"), True) alive.set_help(_("Include only living people in the calendar")) menu.add_option(category_name, "alive", alive) birthdays = BooleanOption(_("Include birthdays"), True) birthdays.set_help(_("Include birthdays in the calendar")) menu.add_option(category_name, "birthdays", birthdays) anniversaries = BooleanOption(_("Include anniversaries"), True) anniversaries.set_help(_("Include anniversaries in the calendar")) menu.add_option(category_name, "anniversaries", anniversaries) surname = BooleanOption(_('Check for wives to use maiden name'), True) surname.set_help(_("Attempt to use maiden names of women")) menu.add_option(category_name, "surname", surname) def __add_notes_options(self, menu): """ Options on the "Months Notes" tabs. """ category_name = _("Jan - Jun Notes") note_jan = StringOption(_('Jan Note'), _('This prints in January')) note_jan.set_help(_("The note for the month of January")) menu.add_option(category_name, "note_jan", note_jan) note_feb = StringOption(_('Feb Note'), _('This prints in February')) note_feb.set_help(_("The note for the month of February")) menu.add_option(category_name, "note_feb", note_feb) note_mar = StringOption(_('Mar Note'), _('This prints in March')) note_mar.set_help(_("The note for the month of March")) menu.add_option(category_name, "note_mar", note_mar) note_apr = StringOption(_('Apr Note'), _('This prints in April')) note_apr.set_help(_("The note for the month of April")) menu.add_option(category_name, "note_apr", note_apr) note_may = StringOption(_('May Note'), _('This prints in May')) note_may.set_help(_("The note for the month of May")) menu.add_option(category_name, "note_may", note_may) note_jun = StringOption(_('Jun Note'), _('This prints in June')) note_jun.set_help(_("The note for the month of June")) menu.add_option(category_name, "note_jun", note_jun) category_name = _("Jul - Dec Notes") note_jul = StringOption(_('Jul Note'), _('This prints in July')) note_jul.set_help(_("The note for the month of July")) menu.add_option(category_name, "note_jul", note_jul) note_aug = StringOption(_('Aug Note'), _('This prints in August')) note_aug.set_help(_("The note for the month of August")) menu.add_option(category_name, "note_aug", note_aug) note_sep = StringOption(_('Sep Note'), _('This prints in September')) note_sep.set_help(_("The note for the month of September")) menu.add_option(category_name, "note_sep", note_sep) note_oct = StringOption(_('Oct Note'), _('This prints in October')) note_oct.set_help(_("The note for the month of October")) menu.add_option(category_name, "note_oct", note_oct) note_nov = StringOption(_('Nov Note'), _('This prints in November')) note_nov.set_help(_("The note for the month of November")) menu.add_option(category_name, "note_nov", note_nov) note_dec = StringOption(_('Dec Note'), _('This prints in December')) note_dec.set_help(_("The note for the month of December")) menu.add_option(category_name, "note_dec", note_dec) def __update_filters(self): """ Update the filter list based on the selected person """ gid = self.__pid.get_value() person = self.__db.get_person_from_gramps_id(gid) filter_list = ReportUtils.get_person_filters(person, False) self.__filter.set_filters(filter_list) def __filter_changed(self): """ Handle filter change. If the filter is not specific to a person, disable the person option """ filter_value = self.__filter.get_value() if filter_value in [1, 2, 3, 4]: # Filters 1, 2, 3 and 4 rely on the center person self.__pid.set_available(True) else: # The rest don't self.__pid.set_available(False) #------------------------------------------------------------------------ # # # #------------------------------------------------------------------------ class Element: """ A parsed XML element """ def __init__(self, name, attributes): 'Element constructor' # The element's tag name self.name = name # The element's attribute dictionary self.attributes = attributes # The element's cdata self.cdata = '' # The element's child element list (sequence) self.children = [] def addChild(self, element): 'Add a reference to a child element' self.children.append(element) def getAttribute(self, key): 'Get an attribute value' return self.attributes.get(key) def getData(self): 'Get the cdata' return self.cdata def getElements(self, name=''): 'Get a list of child elements' #If no tag name is specified, return the all children if not name: return self.children else: # else return only those children with a matching tag name elements = [] for element in self.children: if element.name == name: elements.append(element) return elements def toString(self, level=0): retval = " " * level retval += "<%s" % self.name for attribute in self.attributes: retval += " %s=\"%s\"" % (attribute, self.attributes[attribute]) c = "" for child in self.children: c += child.toString(level+1) if c == "": retval += "/>\n" else: retval += ">\n" + c + ("\n" % self.name) return retval class Xml2Obj: """ XML to Object """ def __init__(self): self.root = None self.nodeStack = [] def StartElement(self, name, attributes): 'SAX start element even handler' # Instantiate an Element object element = Element(name.encode(), attributes) # Push element onto the stack and make it a child of parent if len(self.nodeStack) > 0: parent = self.nodeStack[-1] parent.addChild(element) else: self.root = element self.nodeStack.append(element) def EndElement(self, name): 'SAX end element event handler' self.nodeStack = self.nodeStack[:-1] def CharacterData(self, data): 'SAX character data event handler' if data.strip(): data = data.encode() element = self.nodeStack[-1] element.cdata += data return def Parse(self, filename): # Create a SAX parser Parser = expat.ParserCreate() # SAX event handlers Parser.StartElementHandler = self.StartElement Parser.EndElementHandler = self.EndElement Parser.CharacterDataHandler = self.CharacterData # Parse the XML File ParserStatus = Parser.Parse(open(filename, 'r').read(), 1) return self.root class Holidays: """ Class used to read XML holidays to add to calendar. """ def __init__(self, elements, country="US"): self.debug = 0 self.elements = elements self.country = country self.dates = [] self.initialize() def set_country(self, country): self.country = country self.dates = [] self.initialize() def initialize(self): # parse the date objects for country_set in self.elements.children: if country_set.name == "country" and country_set.attributes["name"] == self.country: for date in country_set.children: if date.name == "date": data = {"value" : "", "name" : "", "offset": "", "type": "", "if": "", } # defaults for attr in date.attributes: data[attr] = date.attributes[attr] self.dates.append(data) def get_daynames(self, y, m, dayname): if self.debug: print "%s's in %d %d..." % (dayname, m, y) retval = [0] dow = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat', 'sun'].index(dayname) for d in range(1, 32): try: date = datetime.date(y, m, d) except ValueError: continue if date.weekday() == dow: retval.append( d ) if self.debug: print "dow=", dow, "days=", retval return retval def check_date(self, date): retval = [] for rule in self.dates: if self.debug: print "Checking ", rule["name"], "..." offset = 0 if rule["offset"] != "": if rule["offset"].isdigit(): offset = int(rule["offset"]) elif rule["offset"][0] in ["-", "+"] and rule["offset"][1:].isdigit(): offset = int(rule["offset"]) else: # must be a dayname offset = rule["offset"] if rule["value"].count("/") == 3: # year/num/day/month, "3rd wednesday in april" y, num, dayname, mon = rule["value"].split("/") if y == "*": y = date.year else: y = int(y) if mon.isdigit(): m = int(mon) elif mon == "*": m = date.month else: m = ['jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'aug', 'sep', 'oct', 'nov', 'dec'].index(mon) + 1 dates_of_dayname = self.get_daynames(y, m, dayname) if self.debug: print "num =", num d = dates_of_dayname[int(num)] elif rule["value"].count("/") == 2: # year/month/day y, m, d = rule["value"].split("/") if y == "*": y = date.year else: y = int(y) if m == "*": m = date.month else: m = int(m) if d == "*": d = date.day else: d = int(d) ndate = datetime.date(y, m, d) if self.debug: print ndate, offset, type(offset) if isinstance(offset, int): if offset != 0: ndate = ndate.fromordinal(ndate.toordinal() + offset) elif isinstance(offset, basestring): dir_ = 1 if offset[0] == "-": dir_ = -1 offset = offset[1:] if offset in ['mon', 'tue', 'wed', 'thu', 'fri', 'sat', 'sun']: # next tuesday you come to, including this one dow = ['mon', 'tue', 'wed', 'thu', 'fri', 'sat', 'sun'].index(offset) ord_ = ndate.toordinal() while ndate.fromordinal(ord_).weekday() != dow: ord_ += dir_ ndate = ndate.fromordinal(ord_) if self.debug: print "ndate:", ndate, "date:", date if ndate == date: if rule["if"] != "": if not eval(rule["if"]): continue retval.append(rule["name"]) return retval # TODO. Try to understand this function and then to replace it with # something more Pythonic, more understandable. # Questions to answer: # * are the file names URLs or native filesystem names? # (This matters if we must use os.path.join(), or build_url_fname()) # * the logic path for "wc" and "yg" and "by" give same subdirs # * why the "./" trailer? def _subdirs(cal, dir_, name): """ This will add the number of subdirs to the filename depending on which calendar is being called: wc = WebCal yg = Year At A Glance by = Printable Pocket Calendar ip = Indiv Pages """ if cal == "wc": subdirs = '.././' elif ((cal == "yg") or (cal == "by")): subdirs = '.././' else: subdirs = '../.././' fname = subdirs + '%s/%s' % (dir_, name) return fname def process_holiday_file(filename): """ This will process a holiday file for country names """ parser = Xml2Obj() element = parser.Parse(filename) country_list = [] for country_set in element.children: if country_set.name == "country": if country_set.attributes["name"] not in country_list: country_list.append(country_set.attributes["name"]) return country_list def _get_countries(): """ Looks in multiple places for holidays.xml files """ locations = [const.PLUGINS_DIR, const.USER_PLUGINS] holiday_file = 'holidays.xml' country_list = [] for dir_ in locations: holiday_full_path = os.path.join(dir_, holiday_file) if os.path.exists(holiday_full_path): cs = process_holiday_file(holiday_full_path) for c in cs: if c not in country_list: country_list.append(c) country_list.sort() country_list.insert(0, _("Don't include holidays")) return country_list # TODO: Only load this once the first time it is actually needed so Gramps # doesn't take so long to start up. _COUNTRIES = _get_countries() # code snippets for Easter and Daylight saving start/ stop # are borrowed from Calendar.py def _easter(year): """ Computes the year/month/day of easter. Based on work by J.-M. Oudin (1940) and is reprinted in the "Explanatory Supplement to the Astronomical Almanac", ed. P. K. Seidelmann (1992). Note: Ash Wednesday is 46 days before Easter Sunday. """ c = year / 100 n = year - 19 * (year / 19) k = (c - 17) / 25 i = c - c / 4 - (c - k) / 3 + 19 * n + 15 i = i - 30 * (i / 30) i = i - (i / 28) * (1 - (i / 28) * (29 / (i + 1)) * ((21 - n) / 11)) j = year + year / 4 + i + 2 - c + c / 4 j = j - 7 * (j / 7) l = i - j month = 3 + (l + 40) / 44 day = l + 28 - 31 * (month / 4) return year, month, day # TODO. Check name prefix, suffix etc. def _get_short_name(person, maiden_name = None): """ Return person's name, unless maiden_name given, unless married_name listed. """ # Get all of a person's names: primary_name = person.get_primary_name() married_name = None names = [primary_name] + person.get_alternate_names() for n in names: if int(n.get_type()) == gen.lib.NameType.MARRIED: married_name = n # Now, decide which to use: if maiden_name is not None: if married_name is not None: first_name, family_name = married_name.get_first_name(), married_name.get_surname() call_name = married_name.get_call_name() else: first_name, family_name = primary_name.get_first_name(), maiden_name call_name = primary_name.get_call_name() else: first_name, family_name = primary_name.get_first_name(), primary_name.get_surname() call_name = primary_name.get_call_name() # If they have a nickname use it if call_name is not None and call_name.strip() != "": first_name = call_name.strip() else: # else just get the first name: first_name = first_name.strip() if " " in first_name: first_name, rest = first_name.split(" ", 1) # just one split max return ("%s %s" % (first_name, family_name)).strip() def _has_webpage_extension(fname): for ext in ('.html', '.htm' '.shtml', '.cgi', '.php', '.php3'): if fname.endswith(ext): return True return False #------------------------------------------------------------------------- # # # #------------------------------------------------------------------------- pmgr = PluginManager.get_instance() pmgr.register_report( name = 'WebCal', category = CATEGORY_WEB, report_class = WebCalReport, options_class = WebCalOptions, modes = MODE_GUI, translated_name = _("Web Calendar"), status = _("Stable"), author_name = "Thom Sturgill", author_email = "thsturgill@yahoo.com", description = _("Produces web (HTML) calendars."), )