Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Refresh Code #3

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Refresh Code #3

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
eeb5f55
Add requirements file
hwwilliams Dec 1, 2023
c72ccee
Auto format and reorder functions
hwwilliams Dec 1, 2023
99b6b5d
Simplify titlization
hwwilliams Dec 1, 2023
804b67b
Cleanup
hwwilliams Dec 1, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
mutagen.mp4
progress.bar
289 changes: 148 additions & 141 deletions title-tagging.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,19 +4,19 @@
import os
import sys


# Run the script with '--help' to see more information.


class bcolors:
BOLD = '\033[1m'
RED = '\033[91m'
RESET = '\033[0m'
YELLOW = '\033[93m'
BOLD = "\033[1m"
RED = "\033[91m"
RESET = "\033[0m"
YELLOW = "\033[93m"


def arguments():
parser = argparse.ArgumentParser(description='''
parser = argparse.ArgumentParser(
description="""
A script to embed title tags in video files.

Dependencies: Python 3, Mutagen, and Progress
Expand All @@ -42,146 +42,120 @@ def arguments():
Example 2:
File name: 01 - Why Learn Python?.mp4
Extracted title tag: Why Learn Python?
''', formatter_class=argparse.RawTextHelpFormatter)
""",
formatter_class=argparse.RawTextHelpFormatter,
)
parser.parse_args()


def ask(confirmation_prompt, default_answer_no=False):
while True:
answer = (input(confirmation_prompt).strip()).lower()
if default_answer_no and answer == '' or answer.startswith('n'):
return False
elif answer == '' or answer.startswith('y'):
return True
elif answer == 'exit':
terminate()
else:
print(f'{bcolors.YELLOW}Invalid Input: Please enter yes or no.{bcolors.RESET}')


def capitalize_title(title_string):
ignore = {'a', 'an', 'and', 'as', 'at', 'but', 'by',
'down', 'for', 'from', 'if', 'in', 'into',
'like', 'near', 'nor', 'of', 'off', 'on',
'once', 'onto', 'or', 'over', 'past', 'so',
'than', 'that', 'the', 'till', 'to', 'upon',
'v', 'v.', 'vs', 'vs.', 'when', 'with', 'yet'}
title = ''
for word in title_string.split(' '):
if word.lower() in ignore:
if word == title_string.split(' ')[0] or word == title_string.split(' ')[-1]:
if not word.isupper():
if not word.endswith('s') & (word.replace('s', '')).isupper():
word = word.capitalize()
else:
word = word.lower()
elif not word.isupper():
if not word.endswith('s') & (word.replace('s', '')).isupper():
if not (word.lower()).startswith('ipv'):
word = word.capitalize()
title += str(f'{word} ')
return title
def example_info():
return f"""
Correct file name format is as follows

Example 1:
File name: Python 3 Tutorial for Beginners - S01E01 - Why Learn Python?.mp4
Extracted title tag: Why Learn Python?

Example 2:
File name: 01 - Why Learn Python?.mp4
Extracted title tag: Why Learn Python?

{bcolors.BOLD}Type "exit" at any prompt to exit the script{bcolors.RESET}

For more information run the script with --help
"""


def terminate():
print("Exiting script.")
sys.exit()


def check_path(directory_path_prompt):
search_directory_set = set([])
while True:
input_directory = input(directory_path_prompt)
if input_directory.lower() == 'exit':
if input_directory.lower() == "exit":
terminate()
if (input_directory.strip()).startswith('.\\'):
input_directory = (input_directory.strip()).replace('.\\', '', 1)
if (input_directory.strip()).startswith(".\\"):
input_directory = (input_directory.strip()).replace(".\\", "", 1)
input_directory = os.path.join((os.getcwd()), input_directory)
if os.path.isdir(input_directory):
search_directory_set.add(input_directory)
if ask('Would you like to add another directory to search (y/N)? ', True):
if ask("Would you like to add another directory to search (y/N)? ", True):
continue
else:
return search_directory_set
else:
print(f'{bcolors.YELLOW}Invalid Directory Path: Please enter a valid directory path.{bcolors.RESET}')
print(
f"{bcolors.YELLOW}Invalid Directory Path: Please enter a valid directory path.{bcolors.RESET}"
)
continue


def check_titles(correct_tags_dictionary, merged_tags_dictionary):
if len(correct_tags_dictionary) >= 1:
print(f'{len(correct_tags_dictionary)} file(s) with correct tag(s).')
if len(merged_tags_dictionary) >= 1:
print(f'{bcolors.YELLOW}Found {len(merged_tags_dictionary)} file(s) with incorrect title tag{bcolors.RESET}.')
for file_path, file_title in merged_tags_dictionary.items():
print(f'{bcolors.YELLOW}Incorrect: Tag for "{file_path}" not successfully set.{bcolors.RESET}')


def example_info():
return f'''
Correct file name format is as follows

Example 1:
File name: Python 3 Tutorial for Beginners - S01E01 - Why Learn Python?.mp4
Extracted title tag: Why Learn Python?

Example 2:
File name: 01 - Why Learn Python?.mp4
Extracted title tag: Why Learn Python?
def valid_extension(extension_prompt):
valid_extension_list = ["mp4"]
while True:
extension_limit = (input(extension_prompt).strip()).lower()
if extension_limit == "exit":
terminate()
if extension_limit.startswith("."):
extension_limit = extension_limit.replace(".", "", 1)
if extension_limit in valid_extension_list:
return extension_limit
else:
print(
f"{bcolors.YELLOW}Invalid Input: Please enter one of the following supported extensions {valid_extension_list}.{bcolors.RESET}"
)
continue

{bcolors.BOLD}Type "exit" at any prompt to exit the script{bcolors.RESET}

For more information run the script with --help
'''
def walk_the_path(valid_directory_set):
path_walked = {}
extension_limit = valid_extension(
"What file extension would you like to use to narrow the search? "
)
for valid_dir in valid_directory_set:
for root, subdirs, files in os.walk(valid_dir):
for file in files:
if file.endswith(extension_limit):
path_walked[file] = root
if len(path_walked) == 0:
print(
f'{bcolors.YELLOW}No files found when searching for files ending with ".{extension_limit}".{bcolors.RESET}'
)
terminate()
elif len(path_walked) >= 1:
return path_walked


def fetch_titles(path_walked_dictionary):
valid_titles = {}
for file, root in path_walked_dictionary.items():
if ' - ' not in file:
print(f'{bcolors.RED}Error: Video File "{os.path.join(root, file)}" is not named correctly.{bcolors.RESET}')
print(f'{bcolors.YELLOW}Run the script with "--help" to see an example of the correct format of a file name.{bcolors.RESET}')
for file_name, root in path_walked_dictionary.items():
if " - " not in file_name:
print(
f'{bcolors.RED}Error: Video file "{os.path.join(root, file_name)}" is not named correctly.{bcolors.RESET}'
)
print(
f'{bcolors.YELLOW}Run the script with "--help" to see an example of the correct format of a file name.{bcolors.RESET}'
)
else:
title = capitalize_title((((file.split(' - '))[-1])[0:-((len(((file.split('.'))[-1]).strip())+1))].strip()))
valid_titles[os.path.join(root, file)] = title.strip()
file_extension = os.path.splitext(file_name)[-1]
valid_titles[os.path.join(root, file_name)] = (
file_name.removesuffix(file_extension).split("-")[-1].strip().title()
)
return valid_titles


def handling_tags(merged_tags_dictionary, correct_tags_dictionary):
if len(merged_tags_dictionary) >= 1:
for file_path, file_title in merged_tags_dictionary.items():
print(f'The title for "{file_path}" would be "{file_title}".')
print(f'Found {len(merged_tags_dictionary)} file(s) with incorrect title tag.')
if len(correct_tags_dictionary) >= 1:
print(f'Found {len(correct_tags_dictionary)} file(s) with correct title tag.')
return True
elif len(merged_tags_dictionary) == 0:
if len(correct_tags_dictionary) >= 1:
print(f'Found {len(correct_tags_dictionary)} file(s) with correct title tag.')
return False


def main():
arguments()
print(example_info())
search_directory_set = check_path('Which directory would you like to search (recursively)? ')
path_walked = walk_the_path(search_directory_set)
valid_titles = fetch_titles(path_walked)
correct_tags, incorrect_tags, empty_tags, merged_tags = sort_tags(valid_titles)
if handling_tags(merged_tags, correct_tags):
if ask('Would you like to update the file(s) as shown above (Y/n)? '):
update_title_progress(merged_tags)
for dict in [correct_tags, incorrect_tags, empty_tags, merged_tags]:
dict.clear()
correct_tags, incorrect_tags, empty_tags, merged_tags = sort_tags(valid_titles)
check_titles(correct_tags, merged_tags)
terminate()


def sort_tags(valid_titles_dictionary):
correct = {}
incorrect = {}
empty = {}
merged = {}
for file_path, file_title in valid_titles_dictionary.items():
video_file = mutagen.mp4.MP4(file_path)
if '©nam' in video_file:
if file_title in video_file.get('©nam'):
if "©nam" in video_file:
if file_title in video_file.get("©nam"):
correct[file_path] = file_title
else:
incorrect[file_path] = file_title
Expand All @@ -192,57 +166,90 @@ def sort_tags(valid_titles_dictionary):
return correct, incorrect, empty, merged


def terminate():
print('Exiting script.')
sys.exit()
def handling_tags(merged_tags_dictionary, correct_tags_dictionary):
if len(merged_tags_dictionary) >= 1:
for file_path, file_title in merged_tags_dictionary.items():
print(f'The title for "{file_path}" would be "{file_title}".')
print(f"Found {len(merged_tags_dictionary)} file(s) with incorrect title tag.")
if len(correct_tags_dictionary) >= 1:
print(
f"Found {len(correct_tags_dictionary)} file(s) with correct title tag."
)
return True
elif len(merged_tags_dictionary) == 0:
if len(correct_tags_dictionary) >= 1:
print(
f"Found {len(correct_tags_dictionary)} file(s) with correct title tag."
)
return False


def ask(confirmation_prompt, default_answer_no=False):
while True:
answer = (input(confirmation_prompt).strip()).lower()
if default_answer_no and answer == "" or answer.startswith("n"):
return False
elif answer == "" or answer.startswith("y"):
return True
elif answer == "exit":
terminate()
else:
print(
f"{bcolors.YELLOW}Invalid Input: Please enter yes or no.{bcolors.RESET}"
)


def update_title(tags_dictionary):
for file_path, file_title in tags_dictionary.items():
video_file = mutagen.mp4.MP4(file_path)
video_file['©nam'] = [file_title]
video_file["©nam"] = [file_title]
video_file.save()


def update_title_progress(tags_dictionary):
count = len(tags_dictionary)
bar = progress.bar.FillingSquaresBar('Processing', max=count, suffix='%(index)d/%(max)d - %(percent).1f%%')
bar = progress.bar.FillingSquaresBar(
"Processing", max=count, suffix="%(index)d/%(max)d - %(percent).1f%%"
)
for tag in range(count):
update_title(tags_dictionary)
count -= 1
bar.next()
bar.finish()


def valid_extension(extension_prompt):
valid_extension_list = ['mp4']
while True:
extension_limit = (input(extension_prompt).strip()).lower()
if extension_limit == 'exit':
terminate()
if extension_limit.startswith('.'):
extension_limit = extension_limit.replace('.', '', 1)
if extension_limit in valid_extension_list:
return extension_limit
else:
print(f'{bcolors.YELLOW}Invalid Input: Please enter one of the following supported extensions {valid_extension_list}.{bcolors.RESET}')
continue
def check_titles(correct_tags_dictionary, merged_tags_dictionary):
if len(correct_tags_dictionary) >= 1:
print(f"{len(correct_tags_dictionary)} file(s) with correct tag(s).")
if len(merged_tags_dictionary) >= 1:
print(
f"{bcolors.YELLOW}Found {len(merged_tags_dictionary)} file(s) with incorrect title tag{bcolors.RESET}."
)
for file_path, file_title in merged_tags_dictionary.items():
print(
f'{bcolors.YELLOW}Incorrect: Tag for "{file_path}" not successfully set.{bcolors.RESET}'
)


def walk_the_path(valid_directory_set):
path_walked = {}
extension_limit = valid_extension('What file extension would you like to use to narrow the search? ')
for valid_dir in valid_directory_set:
for root, subdirs, files in os.walk(valid_dir):
for file in files:
if file.endswith(extension_limit):
path_walked[file] = root
if len(path_walked) == 0:
print(f'{bcolors.YELLOW}No files found when searching for files ending with ".{extension_limit}".{bcolors.RESET}')
terminate()
elif len(path_walked) >= 1:
return path_walked
def main():
arguments()
print(example_info())
search_directory_set = check_path(
"Which directory would you like to search (recursively)? "
)
valid_titles = fetch_titles(walk_the_path(search_directory_set))
correct_tags, incorrect_tags, empty_tags, merged_tags = sort_tags(valid_titles)
if handling_tags(merged_tags, correct_tags):
if ask("Would you like to update the file(s) as shown above (Y/n)? "):
update_title_progress(merged_tags)
for dict in [correct_tags, incorrect_tags, empty_tags, merged_tags]:
dict.clear()
correct_tags, incorrect_tags, empty_tags, merged_tags = sort_tags(
valid_titles
)
check_titles(correct_tags, merged_tags)
terminate()


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