feat: implemented Season.to_str() method

201st-Luka · Aug 25, 2023 · e46aafa · e46aafa
1 parent 0418407
commit e46aafa
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/pyclasher/api/models/season.py b/pyclasher/api/models/season.py
@@ -21,3 +21,13 @@ def from_str(cls, season):
  year = int(season[0])
  month = int(season[1])
  return cls(year, month)
+
+ def to_str(self):
+ return f"{self.year}-{self.month}"
+
+ def __eq__(self, other):
+ if isinstance(other, Season):
+ return self.year == other.year and self.month == other.month
+ if isinstance(other, str):
+ return self == Season.from_str(other)
+ raise NotImplementedError
diff --git a/pyclasher/api/models/season.pyi b/pyclasher/api/models/season.pyi
@@ -13,3 +13,6 @@ class Season:
  @classmethod
  def from_str(cls, season: str):
  ...
+
+ def to_str(self) -> str:
+ ...