# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 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 __author__ = "Douglas Blank " __version__ = "$Revision$" #------------------------------------------------------------------------ # # python modules # #------------------------------------------------------------------------ from TransUtils import sgettext as _ import datetime, time from xml.parsers import expat import const import os import locale #------------------------------------------------------------------------ # # GRAMPS modules # #------------------------------------------------------------------------ import BaseDoc from PluginUtils import Report, ReportOptions, ReportUtils, register_report pt2cm = ReportUtils.pt2cm import GenericFilter from DateHandler import DateDisplay import RelLib # _days could be added to DateDisplay: _codeset = locale.nl_langinfo(locale.CODESET) # slightly different from _months; this is zero based _days = ( unicode(locale.nl_langinfo(locale.DAY_1),_codeset), unicode(locale.nl_langinfo(locale.DAY_2),_codeset), unicode(locale.nl_langinfo(locale.DAY_3),_codeset), unicode(locale.nl_langinfo(locale.DAY_4),_codeset), unicode(locale.nl_langinfo(locale.DAY_5),_codeset), unicode(locale.nl_langinfo(locale.DAY_6),_codeset), unicode(locale.nl_langinfo(locale.DAY_7),_codeset), ) _language, _country = "en", "US" (_language_country, _encoding) = locale.getlocale() if _language_country != None and _language_country.count("_") == 1: (_language, _country) = _language_country.split("_") #------------------------------------------------------------------------ # # The one and only GUI, unfortunately. This will be able to be moved to # Widget once it is created. # #------------------------------------------------------------------------ import gtk #------------------------------------------------------------------------ # # Support functions # #------------------------------------------------------------------------ def easter(year): """ Computes the year/month/day of easter. Currently hardcoded in holidays.xml. 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 "%d/%d/%d" % (year, month, day) #------------------------------------------------------------------------ # # Calendar # #------------------------------------------------------------------------ class Calendar(Report.Report): """ Creates the Calendar object that produces the report. """ def __getitem__(self, item): """ Used to get items from various places. Could be moved up to Report. """ if item in self.doc.style_list: # font is the only element people refer to in writing reports # from the style_list: return self.doc.style_list[item].get_font() elif item in self.options_class.options_dict: # otherwise it is a option: return self.options_class.options_dict[item] else: raise AttributeError, ("no widget named '%s'" % item) def define_graphics_styles(self): """ Set up the report. Could be moved up to Report. """ for widget in self.options_class.widgets: if widget.__class__.__name__ == "StyleWidget": widget.define_graphics_style(self.doc) def get_short_name(self, person, maiden_name = None): """ Is there a better, built-in way of getting a short, personal name? """ nickname = person.get_nick_name() if nickname: first_name = nickname else: first_name = person.get_primary_name().get_first_name() if first_name: first_name = first_name.split()[0] # not middle name if nickname.strip().count(" ") > 0: # HACK: first and last name assumed family_name = "" elif maiden_name != None: family_name = maiden_name else: family_name = person.get_primary_name().get_surname() return ("%s %s" % (first_name, family_name)).strip() def draw_rectangle(self, style, sx, sy, ex, ey): """ This should be in BaseDoc """ self.doc.draw_line(style, sx, sy, sx, ey) self.doc.draw_line(style, sx, sy, ex, sy) self.doc.draw_line(style, ex, sy, ex, ey) self.doc.draw_line(style, sx, ey, ex, ey) ### The rest of these all have to deal with calendar specific things def add_day_item(self, text, year, month, day): 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, year, country = "US"): """ Looks in multiple places for holidays.xml files """ locations = [const.pluginsDir, os.path.expanduser("~/.gramps/plugins")] holiday_file = 'holidays.xml' for dir in locations: holiday_full_path = os.path.join(dir, holiday_file) if os.path.exists(holiday_full_path): self.process_holiday_file(holiday_full_path, year, country) def process_holiday_file(self, filename, year, country): """ This will process a holiday file """ parser = Xml2Obj() element = parser.Parse(filename) calendar = Holidays(element, country) date = datetime.date(year, 1, 1) while date.year == year: holidays = calendar.check_date( date ) for text in holidays: self.add_day_item(text, date.year, date.month, date.day) date = date.fromordinal( date.toordinal() + 1) def write_report(self): """ The short method that runs through each month and creates a page. """ # initialize the dict to fill: self.calendar = {} # get the information, first from holidays: if self["holidays"]: self.get_holidays(self["year"], _country) # currently global # get data from database: self.collect_data() # generate the report: for month in range(1, 13): self.print_page(month) def print_page(self, month): """ This method actually writes the calendar page. """ self.doc.start_page() width = self.doc.get_usable_width() height = self.doc.get_usable_height() header = self.doc.tmargin self.draw_rectangle("border", 0, 0, width, height) self.doc.draw_bar("title", 0, 0, width, header) self.doc.draw_line("border", 0, header, width, header) year = self["year"] title = "%s %d" % (DateDisplay._months[month], year) font_height = pt2cm(1.25 * self["title"].get_size()) self.doc.center_text("title", title, width/2, font_height + self["offset"]) # 1.0 cell_width = width / 7 cell_height = (height - header)/ 6 current_date = datetime.date(year, month, 1) spacing = pt2cm(1.25 * self["text"].get_size()) # 158 if current_date.isoweekday() != 7: # start dow here is 7, sunday current_ord = current_date.toordinal() - current_date.isoweekday() else: current_ord = current_date.toordinal() for day_col in range(7): self.doc.center_text("daynames", _days[day_col], # global day_col * cell_width + cell_width/2, header - font_height/3.0 + self["offset"]) # .35 for week_row in range(6): something_this_week = 0 for day_col in range(7): thisday = current_date.fromordinal(current_ord) if thisday.month == month: something_this_week = 1 self.draw_rectangle("border", day_col * cell_width, header + week_row * cell_height, (day_col + 1) * cell_width, header + (week_row + 1) * cell_height) last_edge = (day_col + 1) * cell_width self.doc.center_text("numbers", str(thisday.day), day_col * cell_width + cell_width/2, header + week_row * cell_height + .5 + self["offset"]) list = self.calendar.get(month, {}).get(thisday.day, []) position = 0.0 for p in list: lines = p.count("\n") + 1 # lines in the text position += (lines * spacing) current = 0 for line in p.split("\n"): self.doc.write_at("text", line, day_col * cell_width + .1, header + (week_row + 1) * cell_height - position + (current * spacing) + self["offset"]) current += 1 current_ord += 1 if not something_this_week: last_edge = 0 font_height = pt2cm(1.25 * self["text1style"].get_size()) self.doc.center_text("text1style", self["text1"], last_edge + (width - last_edge)/2, height - font_height * 3 + self["offset"]) # - 1.5 self.doc.center_text("text2style", self["text2"], last_edge + (width - last_edge)/2, height - font_height * 2 + self["offset"]) # - 0.78 self.doc.center_text("text3style", self["text3"], last_edge + (width - last_edge)/2, height - font_height * 1 + self["offset"]) # - 0.30 self.doc.end_page() def collect_data(self): """ This method runs through the data, and collects the relevant dates and text. """ filter_num = self.options_class.get_filter_number() filters = self.options_class.get_report_filters(self.start_person) filters.extend(GenericFilter.CustomFilters.get_filters()) self.filter = filters[filter_num] people = self.filter.apply(self.database, self.database.get_person_handles(sort_handles=False)) for person_handle in people: person = self.database.get_person_from_handle(person_handle) birth_handle = person.get_birth_handle() birth_date = None if birth_handle: birth_event = self.database.get_event_from_handle(birth_handle) birth_date = birth_event.get_date() if birth_date == "": birth_date = None else: birth_date_obj = birth_event.get_date_object() death_handle = person.get_death_handle() death_date = None if death_handle: death_event = self.database.get_event_from_handle(death_handle) death_date = death_event.get_date() if death_date == "": # BUG: couldn't remove event death_date = None else: death_date_obj = death_event.get_date_object() if death_date == None: alive = True # well, until we get probably_alive in here else: alive = False if self["birthdays"] and birth_date != None and ((self["alive"] and alive) or not self["alive"]): year = birth_date_obj.get_year() month = birth_date_obj.get_month() day = birth_date_obj.get_day() age = self["year"] - year # add some things to handle maiden name: father_lastname = None # husband, actually if self["maiden_name"] == 0: # get husband's last name: if person.get_gender() == RelLib.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 = self.get_short_name(person, father_lastname) self.add_day_item("%s, %d" % (short_name, age), year, month, day) if self["anniversaries"] and ((self["alive"] and alive) 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 = self.get_short_name(spouse) short_name = self.get_short_name(person) if self["alive"]: spouse_death_handle = spouse.get_death_handle() if spouse_death_handle != None: # there is a death event, maybe empty though spouse_death_event = self.database.get_event_from_handle(spouse_death_handle) if spouse_death_event != None: spouse_death_date = spouse_death_event.get_date() if spouse_death_date != "": # BUG: couldn't remove event continue for event_handle in fam.get_event_list(): event = self.database.get_event_from_handle(event_handle) 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 text = _("%(spouse)s and\n %(person)s, %(nyears)d") % { 'spouse' : spouse_name, 'person' : short_name, 'nyears' : years, } self.add_day_item(text, year, month, day) ################################################################################### # These classes are a suggested of how to rework the graphics out of reports. It also # makes these items abstractions, which makes it easy to change the report # infrastructure without having everyone rewrite their reports each time. # # This builds on the current document code, so no changes are needed. ################################################################################### class Widget: """ A Widget virtual base class. This contains no graphics specifics. """ commonDefaults = { "wtype" : None, "name" : None, "label" : None, "help" : None, "wtype" : None, "valid_text" : None, "frame" : None, "value" : None, } defaults = {} def __init__(self, option_object, **args): self.option_object = option_object self.setup(args) self.register() def __getitem__(self, key): if key in self.settings: return self.settings[key] else: raise AttributeError, ("no widget attribute named '%s'" % key) def __setitem__(self, key, value): self.settings[key] = value def setup(self, args = {}): # start with the base defaults common to all: self.settings = self.commonDefaults.copy() # now add those from the subclass: self.settings.update(self.defaults) # ad finally, those from the user: self.settings.update(args) def register(self): className = self.__class__.__name__ if className == "FilterWidget": self.option_object.enable_dict['filter'] = 0 elif className == "StyleWidget": self.option_object[self["name"]] = self["value"] else: self.option_object[self["name"]] = self["value"] self.option_object.options_help[self["name"]] = ( self["wtype"], self["help"], self["valid_text"]) def add_gui(self, dialog): pass def update(self): pass class SpinWidget(Widget): """ A spinner number selector widget for GTK. """ defaults = { "wtype" : "=num", "help" : "Numeric option", "valid_text": "Any number", } def add_gui(self, dialog): keyword = self["name"] obj = self.option_object.__dict__ obj[keyword] = gtk.SpinButton() obj[keyword].set_digits(0) obj[keyword].set_increments(1,2) obj[keyword].set_range(0,2100) obj[keyword].set_numeric(True) obj[keyword].set_value(self.option_object[keyword]) if self["frame"] != None: dialog.add_frame_option(self["frame"], self["label"], obj[keyword]) else: dialog.add_option(self["label"], obj[keyword]) def update(self): dict = self.option_object.__dict__ keyword = self["name"] self.option_object[keyword] = dict[keyword].get_value_as_int() self[keyword] = dict[keyword].get_value_as_int() class CheckWidget(Widget): """ A check box widget for GTK. """ defaults = { "wtype" : "=0/1", "help" : "Yes/No option", "valid_text": "1 for yes, 0 for no", } def add_gui(self, dialog): keyword = self["name"] obj = self.option_object.__dict__ obj[keyword] = gtk.CheckButton(self["label"]) obj[keyword].set_active(self.option_object[keyword]) if self["frame"] != None: dialog.add_frame_option(self["frame"], "", obj[keyword]) else: dialog.add_option("", obj[keyword]) def update(self): dict = self.option_object.__dict__ keyword = self["name"] self.option_object[keyword] = int(dict[keyword].get_active()) self[keyword] = int(dict[keyword].get_active()) class EntryWidget(Widget): """ A text widget for GTK. """ defaults = { "wtype" : "=str", "help" : "String option", "valid_text": "Any textual data", } def add_gui(self, dialog): keyword = self["name"] obj = self.option_object.__dict__ obj[keyword] = gtk.Entry() obj[keyword].set_text(self.option_object[keyword]) if self["frame"] != None: dialog.add_frame_option(self["frame"], self["label"], obj[keyword]) else: dialog.add_option(self["label"], obj[keyword]) def update(self): dict = self.option_object.__dict__ keyword = self["name"] self.option_object[keyword] = unicode(dict[keyword].get_text()) self[keyword] = unicode(dict[keyword].get_text()) class NumberWidget(EntryWidget): """ A number widget for GTK. """ defaults = { "wtype" : "=num", "help" : "Numeric option", "valid_text": "Any number", } def add_gui(self, dialog): keyword = self["name"] obj = self.option_object.__dict__ obj[keyword] = gtk.Entry() obj[keyword].set_text(str(self.option_object[keyword])) if self["frame"] != None: dialog.add_frame_option(self["frame"], self["label"], obj[keyword]) else: dialog.add_option(self["label"], obj[keyword]) def update(self): dict = self.option_object.__dict__ keyword = self["name"] self.option_object[keyword] = float(dict[keyword].get_text()) self[keyword] = float(dict[keyword].get_text()) class StyleWidget(Widget): defaults = { "size" : 8, "bold" : 0, "italics" : 0, "type_face" : BaseDoc.FONT_SERIF, "fill_color": None, } def make_default_style(self, default_style): f = BaseDoc.FontStyle() f.set_size(self["size"]) f.set_italic(self["italics"]) f.set_bold(self["bold"]) f.set_type_face(self["type_face"]) p = BaseDoc.ParagraphStyle() p.set_font(f) p.set_description(self["label"]) default_style.add_style(self["name"],p) def define_graphics_style(self, document): g = BaseDoc.GraphicsStyle() g.set_paragraph_style(self["name"]) if self["fill_color"]: g.set_fill_color(self["fill_color"]) # FIXME: add all other graphics items (color, etc) here document.add_draw_style(self["name"],g) class FilterWidget(Widget): """ A filter widget. This doesn't have the GTK code here, but should. This class takes names of the filters and does everything for you. "all filters" - all of them "everyone" - all people in table "descendents" - direct descendents "descendent familes" - direct descendents and their familes "ancestors" - all ancestors of person "common ancestors" - all common ancestors "calendar attribute" - experimental filter for tagging people """ def get_filters(self, person): """Set up the list of possible content filters.""" if person: name = person.get_primary_name().get_name() gramps_id = person.get_gramps_id() else: name = 'PERSON' gramps_id = '' retval = [] for filter in self["filters"]: if filter in ["everyone", "all filters"]: f = GenericFilter.GenericFilter() f.set_name(_("Entire Database")) f.add_rule(GenericFilter.Everyone([])) retval.append(f) if filter in ["descendants", "all filters"]: f = GenericFilter.GenericFilter() f.set_name(_("Descendants of %s") % name) f.add_rule(GenericFilter.IsDescendantOf([gramps_id,1])) retval.append(f) if filter in ["descendant families", "all filters"]: f = GenericFilter.GenericFilter() f.set_name(_("Descendant Families of %s") % name) f.add_rule(GenericFilter.IsDescendantFamilyOf([gramps_id])) retval.append(f) if filter in ["ancestors", "all filters"]: f = GenericFilter.GenericFilter() f.set_name(_("Ancestors of %s") % name) f.add_rule(GenericFilter.IsAncestorOf([gramps_id,1])) retval.append(f) if filter in ["common ancestors", "all filters"]: f = GenericFilter.GenericFilter() f.set_name(_("People with common ancestor with %s") % name) f.add_rule(GenericFilter.HasCommonAncestorWith([gramps_id])) retval.append(f) if filter in ["calendar attribute", "all filters"]: f = GenericFilter.ParamFilter() f.set_name(_("People with a Calendar attribute")) f.add_rule(GenericFilter.HasTextMatchingSubstringOf(['Calendar',0,0])) retval.append(f) return retval # ----------------------------------------------------------------- # The following could all be moved to the parent class, if you wanted # to adopt this report reworking. Even if you didn't want to use them # it would be ok to put there, because self.widgets would be empty. # ----------------------------------------------------------------- class NewReportOptions(ReportOptions.ReportOptions): """ Defines options and provides code to handling the interface. This is free of any graphics specifics. """ def __getitem__(self, keyword): """ This could be moved up to ReportOptions """ if keyword in self.options_dict: return self.options_dict[keyword] else: raise AttributeError, ("no widget named '%s'" % keyword) def __setitem__(self, keyword, value): """ This could be moved up to ReportOptions """ self.options_dict[keyword] = value def add_user_options(self,dialog): for widget in self.widgets: widget.add_gui(dialog) def parse_user_options(self,dialog): for widget in self.widgets: widget.update() def get_report_filters(self,person): for widget in self.widgets: if widget.__class__.__name__ == "FilterWidget": return widget.get_filters(person) def make_default_style(self,default_style): for widget in self.widgets: if widget.__class__.__name__ == "StyleWidget": widget.make_default_style(default_style) class CalendarOptions(NewReportOptions): def enable_options(self): self.enable_dict = {} self.widgets = [ FilterWidget(self, label = _("Filter"), name = "filter", filters = ["all filters"]), EntryWidget(self, label = _("Text 1"), name = "text1", value = "My Calendar", help = "Large text area", valid_text = "Any text", frame = _("Text Options") ), EntryWidget(self, label = _("Text 2"), name = "text2", value = "Produced with GRAMPS", help = "Medium size text", valid_text = "Any text", frame = _("Text Options") ), EntryWidget(self, label = _("Text 3"), name = "text3", value = "http://gramps-project.org/", help = "Small text area", valid_text = "Any text", frame = _("Text Options") ), SpinWidget(self, label = _("Year of calendar"), name = "year", value = time.localtime()[0], # the current year help = "Year of calendar", valid_text = "Any year", ), CheckWidget(self, label = _("Use maiden names"), name = "maiden_name", value = 1, help = "Use married women's maiden name.", valid_text = "Select to use married women's maiden name.", ), CheckWidget(self, label = _("Only include living people"), name = "alive", value = 1, help = "Include only living people", valid_text = "Select to only include living people", ), CheckWidget(self, label = _("Include birthdays"), name = "birthdays", value = 1, help = "Include birthdays", valid_text = "Select to include birthdays", ), CheckWidget(self, label = _("Include anniversaries"), name = "anniversaries", value = 1, help = "Include anniversaries", valid_text = "Select to include anniversaries", ), CheckWidget(self, label = _("Include holidays"), name = "holidays", value = 1, help = "Include holidays", valid_text = "Select to include holidays", ), NumberWidget(self, label = "Offset", name = "offset", value = -0.5, help = "Distance to move text on page", valid_text = "Any number", frame = "Text Options" ), StyleWidget(self, label = _('Title text and background color.'), name = "title", size = 20, italics = 1, bold = 1, fill_color = (0xEA,0xEA,0xEA), type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Border lines of calendar boxes.'), name = "border", ), StyleWidget(self, label = _('Calendar day numbers.'), name = "numbers", size = 13, bold = 1, type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Daily text display.'), name = "text", size = 9, type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Days of the week text.'), name = "daynames", size = 12, italics = 1, bold = 1, type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Text at bottom, line 1.'), name = "text1style", size = 12, type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Text at bottom, line 2.'), name = "text2style", size = 12, type_face = BaseDoc.FONT_SERIF, ), StyleWidget(self, label = _('Text at bottom, line 3.'), name = "text3style", size = 9, type_face = BaseDoc.FONT_SERIF, ), ] 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 type(offset) == int: if offset != 0: ndate = ndate.fromordinal(ndate.toordinal() + offset) elif type(offset) in [type(u''), str]: 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 #------------------------------------------------------------------------ # # Register this plugin # #------------------------------------------------------------------------ register_report( name = 'calendar', category = Report.CATEGORY_DRAW, report_class = Calendar, options_class = CalendarOptions, modes = Report.MODE_GUI | Report.MODE_BKI | Report.MODE_CLI, translated_name = _("Calendar"), status = _("Experimental"), author_name = "Douglas S. Blank", author_email = "Doug.Blank@gmail.com", description = _("Produces a graphical calendar"), )