70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import re
|
||
|
|
|
||
|
|
def preview_rename(original_name, new_name):
|
||
|
|
print(f"Preview: {original_name} -> {new_name}")
|
||
|
|
|
||
|
|
def perform_rename(original_name, new_name):
|
||
|
|
os.rename(original_name, new_name)
|
||
|
|
print(f"Renamed: {original_name} -> {new_name}")
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# Check if the regex is provided as an argument
|
||
|
|
if len(sys.argv) < 3:
|
||
|
|
print("Please provide a regex pattern with show title, season, and episode groups.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
# Get folder path from command-line argument if provided, else use current directory
|
||
|
|
folder_path = sys.argv[1] if len(sys.argv) > 1 else os.getcwd()
|
||
|
|
|
||
|
|
# Get the regex pattern from the command-line arguments
|
||
|
|
regex_pattern = sys.argv[2]
|
||
|
|
|
||
|
|
for filename in os.listdir(folder_path):
|
||
|
|
# Check if the file is a regular file
|
||
|
|
if os.path.isfile(os.path.join(folder_path, filename)):
|
||
|
|
# Use regex to extract show title, season, episode, and episode name
|
||
|
|
match = re.match(regex_pattern, filename)
|
||
|
|
if match:
|
||
|
|
show_title = match.group(1)
|
||
|
|
season_number = match.group(2).zfill(2)
|
||
|
|
episode_number = match.group(3).zfill(2)
|
||
|
|
|
||
|
|
# Check if there is a fourth group for episode name
|
||
|
|
episode_name_group = match.group(4)
|
||
|
|
episode_name = f" - {episode_name_group}" if episode_name_group else ""
|
||
|
|
|
||
|
|
# Create the new filename in the desired format
|
||
|
|
new_filename = f"{show_title} S{season_number}E{episode_number}{episode_name}.avi"
|
||
|
|
|
||
|
|
# Preview the changes
|
||
|
|
preview_rename(filename, new_filename)
|
||
|
|
|
||
|
|
# Ask for user confirmation
|
||
|
|
user_input = input("Do you want to proceed with the renaming? (y/n): ").lower()
|
||
|
|
if user_input == 'y':
|
||
|
|
# Perform the renaming
|
||
|
|
for filename in os.listdir(folder_path):
|
||
|
|
if os.path.isfile(os.path.join(folder_path, filename)):
|
||
|
|
match = re.match(regex_pattern, filename)
|
||
|
|
if match:
|
||
|
|
show_title = match.group(1)
|
||
|
|
season_number = match.group(2).zfill(2)
|
||
|
|
episode_number = match.group(3).zfill(2)
|
||
|
|
episode_name_group = match.group(4)
|
||
|
|
episode_name = f" - {episode_name_group}" if episode_name_group else ""
|
||
|
|
new_filename = f"{show_title} S{season_number}E{episode_number}{episode_name}.avi"
|
||
|
|
perform_rename(os.path.join(folder_path, filename), os.path.join(folder_path, new_filename))
|
||
|
|
else:
|
||
|
|
print("Renaming canceled.")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) != 3:
|
||
|
|
print("Usage: ./episodes.py <directory> <regex>\nIn the regex, group 1 should be show title, group 2 season and group 3 episode, optionally group 4 for episode title.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
main()
|
||
|
|
|