SolarFM/plugins/youtube_download/__main__.py

# Python imports
import os, threading, subprocess, time

# Gtk imports
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk

# Application imports


# NOTE: Threads WILL NOT die with parent's destruction.
def threaded(fn):
    def wrapper(*args, **kwargs):
        threading.Thread(target=fn, args=args, kwargs=kwargs, daemon=False).start()
    return wrapper

# NOTE: Threads WILL die with parent's destruction.
def daemon_threaded(fn):
    def wrapper(*args, **kwargs):
        threading.Thread(target=fn, args=args, kwargs=kwargs, daemon=True).start()
    return wrapper




class Plugin:
    def __init__(self, fm_builder, fm_event_system):
        self.SCRIPT_PTH         = os.path.dirname(os.path.realpath(__file__))
        self._plugin_name       = "Youtube Download"
        self._plugin_author     = "ITDominator"
        self._plugin_version    = "0.0.1"

        self._fm_builder        = fm_builder
        self._fm_event_system   = fm_event_system
        self._event_sleep_time  = .5
        self._fm_event_message  = None

        self._module_event_observer()

        button = Gtk.Button(label=self._plugin_name)
        button.connect("button-release-event", self._do_download)

        plugin_list = self._fm_builder.get_object("plugin_socket")
        plugin_list.add(button)
        plugin_list.show_all()


    def get_plugin_name(self):
        return self._plugin_name

    def get_plugin_author(self):
        return self._plugin_author

    def get_plugin_version(self):
        return self._plugin_version


    @daemon_threaded
    def _module_event_observer(self):
        while True:
            time.sleep(self._event_sleep_time)
            event = self._fm_event_system.read_module_event()
            if event:
                try:
                    if event[0] is self._plugin_name:
                        target_id, method_target, data = self._fm_event_system.consume_module_event()

                        if not method_target:
                            self._fm_event_message = data
                        else:
                            method = getattr(self.__class__, f"{method_target}")
                            data   = method(*(self, *parameters))
                except Exception as e:
                    print(repr(e))


    @threaded
    def _do_download(self, widget=None, eve=None):
        self._fm_event_system.push_gui_event([self._plugin_name, "get_current_state", ()])
        while not self._fm_event_message:
            pass

        state = self._fm_event_message
        subprocess.Popen([f'{self.SCRIPT_PTH}/download.sh' , state.tab.get_current_directory()])
        self._fm_event_message = None