customize_shortcut.py

import tkinter as tk, util, strings, custom_ui
from tkinter import ttk, messagebox
from PIL import Image

strings.load_language(open(util.user_preferences + "\\language", "r").read())

def show(shortcut_type: str, path: str):
    global shortcut_icon, shortcut_icon_index

    shortcut_icon = ""
    shortcut_icon_index = 0

    window = custom_ui.Toplevel()
    window.title(strings.lang.customize_shortcut)
    window.resizable(False, False)
    window.iconbitmap(util.internal + "icon.ico")
    window.configure(padx = 16)

    use_folder_icon = tk.BooleanVar(value = True)

    shortcut_info = ttk.Frame(window)
    shortcut_info.pack(fill = "x", expand = True, pady = (16, 0))

    icon = tk.Canvas(shortcut_info, width = 32, height = 32, bd = 0, highlightthickness = 0, background = custom_ui.bg)
    icon.pack(side = "left")

    def update_icon(icon_path: str, icon_index: int = 0):
        global shortcut_icon, shortcut_icon_index, img2

        try:
            if icon_path.endswith(".ico") or icon_index == None: img = Image.open(icon_path)
            else: img = util.extract_icon(icon_path, icon_index)
        except: img = util.extract_icon(icon_path, icon_index)

        img.thumbnail((32, 32), Image.Resampling.LANCZOS)
        img.save(util.working_folder + "\\icon.png")

        img2 = tk.PhotoImage(file = util.working_folder + "\\icon.png")

        icon.delete("all")
        icon.create_image(0, 0, image = img2, anchor = "nw")

        shortcut_icon = icon_path
        shortcut_icon_index = icon_index

    name_rame = tk.Frame(shortcut_info, highlightbackground = custom_ui.entry_bd, highlightcolor = custom_ui.entry_focus,
                          highlightthickness = 1)
    name_rame.pack(anchor = "w", padx = (16, 0), side = "left")

    name = tk.Entry(name_rame, width = 50, font = ("Default", 10), background = custom_ui.entry_bg, 
                    foreground = custom_ui.fg, border = 0, highlightthickness = 2, 
                    highlightcolor = custom_ui.entry_bg, highlightbackground = custom_ui.entry_bg, 
                    insertbackground = custom_ui.fg, insertwidth = 1, selectbackground = custom_ui.entry_select,
                    selectforeground = "#FFFFFF")
    name.pack()
    
    path_list = path.split("/")
    name.insert(0, path_list[len(path_list) - 1])

    def show_change_icon_btn():
        if use_folder_icon.get(): 
            change_icon_btn.forget()

            _icon = util.get_folder_icon(path)
            update_icon(_icon[0], _icon[1])
        else: 
            change_icon_btn.pack(side = "left")
            update_icon("C:/Windows/System32/shell32.dll", 4)

    if shortcut_type == "folder":
        ttk.Checkbutton(window, text = strings.lang.use_folder_icon, variable = use_folder_icon, command = show_change_icon_btn).pack(pady = (8, 0), anchor = "w")

    buttons = ttk.Frame(window)
    buttons.pack(pady = 16, fill = "x", expand = True)

    def change_icon():
        shortcut_icon, shortcut_icon_index = util.pick_icon()
        update_icon(shortcut_icon, shortcut_icon_index)

    change_icon_btn = ttk.Button(buttons, text = strings.lang.change_icon, command = change_icon)
    change_icon_btn.pack(side = "left")

    def create_shortcut():
        shortcut_name = util.sanitize_filename(name.get())

        if name.get() != shortcut_name:
            messagebox.showwarning(parent = window, title = strings.lang.shortcut_name_invalid, message = strings.lang.shortcut_name_invalid_description)

        window.destroy()
        
        if shortcut_type == "folder": util.create_folder_shortcut(path, shortcut_name, shortcut_icon, shortcut_icon_index)
        else: util.create_file_shortcut(path, shortcut_name, shortcut_icon, shortcut_icon_index)

    ok_btn = ttk.Button(buttons, text = strings.lang.ok, default = "active", command = create_shortcut).pack(side = "right", padx = (8, 0))
    cancel_btn = ttk.Button(buttons, text = strings.lang.cancel, command = window.destroy).pack(side = "right")

    if shortcut_type == "folder": show_change_icon_btn()
    else: update_icon("C:/Windows/System32/shell32.dll", 0)

    name.focus_set()
    window.bind("<Escape>", lambda event: window.destroy())