add SeasonType enum and update Season model

staciax · Dec 29, 2024 · 8ac952a · 8ac952a
1 parent 9490dc9
commit 8ac952a
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/valorant/__init__.py b/valorant/__init__.py
@@ -23,6 +23,7 @@
     MissionType,
     RelationType,
     RewardType,
+    SeasonType,
     WeaponCategory,
 )
 from .errors import BadRequest, Forbidden, HTTPException, InternalServerError, RateLimited, ValorantError
@@ -42,6 +43,7 @@
     'RateLimited',
     'RelationType',
     'RewardType',
+    'SeasonType',
     'ValorantError',
     'WeaponCategory',
     'models',

diff --git a/valorant/enums.py b/valorant/enums.py
@@ -33,6 +33,7 @@
     'MissionType',
     'RelationType',
     'RewardType',
+    'SeasonType',
     'WeaponCategory',
 )
 
@@ -108,6 +109,10 @@ class RewardType(str, Enum):
     spray = 'Spray'
 
 
+class SeasonType(str, Enum):
+    act = 'EAresSeasonType::Act'
+
+
 class WeaponCategory(str, Enum):
     melee = 'EEquippableCategory::Melee'
     sidearm = 'EEquippableCategory::Sidearm'

diff --git a/valorant/models/seasons.py b/valorant/models/seasons.py
@@ -30,6 +30,7 @@
 
 from pydantic import Field
 
+from ..enums import SeasonType
 from .base import BaseUUIDModel, LocalizedField
 
 __all__ = (
@@ -46,7 +47,7 @@ class Season(BaseUUIDModel):
     # uuid: str
     display_name: LocalizedField = Field(alias='displayName')
     title: LocalizedField | None
-    type: str | None
+    type: SeasonType | None
     start_time: datetime = Field(alias='startTime')
     end_time: datetime = Field(alias='endTime')
     parent_uuid: UUID | None = Field(alias='parentUuid')