89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
# ruff: noqa: N801
|
|
import gi
|
|
|
|
gi.require_version("Gtk", "4.0")
|
|
|
|
import logging
|
|
|
|
from clan_cli.api.directory import FileRequest
|
|
from gi.repository import Gio, GLib, Gtk
|
|
|
|
from clan_app.api import ImplFunc
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
# This implements the abstract function open_file with one argument, file_request,
|
|
# which is a FileRequest object and returns a string or None.
|
|
class open_file(ImplFunc[[FileRequest], str | None]):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
|
|
def async_run(self, file_request: FileRequest) -> bool:
|
|
def on_file_select(file_dialog: Gtk.FileDialog, task: Gio.Task) -> None:
|
|
try:
|
|
gfile = file_dialog.open_finish(task)
|
|
if gfile:
|
|
selected_path = gfile.get_path()
|
|
self.returns(selected_path)
|
|
except Exception as e:
|
|
print(f"Error getting selected file or directory: {e}")
|
|
|
|
def on_folder_select(file_dialog: Gtk.FileDialog, task: Gio.Task) -> None:
|
|
try:
|
|
gfile = file_dialog.select_folder_finish(task)
|
|
if gfile:
|
|
selected_path = gfile.get_path()
|
|
self.returns(selected_path)
|
|
except Exception as e:
|
|
print(f"Error getting selected directory: {e}")
|
|
|
|
def on_save_finish(file_dialog: Gtk.FileDialog, task: Gio.Task) -> None:
|
|
try:
|
|
gfile = file_dialog.save_finish(task)
|
|
if gfile:
|
|
selected_path = gfile.get_path()
|
|
self.returns(selected_path)
|
|
except Exception as e:
|
|
print(f"Error getting selected file: {e}")
|
|
|
|
dialog = Gtk.FileDialog()
|
|
|
|
if file_request.title:
|
|
dialog.set_title(file_request.title)
|
|
|
|
if file_request.filters:
|
|
filters = Gio.ListStore.new(Gtk.FileFilter)
|
|
file_filters = Gtk.FileFilter()
|
|
|
|
if file_request.filters.title:
|
|
file_filters.set_name(file_request.filters.title)
|
|
|
|
if file_request.filters.mime_types:
|
|
for mime in file_request.filters.mime_types:
|
|
file_filters.add_mime_type(mime)
|
|
filters.append(file_filters)
|
|
|
|
if file_request.filters.patterns:
|
|
for pattern in file_request.filters.patterns:
|
|
file_filters.add_pattern(pattern)
|
|
|
|
if file_request.filters.suffixes:
|
|
for suffix in file_request.filters.suffixes:
|
|
file_filters.add_suffix(suffix)
|
|
|
|
filters.append(file_filters)
|
|
dialog.set_filters(filters)
|
|
|
|
# if select_folder
|
|
if file_request.mode == "select_folder":
|
|
dialog.select_folder(callback=on_folder_select)
|
|
elif file_request.mode == "open_file":
|
|
dialog.open(callback=on_file_select)
|
|
elif file_request.mode == "save":
|
|
dialog.save(callback=on_save_finish)
|
|
|
|
return GLib.SOURCE_REMOVE
|
|
|
|
|