music_files.py

############## IMPORTS ##############
import os
import re
import sys
import mutagen
from mutagen.mp3 import MP3
from mutagen.flac import FLAC

############# VARIABLES #############
red      = "\033[1;31m"
green    = "\033[1;32m"
yellow   = "\033[1;33m"
no_color = "\033[0m"

############# FUNCTIONS #############
def lowercase_match(match):
    """Lowercase the whole regular expression match group."""
    return match.group().lower()

def title(value):
    """Make a string title, without some common mistakes."""
    titled = value.title()
    titled = re.sub(r"([a-z])'([A-Z])", lowercase_match, titled)  # Fix Don'T
    titled = re.sub(r"\d([A-Z])", lowercase_match, titled)  # Fix 1St and 2Nd
    return titled

def prettify(str):
    """Get a string and return it stripped and in title format."""
    str = str.strip()
    str = title(str)
    return str 

def rename(path, current_name, new_name):
    """Rename a file."""
    current_file = path + "\\" + current_name
    new_file     = path + "\\" + new_name

    print(f"Renaming '{yellow}{current_name}{no_color}' to '{yellow}{new_name}{no_color}'...")
    os.rename(current_file, new_file)

def handle_mp3(path, track, file):
    """Handle MP3 files."""
    # Artist
    if ("TPE1" in track.keys()):
        artist = str(track.get("TPE1"))
        artist = prettify(artist)

    # Track Title
    if ("TIT2" in track.keys()):
        title = str(track.get("TIT2"))
        title = prettify(title)

    if artist and title:
        new_name = artist + " - " + title + ".mp3"
        rename(path, file, new_name)

def handle_flac(path, track, file):
    """Handle FLAC files."""
    # Artist
    if ("artist" in track.keys()):
        artist = str(track.get("artist")[0])
        artist = prettify(artist)

    # Track Title
    if ("title" in track.keys()):
        title = str(track.get("title")[0])
        title = prettify(title)

    if artist and title:
        new_name = artist + " - " + title + ".flac"
        rename(path, file, new_name)

def main(folder_path, sub_dirs):
    for file_name in os.listdir(folder_path):
        file_path = folder_path + "\\" + file_name

        if os.path.isdir(file_path) and sub_dirs:
            # Recursive Call
            main(file_path, True)
        elif os.path.isfile(file_path):
            file      = mutagen.File(file_path)
            file_type = type(file)

            if file_type == mutagen.mp3.MP3:
                track = MP3(file_path)
                handle_mp3(folder_path, track, file_name)
            elif file_type == mutagen.flac.FLAC:
                track = FLAC(file_path)
                handle_flac(folder_path, track, file_name)
            else:
                print(f"{yellow}File type {type} is not supported yet...{no_color}")

def exit_program():
    print(f"{green}BYE!!{no_color}")
    sys.exit(0)

############### CODE ################
print("#######################################################################################")
while True:
    ui_1 = input(f"Enter the {green}file path{no_color} that contains the files to be renamed or enter {red}exit{no_color} to exit: ")

    if ui_1.lower() != "exit" and not os.path.isdir(ui_1):
        print("Please enter a correct file path: ")
        continue
    else:
        break

if ui_1.lower() == "exit":
    exit_program()
else:
    folder = ui_1

print(f"The directory is set to {yellow}'{folder}'{no_color}.")

while True:
    ui_2 = input(f"Do you want to rename the music files inside all sub directories as well? ({green}y{no_color}/{red}n{no_color}) ")

    if ui_2.lower() == "y" or ui_2.lower() == "n":
        break
    else:
        print(f"Please answer {green}y{no_color} or {red}n{no_color}")
        continue
print("#######################################################################################")

if ui_2.lower() == "exit":
    exit_program()
elif ui_2.lower() == "n":
    main(folder, False)
elif ui_2.lower() == "y":
    main(folder, True)