PyGTK HTML WYSIWYG Editor

readme.md

This is an updated version of the demo located at the links below. Is was updated to use Python 3, WebKit2 and Gtk3.

https://arstechnica.com/information-technology/2009/07/how-to-build-a-desktop-wysiwyg-editor-with-webkit-and-html-5/

https://arstechnica.com/information-technology/2009/07/how-to-build-a-desktop-wysiwyg-editor-with-webkit-and-html-5/2/

HtmlEditor.py

import os
import gi
gi.require_version("Gtk", "3.0")
gi.require_version('WebKit2', '4.0')
from gi.repository import Gtk, Gdk, WebKit2

class HtmlEditor(Gtk.Window):
    def __init__(self):
        super().__init__()
    
        self.set_title("Html Editor")
        self.connect("destroy", Gtk.main_quit)
        self.resize(500, 500)
        self.filename = None
    
        self.editor = WebKit2.WebView()
        self.editor.set_editable(True)
        self.editor.load_html("This is a test", "file:///")
    
        scroll = Gtk.ScrolledWindow()
        scroll.add(self.editor)
        scroll.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
    
        self.ui = self.generate_ui()
        self.add_accel_group(self.ui.get_accel_group())
        self.toolbar1 = self.ui.get_widget("/toolbar_main")
        self.toolbar2 = self.ui.get_widget("/toolbar_format")
        self.menubar = self.ui.get_widget("/menubar_main")
    
        self.layout = Gtk.VBox()
        self.layout.pack_start(self.menubar, False, False, 0)
        self.layout.pack_start(self.toolbar1, False, False, 0)
        self.layout.pack_start(self.toolbar2, False, False, 0)
        self.layout.pack_start(scroll, True, True, 0)
        self.add(self.layout)

    def generate_ui(self):
        ui_def = """
        <ui>
        <menubar name="menubar_main">
            <menu action="menuFile">
            <menuitem action="new" />
            <menuitem action="open" />
            <menuitem action="save" />
            </menu>
            <menu action="menuEdit">
            <menuitem action="cut" />
            <menuitem action="copy" />
            <menuitem action="paste" />
            </menu>
            <menu action="menuInsert">
            <menuitem action="insertimage" />
            </menu>
            <menu action="menuFormat">
            <menuitem action="bold" />
            <menuitem action="italic" />
            <menuitem action="underline" />
            <menuitem action="strikethrough" />
            <separator />
            <menuitem action="font" />
            <menuitem action="color" />
            <separator />
            <menuitem action="justifyleft" />
            <menuitem action="justifyright" />
            <menuitem action="justifycenter" />
            <menuitem action="justifyfull" />
            </menu>
        </menubar>
        <toolbar name="toolbar_main">
            <toolitem action="new" />
            <toolitem action="open" />
            <toolitem action="save" />
            <separator />
            <toolitem action="undo" />
            <toolitem action="redo" />
            <separator />
            <toolitem action="cut" />
            <toolitem action="copy" />
            <toolitem action="paste" />
        </toolbar>
        <toolbar name="toolbar_format">
            <toolitem action="bold" />
            <toolitem action="italic" />
            <toolitem action="underline" />
            <toolitem action="strikethrough" />
            <separator />
            <toolitem action="font" />
            <toolitem action="color" />
            <separator />
            <toolitem action="justifyleft" />
            <toolitem action="justifyright" />
            <toolitem action="justifycenter" />
            <toolitem action="justifyfull" />
            <separator />
            <toolitem action="insertimage" />
            <toolitem action="insertlink" />
        </toolbar>
        </ui>
        """
    
        actions = Gtk.ActionGroup("Actions")
        actions.add_actions([
        ("menuFile", None, "_File"),
        ("menuEdit", None, "_Edit"),
        ("menuInsert", None, "_Insert"),
        ("menuFormat", None, "_Format"),
    
        ("new", Gtk.STOCK_NEW, "_New", None, None, self.on_new),
        ("open", Gtk.STOCK_OPEN, "_Open", None, None, self.on_open),
        ("save", Gtk.STOCK_SAVE, "_Save", None, None, self.on_save),
    
        ("undo", Gtk.STOCK_UNDO, "_Undo", None, None, self.on_action),
        ("redo", Gtk.STOCK_REDO, "_Redo", None, None, self.on_action),
    
        ("cut", Gtk.STOCK_CUT, "_Cut", None, None, self.on_action),
        ("copy", Gtk.STOCK_COPY, "_Copy", None, None, self.on_action),
        ("paste", Gtk.STOCK_PASTE, "_Paste", None, None, self.on_paste),
    
        ("bold", Gtk.STOCK_BOLD, "_Bold", "<ctrl>B", None, self.on_action),
        ("italic", Gtk.STOCK_ITALIC, "_Italic", "<ctrl>I", None, self.on_action),
        ("underline", Gtk.STOCK_UNDERLINE, "_Underline", "<ctrl>U", None, self.on_action),
        ("strikethrough", Gtk.STOCK_STRIKETHROUGH, "_Strike", "<ctrl>T", None, self.on_action),
        ("font", Gtk.STOCK_SELECT_FONT, "Select _Font", "<ctrl>F", None, self.on_select_font),
        ("color", Gtk.STOCK_SELECT_COLOR, "Select _Color", None, None, self.on_select_color),
    
        ("justifyleft", Gtk.STOCK_JUSTIFY_LEFT, "Justify _Left", None, None, self.on_action),
        ("justifyright", Gtk.STOCK_JUSTIFY_RIGHT, "Justify _Right", None, None, self.on_action),
        ("justifycenter", Gtk.STOCK_JUSTIFY_CENTER, "Justify _Center", None, None, self.on_action),
        ("justifyfull", Gtk.STOCK_JUSTIFY_FILL, "Justify _Full", None, None, self.on_action),
    
        ("insertimage", "insert-image", "Insert _Image", None, None, self.on_insert_image),
        ("insertlink", "insert-link", "Insert _Link", None, None, self.on_insert_link),
        ])
    
        actions.get_action("insertimage").set_property("icon-name", "insert-image")
        actions.get_action("insertlink").set_property("icon-name", "insert-link")
    
        ui = Gtk.UIManager()
        ui.insert_action_group(actions)
        ui.add_ui_from_string(ui_def)
        return ui

    def on_action(self, action):
        self.editor.run_javascript("document.execCommand('%s', false, false);" % action.get_name())
    
    def on_paste(self, action):
        self.editor.execute_editing_command(WebKit2.EDITING_COMMAND_PASTE)
    
    def on_new(self, action):
        self.editor.load_html("", "file:///")

    def on_select_font(self, action):
        dialog = Gtk.FontChooserDialog("Select a font")
        if dialog.run() == Gtk.ResponseType.OK:
            fname = dialog.get_font_desc().get_family()
            fsize = dialog.get_font_desc().get_size()
            self.editor.run_javascript("document.execCommand('fontname', null, '%s');" % fname)
            self.editor.run_javascript("document.execCommand('fontsize', null, '%s');" % fsize)
        dialog.destroy()
    
    def on_select_color(self, action):
        dialog = Gtk.ColorChooserDialog("Select Color")
        if dialog.run() == Gtk.ResponseType.OK:
            (r, g, b, a) = dialog.get_rgba()
            color = "#%0.2x%0.2x%0.2x%0.2x" % (
                int(r * 255),
                int(g * 255),
                int(b * 255),
                int(a * 255))
            self.editor.run_javascript("document.execCommand('forecolor', null, '%s');" % color)
        dialog.destroy()
    
    def on_insert_link(self, action):
        dialog = Gtk.Dialog("Enter a URL:", self, 0,
        (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OK, Gtk.ResponseType.OK))
    
        entry = Gtk.Entry()
        dialog.vbox.pack_start(entry, False, False, 0)
        dialog.show_all()
    
        if dialog.run() == Gtk.ResponseType.OK:
            self.editor.run_javascript(
                "document.execCommand('createLink', true, '%s');" % entry.get_text())
        dialog.destroy()

    def on_insert_image(self, action):
        dialog = Gtk.FileChooserDialog("Select an image file", self, Gtk.FileChooserAction.OPEN,
        (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
    
        if dialog.run() == Gtk.ResponseType.OK:
            fn = dialog.get_filename()
            if os.path.exists(fn):
                self.editor.run_javascript(
                "document.execCommand('insertImage', null, '%s');" % fn)
        dialog.destroy()

    def on_open(self, action):
        dialog = Gtk.FileChooserDialog("Select an HTML file", self, Gtk.FileChooserAction.OPEN,
        (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
    
        if dialog.run() == Gtk.ResponseType.OK:
            fn = dialog.get_filename()
            if os.path.exists(fn):
                self.filename = fn
                with open(fn) as fd:
                    self.editor.load_html(fd.read(), "file:///")
        dialog.destroy()

    def on_save(self, action):
        def completion(html, user_data):
            open_mode = user_data
            with open(self.filename, open_mode) as fd:
                fd.write(html)
        
        if self.filename:
            self.get_html(completion, 'w')
        else:
            dialog = Gtk.FileChooserDialog("Select an HTML file", self, Gtk.FileChooserAction.SAVE,
                (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_SAVE, Gtk.ResponseType.OK))
        
            if dialog.run() == Gtk.ResponseType.OK:
                self.filename = dialog.get_filename()
                self.get_html(completion, "w+")
            dialog.destroy()

    def get_html(self, completion_function, user_data):
        def javascript_completion(obj, result, user_data):
            html = self.editor.get_title()
            completion_function(html, user_data)
        self.editor.run_javascript("document.title=document.documentElement.innerHTML;",
                                   None,
                                   javascript_completion,
                                   user_data)

e = HtmlEditor()
e.show_all()
Gtk.main()