mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-09-20 14:31:59 +00:00
58 lines
2.6 KiB
Python
58 lines
2.6 KiB
Python
import logging
|
|
from modules import util
|
|
from modules.util import Failed
|
|
from urllib.parse import urlparse
|
|
|
|
logger = logging.getLogger("Plex Meta Manager")
|
|
|
|
builders = ["mdblist_list"]
|
|
base_url = "https://mdblist.com/lists"
|
|
|
|
headers = {"User-Agent": "Plex-Meta-Manager"}
|
|
|
|
class Mdblist:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
|
|
def validate_mdblist_lists(self, mdb_lists):
|
|
valid_lists = []
|
|
for mdb_dict in util.get_list(mdb_lists, split=False):
|
|
if not isinstance(mdb_dict, dict):
|
|
mdb_dict = {"url": mdb_dict}
|
|
dict_methods = {dm.lower(): dm for dm in mdb_dict}
|
|
if "url" not in dict_methods:
|
|
raise Failed(f"Collection Error: mdb_list url attribute not found")
|
|
elif mdb_dict[dict_methods["url"]] is None:
|
|
raise Failed(f"Collection Error: mdb_list url attribute is blank")
|
|
else:
|
|
mdb_url = mdb_dict[dict_methods["url"]].strip()
|
|
if not mdb_url.startswith(base_url):
|
|
raise Failed(f"Mdblist Error: {mdb_url} must begin with: {base_url}")
|
|
list_count = None
|
|
if "limit" in dict_methods:
|
|
if mdb_dict[dict_methods["limit"]] is None:
|
|
logger.warning(f"Collection Warning: mdb_list limit attribute is blank using 0 as default")
|
|
else:
|
|
try:
|
|
value = int(str(mdb_dict[dict_methods["limit"]]))
|
|
if 0 <= value:
|
|
list_count = value
|
|
except ValueError:
|
|
pass
|
|
if list_count is None:
|
|
logger.warning(f"Collection Warning: mdb_list limit attribute must be an integer 0 or greater using 0 as default")
|
|
if list_count is None:
|
|
list_count = 0
|
|
valid_lists.append({"url": mdb_url, "limit": list_count})
|
|
return valid_lists
|
|
|
|
def get_mdblist_ids(self, method, data):
|
|
if method == "mdblist_list":
|
|
limit_status = f" Limit at: {data['limit']} items" if data['limit'] > 0 else ''
|
|
logger.info(f"Processing Mdblist.com List: {data['url']}{limit_status}")
|
|
parsed_url = urlparse(data["url"])
|
|
url_base = parsed_url._replace(query=None).geturl()
|
|
params = {"limit": data["limit"]} if data["limit"] > 0 else None
|
|
return [(i["imdb_id"], "imdb") for i in self.config.get_json(url_base, headers=headers, params=params)]
|
|
else:
|
|
raise Failed(f"Mdblist Error: Method {method} not supported")
|