main.py

from utils.build_html import build_html
from utils.check_image import compare_images
from utils.utils import pathfinder, hex_to_rgb
from utils.menu import show_menu_base, show_menu_inputfolder, show_menu_outputfolder, show_menu_color

from pathlib import Path

def main():
    # Init vars
    invalidChoice = False
    menuChoice = ""
    inputFolder = {"aFolder": Path(__file__).parent / "images" / "a",
                   "bFolder": Path(__file__).parent / "images" / "b",
                   "emptyString": False}
    outputFolder = {"outputFolder": Path(__file__).parent / "output", "emptyString": False}
    hexColor = "#ff0000"

    while True:
        show_menu_base(invalidChoice, menuChoice)

        menuChoice = input("Make a choice: ")

        if (menuChoice == "1"): menuChoice1(inputFolder)

        elif (menuChoice == "2"): menuChoice2(outputFolder)

        elif (menuChoice == "3"): hexColor = menuChoice3(hexColor)

        elif (menuChoice.lower() == "s") or menuChoice == "4":
            print("\n\nStart...\n\n")
            diff, broken, notExists, diffSize, aFolder, bFolder, outputFolder = (
                compare_images(inputFolder["aFolder"], inputFolder["bFolder"], outputFolder, hexColor))

            build_html(diff, broken, notExists, diffSize, aFolder, bFolder, outputFolder["outputFolder"])

        elif (menuChoice.lower() == "q") or menuChoice == "5": break

        else: invalidChoice = True



def menuChoice1(inputFolder):
    # Init vars
    inputChoice = ""
    invalidChoice = False
    pathInput = ""
    folderCount = 0
    # Def path A and B. emptyString = True -> show def path message

    while True:
        show_menu_inputfolder(invalidChoice, inputChoice, pathInput, inputFolder, folderCount) # Show cool "ui"

        inputChoice = input("Make a choice: ")
        invalidChoice = False # Reset invalid input flag for new user selection
        folderCount = 0 # Reset counter for folder validation in current input

        if inputChoice == "1": # HARD. Change input folder

            pathInput = input("\nThere should be only 2 folders in the specified directory"
                                  "\nWrite path to input folder: ")

            # Def Path: Path("images") / "a". Without folder cleaning. Returns True when ""
            inputFolder["aFolder"], inputFolder["emptyString"] = (
                pathfinder(pathInput, Path("images") / "a", False, 1))

            # Def Path: Path("images") / "b"
            inputFolder["bFolder"], inputFolder["emptyString"] = (
                pathfinder(pathInput, Path("images") / "b", False, 1))

            if inputFolder["emptyString"] == True: continue # Def Path
            elif inputFolder["aFolder"] == pathInput: continue # is not a path to folder

            for iFolder in inputFolder["aFolder"].iterdir(): # Check amount of folders in directory (iFolder = input folder)
                if folderCount == 3: break # Returns 2 iFolders + warning

                if iFolder.is_dir():
                    folderCount += 1

                    if folderCount == 1: inputFolder["aFolder"] = iFolder

                    elif folderCount == 2: inputFolder["bFolder"] = iFolder

                    else: break # Go to "if folderCount == 3", when its actually 3
                else: continue

            if folderCount == 1: # Not enough folders -> set def folders
                inputFolder["aFolder"] = Path(__file__).parent / "images" / "a"
                inputFolder["bFolder"] = Path(__file__).parent / "images" / "b"


        elif inputChoice == "2":

            pathInput = input("Write path to input A folder: ")
            # Def Path: Path("images") / "a". Without folder cleaning. Returns False when ""
            inputFolder["aFolder"], inputFolder["emptyString"] =(
                pathfinder(pathInput, Path("images") / "a",
                           False, 2))

        elif inputChoice == "3":

            # Def Path: Path("images") / "b". Without folder cleaning. Returns False when ""
            pathInput = input("Write path to input B folder: ")
            inputFolder["bFolder"], inputFolder["emptyString"] =(
                pathfinder(pathInput, Path("images") / "b",
                           False, 2))

        elif inputChoice == "4":continue # Show current paths

        elif inputChoice == "test":
            # Def Path: Path("testing") / "test" / "a". Without folder cleaning
            inputFolder["aFolder"] = (
                pathfinder(pathInput, Path("testing") / "test" / "a", False, 3))
            # Def Path: Path("testing") / "test" / "b"
            inputFolder["bFolder"] = (
                pathfinder(pathInput, Path("testing") / "test" / "b", False, 3))

        elif inputChoice == "task":
            # Def Path: Path("testing") / "task" / "a". Without folder cleaning
            inputFolder["aFolder"] = (
                pathfinder(pathInput, Path("testing") / "task" / "a", False, 3))
            # Def Path: Path("testing") / "task" / "b"
            inputFolder["bFolder"] = (
                pathfinder(pathInput, Path("testing") / "task" / "b", False, 3))


        elif inputChoice == "q" or inputChoice == "5":break # Go back

        else: invalidChoice = True # Invalid Choice message



def menuChoice2(outputFolder):
    # Init vars
    outputChoice = ""
    invalidChoice = False

    while True:
        show_menu_outputfolder(invalidChoice, outputChoice, outputFolder)

        outputChoice = input("Make a choice: ")
        invalidChoice = False

        if outputChoice == "1":
            pathInput = input("Write path to output folder: ")
            outputFolder["outputFolder"], outputFolder["emptyString"] = (
                pathfinder(pathInput, Path("images") / "b",
                           False, 1))

        elif outputChoice == "2": continue

        elif outputChoice == "3" or outputChoice == "q": break

        else: invalidChoice = True

def menuChoice3(hexColor):
    # Init vars
    colorChoice = ""
    invalidChoice = False

    while True:
        show_menu_color(invalidChoice, colorChoice, hexColor)

        colorChoice = input("Make a choice: ")
        invalidChoice = False

        if colorChoice == "1":
            hexColor = hex_to_rgb(input("Write color: "))

        elif colorChoice == "2": continue

        elif colorChoice == "3" or colorChoice == "q":
            return hexColor

        else: invalidChoice = True

if (__name__) == "__main__":
    main()