stabbedbybrick/devine-services
3
3
Fork 1
Code Issues Actions Packages Projects Releases Wiki Activity

fix(PLUTO): Fix year label for movies and series

Browse Source
This commit is contained in:
stabbedbybrick 2024-10-21 11:54:16 +02:00
parent 43594fb60d
commit 099374ca0c
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
services/PLUTO/__init__.py
View File

@ -173,7 +173,7 @@ class PLUTO(Service):
season=int(episode.get("season")), season=int(episode.get("season")),
number=int(episode.get("number")), number=int(episode.get("number")),
name=episode.get("name"), name=episode.get("name"),
year=None, year=self.year(episode),
language="en", # self.region, language="en", # self.region,
data=episode, data=episode,
) )
@ -194,10 +194,10 @@ class PLUTO(Service):
Movie( Movie(
id_=movie.get("_id"), id_=movie.get("_id"),
service=self.__class__, service=self.__class__,
year=movie.get("slug", "").split("-")[-3],
name=movie.get("name"), name=movie.get("name"),
language="en", # self.region, language="en", # self.region,
data=movie, data=movie,
year=self.year(movie),
) )
for movie in data for movie in data
] ]
@ -280,3 +280,10 @@ class PLUTO(Service):
) )
return any(ad in text for ad in ads) return any(ad in text for ad in ads)
@staticmethod
def year(data: dict) -> Optional[int]:
title_year = (int(match.group(1)) if (match := re.search(r"\((\d{4})\)", data.get("name", ""))) else None)
slug_year = (int(match.group(1)) if (match := re.search(r"\b(\d{4})\b", data.get("slug", ""))) else None)
return None if title_year else slug_year