# # Gramps - a GTK+/GNOME based genealogy program # # Copyright (C) 2000-2006 Donald N. Allingham # Copyright (C) 2007-2008 Brian G. Matherly # Copyright (C) 2009 Benny Malengier # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # # $Id:HtmlDoc.py 9912 2008-01-22 09:17:46Z acraphae $ #------------------------------------------------------------------------ # # python modules # #------------------------------------------------------------------------ import os import re import time from gettext import gettext as _ #------------------------------------------------------------------------ # # GRAMPS modules # #------------------------------------------------------------------------ from gen.plug import PluginManager import ImgManip import tarfile import const import Errors import BaseDoc from QuestionDialog import ErrorDialog, WarningDialog import Utils #------------------------------------------------------------------------ # # Constant regular expressions # #------------------------------------------------------------------------ t_header_line_re = re.compile( r"(.*)(.*)(.*)", re.DOTALL|re.IGNORECASE|re.MULTILINE) t_keyword_line_re = re.compile( r'(.*name="keywords"\s+content=")([^\"]*)(".*)$', re.DOTALL|re.IGNORECASE|re.MULTILINE) #------------------------------------------------------------------------ # # Default template # #------------------------------------------------------------------------ _top = [ '\n', '\n', '\n', ' \n', ' \n', ' \n', ' \n', ' \n', '\n', '\n', ' \n' ] _bottom = [ ' \n', '\n', '\n' ] #------------------------------------------------------------------------ # # HtmlDoc # #------------------------------------------------------------------------ class HtmlDoc(BaseDoc.BaseDoc,BaseDoc.TextDoc): def __init__(self,styles,type,template): BaseDoc.BaseDoc.__init__(self,styles,None,template) self.year = time.localtime(time.time())[0] self.ext = '.html' self.meta = "" self.copyright = 'Copyright © %d' % (self.year) self.map = None self.f = None self.filename = None self.top = [] self.bottom = [] self.base = "" self.load_template() self.build_header() self.style_declaration = None self.image_dir = "images" def set_extension(self,val): if val[0] != '.': val = "." + val self.ext = val def set_image_dir(self,dirname): self.image_dir = dirname def set_keywords(self,keywords): self.meta = ",".join(keywords) def load_tpkg(self): start = re.compile(r"") stop = re.compile(r"") top_add = 1 bottom_add = 0 archive = tarfile.open(self.template) self.map = {} for tarinfo in archive: self.map[tarinfo.name] = archive.extractfile(tarinfo) templateFile = self.map['template.html'] while 1: line = templateFile.readline() if line == '': break if top_add == 1: self.top.append(line) match = start.search(line) if match: top_add = 0 elif bottom_add == 0: match = stop.search(line) if match is not None: bottom_add = 1 self.bottom.append(line) else: self.bottom.append(line) templateFile.close() archive.close if top_add == 1: mymsg = _("The marker '' was not in the template") ErrorDialog(_("Template Error"),mymsg) def load_html(self): start = re.compile(r"") stop = re.compile(r"") top_add = 1 bottom_add = 0 templateFile = open(self.template,"r") for line in templateFile.readlines(): if top_add == 1: self.top.append(line) match = start.search(line) if match: top_add = 0 elif bottom_add == 0: match = stop.search(line) if match is not None: bottom_add = 1 self.bottom.append(line) else: self.bottom.append(line) templateFile.close() if top_add == 1: mymsg = _("The marker '' was not in the template") ErrorDialog(_("Template Error"),mymsg) def load_template(self): if self.template: try: if self.template[-4:] == 'tpkg': self.load_tpkg() else: self.load_html() except IOError,msg: mymsg = _("Could not open %s\nUsing the default template") % \ self.template WarningDialog(mymsg,str(msg)) self.bottom = _bottom self.top = _top except: mymsg = _("Could not open %s\nUsing the default template") % \ self.template WarningDialog(mymsg) self.bottom = _bottom self.top = _top else: self.bottom = _bottom self.top = _top def process_line(self,line): l = line.replace('$VERSION',const.VERSION) return l.replace('$COPYRIGHT',self.copyright) def open(self,filename): (r,e) = os.path.splitext(filename) if e == self.ext: self.filename = filename else: self.filename = filename + self.ext self.base = os.path.dirname(self.filename) try: self.f = open(self.filename,"w") except IOError,msg: errmsg = "%s\n%s" % (_("Could not create %s") % self.filename, msg) raise Errors.ReportError(errmsg) except: raise Errors.ReportError(_("Could not create %s") % self.filename) if self.meta: match = t_keyword_line_re.match(self.file_header) if match: g = match.groups() line = "%s%s%s" % (g[0],self.meta,g[2]) else: line = self.file_header else: line = self.file_header self.f.write(line) if not self.style_declaration: self.build_style_declaration() self.f.write(self.style_declaration) def build_header(self): self.fix_title("".join(self.top)) def fix_title(self,msg=None): if msg is None: match = t_header_line_re.match(self.file_header) else: match = t_header_line_re.match(msg) if match: m = match.groups() self.file_header = '%s%s%s\n' % (m[0],m[1],m[2]) else: self.file_header = "".join(self.top) self.file_header = self.process_line(self.file_header) def build_style_declaration(self): styles = self.get_style_sheet() text = ['') self.style_declaration = '\n'.join(text) def close(self): for line in self.bottom: self.f.write(self.process_line(line)) self.f.close() self.write_support_files() if self.open_req: Utils.open_file_with_default_application(self.filename) def write_support_files(self): if self.map: for name in self.map.keys(): if name == 'template.html': continue fname = '%s%s%s' % (self.base, os.path.sep, name) try: f = open(fname, 'wb') f.write(self.map[name].read()) f.close() except IOError,msg: errmsg = "%s\n%s" % (_("Could not create %s") % fname, msg) raise Errors.ReportError(errmsg) except: raise Errors.ReportError(_("Could not create %s") % fname) def add_media_object(self, name,pos,x,y,alt=''): self.empty = 0 size = int(max(x,y) * float(150.0/2.54)) refname = "is%s" % os.path.basename(name) if self.image_dir: imdir = '%s%s%s' % (self.base, os.path.sep,self.image_dir) else: imdir = self.base if not os.path.isdir(imdir): try: os.mkdir(imdir) except: return try: ImgManip.resize_to_jpeg(name, refname, size, size) except: return if pos == "right": xtra = ' align="right"' elif pos == "left" : xtra = ' align="left"' else: xtra = '' if self.image_dir: self.f.write('%s\n' % \ (self.image_dir, refname, alt, xtra)) else: self.f.write('%s\n' % (refname, alt, xtra)) def start_table(self, name,style): styles = self.get_style_sheet() self.tbl = styles.get_table_style(style) self.f.write('\n') def end_table(self): self.f.write('
\n') def start_row(self): self.col = 0 self.f.write('\n') def end_row(self): self.f.write('\n') def start_cell(self,style_name,span=1): self.empty = 1 self.f.write(' 1: self.f.write(' colspan="' + str(span) + '"') self.col = self.col + 1 else: self.f.write(' width="') self.f.write(str(self.tbl.get_column_width(self.col))) self.f.write('%"') self.f.write(' class="') self.f.write(style_name) self.f.write('">') self.col = self.col + 1 def end_cell(self): self.f.write('\n') def start_paragraph(self,style_name,leader=None): self.f.write('

') if leader is not None: self.f.write(leader) self.f.write(' ') def end_paragraph(self): if self.empty == 1: self.f.write(' ') self.empty = 0 self.f.write('

\n') def start_bold(self): self.f.write('') def end_bold(self): self.f.write('') def start_superscript(self): self.f.write('') def end_superscript(self): self.f.write('') def write_note(self,text,format,style_name): if format == 1: self.f.write('
' % style_name)
            self.write_text(text)
            self.f.write('
') elif format == 0: for line in text.split('\n\n'): self.start_paragraph(style_name) self.write_text(line.strip().replace('\n',' ')) self.end_paragraph() def write_text(self,text,mark=None): text = text.replace('&','&'); # Must be first text = text.replace('<','<'); text = text.replace('>','>'); text = text.replace('\n','
') if text != "": self.empty = 0 self.f.write(text) def page_break(self): pass #------------------------------------------------------------------------ # # Register the document generator with the GRAMPS plugin system # #------------------------------------------------------------------------ pmgr = PluginManager.get_instance() pmgr.register_text_doc(_('HTML'), HtmlDoc, 0, 1, ".html")