You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
#!/usr/bin/python3
|
|
|
|
import os
|
|
import re
|
|
import json
|
|
|
|
ONE_PIECE_SEASON_COUNT = [
|
|
8, # Season 1
|
|
22, # Season 2
|
|
17, # Season 3
|
|
13, # Season 4
|
|
9, # Season 5
|
|
22, # Season 6
|
|
39, # Season 7
|
|
13, # Season 8
|
|
52, # Season 9
|
|
31, # Season 10
|
|
99, # Season 11
|
|
56, # Season 12
|
|
100, # Season 13
|
|
35, # Season 14
|
|
62, # Season 15
|
|
# hack to deal with the missing episode 590 in season 16. The actual count is 49
|
|
50, # Season 16
|
|
118, # Season 17
|
|
33, # Season 18
|
|
98, # Season 19
|
|
14, # Season 20
|
|
179, # Season 21
|
|
]
|
|
|
|
ONE_PIECE_DIR = "/mnt/mergerfs/media/tv/One Piece"
|
|
|
|
|
|
def _get_absolute_episode_number(filename):
|
|
# [HorribleSubs] One Piece - 260 [1080p].mkv -> 260
|
|
return int(re.findall(r"\s(\d+)\s", filename)[0])
|
|
|
|
|
|
def _get_episode_season(absolute_number):
|
|
season = 1
|
|
episodes_in_season = ONE_PIECE_SEASON_COUNT[season - 1]
|
|
while absolute_number > episodes_in_season:
|
|
season += 1
|
|
episodes_in_season += ONE_PIECE_SEASON_COUNT[season - 1]
|
|
return season
|
|
|
|
|
|
def _get_season_episode_number(season, absolute_number):
|
|
relative_number = absolute_number - sum(ONE_PIECE_SEASON_COUNT[:season - 1])
|
|
# season 16 is missing episode 590 so we need to offset the number for just this season.
|
|
if season == 16 and absolute_number > 590:
|
|
relative_number -= 1
|
|
return relative_number
|
|
|
|
|
|
def _get_new_name(relative_episode_number, actual_season):
|
|
name = "[HorribleSubs] One Piece - S"
|
|
if actual_season < 10:
|
|
name += "0"
|
|
name += f"{actual_season}E"
|
|
if relative_episode_number < 10:
|
|
name += "0"
|
|
return name + f"{relative_episode_number} [1080p].mkv"
|
|
|
|
|
|
def main():
|
|
backup_json = []
|
|
all_files = os.listdir(ONE_PIECE_DIR)
|
|
for f in all_files:
|
|
# skip files that have a season associated with them
|
|
if "S0" in f:
|
|
print(f"Skipping {f} as it is already in the correct format")
|
|
continue
|
|
absolute_episode_number = _get_absolute_episode_number(f)
|
|
season = _get_episode_season(absolute_episode_number)
|
|
season_episode_number = _get_season_episode_number(season, absolute_episode_number)
|
|
new_name = _get_new_name(season_episode_number, season)
|
|
backup_json.append({"old": f, "new": new_name})
|
|
|
|
# write a backup file in case we need to revert
|
|
with open("/home/cianhatton/rename-one-piece-backup.json", "w") as f:
|
|
json.dump(backup_json, f, indent=4)
|
|
|
|
for item in backup_json:
|
|
os.rename(f"{ONE_PIECE_DIR}/{item['old']}", f"{ONE_PIECE_DIR}/{item['new']}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|