568 lines
21 KiB
Python
568 lines
21 KiB
Python
#
|
|
# Gramps - a GTK+/GNOME based genealogy program
|
|
#
|
|
# Copyright (C) 2000-2006 Donald N. Allingham
|
|
# Copyright (C) 2009 Gary Burton
|
|
#
|
|
# 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$
|
|
|
|
"""
|
|
Provide the managed window interface, which allows GRAMPS to track
|
|
the create/deletion of dialog windows.
|
|
"""
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Standard python modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
from cStringIO import StringIO
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# GNOME/GTK
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import gtk
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Gramps modules
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
import const
|
|
import Errors
|
|
import Config
|
|
from glade import Glade
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Window manager
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
|
|
_win_top = '<ui><menubar name="MenuBar"><menu action="WindowsMenu">'
|
|
_win_btm = '</menu></menubar></ui>'
|
|
DISABLED = -1
|
|
|
|
#-----------------------------------------------------------------------
|
|
#
|
|
# Helper function
|
|
#
|
|
#-----------------------------------------------------------------------
|
|
|
|
def get_object(self,value):
|
|
if self.get_name() == value:
|
|
return self
|
|
elif hasattr(self,'get_children'):
|
|
for child in self.get_children():
|
|
object = get_object(child, value)
|
|
if object is not None:
|
|
return object
|
|
return None
|
|
|
|
class GrampsWindowManager:
|
|
"""
|
|
Manage hierarchy of open GRAMPS windows.
|
|
|
|
This class's purpose is to manage the hierarchy of open windows.
|
|
The idea is to maintain the tree of branches and leaves.
|
|
A leaf does not have children and corresponds to a single open window.
|
|
A branch has children and corresponds to a group of windows.
|
|
|
|
We will follow the convention of having first leaf in any given
|
|
branch represent a parent window of the group, and the rest of the
|
|
children leaves/branches represent windows spawned from the parent.
|
|
|
|
The tree structure is maintained as a list of items.
|
|
Items which are lists are branches.
|
|
Items which are not lists are leaves.
|
|
|
|
Lookup of an item is done via track sequence. The elements of
|
|
the track sequence specify the lookup order: [2,3,1] means
|
|
'take the second item of the tree, take its third child, and
|
|
then the first child of that child'.
|
|
|
|
Lookup can be also done by ID for windows that are identifiable.
|
|
"""
|
|
|
|
def __init__(self, uimanager):
|
|
# initialize empty tree and lookup dictionary
|
|
self.uimanager = uimanager
|
|
self.window_tree = []
|
|
self.id2item = {}
|
|
self.action_group = gtk.ActionGroup('WindowManger')
|
|
self.active = DISABLED
|
|
self.ui = _win_top + _win_btm
|
|
|
|
def disable(self):
|
|
"""
|
|
Remove the UI and action groups if the navigation is enabled
|
|
"""
|
|
if self.active != DISABLED:
|
|
self.uimanager.remove_ui(self.active)
|
|
self.uimanager.remove_action_group(self.action_group)
|
|
self.active = DISABLED
|
|
|
|
def enable(self):
|
|
"""
|
|
Enables the UI and action groups
|
|
"""
|
|
self.uimanager.insert_action_group(self.action_group, 1)
|
|
self.active = self.uimanager.add_ui_from_string(self.ui)
|
|
self.uimanager.ensure_update()
|
|
|
|
def get_item_from_track(self, track):
|
|
# Recursively find an item given track sequence
|
|
item = self.window_tree
|
|
for index in track:
|
|
item = item[index]
|
|
return item
|
|
|
|
def get_item_from_id(self, item_id):
|
|
# Find an item given its ID
|
|
# Return None if the ID is not found
|
|
return self.id2item.get(item_id, None)
|
|
|
|
def close_track(self, track):
|
|
# This is called when item needs to be closed
|
|
# Closes all its children and then removes the item from the tree.
|
|
try:
|
|
item = self.get_item_from_track(track)
|
|
self.recursive_action(item, self.close_item)
|
|
# This only needs to be run once for the highest level point
|
|
# to remove.
|
|
self.remove_item(track)
|
|
except IndexError:
|
|
print "Missing item from window manager", track, self.close_item
|
|
|
|
def recursive_action(self, item, func, *args):
|
|
# This function recursively calls itself over the child items
|
|
# starting with the given item.
|
|
# Eventualy, every non-list item (leaf) will be reached
|
|
# and the func(item,*args) will be called on that item.
|
|
if isinstance(item, list):
|
|
# If this item is a branch
|
|
# close the children except for the first one
|
|
for sub_item in item[1:]:
|
|
self.recursive_action(sub_item, func, *args)
|
|
# return the first child
|
|
last_item = item[0]
|
|
else:
|
|
# This item is a leaf -- no children to close
|
|
# return itself
|
|
last_item = item
|
|
func(last_item, *args)
|
|
|
|
def close_item(self, item, *args):
|
|
# Given an item, close its window and remove it's ID from the dict
|
|
if item.window_id:
|
|
del self.id2item[item.window_id]
|
|
if item.window:
|
|
item.window.destroy()
|
|
|
|
def remove_item(self, track):
|
|
# We need the whole gymnastics below because our item
|
|
# may actually be a list consisting of a single real
|
|
# item and empty lists.
|
|
|
|
# find the track corresponding to the parent item
|
|
parent_track = track[:-1]
|
|
# find index of our item in parent
|
|
child_in_parent = track[-1:][0]
|
|
# obtain parent item and remove our item from it
|
|
parent_item = self.get_item_from_track(parent_track)
|
|
parent_item.pop(child_in_parent)
|
|
# Adjust each item following the removed one
|
|
# so that it's track is down by one on this level
|
|
for ix in range(child_in_parent, len(parent_item)):
|
|
item = parent_item[ix]
|
|
self.recursive_action(item, self.move_item_down, len(track)-1)
|
|
# Rebuild menu
|
|
self.build_windows_menu()
|
|
|
|
def move_item_down(self, item, *args):
|
|
# Given an item and an index, adjust the item's track
|
|
# by subtracting 1 from that index's level
|
|
index = args[0]
|
|
item.track[index] -= 1
|
|
|
|
def add_item(self, track, item):
|
|
# if the item is identifiable then we need to remember
|
|
# its id so that in the future we recall this window
|
|
# instead of spawning a new one
|
|
# So people can make as many windows of type item.window_id = None
|
|
# Use this for add dialogs, where users may add as many values
|
|
# simultaneously as they want.
|
|
# Actually, we should do this away, as requiring at least id(obj)
|
|
# is not a big requirement ?
|
|
if item.window_id:
|
|
self.id2item[item.window_id] = item
|
|
|
|
# Make sure we have a track
|
|
parent_item = self.get_item_from_track(track)
|
|
assert isinstance(parent_item, list) or track == [], \
|
|
"Gwm: add_item: Incorrect track - Is parent not a leaf?"
|
|
|
|
# Prepare a new item, depending on whether it is branch or leaf
|
|
if item.submenu_label:
|
|
# This is an item with potential children -- branch
|
|
new_item = [item]
|
|
else:
|
|
# This is an item without children -- leaf
|
|
new_item = item
|
|
|
|
# append new item to the parent
|
|
parent_item.append(new_item)
|
|
|
|
# rebuild the Windows menu based on the new tree
|
|
self.build_windows_menu()
|
|
|
|
# prepare new track corresponding to the added item and return it
|
|
new_track = track + [len(parent_item)-1]
|
|
return new_track
|
|
|
|
def call_back_factory(self, item):
|
|
if not isinstance(item, list):
|
|
def func(obj):
|
|
if item.window_id and self.id2item.get(item.window_id):
|
|
self.id2item[item.window_id].present()
|
|
else:
|
|
def func(obj):
|
|
pass
|
|
return func
|
|
|
|
def generate_id(self, item):
|
|
return str(item.window_id)
|
|
|
|
def display_menu_list(self, data, action_data, mlist):
|
|
if isinstance(mlist, (list, tuple)):
|
|
i = mlist[0]
|
|
idval = self.generate_id(i)
|
|
data.write('<menu action="M:%s">' % idval)
|
|
action_data.append(("M:"+idval, None, i.submenu_label,
|
|
None, None, None))
|
|
else:
|
|
i = mlist
|
|
idval = self.generate_id(i)
|
|
|
|
data.write('<menuitem action="%s"/>' % idval)
|
|
action_data.append((idval, None, i.menu_label, None, None,
|
|
self.call_back_factory(i)))
|
|
|
|
if isinstance(mlist, (list, tuple)) and (len(mlist) > 1):
|
|
for i in mlist[1:]:
|
|
if isinstance(i, list):
|
|
self.display_menu_list(data, action_data, i)
|
|
else:
|
|
idval = self.generate_id(i)
|
|
data.write('<menuitem action="%s"/>'
|
|
% self.generate_id(i))
|
|
action_data.append((idval, None, i.menu_label,
|
|
None, None,
|
|
self.call_back_factory(i)))
|
|
if isinstance(mlist, (list, tuple)):
|
|
data.write('</menu>')
|
|
|
|
def build_windows_menu(self):
|
|
if self.active != DISABLED:
|
|
self.uimanager.remove_ui(self.active)
|
|
self.uimanager.remove_action_group(self.action_group)
|
|
|
|
self.action_group = gtk.ActionGroup('WindowManger')
|
|
action_data = []
|
|
|
|
data = StringIO()
|
|
data.write(_win_top)
|
|
for i in self.window_tree:
|
|
self.display_menu_list(data, action_data, i)
|
|
data.write(_win_btm)
|
|
self.ui = data.getvalue()
|
|
data.close()
|
|
self.action_group.add_actions(action_data)
|
|
self.enable()
|
|
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Gramps Managed Window class
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
class ManagedWindow:
|
|
"""
|
|
Managed window base class.
|
|
|
|
This class provides all the goodies necessary for user-friendly window
|
|
management in GRAMPS: registering the menu item under the Windows
|
|
menu, keeping track of child windows, closing them on close/delete
|
|
event, and presenting itself when selected or attempted to create again.
|
|
"""
|
|
def __init__(self, uistate, track, obj):
|
|
"""
|
|
Create child windows and add itself to menu, if not there already.
|
|
|
|
|
|
The usage from derived classes is envisioned as follows:
|
|
|
|
|
|
import ManagedWindow
|
|
class SomeWindowClass(ManagedWindow.ManagedWindow):
|
|
def __init__(self,uistate,dbstate,track):
|
|
window_id = self # Or e.g. window_id = person.handle
|
|
submenu_label = None # This window cannot have children
|
|
menu_label = 'Menu label for this window'
|
|
ManagedWindow.ManagedWindow.__init__(self,
|
|
uistate,
|
|
track,
|
|
window_id,
|
|
submenu_label,
|
|
menu_label)
|
|
# Proceed with the class.
|
|
...
|
|
|
|
@param uistate gramps uistate
|
|
@param track {list of parent windows, [] if the main GRAMPS window
|
|
is the parent}
|
|
@param obj The object that is used to id the managed window,
|
|
The inheriting object needs a method build_menu_names(self, obj)
|
|
which works on this obj and creates menu labels
|
|
for use in the Gramps Window Menu.
|
|
If self.submenu_label ='' then leaf, otherwise branch
|
|
|
|
|
|
"""
|
|
window_key = self.build_window_key(obj)
|
|
menu_label, submenu_label = self.build_menu_names(obj)
|
|
self._gladeobj = None
|
|
self.isWindow = None
|
|
self.width_key = None
|
|
self.height_key = None
|
|
self.__refs_for_deletion = []
|
|
|
|
if uistate.gwm.get_item_from_id(window_key):
|
|
uistate.gwm.get_item_from_id(window_key).present()
|
|
raise Errors.WindowActiveError('This window is already active')
|
|
else:
|
|
self.window_id = window_key
|
|
self.submenu_label = submenu_label
|
|
self.menu_label = menu_label
|
|
self.uistate = uistate
|
|
self.track = self.uistate.gwm.add_item(track, self)
|
|
# Work out parent_window
|
|
if len(self.track) > 1:
|
|
# We don't belong to the lop level
|
|
if self.track[-1] > 0:
|
|
# If we are not the first in the group,
|
|
# then first in that same group is our parent
|
|
parent_item_track = self.track[:-1]
|
|
parent_item_track.append(0)
|
|
else:
|
|
# If we're first in the group, then our parent
|
|
# is the first in the group one level up
|
|
parent_item_track = self.track[:-2]
|
|
parent_item_track.append(0)
|
|
|
|
# Based on the track, get item and then window object
|
|
self.parent_window = self.uistate.gwm.get_item_from_track(
|
|
parent_item_track).window
|
|
else:
|
|
# On the top level: we use gramps top window
|
|
self.parent_window = self.uistate.window
|
|
|
|
def set_window(self, window, title, text, msg=None, isWindow=False):
|
|
"""
|
|
Set the window that is managed.
|
|
|
|
@param window if isWindow=False window must be a gtk.Window() object, otherwise None
|
|
@param title a label widget in which to write the title, None if not needed
|
|
@param text text to use as title of window and in title param
|
|
@param msg if not None, use msg as title of window instead of text
|
|
@param isWindow {if isWindow than self is the window
|
|
(so self inherits from gtk.Window and
|
|
from ManagedWindow.ManagedWindow)
|
|
if not isWindow, than window is the Window to manage,
|
|
and after this method self.window stores it.
|
|
}
|
|
|
|
"""
|
|
self.isWindow = isWindow
|
|
self.msg = msg
|
|
self.titlelabel = title
|
|
if self.isWindow :
|
|
set_titles(self, title, text, msg)
|
|
else :
|
|
set_titles(window, title, text, msg)
|
|
#closing the gtk.Window must also close ManagedWindow
|
|
self.window = window
|
|
self.window.connect('delete-event', self.close)
|
|
|
|
def update_title(self, text):
|
|
if self.isWindow:
|
|
set_titles(self, self.titlelabel, text, self.msg)
|
|
else:
|
|
set_titles(self.window, self.titlelabel, text, self.msg)
|
|
|
|
def build_menu_names(self, obj):
|
|
return ('Undefined Menu','Undefined Submenu')
|
|
|
|
def build_window_key(self, obj):
|
|
return id(obj)
|
|
|
|
def define_glade(self, top_module, glade_file=None):
|
|
if glade_file is None:
|
|
glade_file = const.GLADE_FILE
|
|
builder = gtk.Builder()
|
|
builder.add_from_file(glade_file)
|
|
self._gladeobj = builder.get_object(top_module)
|
|
self._gladeobj.get_object = get_object
|
|
return self._gladeobj
|
|
|
|
def get_widget(self, name):
|
|
assert(self._gladeobj)
|
|
object = get_object(self._gladeobj,name)
|
|
if object is not None:
|
|
return object
|
|
raise ValueError, (
|
|
'ManagedWindow.get_widget: "%s" widget not found in "%s"' %
|
|
(name, self._gladeobj.get_name())
|
|
)
|
|
return object
|
|
|
|
def connect_button(self, button_name, function):
|
|
assert(self._gladeobj)
|
|
self.get_widget(button_name).connect('clicked', function)
|
|
|
|
def show(self):
|
|
if self.isWindow :
|
|
self.set_transient_for(self.parent_window)
|
|
self.opened = True
|
|
self.show_all()
|
|
|
|
else :
|
|
assert self.window, "ManagedWindow: self.window does not exist!"
|
|
self.window.set_transient_for(self.parent_window)
|
|
self.opened = True
|
|
self.window.show_all()
|
|
|
|
def modal_call(self, after_ok_func=None):
|
|
"""
|
|
Method to do modal run of the ManagedWindow.
|
|
Connect the OK button to a method that checks if all is ok,
|
|
Do not call close, close is called here.
|
|
(if not ok, do self.window.run() to obtain new response )
|
|
TODO: remove close here and do close in ReportDialog,
|
|
this can only be done, once all methods use modal_call()
|
|
instead of their own implementation
|
|
Connect Cancel to do close, delete event is connected to close
|
|
here in ManagedWindow.
|
|
Do not generete RESPONSE_OK/CANCEL/DELETE_EVENT on button clicks
|
|
of other buttons
|
|
after_ok_func is called on ok click in this method
|
|
"""
|
|
#self.show()
|
|
while True:
|
|
response = self.window.run()
|
|
if response == gtk.RESPONSE_OK:
|
|
# dialog will be closed by connect, now continue work while
|
|
# rest of dialog is unresponsive, release when finished
|
|
self.close()
|
|
if after_ok_func is not None:
|
|
after_ok_func()
|
|
break
|
|
elif (response == gtk.RESPONSE_DELETE_EVENT or
|
|
response == gtk.RESPONSE_CANCEL):
|
|
# connect buttons generating this to a close call
|
|
break
|
|
|
|
def close(self, *obj):
|
|
"""
|
|
Close itself.
|
|
|
|
Takes care of closing children and removing itself from menu.
|
|
"""
|
|
self._save_size()
|
|
self.clean_up()
|
|
self.uistate.gwm.close_track(self.track)
|
|
self.opened = False
|
|
self.parent_window.present()
|
|
|
|
def present(self):
|
|
"""
|
|
Present window (unroll/unminimize/bring to top).
|
|
"""
|
|
if self.isWindow :
|
|
self.present(self)
|
|
else :
|
|
assert hasattr(self, 'window'), \
|
|
"ManagedWindow: self.window does not exist!"
|
|
self.window.present()
|
|
|
|
def _set_size(self):
|
|
"""
|
|
Set the dimensions of the window
|
|
"""
|
|
if self.width_key is not None:
|
|
width = Config.get(self.width_key)
|
|
height = Config.get(self.height_key)
|
|
self.window.resize(width, height)
|
|
|
|
def _save_size(self):
|
|
"""
|
|
Save the dimensions of the window to the config file
|
|
"""
|
|
if self.width_key is not None:
|
|
(width, height) = self.window.get_size()
|
|
Config.set(self.width_key, width)
|
|
Config.set(self.height_key, height)
|
|
Config.sync()
|
|
|
|
def track_ref_for_deletion(self, ref):
|
|
"""
|
|
Record references of instance variables that need to be removed
|
|
from scope so that the class can be garbage collected
|
|
"""
|
|
if ref not in self.__refs_for_deletion:
|
|
self.__refs_for_deletion.append(ref)
|
|
|
|
def clean_up(self):
|
|
"""
|
|
Remove any instance variables from scope which point to non-glade
|
|
GTK objects so that the class can be garbage collected.
|
|
If the object is a Gramps widget then it should have a clean_up method
|
|
which can be called that removes any other GTK object it contains.
|
|
"""
|
|
while len(self.__refs_for_deletion):
|
|
attr = self.__refs_for_deletion.pop()
|
|
obj = getattr(self, attr)
|
|
if hasattr(obj, 'clean_up'):
|
|
obj.clean_up()
|
|
delattr(self, attr)
|
|
#-------------------------------------------------------------------------
|
|
#
|
|
# Helper functions
|
|
#
|
|
#-------------------------------------------------------------------------
|
|
def set_titles(window, title, text, msg=None):
|
|
if title:
|
|
title.set_text('<span weight="bold" size="larger">%s</span>' % text)
|
|
title.set_use_markup(True)
|
|
if msg:
|
|
window.set_title('%s - GRAMPS' % msg)
|
|
else:
|
|
window.set_title('%s - GRAMPS' % text)
|
|
window.set_icon_from_file(const.ICON)
|