Plex-Meta-Manager/modules/mdblist.py

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()
            url_base = url_base if url_base.endswith("/") else f"{url_base}/"
            url_base = url_base if url_base.endswith("json/") else f"{url_base}json/"
            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")