115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
# Python imports
|
|
|
|
# Lib imports
|
|
import gi
|
|
gi.require_version('Gtk', '3.0')
|
|
from gi.repository import Gtk
|
|
from gi.repository import Gio
|
|
from gi.repository import GLib
|
|
|
|
# Application imports
|
|
from mixins.tree_nixin import TreeMixin
|
|
|
|
|
|
|
|
class ImagesList(TreeMixin, Gtk.ScrolledWindow):
|
|
def __init__(self):
|
|
super(ImagesList, self).__init__()
|
|
|
|
self._tree_view = None
|
|
self._store = None
|
|
self._dir_watcher = None
|
|
self._watch_dir = None
|
|
|
|
self._set_file_watcher()
|
|
self._setup_styling()
|
|
self._setup_signals()
|
|
self._subscribe_to_events()
|
|
self._load_widgets()
|
|
|
|
self.show_all()
|
|
|
|
|
|
def _setup_styling(self):
|
|
self.set_size_request(200, -1)
|
|
self.set_hexpand(False)
|
|
self.set_vexpand(True)
|
|
self.set_margin_top(15)
|
|
|
|
def _setup_signals(self):
|
|
...
|
|
|
|
def _subscribe_to_events(self):
|
|
...
|
|
|
|
def _load_widgets(self):
|
|
scroll, self._store = self._create_treeview_widget("Images")
|
|
|
|
self._tree_view = scroll.get_children()[0]
|
|
self._tree_view.connect("button-press-event", self._handle_clicks)
|
|
self._tree_view.connect("row-activated", self._set_active_image)
|
|
|
|
self._tree_view.set_property("activate-on-single-click", True)
|
|
|
|
self.referesh_directory_list()
|
|
self.add(scroll)
|
|
|
|
def _set_file_watcher(self):
|
|
if settings.is_debug():
|
|
logger.debug(f"Watcher Will Not Be Set...")
|
|
return
|
|
|
|
images_dir = settings.get_screenshots_dir()
|
|
self._watch_dir = Gio.File.new_for_uri(f"file://{images_dir}")
|
|
self._dir_watcher = self._watch_dir.monitor_directory(Gio.FileMonitorFlags.WATCH_MOVES, Gio.Cancellable())
|
|
|
|
self._dir_watcher.connect("changed", self._dir_watch_updates, ())
|
|
|
|
def _dir_watch_updates(self, file_monitor, file, other_file = None, eve_type = None, data = None):
|
|
if eve_type in [Gio.FileMonitorEvent.CREATED, Gio.FileMonitorEvent.DELETED,
|
|
Gio.FileMonitorEvent.RENAMED, Gio.FileMonitorEvent.MOVED_IN,
|
|
Gio.FileMonitorEvent.MOVED_OUT]:
|
|
|
|
self._store.clear()
|
|
self.referesh_directory_list()
|
|
|
|
def _set_active_image(self, tree_view = None, path = None, column = None):
|
|
file = self.get_selected()
|
|
event_system.emit("set_image_to_view", (file,))
|
|
event_system.emit("set_revert_data", (file,))
|
|
|
|
def _handle_clicks(self, widget = None, eve = None):
|
|
if eve.button == 1 and eve.type == 5:
|
|
file = self.get_selected()
|
|
event_system.emit("set_revert_data", (file,))
|
|
event_system.emit("open_file")
|
|
return
|
|
|
|
if eve.button == 1 and eve.type == 4:
|
|
self._set_active_image()
|
|
return
|
|
|
|
if eve.button == 3:
|
|
file = self.get_selected()
|
|
|
|
event_system.emit("set_revert_data", (file,))
|
|
event_system.emit("show_menu", (file,))
|
|
return
|
|
|
|
@threaded
|
|
def referesh_directory_list(self):
|
|
images = settings.get_directory_list()
|
|
images.sort()
|
|
for image in images:
|
|
GLib.idle_add(self.add_to_store, (image,))
|
|
|
|
def add_to_store(self, image):
|
|
self._store.append(image)
|
|
|
|
def get_selected(self):
|
|
model, treeiter = self._tree_view.get_selection().get_selected()
|
|
if treeiter != None:
|
|
return model[treeiter][0]
|
|
|
|
return None
|