#!/bin/python

# rename folder in Python

# you can run this script with: python3 renameFolderInPython.py "< path to folder >" "< folder name >" "< new folder name >"

# haven't successfully tested this script on Windows yet 

import colorama, os, sys, traceback
from colorama import Fore, Style
from datetime import datetime
colorama.init()


def checkOs(): 
    print("Started checking operating system at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))

    if sys.platform == "win32": 
        print(Fore.GREEN + "Operating System:", end=""); sys.stdout.flush()
        os.system('ver')
        print(Style.RESET_ALL, end="")
        operatingSystem = "Windows"

    elif sys.platform == "darwin": 
        print(Fore.GREEN + "Operating System:")
        os.system('sw_vers')
        print(Style.RESET_ALL)
        operatingSystem = "macOS"

    elif sys.platform == "linux": 
        print(Fore.GREEN + "Operating System:")
        os.system('uname -r')
        print(Style.RESET_ALL)
        operatingSystem = "Linux"

    print("Finished checking operating system at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))

    print("")
    return operatingSystem


def getPathToFolder(operatingSystem): 
    if operatingSystem == "Windows": 

        pathToFolder = str(input("Please type the path to the folder you wish to rename and press the \"Enter\" key (Example: C:\\Users\\%USERNAME%\\Desktop): "))

    else: 

        pathToFolder = str(input("Please type the path to the folder you wish to rename and press the \"return\" key (Example: /Users/$USER/Desktop): "))

    pathToFolder = os.path.expandvars(pathToFolder)
    print("")
    return pathToFolder


def getFolderName(operatingSystem): 
    if operatingSystem == "Windows": 

        folderName = str(input("Please type the name of the folder you wish to rename and press the \"Enter\" key (Example: Exes): "))

    else: 

        folderName = str(input("Please type the name of the folder you wish to rename and press the \"return\" key (Example: Exes): "))

    print("")
    return folderName


def getNewFolderName(operatingSystem):
    if operatingSystem == "Windows": 

        newFolderName = str(input("Please type the new folder name and press the \"Enter\" key (Example: Forgotten): "))

    else:

        newFolderName = str(input("Please type the new folder name and press the \"return\" key (Example: Forgotten): "))

    print("")
    return newFolderName


def checkParameters(pathToFolder, folderName, newFolderName): 
    print("Started checking parameter(s) at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))
    valid = True

    print("Parameter(s):")
    print("----------------------------------------")
    print("pathToFolder : {0}".format(pathToFolder))
    print("folderName   : {0}".format(folderName))
    print("newFolderName: {0}".format(newFolderName))
    print("----------------------------------------")

    if pathToFolder == None or pathToFolder == "": 
        print(Fore.RED + "pathToFolder is not set." + Style.RESET_ALL)
        valid = False

    if folderName == None or folderName == "": 
        print(Fore.RED + "folderName is not set." + Style.RESET_ALL)
        valid = False

    if newFolderName == None or newFolderName == "":
        print(Fore.RED + "newFolderName is not set." + Style.RESET_ALL)
        valid = False

    if valid == True: 
        print(Fore.GREEN + "All parameter check(s) passed." + Style.RESET_ALL)

        print("Finished checking parameter(s) at", datetime.now().strftime("%m-%d-%Y %I:%M %p"))

        print("")

    else: 
        raise Exception("One or more parameter(s) are incorrect.")


def renameFolder(): 
    print("\nRename folder in Python.\n")
    operatingSystem = checkOs()

    if len(sys.argv) >= 2: 
        pathToFolder  = str(sys.argv[1])
        folderName    = str(sys.argv[2])
        newFolderName = str(sys.argv[3])

    else: 
        pathToFolder  = getPathToFolder(operatingSystem)
        folderName    = getFolderName(operatingSystem)
        newFolderName = getNewFolderName(operatingSystem)

    checkParameters(pathToFolder, folderName, newFolderName)

    try: 
        startDateTime = datetime.now()
        
        print("Started renaming \"{0}\" folder at {1}".format(folderName, startDateTime.strftime("%m-%d-%Y %I:%M %p")))

        folderPath = "{0}/{1}".format(pathToFolder, folderName)
        newFolderNamePath = "{0}/{1}".format(pathToFolder, newFolderName)
        os.rename(folderPath, newFolderNamePath)

        print(Fore.GREEN + "Successfully renamed \"{0}\" folder to \"{1}\" folder.".format(folderName, newFolderName) + Style.RESET_ALL)

        finishDateTime = datetime.now()

        print("Finished renaming \"{0}\" folder at {1}".format(folderName, finishDateTime.strftime("%m-%d-%Y %I:%M %p")))

        duration = finishDateTime - startDateTime
        print("Total execution time: {0} second(s)".format(duration.seconds))
        print("")

    except Exception as e:
        print(Fore.RED + "Failed to rename \"{0}\" folder.".format(folderName))
        traceback.print_exc()
        exit("" + Style.RESET_ALL)


renameFolder()