BulkR/src/core/change_view.py

148 lines
4.4 KiB
Python

# Python imports
import os
import re
# Lib imports
import gi
gi.require_version('Gtk', '3.0')
gi.require_version('Gdk', '3.0')
from gi.repository import Gtk
from gi.repository import Gdk
# Application imports
from mixins import CommonWidgetGeneratorMixin
from mixins import CommonActionsMixin
class ChangeView(Gtk.Box, CommonWidgetGeneratorMixin, CommonActionsMixin):
def __init__(self):
super(ChangeView, self).__init__()
self._active_path = None
self._from_store = None
self._to_store = None
self._from_changes = []
self._to_changes = []
self._setup_styling()
self._setup_signals()
self._subscribe_to_events()
self._load_widgets()
self.show_all()
def _setup_styling(self):
self.set_spacing(20)
self.set_border_width(2)
self.set_homogeneous(True)
def _setup_signals(self):
...
def _subscribe_to_events(self):
event_system.subscribe("reset-from-view", self.reset_from_view)
event_system.subscribe("reset-to-view", self.reset_to_view)
event_system.subscribe("get-from", self.get_from_list)
event_system.subscribe("get-to", self.get_to_list)
event_system.subscribe("set-from", self.set_from_list)
event_system.subscribe("set-to", self.set_to_list)
event_system.subscribe("update-from", self.update_from_list)
event_system.subscribe("update-to", self.update_to_list)
event_system.subscribe("get-active-path", self._get_active_path)
event_system.subscribe("set-active-path", self._set_active_path)
def _load_widgets(self):
from_container = Gtk.Box()
to_container = Gtk.Box()
from_scroll_vw, \
self._from_store = self._create_treeview_widget(title="From:")
to_scroll_vw, \
self._to_store = self._create_treeview_widget(title="To:")
from_container.add(from_scroll_vw)
to_container.add(to_scroll_vw)
from_container.set_orientation(1)
to_container.set_orientation(1)
fchild = from_scroll_vw.get_children()[0]
fchild.connect("drag-data-received", self._on_drag_data_received)
URI_TARGET_TYPE = 80
uri_target = Gtk.TargetEntry.new('text/uri-list', Gtk.TargetFlags(0), URI_TARGET_TYPE)
targets = [ uri_target ]
action = Gdk.DragAction.COPY
fchild.enable_model_drag_dest(targets, action)
fchild.enable_model_drag_source(0, targets, action)
self.add(from_container)
self.add(to_container)
def _on_drag_data_received(self, widget, drag_context, x, y, data, info, time):
if info == 80:
uri = data.get_uris()[0].split("file://")[1]
self._set_active_path(uri)
def _get_active_path(self):
return self._active_path
def _set_active_path(self, _file):
if os.path.isdir(_file) :
self._from_changes.clear()
self._active_path = _file
for f in os.listdir(_file):
self._from_changes.append(f)
self._from_changes.sort(key=self._natural_keys)
self._to_changes = self._from_changes
event_system.emit("update-from")
event_system.emit("update-to")
def get_from_list(self):
return self._from_changes
def get_to_list(self):
return self._to_changes
def set_from_list(self, from_list):
self._from_changes = from_list
def set_to_list(self, to_list):
self._to_changes = to_list
def update_from_list(self):
print("Updating From List...")
if self._from_store:
self._from_store.clear()
for i, change in enumerate(self._from_changes):
self._from_store.insert(i, [change])
def update_to_list(self):
print("Updating To List...")
if self._to_store:
self._to_store.clear()
for i, change in enumerate(self._to_changes):
self._to_store.insert(i, [change])
def reset_to_view(self):
self._to_changes = self._from_changes
event_system.emit("update-to")
def reset_from_view(self):
self._set_active_path(self._active_path)
def _atoi(self, text):
return int(text) if text.isdigit() else text
def _natural_keys(self, text):
return [ self._atoi(c) for c in re.split('(\d+)',text) ]