308 lines
9.0 KiB
Python
308 lines
9.0 KiB
Python
#!/usr/bin/env python3
|
|
import os
|
|
import tkinter as tk
|
|
from tkinter import scrolledtext
|
|
import tkinter.font as tkfont
|
|
import shutil
|
|
import configparser
|
|
|
|
|
|
# ============================================================
|
|
# CONFIG:
|
|
config = configparser.ConfigParser()
|
|
config.read(r"C:\ProgramData\Program-Search\config.ini")
|
|
SEARCH_ROOT = config["Paths"]["SEARCH_ROOT"]
|
|
TRASH_DIR = config["Paths"]["TRASH_DIR"]
|
|
EDITOR_ENABLED = config.getboolean("Features", "EDITOR_ENABLED")
|
|
AUTO_CLOSE_SECONDS = config.getint("Features", "AUTO_CLOSE_SECONDS")
|
|
CLOSE_ALL_ENABLED = config.getboolean("Features", "CLOSE_ALL_ENABLED")
|
|
# ============================================================
|
|
|
|
os.makedirs(TRASH_DIR, exist_ok=True)
|
|
|
|
|
|
def custom_messagebox(title, message, kind="info"):
|
|
win = tk.Toplevel()
|
|
win.title(title)
|
|
win.geometry("600x300")
|
|
win.grab_set()
|
|
win.focus_force()
|
|
center_window(win)
|
|
|
|
frame = tk.Frame(win, padx=20, pady=20)
|
|
frame.pack(expand=True, fill="both")
|
|
|
|
tk.Label(frame, text=message, font=("Consolas", 24), wraplength=550).pack(pady=20)
|
|
|
|
if kind == "yesno":
|
|
result = {"value": False}
|
|
|
|
def yes():
|
|
result["value"] = True
|
|
win.destroy()
|
|
|
|
def no():
|
|
result["value"] = False
|
|
win.destroy()
|
|
|
|
tk.Button(frame, text="Yes", font=("Consolas", 24), padx=10, pady=10, command=yes).pack(side="left", padx=16)
|
|
tk.Button(frame, text="No", font=("Consolas", 24), padx=10, pady=10, command=no).pack(side="right", padx=16)
|
|
|
|
win.wait_window()
|
|
return result["value"]
|
|
|
|
else:
|
|
def close():
|
|
win.destroy()
|
|
|
|
tk.Button(frame, text="OK", font=("Consolas", 24), padx=10, pady=10, command=close).pack(pady=20)
|
|
win.bind("<Return>", lambda event: close())
|
|
win.wait_window()
|
|
|
|
|
|
def center_window(win):
|
|
win.update_idletasks()
|
|
w = win.winfo_width()
|
|
h = win.winfo_height()
|
|
sw = win.winfo_screenwidth()
|
|
sh = win.winfo_screenheight()
|
|
x = (sw // 2) - (w // 2)
|
|
y = (sh // 2) - (h // 2)
|
|
win.geometry(f"{w}x{h}+{x}+{y}")
|
|
|
|
|
|
def popup_search_input():
|
|
win = tk.Toplevel()
|
|
win.title("Search Trash")
|
|
win.geometry("600x300")
|
|
center_window(win)
|
|
|
|
tk.Label(win, text="Enter part number or name:", font=("Consolas", 18)).pack(pady=20)
|
|
|
|
entry = tk.Entry(win, font=("Consolas", 18))
|
|
entry.pack(pady=10)
|
|
entry.focus_set()
|
|
|
|
result = {"value": None}
|
|
|
|
def submit():
|
|
result["value"] = entry.get().strip().lower()
|
|
win.destroy()
|
|
|
|
entry.bind("<Return>", lambda event: submit())
|
|
win.bind("<Return>", lambda event: submit())
|
|
|
|
tk.Button(win, text="Search", font=("Consolas", 18), padx=20, pady=10, command=submit).pack(pady=40)
|
|
|
|
win.wait_window()
|
|
return result["value"]
|
|
|
|
|
|
def search_files(term):
|
|
matches = []
|
|
|
|
for root, dirs, files in os.walk(TRASH_DIR):
|
|
for fname in files:
|
|
path = os.path.join(root, fname)
|
|
fname_lower = fname.lower()
|
|
|
|
filename_match = term in fname_lower
|
|
contents_match = False
|
|
|
|
try:
|
|
with open(path, "r", errors="ignore") as f:
|
|
if term in f.read().lower():
|
|
contents_match = True
|
|
except:
|
|
pass
|
|
|
|
if filename_match or contents_match:
|
|
matches.append(path)
|
|
|
|
return matches
|
|
|
|
|
|
def popup_results_list(matches):
|
|
win = tk.Toplevel()
|
|
win.title("Matches Found")
|
|
win.geometry("1200x1000")
|
|
center_window(win)
|
|
|
|
tk.Label(win, text="Select files to preview:", font=("Consolas", 18)).pack(pady=20)
|
|
|
|
lb = tk.Listbox(win, selectmode=tk.MULTIPLE, font=("Consolas", 18))
|
|
lb.pack(expand=True, fill="both")
|
|
|
|
for m in matches:
|
|
lb.insert(tk.END, m)
|
|
|
|
result = {"selected": []}
|
|
|
|
def select_all():
|
|
lb.select_set(0, tk.END)
|
|
|
|
def submit():
|
|
sel = lb.curselection()
|
|
result["selected"] = [matches[i] for i in sel]
|
|
win.destroy()
|
|
|
|
lb.bind("<Return>", lambda event: submit())
|
|
win.bind("<Return>", lambda event: submit())
|
|
|
|
tk.Button(win, text="Open Selected", font=("Consolas", 18), padx=20, pady=10, command=submit).pack(side="right", padx=20, pady=20)
|
|
tk.Button(win, text="Select All", font=("Consolas", 18), padx=20, pady=10, command=select_all).pack(side="left", padx=20, pady=20)
|
|
|
|
win.wait_window()
|
|
return result["selected"]
|
|
|
|
|
|
def popup_file_preview(path, on_close_callback):
|
|
win = tk.Toplevel()
|
|
win.title(path)
|
|
win.geometry("900x1000")
|
|
center_window(win)
|
|
|
|
box = scrolledtext.ScrolledText(win, wrap=tk.WORD, font=("Consolas", 18))
|
|
box.pack(expand=True, fill="both")
|
|
|
|
try:
|
|
with open(path, "r", errors="ignore") as f:
|
|
original_text = f.read()
|
|
box.insert(tk.END, original_text)
|
|
box.edit_modified(False)
|
|
except Exception as e:
|
|
box.insert(tk.END, f"Error reading file:\n{e}")
|
|
|
|
modified = {"changed": False}
|
|
|
|
def mark_modified(event=None):
|
|
if box.edit_modified():
|
|
modified["changed"] = True
|
|
box.edit_modified(False)
|
|
|
|
if EDITOR_ENABLED:
|
|
box.bind("<<Modified>>", mark_modified)
|
|
else:
|
|
box.configure(state="disabled")
|
|
|
|
frame = tk.Frame(win)
|
|
frame.pack(fill="x", pady=10)
|
|
|
|
def save_file():
|
|
try:
|
|
with open(path, "w", errors="ignore") as f:
|
|
f.write(box.get("1.0", tk.END))
|
|
modified["changed"] = False
|
|
custom_messagebox("Saved", f"File saved:\n{path}")
|
|
except Exception as e:
|
|
custom_messagebox("Save Error", str(e))
|
|
|
|
if EDITOR_ENABLED:
|
|
tk.Button(frame, text="Save", fg="green", font=("Consolas", 18), padx=15, pady=8, command=save_file).pack(side="left", padx=10, pady=10)
|
|
|
|
def restore_file():
|
|
if EDITOR_ENABLED and modified["changed"]:
|
|
ask = custom_messagebox("Unsaved Changes", "You have unsaved changes.\nSave before restoring?", kind="yesno")
|
|
if ask:
|
|
save_file()
|
|
|
|
try:
|
|
rel_path = os.path.relpath(path, TRASH_DIR)
|
|
dest_path = os.path.join(SEARCH_ROOT, rel_path)
|
|
|
|
os.makedirs(os.path.dirname(dest_path), exist_ok=True)
|
|
shutil.move(path, dest_path)
|
|
|
|
custom_messagebox("Restored", f"File restored to:\n{dest_path}")
|
|
win.destroy()
|
|
on_close_callback()
|
|
|
|
except Exception as e:
|
|
custom_messagebox("Restore Error", str(e))
|
|
|
|
tk.Button(frame, text="Restore File", fg="blue", font=("Consolas", 18), padx=15, pady=8, command=restore_file).pack(side="left", padx=10, pady=10)
|
|
|
|
def close():
|
|
if EDITOR_ENABLED and modified["changed"]:
|
|
ask = custom_messagebox("Unsaved Changes", "You have unsaved changes.\nSave before closing?", kind="yesno")
|
|
if ask:
|
|
save_file()
|
|
|
|
win.destroy()
|
|
on_close_callback()
|
|
|
|
tk.Button(frame, text="Close", font=("Consolas", 18), padx=15, pady=8, command=close).pack(side="right", padx=10, pady=10)
|
|
|
|
def close_all_button():
|
|
ask = custom_messagebox("WARNING!", "Are you sure you want to close all windows?", kind="yesno")
|
|
if ask:
|
|
close_all(win.master)
|
|
|
|
if CLOSE_ALL_ENABLED:
|
|
tk.Button(frame, text="Close All", font=("Consolas", 18), padx=15, pady=8, command=close_all_button).pack(side="right", padx=10, pady=10)
|
|
|
|
def auto_close():
|
|
if win.winfo_exists():
|
|
win.destroy()
|
|
on_close_callback()
|
|
|
|
win.protocol("WM_DELETE_WINDOW", close)
|
|
win.after(AUTO_CLOSE_SECONDS * 1000, auto_close)
|
|
|
|
|
|
def close_all(root):
|
|
for w in root.winfo_children():
|
|
try:
|
|
w.destroy()
|
|
except:
|
|
pass
|
|
root.destroy()
|
|
|
|
|
|
def main():
|
|
root = tk.Tk()
|
|
root.withdraw()
|
|
|
|
default_font = tkfont.nametofont("TkDefaultFont")
|
|
default_font.configure(family="Consolas", size=18)
|
|
|
|
text_font = tkfont.nametofont("TkTextFont")
|
|
text_font.configure(family="Consolas", size=18)
|
|
|
|
fixed_font = tkfont.nametofont("TkFixedFont")
|
|
fixed_font.configure(family="Consolas", size=18)
|
|
|
|
while True:
|
|
search_term = popup_search_input()
|
|
if not search_term:
|
|
root.destroy()
|
|
return
|
|
|
|
matches = search_files(search_term)
|
|
|
|
if matches:
|
|
break
|
|
|
|
custom_messagebox("No Results", "No matching files found.")
|
|
|
|
selected = popup_results_list(matches)
|
|
if not selected:
|
|
root.destroy()
|
|
return
|
|
|
|
open_windows = {"count": len(selected)}
|
|
|
|
def window_closed():
|
|
open_windows["count"] -= 1
|
|
if open_windows["count"] == 0:
|
|
root.destroy()
|
|
|
|
for path in selected:
|
|
popup_file_preview(path, window_closed)
|
|
|
|
root.mainloop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|