mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-11-26 22:40:26 +00:00
158 lines
9.2 KiB
Python
158 lines
9.2 KiB
Python
import logging
|
|
from modules import util
|
|
from modules.util import Failed
|
|
|
|
logger = logging.getLogger("Plex Meta Manager")
|
|
|
|
builders = ["flixpatrol_url", "flixpatrol_demographics", "flixpatrol_popular", "flixpatrol_top"]
|
|
generations = ["all", "boomers", "x", "y", "z"]
|
|
generations_translation = {"all": "all-generations", "boomers": "baby-boomers", "x": "generation-x", "y": "generation-y", "z": "generation-z"}
|
|
generations_pretty = {"all": "All generations", "boomers": "Baby Boomers", "x": "Generation X", "y": "Generation Y (Millenials)", "z": "Generation Z"}
|
|
gender = ["all", "men", "women"]
|
|
demo_locations = ["world", "brazil", "canada", "france", "germany", "india", "mexico", "united_kingdom", "united_states"]
|
|
locations = [
|
|
"world", "albania", "argentina", "armenia", "australia", "austria", "azerbaijan", "bahamas", "bahrain",
|
|
"bangladesh", "belarus", "belgium", "belize", "benin", "bolivia", "bosnia_and_herzegovina", "botswana", "brazil",
|
|
"bulgaria", "burkina_faso", "cambodia", "canada", "chile", "colombia", "costa_rica", "croatia", "cyprus",
|
|
"czech_republic", "denmark", "dominican_republic", "ecuador", "egypt", "estonia", "finland", "france", "gabon",
|
|
"germany", "ghana", "greece", "guatemala", "guinea_bissau", "haiti", "honduras", "hong_kong", "hungary", "iceland",
|
|
"india", "indonesia", "ireland", "israel", "italy", "ivory_coast", "jamaica", "japan", "jordan", "kazakhstan",
|
|
"kenya", "kuwait", "kyrgyzstan", "laos", "latvia", "lebanon", "lithuania", "luxembourg", "malaysia", "maldives",
|
|
"mali", "malta", "mexico", "moldova", "mongolia", "montenegro", "morocco", "mozambique", "namibia", "netherlands",
|
|
"new_zealand", "nicaragua", "niger", "nigeria", "north_macedonia", "norway", "oman", "pakistan", "panama",
|
|
"papua_new_guinea", "paraguay", "peru", "philippines", "poland", "portugal", "qatar", "romania", "russia",
|
|
"rwanda", "salvador", "saudi_arabia", "senegal", "serbia", "singapore", "slovakia", "slovenia", "south_africa",
|
|
"south_korea", "spain", "sri_lanka", "sweden", "switzerland", "taiwan", "tajikistan", "tanzania", "thailand",
|
|
"togo", "trinidad_and_tobago", "turkey", "turkmenistan", "uganda", "ukraine", "united_arab_emirates",
|
|
"united_kingdom", "united_states", "uruguay", "uzbekistan", "venezuela", "vietnam", "zambia", "zimbabwe"
|
|
]
|
|
popular = ["movie_db", "facebook", "google", "twitter", "twitter_trends", "instagram", "instagram_trends", "youtube", "imdb", "letterboxd", "rotten_tomatoes", "tmdb", "trakt"]
|
|
platforms = ["netflix", "hbo", "disney", "amazon", "itunes", "google", "paramount_plus", "hulu", "vudu", "imdb", "amazon_prime", "star_plus"]
|
|
base_url = "https://flixpatrol.com"
|
|
urls = {
|
|
"top10": f"{base_url}/top10/",
|
|
"popular_movies": f"{base_url}/popular/movies/",
|
|
"popular_shows": f"{base_url}/popular/tv-shows/",
|
|
"demographics": f"{base_url}/demographics/"
|
|
}
|
|
|
|
class FlixPatrol:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
|
|
def _request(self, url, language, xpath):
|
|
if self.config.trace_mode:
|
|
logger.debug(f"URL: {url}")
|
|
return self.config.get_html(url, headers=util.header(language)).xpath(xpath)
|
|
|
|
def _tmdb(self, flixpatrol_url, language):
|
|
ids = self._request(flixpatrol_url, language, "//script[@type='application/ld+json']/text()")
|
|
if len(ids) > 0 and ids[0]:
|
|
if "https://www.themoviedb.org" in ids[0]:
|
|
return util.regex_first_int(ids[0].split("https://www.themoviedb.org")[1], "TMDB Movie ID")
|
|
raise Failed(f"FlixPatrol Error: TMDb Movie ID not found at {flixpatrol_url}")
|
|
|
|
def _parse_list(self, list_url, language, is_movie, limit=0):
|
|
flixpatrol_urls = []
|
|
if list_url.startswith(urls["top10"]):
|
|
platform = list_url[len(urls["top10"]):].split("/")[0]
|
|
flixpatrol_urls = self._request(
|
|
list_url, language,
|
|
f"//div[@id='{platform}-{'1' if is_movie else '2'}']//a[@class='hover:underline']/@href"
|
|
)
|
|
logger.info(flixpatrol_urls)
|
|
if not flixpatrol_urls:
|
|
flixpatrol_urls = self._request(
|
|
list_url, language,
|
|
f"//h3[text() = '{'TOP 10 Movies' if is_movie else 'TOP 10 TV Shows'}']/following-sibling::div//a[@class='hover:underline']/@href"
|
|
)
|
|
logger.info(flixpatrol_urls)
|
|
elif list_url.startswith(tuple([v for k, v in urls.items()])):
|
|
flixpatrol_urls = self._request(
|
|
list_url, language,
|
|
f"//a[@class='flex group' and .//span[.='{'Movie' if is_movie else 'TV Show'}']]/@href"
|
|
)
|
|
return flixpatrol_urls if limit == 0 else flixpatrol_urls[:limit]
|
|
|
|
def validate_flixpatrol_lists(self, flixpatrol_lists, language, is_movie):
|
|
valid_lists = []
|
|
for flixpatrol_list in util.get_list(flixpatrol_lists, split=False):
|
|
list_url = flixpatrol_list.strip()
|
|
if not list_url.startswith(tuple([v for k, v in urls.items()])):
|
|
fails = "\n".join([f"{v} (For {k.replace('_', ' ').title()})" for k, v in urls.items()])
|
|
raise Failed(f"FlixPatrol Error: {list_url} must begin with either:\n{fails}")
|
|
elif len(self._parse_list(list_url, language, is_movie)) > 0:
|
|
valid_lists.append(list_url)
|
|
else:
|
|
raise Failed(f"FlixPatrol Error: {list_url} failed to parse")
|
|
return valid_lists
|
|
|
|
def validate_flixpatrol_dict(self, method, data, language, is_movie):
|
|
return len(self.validate_flixpatrol_lists(self.get_url(method, data, is_movie), language, is_movie)) > 0
|
|
|
|
def get_url(self, method, data, is_movie):
|
|
if method == "flixpatrol_demographics":
|
|
return f"{urls['demographics']}" \
|
|
f"{generations_translation[data['generation']]}/" \
|
|
f"{'all-genders' if data['gender'] == 'all' else data['gender']}/" \
|
|
f"{data['location'].replace('_', '-')}/"
|
|
elif method == "flixpatrol_popular":
|
|
return f"{urls['popular_movies'] if is_movie else urls['popular_shows']}" \
|
|
f"{data['source'].replace('_', '-')}/" \
|
|
f"{util.time_window(data['time_window'])}/"
|
|
elif method == "flixpatrol_top":
|
|
return f"{urls['top10']}" \
|
|
f"{data['platform'].replace('_', '-')}/" \
|
|
f"{data['location'].replace('_', '-')}/" \
|
|
f"{util.time_window(data['time_window'])}/full/"
|
|
elif method == "flixpatrol_url":
|
|
return data
|
|
else:
|
|
raise Failed(f"FlixPatrol Error: Method {method} not supported")
|
|
|
|
def get_flixpatrol_ids(self, method, data, language, is_movie):
|
|
if method == "flixpatrol_demographics":
|
|
logger.info("Processing FlixPatrol Demographics:")
|
|
logger.info(f"\tGeneration: {generations_pretty[data['generation']]}")
|
|
logger.info(f"\tGender: {'All genders' if data['gender'] == 'all' else data['gender'].capitalize()}")
|
|
logger.info(f"\tLocation: {data['location'].replace('_', ' ').title()}")
|
|
logger.info(f"\tLimit: {data['limit']}")
|
|
elif method == "flixpatrol_popular":
|
|
logger.info("Processing FlixPatrol Popular:")
|
|
logger.info(f"\tSource: {data['source'].replace('_', ' ').title()}")
|
|
logger.info(f"\tTime Window: {data['time_window'].replace('_', ' ').title()}")
|
|
logger.info(f"\tLimit: {data['limit']}")
|
|
elif 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"\tTime Window: {data['time_window'].replace('_', ' ').title()}")
|
|
logger.info(f"\tLimit: {data['limit']}")
|
|
elif method == "flixpatrol_url":
|
|
logger.info(f"Processing FlixPatrol URL: {data}")
|
|
url = self.get_url(method, data, is_movie)
|
|
|
|
items = self._parse_list(url, language, is_movie, limit=data["limit"] if isinstance(data, dict) else 0)
|
|
media_type = "movie" if is_movie else "show"
|
|
total_items = len(items)
|
|
if total_items > 0:
|
|
ids = []
|
|
for i, item in enumerate(items, 1):
|
|
util.print_return(f"Finding TMDb ID {i}/{total_items}")
|
|
tmdb_id = None
|
|
expired = None
|
|
if self.config.Cache:
|
|
tmdb_id, expired = self.config.Cache.query_flixpatrol_map(item, media_type)
|
|
if not tmdb_id or expired is not False:
|
|
try:
|
|
tmdb_id = self._tmdb(f"{base_url}{item}", language)
|
|
except Failed as e:
|
|
logger.error(e)
|
|
continue
|
|
if self.config.Cache:
|
|
self.config.Cache.update_flixpatrol_map(expired, item, tmdb_id, media_type)
|
|
ids.append((tmdb_id, "tmdb" if is_movie else "tmdb_show"))
|
|
logger.info(util.adjust_space(f"Processed {total_items} TMDb IDs"))
|
|
return ids
|
|
else:
|
|
raise Failed(f"FlixPatrol Error: No List Items found in {data}")
|