Rare/rare/components/tabs/store/shop_models.py

import datetime
from dataclasses import dataclass
from typing import List, Dict
import epicstore_api.queries as egs_query


class ImageUrlModel:
    def __init__(
        self,
        front_tall: str = "",
        offer_image_tall: str = "",
        thumbnail: str = "",
        front_wide: str = "",
        offer_image_wide: str = "",
        product_logo: str = "",
    ):
        self.front_tall = front_tall
        self.offer_image_tall = offer_image_tall
        self.thumbnail = thumbnail
        self.front_wide = front_wide
        self.offer_image_wide = offer_image_wide
        self.product_logo = product_logo

    @classmethod
    def from_json(cls, json_data: list):
        tmp = cls()
        for item in json_data:
            if item["type"] == "Thumbnail":
                tmp.thumbnail = item["url"]
            elif item["type"] == "DieselStoreFrontTall":
                tmp.front_tall = item["url"]
            elif item["type"] == "DieselStoreFrontWide":
                tmp.front_wide = item["url"]
            elif item["type"] == "OfferImageTall":
                tmp.offer_image_tall = item["url"]
            elif item["type"] == "OfferImageWide":
                tmp.offer_image_wide = item["url"]
            elif item["type"] == "ProductLogo":
                tmp.product_logo = item["url"]
        return tmp


class ShopGame:
    # TODO: Copyrights etc
    def __init__(
        self,
        title: str = "",
        id: str = "",
        image_urls: ImageUrlModel = None,
        social_links: Dict = None,
        langs: Dict = None,
        reqs: Dict = None,
        publisher: str = "",
        developer: str = "",
        original_price: str = "",
        discount_price: str = "",
        tags: List = None,
        namespace: str = "",
        offer_id: str = "",
    ):
        self.title = title
        self.id = id
        self.image_urls = image_urls
        self.links = []
        if social_links:
            for item in social_links:
                if item.startswith("link"):
                    self.links.append(
                        tuple((item.replace("link", ""), social_links[item]))
                    )
        else:
            self.links = []
        self.languages = langs if langs is not None else {}
        self.reqs = reqs if reqs is not None else {}
        self.publisher = publisher
        self.developer = developer
        self.price = original_price
        self.discount_price = discount_price
        self.tags = tags if tags is not None else []
        self.namespace = namespace
        self.offer_id = offer_id

    @classmethod
    def from_json(cls, api_data: dict, search_data: dict):
        if isinstance(api_data, list):
            for product in api_data:
                if product["_title"] == "home":
                    api_data = product
                    break
        if "pages" in api_data.keys():
            for page in api_data["pages"]:
                if page["_slug"] == "home":
                    api_data = page
                    break
        tmp = cls()
        if search_data:
            tmp.title = search_data.get("title", "Fail")
            tmp.id = search_data.get("id")
            tmp.image_urls = ImageUrlModel.from_json(search_data["keyImages"])
            if not tmp.developer:
                for i in search_data["customAttributes"]:
                    if i["key"] == "developerName":
                        tmp.developer = i["value"]
            tmp.price = search_data["price"]["totalPrice"]["fmtPrice"]["originalPrice"]
            tmp.discount_price = search_data["price"]["totalPrice"]["fmtPrice"][
                "discountPrice"
            ]
            tmp.namespace = search_data["namespace"]
            tmp.offer_id = search_data["id"]

        if api_data:
            links = api_data["data"]["socialLinks"]
            tmp.links = []
            for item in links:
                if item.startswith("link"):
                    tmp.links.append(tuple((item.replace("link", ""), links[item])))
            tmp.available_voice_langs = api_data["data"]["requirements"].get(
                "languages", "Failed"
            )
            tmp.reqs = {}
            for i, system in enumerate(api_data["data"]["requirements"].get("systems", [])):
                try:
                    tmp.reqs[system["systemType"]] = {}
                except KeyError:
                    continue
                for req in system["details"]:
                    try:
                        tmp.reqs[system["systemType"]][req["title"]] = (
                            req["minimum"],
                            req["recommended"],
                        )
                    except KeyError:
                        pass
            tmp.publisher = api_data["data"]["meta"].get("publisher", "")
            tmp.developer = api_data["data"]["meta"].get("developer", "")
            tmp.tags = [
                i.replace("_", " ").capitalize()
                for i in api_data["data"]["meta"].get("tags", [])
            ]

        return tmp


@dataclass
class BrowseModel:
    category: str = "games/edition/base|bundles/games|editors|software/edition/base"
    count: int = 30
    language_code: str = "en"
    country_code: str = "US"
    keywords: str = ""
    sortDir: str = "DESC"
    start: int = 0
    tag: str = ""
    withMapping: bool = True
    withPrice: bool = True
    date: str = (
        f"[,{datetime.datetime.strftime(datetime.datetime.now(), '%Y-%m-%dT%H:%M:%S')}.420Z]"
    )
    price: str = ""
    onSale: bool = False

    def __post_init__(self):
        self.locale = f"{self.language_code}-{self.country_code}"

    @property
    def __dict__(self):
        payload = {
            "count": self.count,
            "category": self.category,
            "allowCountries": self.country_code,
            "namespace": "",
            "sortBy": "releaseDate",
            "sortDir": self.sortDir,
            "start": self.start,
            "keywords": self.keywords,
            "tag": self.tag,
            "priceRange": self.price,
            "releaseDate": self.date,
            "withPrice": self.withPrice,
            "locale": self.locale,
            "country": self.country_code,
        }
        if self.price == "free":
            payload["freeGame"] = True
            payload.pop("priceRange")
        elif self.price.startswith("<price>"):
            payload["priceRange"] = self.price.replace("<price>", "")
        if self.onSale:
            payload["onSale"] = True

        if self.price:
            payload["effectiveDate"] = self.date
        else:
            payload.pop("priceRange")
        return payload