Global threading decorators; Endpoint registry creation

plugins/vod_thumbnailer/plugin.py

@@ -0,0 +1,173 @@
+# Python imports
+import os, threading, subprocess, time, inspect, hashlib
+from datetime import datetime
+
+# Gtk imports
+import gi
+gi.require_version('Gtk', '3.0')
+gi.require_version('GdkPixbuf', '2.0')
+from gi.repository import Gtk, GLib, Gio, GdkPixbuf
+
+# 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 Manifest:
+    path: str     = os.path.dirname(os.path.realpath(__file__))
+    name: str     = "VOD Thumbnailer"
+    author: str   = "ITDominator"
+    version: str  = "0.0.1"
+    support: str  = ""
+    requests: {}  = {
+        'ui_target': "context_menu",
+        'pass_fm_events': "true"
+    }
+
+class Plugin(Manifest):
+    def __init__(self):
+        self._GLADE_FILE            = f"{self.path}/re_thumbnailer.glade"
+        self._builder               = None
+        self._thumbnailer_dialog    = None
+        self._thumbnail_preview_img = None
+        self._file_name             = None
+        self._file_location         = None
+        self._file_hash             = None
+        self._state                 = None
+
+        self._event_system          = None
+        self._event_sleep_time      = .5
+        self._event_message         = None
+
+
+
+    def get_ui_element(self):
+        self._builder           = Gtk.Builder()
+        self._builder.add_from_file(self._GLADE_FILE)
+
+        classes  = [self]
+        handlers = {}
+        for c in classes:
+            methods = None
+            try:
+                methods = inspect.getmembers(c, predicate=inspect.ismethod)
+                handlers.update(methods)
+            except Exception as e:
+                print(repr(e))
+
+        self._builder.connect_signals(handlers)
+
+        self._thumbnailer_dialog    = self._builder.get_object("thumbnailer_dialog")
+        self._file_name             = self._builder.get_object("file_name")
+        self._file_location         = self._builder.get_object("file_location")
+        self._thumbnail_preview_img = self._builder.get_object("thumbnail_preview_img")
+        self._file_hash             = self._builder.get_object("file_hash")
+
+        button = Gtk.Button(label=self.name)
+        button.connect("button-release-event", self._show_thumbnailer_page)
+        return button
+
+    def set_fm_event_system(self, fm_event_system):
+        self._event_system = fm_event_system
+
+    def run(self):
+        self._module_event_observer()
+
+
+
+
+    @threaded
+    def _show_thumbnailer_page(self, widget=None, eve=None):
+        self._event_system.push_gui_event([self.name, "get_current_state", ()])
+        self.wait_for_fm_message()
+
+        state               = self._event_message
+        self._event_message = None
+
+        GLib.idle_add(self._process_changes, (state))
+
+    def _process_changes(self, state):
+        self._state = None
+
+        if len(state.selected_files) == 1:
+            if state.selected_files[0].lower().endswith(state.tab.fvideos):
+                self._state = state
+                self._set_ui_data()
+                response   = self._thumbnailer_dialog.run()
+                if response in [Gtk.ResponseType.CANCEL, Gtk.ResponseType.DELETE_EVENT]:
+                    self._thumbnailer_dialog.hide()
+
+
+    def _regenerate_thumbnail(self, widget=None, eve=None):
+        print("Regenerating thumbnail...")
+        file      = self._file_name.get_text()
+        dir       = self._file_location.get_text()
+        file_hash = self._file_hash.get_text()
+
+        hash_img_pth = f"{self._state.tab.ABS_THUMBS_PTH}/{file_hash}.jpg"
+        try:
+            if os.path.isfile(hash_img_pth):
+                os.remove(hash_img_pth)
+
+            img_pixbuf = self._state.tab.create_icon(dir, file)
+            self._thumbnail_preview_img.set_from_pixbuf(img_pixbuf)
+        except Exception as e:
+            print("Couldn't regenerate thumbnail!")
+
+    def _use_selected_thumbnail(self, widget=None, eve=None):
+        print("_use_selected_thumbnail stub...")
+
+
+    def _set_ui_data(self):
+        uri          = self._state.selected_files[0]
+        path         = self._state.tab.get_current_directory()
+        parts        = uri.split("/")
+
+        file_hash    = hashlib.sha256(str.encode(uri)).hexdigest()
+        hash_img_pth = f"{self._state.tab.ABS_THUMBS_PTH}/{file_hash}.jpg"
+        img_pixbuf   = GdkPixbuf.Pixbuf.new_from_file(hash_img_pth)
+
+        self._thumbnail_preview_img.set_from_pixbuf(img_pixbuf)
+        self._file_name.set_text(parts[ len(parts) - 1 ])
+        self._file_location.set_text(path)
+        self._file_hash.set_text(file_hash)
+
+
+
+    def wait_for_fm_message(self):
+        while not self._event_message:
+            pass
+
+    @daemon_threaded
+    def _module_event_observer(self):
+        while True:
+            time.sleep(self._event_sleep_time)
+            event = self._event_system.read_module_event()
+            if event:
+                try:
+                    if event[0] == self.name:
+                        target_id, method_target, data = self._event_system.consume_module_event()
+
+                        if not method_target:
+                            self._event_message = data
+                        else:
+                            method = getattr(self.__class__, f"{method_target}")
+                            if data:
+                                data = method(*(self, *data))
+                            else:
+                                method(*(self,))
+                except Exception as e:
+                    print(repr(e))