icon_apply_advanced.py

"""
Contains logic of building GUI for the application.
"""
import os

from tkinter import Tk, ttk, messagebox
from tkinter.filedialog import askdirectory

from icon_apply import seticon
from icon_online_search import find_and_convert

class CreateGUI():
    """
    Creates the GUI using tkinter module.
    """
    dir_name = None
    def __init__(self, root_obj):
        """
        Initializes the Tk() root object for the class and calls the 'construct_dialog' method.
        """
        self.root_obj = root_obj
        self.construct_dialog()
        
    def bulk_icon_apply(self, directory):
        """
        Walks recursively through the directory and its subdirectories and sets icons
        to each respective folder fetched by the 'get_ico_file' method.
        """
        cnt = 0
        root, name = os.path.split(directory)
        main_directory_icon = self.get_ico_file(root, name)
        if main_directory_icon:
            seticon(directory, main_directory_icon, 0)
            cnt += 1
        
        for root, dirs, _ in os.walk(directory):
            for name in dirs:
                subdir_icon = self.get_ico_file(root, name)
                if subdir_icon:
                    seticon(os.path.join(root, name), subdir_icon, 0)
                    cnt += 1

        messagebox.showinfo("Success", "Icons applied successfully to {} folders.".format(cnt))
        
    def get_directory(self):
        """
        Validates the entered directory name and calls the 'bulk_icon_apply' method.
        """
        self.dir_name = askdirectory(title="Choose directory")
        if self.dir_name:
            if os.path.isdir(self.dir_name):
                try:
                    self.bulk_icon_apply(self.dir_name)
                except IOError as err:
                    messagebox.showerror("Error", err)
            else:
                messagebox.showerror("Error", "Invalid directory.")
    
    @staticmethod           
    def get_ico_file(root, name):
        """
        Fetches the icon to be applied to the folder.
        """
        ico_file = find_and_convert(root, name)
        return ico_file if os.path.isfile(ico_file) else None
        
    def construct_dialog(self):
        """
        Constructs the GUI dialog form.
        """
        self.root_obj.title("Icon apply - advanced")
        self.root_obj.geometry('{}x{}'.format(250, 60))
        label = ttk.Label(self.root_obj, text="Search and apply icons", font=("Arial", 10))
        label.pack()
        button = ttk.Button(self.root_obj, text='Select directory', command=self.get_directory)
        button.pack()

def main():
    """
    Entry-point of the function.
    """
    root = Tk()
    CreateGUI(root)
    root.mainloop()
    
if __name__ == "__main__":
    main()