xmledit.py

#!/usr/bin/env python2

import signal
signal.signal(signal.SIGINT, signal.SIG_DFL)

import os, sys
os.environ['GTK_THEME'] = 'light'

from gi.repository import GObject, Gst, Gtk, Gdk, GtkSource

GObject.threads_init()
Gdk.threads_init()

import command, validate, dialog

class SourceView(Gtk.Window):
    def __init__(self, filename=None):
        super(SourceView, self).__init__()

        self.filename = 'unsaved'

        self.connect("destroy", self.quit)
        self.set_default_size(1024, 768)
        self.set_border_width(10)
        self.set_title('Homeostasis editor')

        tabs = Gtk.Notebook()
        self.tabs = tabs

        vbox = Gtk.VBox() # contains tabs and buttons

        self.source_buffer = GtkSource.Buffer()
        lang_manager = GtkSource.LanguageManager()
        self.source_buffer.set_language(lang_manager.get_language('xml'))
        self.source_buffer.set_modified(False)
        self.source_view = GtkSource.View.new_with_buffer(self.source_buffer)
        self.source_view.set_show_line_numbers(True)
        self.source_view.set_tab_width(4)
        self.source_view.set_wrap_mode(Gtk.WrapMode.WORD)
        self.scrolled = Gtk.ScrolledWindow()
        self.scrolled.add(self.source_view)
        #vbox.pack_start(self.scrolled, True, True, 5)

        self.command = command.Command()
        #vbox.pack_start(self.command, False, True, 5)

        hbox = Gtk.HBox()
        new_button = Gtk.Button('New')
        new_button.connect('clicked', self.new)
        hbox.pack_start(new_button, False, False, 5)

        load_button = Gtk.Button('Load')
        load_button.connect('clicked', lambda *args: self.show_filechooser("load"))
        hbox.pack_start(load_button, False, False, 5)

        validate_button = Gtk.Button('Validate')
        validate_button.connect('clicked', self.validate)
        hbox.pack_start(validate_button, False, False, 5)

        compile_button = Gtk.Button('Compile')
        compile_button.connect('clicked', self.compile)
        hbox.pack_start(compile_button, False, False, 5)

        save_button = Gtk.Button('Save')
        save_button.connect('clicked', self.save)
        hbox.pack_start(save_button, False, False, 5)

        saveas_button = Gtk.Button('Save as')
        saveas_button.connect('clicked', lambda *args: self.show_filechooser("saveas"))
        hbox.pack_start(saveas_button, False, False, 5)

        vbox.pack_start(tabs, True, True, 5)
        vbox.pack_start(hbox, False, True, 5)

        tabs.append_page(self.scrolled, Gtk.Label(self.filename))
        tabs.append_page(self.command, Gtk.Label('Compilation result'))
        tabs.set_show_border(False)

        if filename != None:
            self.set_filename(filename)
            self.load()

        self.add(vbox)
        self.show_all()

        self.command.add_text('READY')

    def new(self, *args):
        if not self.source_buffer.get_modified() or dialog.confirm('Discard changes to "%s"?' % self.filename):
            self.source_buffer.set_text("""<homeostasis version="25-11-2014">
    <liste_section sequences="1" ordre="variable" repetition="oui" action="exclusive">
        <section id="2" action="example1">
            <sequence ordre="strict" repetition="non" action="" lang="esp">
                <keyword action="example2" lang="eng"> hello </keyword>
            </sequence>
        </section>
    </liste_section>
</homeostasis>""")
            self.set_filename('unsaved')

    def set_filename(self, filename):
        self.filename = filename
        self.tabs.set_tab_label(self.scrolled, Gtk.Label(os.path.basename(filename)))

    def show_filechooser(self, dialog_type="load"):
        if dialog_type == "load":
            chooser = Gtk.FileChooserDialog("Please choose a file to load", self, Gtk.FileChooserAction.OPEN, (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN, Gtk.ResponseType.OK))
        else:
            chooser = Gtk.FileChooserDialog("Please set a file to save", self, Gtk.FileChooserAction.SAVE, (Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_SAVE, Gtk.ResponseType.OK))
        filter_text = Gtk.FileFilter()
        filter_text.set_name("XML files")
        filter_text.add_mime_type("text/xml")
        chooser.add_filter(filter_text)
        chooser.set_current_folder('%s/data' % (os.path.dirname(__file__) or '.'))

        response = chooser.run()
        if response == -5:
            self.filename = chooser.get_filename()
            print self.filename
            self.set_filename(self.filename)
            chooser.destroy()
            if dialog_type == "load":
                return self.load()
            else:
                self.source_buffer.set_modified(True) # force to save
                return self.save()
        chooser.destroy()

    def load(self, *args):
        if not self.source_buffer.get_modified() or dialog.confirm('Discard changes to "%s"?' % self.filename):
            try:
                with open(self.filename) as fp:
                    self.source_buffer.set_text(fp.read())
                self.source_buffer.set_modified(False)
            except Exception as e:
                print e
                self.command.add_text('FAILED to load "%s"' % self.filename)
                return False
            return True
        return False

    def save(self, *args):
        if not self.source_buffer.get_modified(): # no need to save
            return True
        if os.path.exists(self.filename) and not dialog.confirm('Overwrite "%s"?' % self.filename):
            return False
        if self.filename == 'unsaved':
            return self.show_filechooser("save")
        else:
            self.command.clear()
            try:
                with open(self.filename, 'w') as fp:
                    bounds = self.source_buffer.get_bounds()
                    fp.write(self.source_buffer.get_text(bounds[0], bounds[1], True))
                self.source_buffer.set_modified(False)
            except Exception as e:
                print e
                self.command.add_text('FAILED to save as "%s"' % self.filename)
                return False
        return True

    def validate(self, *args):
        #self.command.clear()
        if self.save():
            self.command.add_text('VALIDATE...')
            self.tabs.set_current_page(1)
            result, message = validate.validate_xml(self.filename)
            self.command.add_text(message)
            if result:
                self.command.add_text('SUCCESS')
            else:
                self.command.add_text('FAILED')
            return result

    def compile(self, *args):
        #self.command.clear()
        self.tabs.set_current_page(1)
        if self.save():
            if self.validate():
                self.command.add_text('COMPILE...')
                self.command.run('./asr/tools/compile.sh "%s"' % self.filename)

    def quit(self, window):
        #if not self.source_buffer.get_modified() or dialog.confirm('Discard changes to "%s"?' % self.filename):
        Gtk.main_quit()
        #    return True
        #return False

if __name__ == '__main__':
    if len(sys.argv) > 1:
        app = SourceView(sys.argv[1])
    else:
        app = SourceView()
    Gtk.main()