diff options
Diffstat (limited to 'services')
| -rw-r--r-- | services/tmdb.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/services/tmdb.py b/services/tmdb.py index 3574f57..27f9f5b 100644 --- a/services/tmdb.py +++ b/services/tmdb.py @@ -46,6 +46,12 @@ def countries(movie: dict) -> str | None: return ", ".join(names) if names else None +def genres(movie: dict) -> str | None: + names = [genre.get("name") for genre in movie.get("genres", [])] + names = [name for name in names if name] + return ", ".join(names) if names else None + + def languages(movie: dict) -> str | None: names = [ language.get("english_name") or language.get("name") @@ -115,6 +121,7 @@ def movie_payload(movie: dict, fallback: dict | None = None) -> dict: "release_date": release_date, "director": directors(movie), "country": countries(movie), + "genre": genres(movie), "language": languages(movie), "runtime": movie.get("runtime"), "overview": movie.get("overview") or fallback.get("overview"), @@ -275,6 +282,7 @@ def apply_metadata_to_film(film, metadata: dict) -> bool: "director": metadata.get("director"), "year": metadata.get("year"), "country": metadata.get("country"), + "genre": metadata.get("genre"), "language": metadata.get("language"), "runtime": metadata.get("runtime"), } |