c962d5e6e1
* various (294 files) pep8 doc fixes & pylint fixes svn: r10103
341 lines
14 KiB
Python
341 lines
14 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2001-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
|
|
#
|
|
|
|
# $Id$
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import os
|
|
from gettext import gettext as _
|
|
|
|
import logging
|
|
|
|
log = logging.getLogger(".")
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GTK+ modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GRAMPS modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import Config
|
|
import Errors
|
|
from QuestionDialog import ErrorDialog, OptionDialog
|
|
from _Constants import (CATEGORY_TEXT, CATEGORY_DRAW, CATEGORY_BOOK,
|
|
CATEGORY_VIEW, CATEGORY_CODE, CATEGORY_WEB,
|
|
CATEGORY_GRAPHVIZ, standalone_categories)
|
|
from _BareReportDialog import BareReportDialog
|
|
from _FileEntry import FileEntry
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# ReportDialog class
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
class ReportDialog(BareReportDialog):
|
|
"""
|
|
The ReportDialog base class. This is a base class for generating
|
|
customized dialogs to solicit options for a report. It cannot be
|
|
used as is, but it can be easily sub-classed to create a functional
|
|
dialog for a stand-alone report.
|
|
"""
|
|
|
|
def __init__(self, dbstate, uistate, option_class, name, trans_name,
|
|
track=[]):
|
|
"""Initialize a dialog to request that the user select options
|
|
for a basic *stand-alone* report."""
|
|
|
|
self.style_name = "default"
|
|
self.page_html_added = False
|
|
BareReportDialog.__init__(self, dbstate, uistate, option_class,
|
|
name, trans_name, track)
|
|
|
|
# Allow for post processing of the format frame, since the
|
|
# show_all task calls events that may reset values
|
|
|
|
def init_interface(self):
|
|
BareReportDialog.init_interface(self)
|
|
|
|
def get_title(self):
|
|
"""The window title for this dialog"""
|
|
name = self.report_name
|
|
category = standalone_categories[self.category]
|
|
return "%s - %s - GRAMPS" % (name,category)
|
|
|
|
def get_header(self, name):
|
|
"""The header line to put at the top of the contents of the
|
|
dialog box. By default this will just be the name of the
|
|
report for the selected person. """
|
|
return self.report_name
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Customization hooks for subclasses
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def get_target_is_directory(self):
|
|
"""Is the user being asked to input the name of a file or a
|
|
directory in the 'Save As' File Entry widget. This item
|
|
currently only selects the Filename/Directory prompt, and
|
|
whether or not the browser accepts filenames. In the future it
|
|
may also control checking of the selected filename."""
|
|
return None
|
|
|
|
def get_default_basename(self):
|
|
"""What should the default name be?
|
|
"""
|
|
spath = self.options.handler.get_stylesheet_savefile()
|
|
return spath.split('.')[0]
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Functions related getting/setting the default directory for a dialog.
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def get_default_directory(self):
|
|
"""Get the name of the directory to which the target dialog
|
|
box should default. This value can be set in the preferences
|
|
panel."""
|
|
return Config.get(Config.REPORT_DIRECTORY)
|
|
|
|
def set_default_directory(self, value):
|
|
"""Save the name of the current directory, so that any future
|
|
reports will default to the most recently used directory.
|
|
This also changes the directory name that will appear in the
|
|
preferences panel, but does not change the preference in disk.
|
|
This means that the last directory used will only be
|
|
remembered for this session of gramps unless the user saves
|
|
his/her preferences."""
|
|
Config.set(Config.REPORT_DIRECTORY,value)
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Functions related to setting up the dialog window.
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def setup_target_frame(self):
|
|
"""Set up the target frame of the dialog. This function
|
|
relies on several target_xxx() customization functions to
|
|
determine whether the target is a directory or file, what the
|
|
title of any browser window should be, and what default
|
|
directory should be used."""
|
|
|
|
# Save Frame
|
|
|
|
label = gtk.Label("<b>%s</b>" % _('Document Options'))
|
|
label.set_use_markup(1)
|
|
label.set_alignment(0.0,0.5)
|
|
self.tbl.set_border_width(12)
|
|
self.tbl.attach(label, 0, 4, self.col, self.col+1, gtk.FILL)
|
|
self.col += 1
|
|
|
|
hid = self.get_stylesheet_savefile()
|
|
if hid[-4:]==".xml":
|
|
hid = hid[0:-4]
|
|
self.target_fileentry = FileEntry(hid,_("Save As"))
|
|
|
|
if self.get_target_is_directory():
|
|
self.target_fileentry.set_directory_entry(1)
|
|
self.doc_label = gtk.Label("%s:" % _("Directory"))
|
|
else:
|
|
self.doc_label = gtk.Label("%s:" % _("Filename"))
|
|
self.doc_label.set_alignment(0.0,0.5)
|
|
|
|
self.tbl.attach(self.doc_label, 1, 2, self.col, self.col+1,
|
|
xoptions=gtk.SHRINK|gtk.FILL,yoptions=gtk.SHRINK)
|
|
self.tbl.attach(self.target_fileentry, 2, 4, self.col, self.col+1,
|
|
xoptions=gtk.EXPAND|gtk.FILL,yoptions=gtk.SHRINK)
|
|
self.col += 1
|
|
|
|
spath = self.get_default_directory()
|
|
|
|
self.target_fileentry.set_filename(spath)
|
|
self.target_fileentry.gtk_entry().set_position(len(spath))
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Functions related to retrieving data from the dialog window
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def parse_target_frame(self):
|
|
"""Parse the target frame of the dialog. If the target
|
|
filename is empty this routine returns a special value of None
|
|
to tell the calling routine to give up. This function also
|
|
saves the current directory so that any future reports will
|
|
default to the most recently used directory."""
|
|
self.target_path = self.target_fileentry.get_full_path(0)
|
|
if not self.target_path:
|
|
return None
|
|
|
|
# First we check whether the selected path exists
|
|
if os.path.exists(self.target_path):
|
|
|
|
# selected path is an existing dir and we need a dir
|
|
if os.path.isdir(self.target_path) \
|
|
and self.get_target_is_directory():
|
|
|
|
# check whether the dir has rwx permissions
|
|
if not os.access(self.target_path, os.R_OK|os.W_OK|os.X_OK):
|
|
ErrorDialog(_('Permission problem'),
|
|
_("You do not have permission to write "
|
|
"under the directory %s\n\n"
|
|
"Please select another directory or correct "
|
|
"the permissions." % self.target_path)
|
|
)
|
|
return None
|
|
|
|
# selected path is an exsting file and we need a file
|
|
if os.path.isfile(self.target_path) \
|
|
and not self.get_target_is_directory():
|
|
a = OptionDialog(_('File already exists'),
|
|
_('You can choose to either overwrite the '
|
|
'file, or change the selected filename.'),
|
|
_('_Overwrite'),None,
|
|
_('_Change filename'),None)
|
|
|
|
if a.get_response() == gtk.RESPONSE_YES:
|
|
return None
|
|
|
|
# selected path does not exist yet
|
|
else:
|
|
# we will need to create the file/dir
|
|
# need to make sure we can create in the parent dir
|
|
parent_dir = os.path.dirname(os.path.normpath(self.target_path))
|
|
if not os.access(parent_dir, os.W_OK):
|
|
ErrorDialog(_('Permission problem'),
|
|
_("You do not have permission to create "
|
|
"%s\n\n"
|
|
"Please select another path or correct "
|
|
"the permissions." % self.target_path)
|
|
)
|
|
return None
|
|
|
|
self.set_default_directory(os.path.dirname(self.target_path) + os.sep)
|
|
self.options.handler.output = self.target_path
|
|
return 1
|
|
|
|
def on_ok_clicked(self, obj):
|
|
"""The user is satisfied with the dialog choices. Validate
|
|
the output file name before doing anything else. If there is
|
|
a file name, gather the options and create the report."""
|
|
|
|
# Is there a filename? This should also test file permissions, etc.
|
|
if not self.parse_target_frame():
|
|
self.window.run()
|
|
|
|
# Preparation
|
|
self.parse_style_frame()
|
|
self.parse_user_options()
|
|
|
|
# Save options
|
|
self.options.handler.save_options()
|
|
|
|
#------------------------------------------------------------------------
|
|
#
|
|
# Generic task function a standalone GUI report
|
|
#
|
|
#------------------------------------------------------------------------
|
|
def report(dbstate,uistate,person,report_class, options_class,
|
|
trans_name, name,category, require_active):
|
|
"""
|
|
report - task starts the report. The plugin system requires that the
|
|
task be in the format of task that takes a database and a person as
|
|
its arguments.
|
|
"""
|
|
|
|
if require_active and not person:
|
|
ErrorDialog(
|
|
_('Active person has not been set'),
|
|
_('You must select an active person for this report to work '
|
|
'properly.'))
|
|
return
|
|
|
|
if category == CATEGORY_TEXT:
|
|
from _TextReportDialog import TextReportDialog
|
|
dialog_class = TextReportDialog
|
|
elif category == CATEGORY_DRAW:
|
|
from _DrawReportDialog import DrawReportDialog
|
|
dialog_class = DrawReportDialog
|
|
elif category == CATEGORY_GRAPHVIZ:
|
|
from _GraphvizReportDialog import GraphvizReportDialog
|
|
dialog_class = GraphvizReportDialog
|
|
elif category == CATEGORY_WEB:
|
|
from _WebReportDialog import WebReportDialog
|
|
dialog_class = WebReportDialog
|
|
elif category in (CATEGORY_BOOK,CATEGORY_CODE,CATEGORY_VIEW):
|
|
try:
|
|
report_class(dbstate,uistate)
|
|
except Errors.WindowActiveError:
|
|
pass
|
|
return
|
|
else:
|
|
dialog_class = ReportDialog
|
|
|
|
dialog = dialog_class(dbstate, uistate, options_class, name, trans_name)
|
|
|
|
while True:
|
|
response = dialog.window.run()
|
|
if response == gtk.RESPONSE_OK:
|
|
dialog.close()
|
|
try:
|
|
MyReport = report_class(dialog.db, dialog.options)
|
|
MyReport.doc.init()
|
|
MyReport.begin_report()
|
|
MyReport.write_report()
|
|
MyReport.end_report()
|
|
except Errors.FilterError, msg:
|
|
(m1,m2) = msg.messages()
|
|
ErrorDialog(m1,m2)
|
|
except IOError, msg:
|
|
ErrorDialog(_("Report could not be created"),str(msg))
|
|
except Errors.ReportError, msg:
|
|
(m1,m2) = msg.messages()
|
|
ErrorDialog(m1,m2)
|
|
except Errors.DatabaseError,msg:
|
|
ErrorDialog(_("Report could not be created"),str(msg))
|
|
# The following except statement will catch all "NoneType" exceptions.
|
|
# This is useful for released code where the exception is most likely
|
|
# a corrupt database. But it is less useful for developing new reports
|
|
# where the execption is most likely a report bug.
|
|
# except AttributeError,msg:
|
|
# if str(msg).startswith("'NoneType' object has no attribute"):
|
|
# # "'NoneType' object has no attribute ..." usually means
|
|
# # database corruption
|
|
# RunDatabaseRepair(str(msg))
|
|
# else:
|
|
# raise
|
|
raise
|
|
except:
|
|
log.error("Failed to run report.", exc_info=True)
|
|
break
|
|
elif (response == gtk.RESPONSE_DELETE_EVENT or
|
|
response == gtk.RESPONSE_CANCEL):
|
|
dialog.close()
|
|
break
|