add ShopCategory enum and update ShopData model to use it

staciax · Dec 29, 2024 · 58e5f0b · 58e5f0b
1 parent ed0f38d
commit 58e5f0b
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 3 deletions.
diff --git a/valorant/__init__.py b/valorant/__init__.py
@@ -24,6 +24,7 @@
     RelationType,
     RewardType,
     SeasonType,
+    ShopCategory,
     WeaponCategory,
 )
 from .errors import BadRequest, Forbidden, HTTPException, InternalServerError, RateLimited, ValorantError
@@ -44,6 +45,7 @@
     'RelationType',
     'RewardType',
     'SeasonType',
+    'ShopCategory',
     'ValorantError',
     'WeaponCategory',
     'models',

diff --git a/valorant/enums.py b/valorant/enums.py
@@ -34,6 +34,7 @@
     'RelationType',
     'RewardType',
     'SeasonType',
+    'ShopCategory',
     'WeaponCategory',
 )
 
@@ -113,6 +114,16 @@ class SeasonType(str, Enum):
     act = 'EAresSeasonType::Act'
 
 
+class ShopCategory(str, Enum):
+    armor = 'Armor'
+    pistols = 'Pistols'
+    smgs = 'SMGs'
+    shotguns = 'Shotguns'
+    heavy_weapons = 'Heavy Weapons'
+    rifles = 'Rifles'
+    sniper_rifles = 'Sniper Rifles'
+
+
 class WeaponCategory(str, Enum):
     melee = 'EEquippableCategory::Melee'
     sidearm = 'EEquippableCategory::Sidearm'

diff --git a/valorant/models/gear.py b/valorant/models/gear.py
@@ -26,6 +26,7 @@
 
 from pydantic import Field
 
+from ..enums import ShopCategory
 from .base import BaseModel, BaseUUIDModel, LocalizedField
 
 __all__ = (
@@ -42,7 +43,7 @@ class Detail(BaseModel):
 
 class ShopData(BaseModel):
     cost: int
-    category: str
+    category: ShopCategory
     shop_order_priority: int = Field(alias='shopOrderPriority')
     category_text: LocalizedField = Field(alias='categoryText')
     grid_position: Any = Field(alias='gridPosition')

diff --git a/valorant/models/weapons.py b/valorant/models/weapons.py
@@ -26,7 +26,7 @@
 
 from pydantic import Field
 
-from ..enums import WeaponCategory
+from ..enums import ShopCategory, WeaponCategory
 from .base import BaseModel, BaseUUIDModel, LocalizedField
 
 __all__ = (
@@ -91,7 +91,7 @@ class GridPosition(BaseModel):
 
 class ShopData(BaseModel):
     cost: int
-    category: str
+    category: ShopCategory
     shop_order_priority: int = Field(alias='shopOrderPriority')
     category_text: LocalizedField = Field(alias='categoryText')
     grid_position: GridPosition | None = Field(alias='gridPosition')