catalog: init

Author: tomodachi94

roblox/catalog.py

@@ -0,0 +1,151 @@
+"""
+
+This module contains classes intended to parse and deal with data from Roblox catalog endpoint.
+
+"""
+from __future__ import annotations
+from datetime import datetime
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .client import Client
+    from typing import Optional
+from .bases.basecatalogitem import BaseCatalogItem
+from .bases.baseuser import BaseUser
+from .assets import AssetType
+from .partials.partialgroup import PartialGroup
+from .partials.partialuser import CatalogCreatorPartialUser
+
+
+item_type_names = {
+    1: "Asset",
+    2: "Bundle"
+}
+
+class CatalogItemType:
+    """
+    Represents a catalog item type.
+
+    Attributes:
+        id: Id of the item type
+        name: Name of the item type
+    """
+
+    def __init__(self, type_id: int):
+        """
+        Arguments:
+            type_id: The itemType ID to instantiate this CatalogItemType object with.
+                     This is used to determine the name of the CatalogItemType.
+        """
+
+        self.id: int = type_id
+        self.name: Optional[str] = item_type_names.get(type_id)
+
+    def __repr__(self):
+        return f"<{self.__class__.__name__} id={self.id} name={self.name!r}>"
+
+
+class CatalogItemPremiumPricing:
+    """
+    Represents a catalog item's pricing for Roblox Premium subscribers.
+
+    Attributes:
+        robux: Price of the
+        name: Name of the item type
+    """
+
+    def __init__(self, robux: int, percentage: int):
+        """
+        Arguments:
+            type_id: The itemType ID to instantiate this CatalogItemType object with.
+                     This is used to determine the name of the CatalogItemType.
+        """
+
+        self.robux: int = robux
+        self.percentage: int = percentage
+
+    def __repr__(self):
+        return f"<{self.__class__.__name__} robux={self.robux} percentage={self.percentage}>"
+
+
+
+class CatalogItem(BaseCatalogItem):
+    """
+    Represents a Catalog/Avatar Shop/Marketplace item.
+
+    Attributes:
+        id: The item's ID.
+        name: The item's name.
+        item_type: The item's type as an instance of CatalogItemType.
+        asset_type: The asset's type as an instance of AssetType
+        description: The item's description.
+        image_url: A link to the item's image.
+        creator: A class representing the creator of the item.
+        price: The price of the item, in Robux.
+    """
+
+    def __init__(self, client: Client, data: dict):
+        self._client: Client = client
+        self.id: int = data["id"]
+        self.item_type = CatalogItemType(data["itemType"])
+        super().__init__(client=self._client, catalog_item_id=self.id, catalog_item_type=self.item_type)
+
+        self.name: str = data["name"]
+        self.description: str = data["description"]
+
+        self.asset_type: AssetType = AssetType(type_id=data["assetType"])
+
+        self.is_offsale: bool = data["isOffsale"]
+
+        # Creator
+        self.creator: CatalogCreatorPartialUser or CatalogCreatorPartialGroup
+        if data["creatorType"] == "User":
+            self.creator = CatalogCreatorPartialUser(client=client, data=data)
+        elif data["creatorType"] == "Group":
+            self.creator = CatalogCreatorPartialGroup(client=client, group_id=data)
+
+        self.price: int = data["price"]
+        self.purchase_count: int = data["purchaseCount"]
+        self.favorite_count: int = data["favoriteCount"]
+        self.sale_location_type: str = data["saleLocationType"]
+
+
+
+        if data["premiumPricing"]:
+            self.premium_pricing = {}
+            self.premium_pricing.in_robux: int = data["premiumPricing"]["premiumPriceInRobux"]
+            self.premium_pricing.discount_percentage: int = data["premiumPricing"]["premiumDiscountPercentage"]
+
+
+    def __repr__(self):
+        return f"<{self.__class__.__name__} name={self.name!r}>"
+
+
+class LimitedCatalogItem(CatalogItem):
+    """
+    Represents a limited Catalog/Avatar Shop/Marketplace item.
+
+    Attributes:
+        id: The item's ID.
+        name: The item's name.
+        item_type: The item's type as an instance of CatalogItemType.
+        asset_type: The asset's type as an instance of AssetType
+        description: The item's description.
+        image_url: A link to the item's image.
+        creator: A class representing the creator of the item.
+        price: The price of the item, in Robux.
+    """
+
+    def __init__(self, client=client, data=data):
+        super.__init__(client=client, data=data)
+
+        self.collectible_item_id: str = data["collectibleItemId"]
+        self.quantity_limit_per_user: int = data["quantityLimitPerUser"]
+        self.units_available_for_consumption: int = data["unitsAvailableForConsumption"]
+        self.total_quantity: int = data["totalQuantity"]
+        self.has_resellers: bool = data["hasResellers"]
+        self.offsale_deadline: Optional[datetime] = datetime.fromtimestamp(data["offsaleDeadline"])
+        self.lowest_price: int = data["lowestPrice"]
+        self.lowest_resale_price: int = data["lowestResalePrice"]
+        self.price_status: str = data["priceStatus"]

roblox/client.py

@@ -4,7 +4,7 @@
 
 """
 
-from typing import Union, List, Optional
+from typing import Union, List, Optional, Literal, TypedDict
 
 from .account import AccountProvider
 from .assets import EconomyAsset
@@ -550,3 +550,47 @@ def get_base_gamepass(self, gamepass_id: int) -> BaseGamePass:
         Returns: A BaseGamePass.
         """
         return BaseGamePass(client=self, gamepass_id=gamepass_id)
+
+    # Catalog
+    def get_catalog_items(self, catalog_item_array: List[TypedDict[catalog_id: int, catalog_item_type: Literal[1, 2]]]) -> List[CatalogItem]:
+        """
+        Gets a catalog item with the passed ID.
+
+        The catalog is also known as the Avatar Shop or the Marketplace.
+
+        Arguments:
+            catalog_id: A Roblox catalog item ID.
+            catalog_item_type: The type of item. 1 for an asset, and 2 for a bundle.
+
+        Returns:
+            A list of CatalogItem.
+        """
+        try:
+            catalog_item_response = await self._requests.post(
+                url=self._url_generator.get_url(
+                    "catalog", "v1/catalog/items/details"
+                ),
+                data={"data": catalog_item_array}
+            )
+        except NotFound as exception:
+            raise CatalogItemNotFound(
+                message="Invalid catalog item.",
+                response=exception.response
+            ) from None
+        catalog_item_data = catalog_item_response.json()
+        catalog_list: Literal[CatalogItem] = []
+        for catalog_item in catalog_item_data:
+            if data["collectibleItemId"]: # This is the only consistent indicator of an item's limited status
+                catalog_list.append(LimitedCatalogItem(client=self, data=catalog_item))
+            else:
+                catalog_list.append(CatalogItem(client=self, data=catalog_item))
+
+        return catalog_list
+
+    def get_base_catalog_items(self, catalog_item_array: List[TypedDict[catalog_id: int, catalog_item_type: Literal[1, 2]]]) -> List[CatalogItem]:
+        catalog_list: Literal[CatalogItem] = []
+
+        for catalog_item in catalog_item_array:
+            catalog_list.append(BaseCatalogItem(client=self, data=catalog_item))
+
+        return catalog_list