openmedialibrary/oml/gtkstatus.py

#!/usr/bin/env python3
import os
from os.path import dirname, abspath
import subprocess
import webbrowser
import _thread

import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk, GLib
try:
    gi.require_version('AppIndicator3', '0.1')
    from gi.repository import AppIndicator3 as appindicator
except:
    appindicator = None


base = dirname(dirname(dirname(abspath(__file__))))
icon = os.path.join(base, 'openmedialibrary/static/png/oml.png')
title = "Open Media Library"
ctl = base + '/ctl'


class OMLIcon:
    menu = None
    icon = None
    indicator = None

    def __init__(self):
        if appindicator:
            self.indicator = appindicator.Indicator.new("OML", icon,
                appindicator.IndicatorCategory.APPLICATION_STATUS)
            self.indicator.set_status(appindicator.IndicatorStatus.ACTIVE)
            self.menu = self.get_menu()
            self.indicator.set_menu(self.menu)
        else:
            self.icon = Gtk.StatusIcon()
            self.icon.set_from_file(icon)
            self.icon.set_title(title)
            self.icon.connect("activate", self._click)
            self.icon.connect("popup-menu", self._click)
        self._p = subprocess.Popen([ctl, 'start'])
        GLib.timeout_add_seconds(1, self._open, None)
        _thread.start_new_thread(self._wait, ())

    def _wait(self):
        self._p.wait()
        self._quit(None)

    def get_menu(self):
        menu = Gtk.Menu()
        about = Gtk.MenuItem(label="Open")
        about.connect("activate", self._open)
        menu.append(about)
        quit = Gtk.MenuItem(label="Quit")
        quit.connect("activate", self._quit)
        menu.append(quit)
        menu.show_all()
        return menu

    def _click(self, icon, button=None, time=None):
        if self.menu:
            self.menu.destroy()
            self.menu = None
        else:
            self.menu = self.get_menu()
            button = 1
            time = Gtk.get_current_event_time()
            self.menu.popup(None, None, self.icon.position_menu, icon, button, time)

    def _quit(self, q):
        Gtk.main_quit()
        p = subprocess.Popen([ctl, 'stop'])
        self.menu = None

    def _open(self, *args):
        url = 'file://' + base + '/openmedialibrary/static/html/load.html'
        webbrowser.open_new_tab(url)
        self.menu = None

OMLIcon()
Gtk.main()