147 lines
5.7 KiB
Python
147 lines
5.7 KiB
Python
import datetime
|
|
import random
|
|
from dataclasses import dataclass
|
|
|
|
|
|
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 = "", image_urls: ImageUrlModel = None, social_links: dict = None,
|
|
langs: list = 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.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
|
|
self.reqs = reqs
|
|
self.publisher = publisher
|
|
self.developer = developer
|
|
self.price = original_price
|
|
self.discount_price = discount_price
|
|
self.tags = tags
|
|
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()
|
|
tmp.title = search_data.get("title", "Fail")
|
|
tmp.image_urls = ImageUrlModel.from_json(search_data["keyImages"])
|
|
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", "")
|
|
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.tags = [i.replace("_", " ").capitalize() for i in api_data["data"]["meta"].get("tags", [])]
|
|
tmp.namespace = search_data["namespace"]
|
|
tmp.offer_id = search_data["id"]
|
|
|
|
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%X')}.{str(random.randint(0, 999)).zfill(3)}Z]"
|
|
price: str = ""
|
|
onSale: bool = False
|
|
|
|
@property
|
|
def __dict__(self):
|
|
payload = {"category": self.category,
|
|
"count": self.count,
|
|
"country": self.country_code,
|
|
"keywords": self.keywords,
|
|
"locale": self.language_code + "-" + self.country_code,
|
|
"sortDir": self.sortDir,
|
|
"allowCountries": self.country_code,
|
|
"start": self.start,
|
|
"tag": self.tag,
|
|
"withMapping": self.withMapping,
|
|
"withPrice": self.withPrice,
|
|
"releaseDate": self.date,
|
|
"effectiveDate": self.date,
|
|
}
|
|
|
|
if self.price == "free":
|
|
payload["freeGame"] = True
|
|
elif self.price.startswith("<price>"):
|
|
payload["priceRange"] = self.price.replace("<price>", "")
|
|
if self.onSale:
|
|
payload["onSale"] = True
|
|
return payload
|