Set Stookalert to boolean

fwestenberg · Nov 3, 2023 · 3de6902 · 3de6902
1 parent ad6c764
commit 3de6902
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 setup(
   name = 'stookwijzer',
   packages = ['stookwijzer'],
-  version = '1.4.0',
+  version = '1.4.1',
   license='MIT',
   description = 'Stookwijzer package',
   author = 'fwestenberg',

diff --git a/stookwijzer/example.py b/stookwijzer/example.py
@@ -9,4 +9,5 @@
     print(sw.windspeed_bft)
     print(sw.windspeed_ms)
     print(sw.lki)
-    print(sw.forecast)
+    print(sw.forecast)
+
diff --git a/stookwijzer/stookwijzerapi.py b/stookwijzer/stookwijzerapi.py
@@ -55,8 +55,7 @@ def forecast(self):
 
         for offset in range(2, 25, 2):
             forecast.append(self.get_forecast_at_offset(localdt, offset))
-
-        return {"forecast": forecast}
+        return forecast
 
     @property
     def last_updated(self):
@@ -70,15 +69,15 @@ def update(self):
         advice = self.get_property("advies_0")
         if advice:
             self._state = self.get_color(advice)
-            self._alert = self.get_property("alert_0")
+            self._alert = self.get_property("alert_0") == '1'
             self._last_updated = datetime.now()
 
     def get_forecast_at_offset(self, runtime: datetime, offset: int) -> dict:
         """Get forecast at a certain offset."""
         return {
             "datetime": (runtime + timedelta(hours=offset)).isoformat(),
             "advice": self.get_color(self.get_property("advies_" + str(offset))),
-            "alert": self.get_property("alert_" + str(offset)),
+            "alert": self.get_property("alert_" + str(offset))  == '1',
         }
 
     def get_boundary_box(self, latitude: float, longitude: float) -> str: