Ordered StrEnum members to allow sorting Levels

1 files changed, 23 insertions, 0 deletions

diff --git a/windy/point_forecast.py b/windy/point_forecast.py
index 743d85f..54a10c1 100644
--- a/windy/point_forecast.py
+++ b/windy/point_forecast.py
@@ -19,6 +19,29 @@ def _convert_notation(unit):
 
 
 class _StrEnum(Enum):
+	def __init__(self, value):
+		self._index = len(self.__class__.__members__)
+
+	def __lt__(self, other):
+		if self.__class__ is other.__class__:
+			return self._index < other._index
+		return NotImplemented
+
+	def __le__(self, other):
+		if self.__class__ is other.__class__:
+			return self._index <= other._index
+		return NotImplemented
+
+	def __gt__(self, other):
+		if self.__class__ is other.__class__:
+			return self._index > other._index
+		return NotImplemented
+
+	def __ge__(self, other):
+		if self.__class__ is other.__class__:
+			return self._index >= other._index
+		return NotImplemented
+
 	def __str__(self):
 		return self.value