|
|
|
|
from modules import util
|
|
|
|
|
from modules.util import Failed
|
|
|
|
|
|
|
|
|
|
logger = util.logger
|
|
|
|
|
|
|
|
|
|
builders = ["reciperr_list", "stevenlu_popular"]
|
|
|
|
|
|
|
|
|
|
stevenlu_url = "https://s3.amazonaws.com/popular-movies/movies.json"
|
|
|
|
|
|
|
|
|
|
class Reciperr:
|
|
|
|
|
def __init__(self, config):
|
|
|
|
|
self.config = config
|
|
|
|
|
|
|
|
|
|
def _request(self, url, name="Reciperr"):
|
|
|
|
|
response = self.config.get(url)
|
|
|
|
|
if response.status_code >= 400:
|
|
|
|
|
raise Failed(f"{name} Error: JSON not found at {url}")
|
|
|
|
|
return response.json()
|
|
|
|
|
|
|
|
|
|
def validate_list(self, data):
|
|
|
|
|
valid_lists = []
|
|
|
|
|
for reciperr_list in util.get_list(data, split=False):
|
|
|
|
|
if "imdb_id" not in self._request(reciperr_list)[0]:
|
|
|
|
|
raise Failed(f"Reciperr Error: imdb_id not found in the JSON at {reciperr_list}")
|
|
|
|
|
valid_lists.append(reciperr_list)
|
|
|
|
|
return valid_lists
|
|
|
|
|
|
|
|
|
|
def get_imdb_ids(self, method, data):
|
|
|
|
|
name = "StevenLu" if method == "stevenlu_popular" else "Reciperr"
|
|
|
|
|
logger.info(f"Processing {name} Movies")
|
|
|
|
|
if method == "reciperr_list":
|
|
|
|
|
ids = [(i["imdb_id"], "imdb") for i in self._request(data)]
|
|
|
|
|
elif method == "stevenlu_popular":
|
|
|
|
|
ids = [(i["imdb_id"], "imdb") for i in self._request(stevenlu_url, name="StevenLu")]
|
|
|
|
|
else:
|
|
|
|
|
raise Failed(f"Config Error: Method {method} not supported")
|
|
|
|
|
if not ids:
|
|
|
|
|
raise Failed(f"{name} Error: No IDs found.")
|
|
|
|
|
return ids
|
