87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
import os
|
|
import pathlib
|
|
import sys
|
|
import tkinter as tk
|
|
from tkinter import filedialog, messagebox
|
|
|
|
from library.common_gui_functions import tk_context
|
|
from library.common_utilities import CommonUtilities
|
|
|
|
|
|
class TkGui:
|
|
def __init__(self):
|
|
self.file_types = None
|
|
|
|
def open_file_dialog(self, initial_dir=None, initial_file=None, file_types="all"):
|
|
with tk_context():
|
|
self.file_types = file_types
|
|
if self.file_types in CommonUtilities.file_filters:
|
|
filters = CommonUtilities.file_filters[self.file_types]
|
|
else:
|
|
filters = CommonUtilities.file_filters["all"]
|
|
|
|
if self.file_types == "directory":
|
|
result = filedialog.askdirectory(initialdir=initial_dir)
|
|
else:
|
|
result = filedialog.askopenfilename(initialdir=initial_dir, initialfile=initial_file, filetypes=filters)
|
|
|
|
# Return a tuple (file_path, canceled)
|
|
# file_path: the selected file path or an empty string if no file is selected
|
|
# canceled: True if the user pressed the cancel button, False otherwise
|
|
return result, result == ""
|
|
|
|
def save_file_dialog(self, initial_dir, initial_file, file_types="all"):
|
|
self.file_types = file_types
|
|
|
|
# Use the tk_context function with the 'with' statement
|
|
with tk_context():
|
|
if self.file_types in CommonUtilities.file_filters:
|
|
filters = CommonUtilities.file_filters[self.file_types]
|
|
else:
|
|
filters = CommonUtilities.file_filters["all"]
|
|
|
|
save_file_path = filedialog.asksaveasfilename(initialdir=initial_dir, initialfile=initial_file,
|
|
filetypes=filters, defaultextension=".safetensors")
|
|
|
|
return save_file_path
|
|
|
|
def show_message_box(_message, _title="Message", _level="info"):
|
|
with tk_context():
|
|
message_type = {
|
|
"warning": messagebox.showwarning,
|
|
"error": messagebox.showerror,
|
|
"info": messagebox.showinfo,
|
|
"question": messagebox.askquestion,
|
|
"okcancel": messagebox.askokcancel,
|
|
"retrycancel": messagebox.askretrycancel,
|
|
"yesno": messagebox.askyesno,
|
|
"yesnocancel": messagebox.askyesnocancel
|
|
}
|
|
|
|
if _level in message_type:
|
|
message_type[_level](_title, _message)
|
|
else:
|
|
messagebox.showinfo(_title, _message)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
mode = sys.argv[1]
|
|
|
|
if mode == 'file_dialog':
|
|
starting_dir = sys.argv[2] if len(sys.argv) > 2 else None
|
|
starting_file = sys.argv[3] if len(sys.argv) > 3 else None
|
|
file_class = sys.argv[4] if len(sys.argv) > 4 else None # Update this to sys.argv[4]
|
|
gui = TkGui()
|
|
file_path = gui.open_file_dialog(starting_dir, starting_file, file_class)
|
|
print(file_path) # Make sure to print the result
|
|
|
|
elif mode == 'msgbox':
|
|
message = sys.argv[2]
|
|
title = sys.argv[3] if len(sys.argv) > 3 else ""
|
|
gui = TkGui()
|
|
gui.show_message_box(message, title)
|
|
except Exception as e:
|
|
print(f"Error: {e}", file=sys.stderr)
|
|
sys.exit(1)
|