done

rename-episodes.py

@@ -0,0 +1,98 @@
+import os, sys
+from typing import final
+
+
+# Getting the folder
+folder = ''
+
+if folder == '':
+    folder = os.getcwd()
+
+
+folderChoice = input(f'The selected folder is \n{folder}\nProceed with the folder above? (Y/n): ')
+
+if folderChoice.lower() == 'n':
+    folder = input('Enter path where you want to rename files:\n')
+elif folderChoice.upper() != 'Y':
+    print('What am I supposed to do now?!!!')
+    assert False
+
+
+# Getting all the files
+os.chdir(folder)
+print(f'Proceeding with {folder}')
+ls = os.listdir()
+files = []
+folders = []
+for item in ls:
+    if os.path.isfile(item):
+        files.append(item)
+    else:
+        print(f'WARNING: {item} is not a file!')
+
+# for file in files: print(file)
+files.sort()
+# print(f'\n{"="*30}')
+# for file in files: print(file)
+
+
+# Getting season
+def getSeason(folder):
+    # print(f'folder: {folder}')
+    # print(os.path.split(folder)[-1])
+    if os.path.split(folder)[-1] != '':
+        # print('if is working')
+        path = os.path.split(folder)[-1]
+        # print(f'path: {path}')
+        return path
+    else:
+        # print('else is working')
+        # getSeason(os.path.split(folder)[-2])
+        # print(f'os.path.split(folder)[-2]: {os.path.split(folder)[-2]}')
+        return getSeason(os.path.split(folder)[-2])
+
+season = getSeason(folder)
+season = season[7:9]
+
+seasonConfirmation = input(f'The season seems to be: "{season}". Is it correct? (Y/n)? ')
+
+if seasonConfirmation.lower() == 'n':
+    season = input('Please enter the season(Make it double digit, add a 0 in the front if needed): ')
+elif seasonConfirmation.upper() != 'Y':
+    print('What do I do now?!!!')
+    assert False
+
+print(f'Proceeding with season {season}')
+
+# Get new name string
+# f'{filename[:16]} test{filename[16:]}'
+def renameTo(filename, season, episode):
+    return f'{filename[:16]} S{season}E{episode}{filename[16:]}'
+
+# Dry run
+print(f'\nThe files will be renamed like this:\n')
+renameInfo = []
+episode = 1
+for file in files:
+    print(f'{file} => {renameTo(file, season, episode)}')
+    episode += 1
+
+finalConfirmation = input('Do you want to proceed with the changes listed above? (Y/n): ')
+
+
+# Final confirmation logic
+if finalConfirmation.lower() == 'n':
+    print('got it, terminating')
+    assert False
+elif finalConfirmation.upper() != 'Y':
+    print('What do I do now? :\'(')
+    assert False
+elif finalConfirmation.upper == 'Y':
+    print('renaming files...')
+
+# Rename!
+episode = 1
+for file in files:
+    os.rename(file, renameTo(file, season, episode))
+    episode += 1
+    print(f'Renamed {file}')