source: epiphany-session-manager/session-manager.py @ 51

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Session save/restore facility for Epiphany
#
# Copyright © 2007 Simon E. Ward <simon@bleah.co.uk>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of version 2 of the GNU General Public
# License as published by the Free Software Foundation.
#
# 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., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA

import epiphany
import gtk
import os.path

_ui_str = """
<ui>
        <menubar name="menubar">
                <menu name="ToolsMenu" action="Tools">
                        <separator/>
                        <menuitem name="SaveSession" action="SaveSession"/>
                        <menuitem name="LoadSession" action="LoadSession"/>
                        <separator/>
                </menu>
        </menubar>
</ui>
"""

def _warn_on_overwrite(parent, filename):
        dirname, basename = os.path.split(filename)
        msg = 'A file named "%s" already exists in the directory "%s".'\
              'Do you want to overwrite it?' % (basename, dirname)
        dialog = gtk.MessageDialog(
                        parent=parent,
                        flags=gtk.DIALOG_MODAL
                             |gtk.DIALOG_DESTROY_WITH_PARENT,
                        type=gtk.MESSAGE_WARNING,
                        buttons=gtk.BUTTONS_YES_NO,
                        message_format=msg)
        overwrite = dialog.run()
        dialog.destroy()
        return overwrite

def _show_dialog(title, action):
        if action == gtk.FILE_CHOOSER_ACTION_SAVE:
                ok_button = gtk.STOCK_SAVE
        else:
                ok_button = gtk.STOCK_OPEN

        dialog = gtk.FileChooserDialog(
                        title=title,
                        action=action,
                        buttons=(
                                gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
                                ok_button, gtk.RESPONSE_OK
                                )
                        )
        dialog.set_default_response(gtk.RESPONSE_OK)

        response = dialog.run()
        if response == gtk.RESPONSE_OK:
                filename = dialog.get_filename()

                if action == gtk.FILE_CHOOSER_ACTION_SAVE:
                        while os.path.exists(filename):
                                if _warn_on_overwrite(dialog, filename) \
                                        == gtk.RESPONSE_YES:
                                        break
                                response = dialog.run()
                                if response != gtk.RESPONSE_OK:
                                        filename = None
                                        break
                                filename = dialog.get_filename()
        else:
                filename = None

        dialog.destroy()
        return filename

def _get_session():
        shell = epiphany.ephy_embed_shell_get_default()
        return shell.get_session()

def _save_session(action, session):
        filename = _show_dialog('Save Session',
                                gtk.FILE_CHOOSER_ACTION_SAVE)
        if filename:
                session.save(filename)

def _load_session(action, session):
        filename = _show_dialog('Load Session',
                                gtk.FILE_CHOOSER_ACTION_OPEN)
        if filename:
                session.load(filename, 0)

_actions = [
        ('SaveSession', None, 'Save Session…', None, None, _save_session),
        ('LoadSession', None, 'Load Session…', None, None, _load_session)
        ]

def attach_window(window):
        session = _get_session()
        ui_manager = window.get_ui_manager()
        group = gtk.ActionGroup('SessionManager')
        group.add_actions(_actions, user_data=session)
        ui_manager.insert_action_group(group, -1)
        ui_id = ui_manager.add_ui_from_string(_ui_str)

        window._session_manager_data = (group, ui_id)

def detach_window(window):
        group, ui_id = window._session_manager_data
        del window._session_manager_data

        ui_manager = window.get_ui_manager()
        ui_manager.remove_ui(ui_id)
        ui_manager.remove_action_group(group)
        ui_manager.ensure_update()