214 lines
6.1 KiB
Python
Executable File
214 lines
6.1 KiB
Python
Executable File
#!/usr/sbin/python
|
|
|
|
# GTK Imports
|
|
import gi, faulthandler, signal
|
|
gi.require_version('Gtk', '3.0')
|
|
|
|
from gi.repository import Gtk
|
|
from gi.repository import GLib
|
|
|
|
|
|
# Python Imports
|
|
import os, threading, time
|
|
|
|
from setproctitle import setproctitle
|
|
|
|
|
|
|
|
app_name = "PATH Edit Tool"
|
|
|
|
|
|
|
|
def threaded(fn):
|
|
def wrapper(*args, **kwargs):
|
|
threading.Thread(target=fn, args=args, kwargs=kwargs).start()
|
|
return wrapper
|
|
|
|
|
|
class Main(Gtk.Window):
|
|
def __init__(self):
|
|
super(Main, self).__init__()
|
|
self._USER_HOME = os.path.expanduser('~')
|
|
PREFERED_BASH_PATH = f"{self._USER_HOME}/.bash_paths"
|
|
|
|
icon_size = 16
|
|
box = Gtk.Box()
|
|
box2 = Gtk.Box()
|
|
separator = Gtk.Separator()
|
|
self._insert_entry = Gtk.Entry()
|
|
self.message_widget = Gtk.Popover.new(separator)
|
|
self.message_label = Gtk.Label()
|
|
add_button = Gtk.Button.new_from_icon_name("gtk-add", icon_size)
|
|
delete_button = Gtk.Button.new_from_icon_name("gtk-delete", icon_size)
|
|
save_button = Gtk.Button.new_from_icon_name("gtk-save", icon_size)
|
|
|
|
scroll_vw, grid, self.store = self._create_treeview_widget(title="PATHs")
|
|
tree_selection = grid.get_selection()
|
|
|
|
box2.add(self._insert_entry)
|
|
box2.add(add_button)
|
|
box2.add(delete_button)
|
|
|
|
box.add(separator)
|
|
box.add(scroll_vw)
|
|
box.add(box2)
|
|
box.add(save_button)
|
|
|
|
self.message_widget.add(self.message_label)
|
|
self.add(box)
|
|
self.add(self.message_widget)
|
|
|
|
self.message_widget.set_default_widget(self.message_label)
|
|
self._insert_entry.set_hexpand(True)
|
|
self._insert_entry.set_placeholder_text("Path...")
|
|
save_button.set_label("Save")
|
|
scroll_vw.set_vexpand(True)
|
|
box.set_orientation(1)
|
|
box.set_vexpand(True)
|
|
self.set_default_size(480, 560)
|
|
self.set_title(f"{app_name}")
|
|
self.set_icon_name("applications-accessories")
|
|
|
|
add_button.connect("clicked", self.add_entry)
|
|
delete_button.connect("clicked", self.delete_entry)
|
|
save_button.connect("clicked", self.save_enteries)
|
|
tree_selection.connect("changed", self._set_selected)
|
|
|
|
self.connect("delete-event", Gtk.main_quit)
|
|
GLib.unix_signal_add(GLib.PRIORITY_DEFAULT, signal.SIGINT, Gtk.main_quit)
|
|
|
|
self.success = "#88cc27"
|
|
self.warning = "#ffa800"
|
|
self.error = "#ff0000"
|
|
self.selected = None
|
|
|
|
if os.path.isfile(PREFERED_BASH_PATH):
|
|
self.bashrc_path = PREFERED_BASH_PATH
|
|
else:
|
|
self.bashrc_path = f"{self._USER_HOME}/.bashrc"
|
|
|
|
self._load_paths_data()
|
|
|
|
self.message_widget.show_all()
|
|
self.message_widget.popdown()
|
|
box.show_all()
|
|
box2.show_all()
|
|
self.show_all()
|
|
|
|
|
|
|
|
|
|
def add_entry(self, widget):
|
|
path = self._insert_entry.get_text().strip()
|
|
|
|
if os.path.isdir(path):
|
|
self.store.append([path])
|
|
else:
|
|
self.display_message(self.warning, "Not a directory!")
|
|
|
|
def delete_entry(self, widget):
|
|
self.store.remove(self.selected)
|
|
|
|
def save_enteries(self, widget):
|
|
try:
|
|
paths = list()
|
|
iter = self.store.get_iter_first()
|
|
while iter != None:
|
|
pth = self.store.get_value(iter, 0)
|
|
pth = pth.replace(self._USER_HOME, "$HOME")
|
|
paths.append(pth)
|
|
iter = self.store.iter_next(iter)
|
|
|
|
|
|
toExport = "export PATH=\"" + ':'.join(paths) + "\"\n\n"
|
|
file = open(self.bashrc_path, mode='r')
|
|
for line in file:
|
|
if "export PATH=" in line:
|
|
continue
|
|
else:
|
|
toExport += line
|
|
|
|
file.close()
|
|
file = open(self.bashrc_path, mode='w')
|
|
file.write(toExport)
|
|
file.close()
|
|
self.display_message(self.success, "Successfully saved file!")
|
|
except Exception as e:
|
|
self.display_message(self.error, "Opening/Writing to file failed!")
|
|
print("Opening/Writing to file failed with the following:\n\n")
|
|
print(e)
|
|
|
|
|
|
def _set_selected(self, user_data):
|
|
selected = user_data.get_selected()[1]
|
|
if selected:
|
|
self.selected = selected
|
|
|
|
def _load_paths_data(self):
|
|
PATH_str = os.getenv("PATH")
|
|
|
|
# If path exists in bashrc replace default selection...
|
|
file = open(self.bashrc_path, mode='r')
|
|
for line in file:
|
|
if "export PATH=" in line:
|
|
part = line.replace("export PATH=", "")
|
|
cleaned = part.replace("\"", "")
|
|
PATH_str = cleaned.strip()
|
|
|
|
# Split string into list/tuple and add parts to store
|
|
paths = PATH_str.split(":")
|
|
for path in paths:
|
|
self.store.append([path])
|
|
|
|
|
|
|
|
|
|
def display_message(self, type, text):
|
|
markup = f"<span foreground='{type}'>{text}</span>"
|
|
self.message_label.set_markup(markup)
|
|
self.message_widget.popup()
|
|
self.hide_message_timed()
|
|
|
|
@threaded
|
|
def hide_message_timed(self):
|
|
time.sleep(3)
|
|
GLib.idle_add(self.message_widget.popdown)
|
|
|
|
|
|
|
|
|
|
def _create_treeview_widget(self, title = "Not Set"):
|
|
scroll = Gtk.ScrolledWindow()
|
|
grid = Gtk.TreeView()
|
|
store = Gtk.ListStore(str)
|
|
column = Gtk.TreeViewColumn(title)
|
|
name = Gtk.CellRendererText()
|
|
selec = grid.get_selection()
|
|
|
|
grid.set_model(store)
|
|
selec.set_mode(2)
|
|
|
|
column.pack_start(name, True)
|
|
column.add_attribute(name, "text", 0)
|
|
column.set_expand(False)
|
|
|
|
grid.append_column(column)
|
|
grid.set_search_column(0)
|
|
grid.set_headers_visible(True)
|
|
grid.set_enable_tree_lines(False)
|
|
|
|
grid.show_all()
|
|
scroll.add(grid)
|
|
grid.columns_autosize()
|
|
return scroll, grid, store
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
faulthandler.enable() # For better debug info
|
|
setproctitle(f"{app_name}")
|
|
|
|
main = Main()
|
|
Gtk.main()
|