mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-11-10 06:54:21 +00:00
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
from modules import util
|
|
from modules.util import Failed
|
|
|
|
logger = util.logger
|
|
|
|
ids_url = "https://raw.githubusercontent.com/meisnate12/PMM-TOP10/master/top10.yml"
|
|
builders = ["flixpatrol_top"]
|
|
|
|
|
|
class FlixPatrol:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self._data = None
|
|
self._platforms = None
|
|
self._locations = None
|
|
|
|
@property
|
|
def data(self):
|
|
if self._data is None:
|
|
self._data = self.config.load_yaml(ids_url)
|
|
return self._data
|
|
|
|
@property
|
|
def platforms(self):
|
|
if self._platforms is None:
|
|
self._platforms = [platform for platform in self.data]
|
|
self._platforms.sort()
|
|
return self._platforms
|
|
|
|
@property
|
|
def locations(self):
|
|
if self._locations is None:
|
|
self._locations = []
|
|
for platform in self.data:
|
|
self._locations.extend([loc for loc in self.data[platform] if loc not in self._locations and loc != "world"])
|
|
self._locations.sort()
|
|
self._locations = ["world"] + self._locations
|
|
return self._locations
|
|
|
|
def validate_builder(self, method, data, is_movie):
|
|
builder_type = "movies" if is_movie else "shows"
|
|
if method == "flixpatrol_top" and data["location"] in self.data[data["platform"]] and builder_type in self.data[data["platform"]][data["location"]]:
|
|
return data
|
|
raise Failed(f"FlixPatrol Error: No {builder_type[:-1].capitalize()} Data Found for {data['platform']} in {data['location']}")
|
|
|
|
def get_tmdb_ids(self, method, data, is_movie):
|
|
flix_items = []
|
|
if method == "flixpatrol_top":
|
|
logger.info("Processing FlixPatrol Top:")
|
|
logger.info(f"\tPlatform: {data['platform'].replace('_', ' ').title()}")
|
|
logger.info(f"\tLocation: {data['location'].replace('_', ' ').title()}")
|
|
logger.info(f"\tLimit: {data['limit']}")
|
|
flix_items = self.data[data["platform"]][data["location"]]["movies" if is_movie else "shows"][:data["limit"]]
|
|
items = [(i, "tmdb" if is_movie else "tmdb_show") for i in flix_items]
|
|
if len(items) > 0:
|
|
logger.info(f"Processed {len(items)} TMDb IDs")
|
|
return items
|
|
else:
|
|
raise Failed(f"FlixPatrol Error: No List Items found in {data}")
|