363 lines
9.7 KiB
Python
363 lines
9.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import re
|
|
import sys
|
|
import requests
|
|
import argparse
|
|
from pathlib import Path
|
|
from difflib import SequenceMatcher
|
|
|
|
|
|
TMDB_TOKEN = "eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiJlOTIwMzJlMDk1NmEzMWEyZjQ4MDA4NGI0MTg4OTIyZCIsIm5iZiI6MTc5MDQzMjc5NS42NjcsInN1YiI6IjZhYjdkNjFiZDY5NmI1ZGFjNGM5NDRkMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.IjddKtQ5X62A63kYslH24Pxg_TFCzu-QpGak_Pemgr8"
|
|
|
|
if not TMDB_TOKEN:
|
|
print("ERROR: TMDB_API_TOKEN is not set.")
|
|
print()
|
|
print("Example:")
|
|
print(" export TMDB_API_TOKEN='your_token'")
|
|
sys.exit(1)
|
|
|
|
|
|
BASE_URL = "https://api.themoviedb.org/3"
|
|
|
|
HEADERS = {
|
|
"Authorization": f"Bearer {TMDB_TOKEN}",
|
|
"accept": "application/json",
|
|
}
|
|
|
|
|
|
def tmdb_get(endpoint, params=None):
|
|
"""Perform a GET request against TMDb."""
|
|
url = BASE_URL + endpoint
|
|
print(f"Set of Tmdb query: {url}")
|
|
|
|
response = requests.get(
|
|
url,
|
|
headers=HEADERS,
|
|
params=params,
|
|
timeout=15,
|
|
)
|
|
|
|
response.raise_for_status()
|
|
|
|
return response.json()
|
|
|
|
def find_by_title(title, media_type="tv", year=None, language="de-DE"):
|
|
"""
|
|
Search TMDB by title.
|
|
|
|
media_type:
|
|
tv | movie
|
|
"""
|
|
|
|
params = {
|
|
"query": title,
|
|
"language": language,
|
|
}
|
|
|
|
if year:
|
|
if media_type == "tv":
|
|
params["first_air_date_year"] = year
|
|
else:
|
|
params["year"] = year
|
|
|
|
return tmdb_get(
|
|
f"/search/{media_type}",
|
|
params,
|
|
)
|
|
|
|
|
|
def find_by_tmdb_id(tmdb_id, language="de-DE"):
|
|
"""
|
|
Try to determine whether a TMDB ID is a TV show or movie.
|
|
|
|
Returns:
|
|
("tv", data)
|
|
("movie", data)
|
|
(None, None)
|
|
"""
|
|
|
|
try:
|
|
data = get_tv_details(tmdb_id, language)
|
|
return "tv", data
|
|
except requests.HTTPError:
|
|
pass
|
|
|
|
try:
|
|
data = get_movie_details(tmdb_id, language)
|
|
return "movie", data
|
|
except requests.HTTPError:
|
|
pass
|
|
|
|
return None, None
|
|
|
|
def get_tv_details(tmdb_id , language="de-DE"):
|
|
"""
|
|
Retrieve TV show details in German.
|
|
"""
|
|
|
|
return tmdb_get(
|
|
f"/tv/{tmdb_id}",
|
|
{
|
|
"language": language,
|
|
},
|
|
)
|
|
|
|
|
|
def get_movie_details(tmdb_id, language="de-DE"):
|
|
"""
|
|
Retrieve movie details in German.
|
|
"""
|
|
|
|
return tmdb_get(
|
|
f"/movie/{tmdb_id}",
|
|
{
|
|
"language": language,
|
|
},
|
|
)
|
|
|
|
def get_season(tmdb_id, season_number):
|
|
return tmdb_get(
|
|
f"/tv/{tmdb_id}/season/{season_number}",
|
|
{
|
|
"language": "de-DE",
|
|
},
|
|
)
|
|
|
|
def get_episode_list(show,series_path: Path):
|
|
#_, show = find_by_tmdb_id(tmdb_id)
|
|
file_seasons = sorted({
|
|
int(m.group(1))
|
|
for d in series_path.iterdir()
|
|
if d.is_dir()
|
|
and (m := re.match(r"^Season\s+0*(\d+)$", d.name, re.I))
|
|
})
|
|
episodes = []
|
|
|
|
for season in show.get("seasons", []):
|
|
season_number = season["season_number"]
|
|
|
|
# skip specials if desired
|
|
if season_number == 0 or season_number not in file_seasons:
|
|
continue
|
|
|
|
season_data = get_season(show.get("id"), season_number)
|
|
|
|
for episode in season_data.get("episodes", []):
|
|
episodes.append({
|
|
"season": season_number,
|
|
"episode": episode["episode_number"],
|
|
"title": episode["name"],
|
|
"air_date": episode["air_date"],
|
|
"runtime": episode.get("runtime"),
|
|
"tmdb_id": episode["id"],
|
|
})
|
|
|
|
return episodes
|
|
|
|
|
|
def print_results(result):
|
|
media_type, data = result
|
|
|
|
print(f"Type : {media_type}")
|
|
print(f"TMDB ID : {data.get('id')}")
|
|
print(f"Title : {data.get('name') or data.get('title')}")
|
|
print(f"Original : {data.get('original_name') or data.get('original_title')}")
|
|
print(f"Year : {(data.get('first_air_date') or data.get('release_date') or '')[:4]}")
|
|
print(f"Rating : {data.get('vote_average')}")
|
|
print(f"Votes : {data.get('vote_count')}")
|
|
print(f"Status : {data.get('status')}")
|
|
print(f"Tagline : {data.get('tagline')}")
|
|
print(f"Overview : {data.get('overview')}")
|
|
|
|
genres = ", ".join(g["name"] for g in data.get("genres", []))
|
|
print(f"Genres : {genres}")
|
|
|
|
print(f"Poster : https://image.tmdb.org/t/p/original{data.get('poster_path', '')}")
|
|
print(f"Backdrop : https://image.tmdb.org/t/p/original{data.get('backdrop_path', '')}")
|
|
|
|
if media_type == "tv":
|
|
print(f"Seasons : {data.get('number_of_seasons')}")
|
|
print(f"Episodes : {data.get('number_of_episodes')}")
|
|
|
|
def normalize(text, title, year):
|
|
text = text.lower()
|
|
|
|
# remove series title
|
|
text = re.sub(re.escape(title.lower()), " ", text)
|
|
|
|
# remove year
|
|
text = re.sub(rf"\b{year}\b", " ", text)
|
|
|
|
# remove anything in [] including the brackets
|
|
text = re.sub(r"\[[^\]]*\]", " ", text)
|
|
|
|
# remove punctuation
|
|
text = re.sub(r"[^\w\s]", " ", text)
|
|
|
|
# remove episode numbers like S01E01
|
|
text = re.sub(r"\bs\d+e\d+\b", " ", text, flags=re.I)
|
|
|
|
# collapse spaces
|
|
text = re.sub(r"\s+", " ", text).strip()
|
|
|
|
return text
|
|
|
|
def best_episode_match(filename, episodes,title,year):
|
|
filename_norm = normalize(Path(filename).stem, title, year)
|
|
|
|
best = None
|
|
best_score = 0
|
|
|
|
for ep in episodes:
|
|
score = SequenceMatcher(
|
|
None,
|
|
filename_norm,
|
|
normalize(ep["title"], title, year)
|
|
).ratio()
|
|
|
|
if score > best_score:
|
|
best_score = score
|
|
best = ep
|
|
|
|
return best, best_score
|
|
|
|
def rename_media(media_data,episodes, series_path,file_title):
|
|
series=series_path.name
|
|
episode_map = {
|
|
(ep["season"], ep["episode"]): ep
|
|
for ep in episodes
|
|
}
|
|
file_list = []
|
|
new_file_list = []
|
|
for file in series_path.rglob("*"):
|
|
match_SE_name = re.search(r"S(\d{2})E(\d{2})", file.name, re.I)
|
|
if not match_SE_name:
|
|
match_season_folder = re.search(r"Season (\d{2})", file.name, re.I)
|
|
if match_season_folder:
|
|
continue # witchout bothering anyone
|
|
print(f"No season/episode number found: {file.name}")
|
|
continue
|
|
|
|
season = int(match_SE_name.group(1))
|
|
episode= int(match_SE_name.group(2))
|
|
|
|
episode_number = episode_map.get((season, episode))
|
|
|
|
if not episode_number:
|
|
print(f"Episode not found: S{season:02d}E{episode:02d}")
|
|
continue
|
|
episode_season_number_match = re.match(
|
|
r"^(.*?S\d{2}E\d{2})\s*-\s*(.*?)\s*(\[[^\]]+\].*)?$",
|
|
file.stem,
|
|
re.I,
|
|
)
|
|
if not episode_season_number_match:
|
|
print(f"S{season:02d}E{episode:02d} not found")
|
|
continue
|
|
prefix = episode_season_number_match.group(1).replace(file_title,(media_data.get('name')
|
|
or media_data.get('title'))) # Andor (2022) - S01E07
|
|
quality = episode_season_number_match.group(3) or "" # [WEBRip-1080p][AAC 5.1][h265]
|
|
|
|
new_name = (
|
|
f"{prefix} - {episode_number['title']}"
|
|
f"{(' ' + quality) if quality else ''}"
|
|
f"{file.suffix}"
|
|
)
|
|
|
|
print(f"{file.name} -> {new_name}")
|
|
|
|
file_list.append(file)
|
|
new_file_list.append(new_name)
|
|
|
|
title=(f"{(media_data.get('name') or media_data.get('title'))} "
|
|
f"({(media_data.get('first_air_date') or media_data.get('release_date') or '')[:4]}) "
|
|
f"[{media_data.get('id')}]")
|
|
print(f"{series} -> {title}")
|
|
|
|
file_list.append(series_path)
|
|
new_file_list.append(title)
|
|
has_changes = False
|
|
for file, new_file in zip(file_list, new_file_list):
|
|
if file.name != new_file:
|
|
has_changes = True
|
|
break
|
|
if not has_changes:
|
|
print("No changes necessary.")
|
|
return
|
|
answer = input("Rename files? [y/N]: ").strip().lower()
|
|
if answer in ("y", "yes"):
|
|
print("Proceeding...")
|
|
for file, new_file in zip(file_list,new_file_list):
|
|
file.rename(file.with_name(new_file))
|
|
print("All files renamed")
|
|
return
|
|
print("Cancelled.")
|
|
return
|
|
|
|
def process_series(series_path: Path):
|
|
series = series_path.name
|
|
|
|
m = re.match(r"^(.*?)\s*\((\d{4})\)\s*\[(\d+)\]$", series)
|
|
|
|
if not m:
|
|
print(f"Skipping invalid folder: {series}")
|
|
return
|
|
|
|
file_title = m.group(1).strip()
|
|
#file_year = int(m.group(2))
|
|
file_tmdb_id = int(m.group(3))
|
|
|
|
if file_tmdb_id == 0:
|
|
print("TMDB ID required")
|
|
return
|
|
|
|
try:
|
|
data = find_by_tmdb_id(file_tmdb_id)
|
|
|
|
if not data:
|
|
print("TMDB lookup failed")
|
|
return
|
|
|
|
media_type, media_data = data
|
|
|
|
#print_results(data)
|
|
|
|
episodes = get_episode_list(media_data,series_path)
|
|
|
|
rename_media(
|
|
media_data,
|
|
episodes,
|
|
series_path,
|
|
file_title
|
|
)
|
|
|
|
except Exception as e:
|
|
print(f"Error processing {series}: {e}")
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"path",
|
|
help='Either a series folder "Maya the Bee (1975) [10620]" or a root folder containing such folders'
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
input_path = Path(args.path)
|
|
|
|
if not input_path.is_dir():
|
|
parser.error(f"Directory not found: {input_path}")
|
|
|
|
# Input is already a series folder
|
|
if re.match(r"^(.*?)\s*\((\d{4})\)\s*\[(\d+)\]$", input_path.name):
|
|
process_series(input_path)
|
|
|
|
# Input is a root folder containing series folders
|
|
else:
|
|
for series_path in sorted(input_path.iterdir()):
|
|
if series_path.is_dir():
|
|
process_series(series_path)
|
|
|
|
if __name__ == "__main__":
|
|
main() |