Extended Enums for Levels and Models

1 files changed, 29 insertions, 3 deletions

diff --git a/windy.py b/windy.py
index 3938ed6..d7567ff 100644
--- a/windy.py
+++ b/windy.py
@@ -5,13 +5,39 @@ import requests
 API = "https://api.windy.com/api"
 
 
-class Model(Enum):
-	ICONEU = "iconEu"
-
+class _StrEnum(Enum):
 	def __str__(self):
 		return self.value
 
 
+class Model(_StrEnum):
+	AROME = "arome"
+	GEOS5 = "geos5"
+	GFS = "gfs"
+	GFSWAVE = "gfsWave"
+	ICONEU = "iconEu"
+	NAMALASKA = "namAlaska"
+	NAMCONUS = "namConus"
+	NAMHAWAII = "namHawaii"
+
+
+class Level(_StrEnum):
+	SURFACE = "surface"
+	H1000 = "1000h"
+	H950 = "950h"
+	H925 = "925h"
+	H900 = "900h"
+	H850 = "850h"
+	H800 = "800h"
+	H700 = "700h"
+	H600 = "600h"
+	H500 = "500h"
+	H400 = "400h"
+	H300 = "300h"
+	H200 = "200h"
+	H150 = "150h"
+
+
 class PointForecast:
 	ENDPOINT = "/point-forecast/v2"