From 25d822372e788687cbb9238caf7fb2a83264b109 Mon Sep 17 00:00:00 2001
From: David Bures <12524436+PiDiBi@users.noreply.github.com>
Date: Mon, 1 Jul 2024 11:13:00 -0700
Subject: [PATCH] weather alerts (#2)

---
 tests/test_weather_bot.py | 13 ++++++++++++
 weather_bot.py            | 43 +++++++++++++++++++++++++++++++++++++--
 2 files changed, 54 insertions(+), 2 deletions(-)

diff --git a/tests/test_weather_bot.py b/tests/test_weather_bot.py
index c67a98e..a0c923b 100644
--- a/tests/test_weather_bot.py
+++ b/tests/test_weather_bot.py
@@ -190,3 +190,16 @@ def test_get_hf_band_conditions(mock_interface):
     assert "20m" in hf
     assert "15m" in hf
     assert "10m" in hf
+
+@patch("meshtastic.serial_interface.SerialInterface")
+def test_get_wx_alerts(mock_interface):
+    # Mock the SerialInterface to avoid actual serial communication
+    mock_interface = MagicMock()
+    mock_interface.return_value = mock_interface
+    bot = WeatherBot(mock_interface)
+    lat = "37.7749"
+    lon = "-122.4194"
+    alerts = bot.get_wx_alerts(lat, lon)
+    print(f"alerts: {alerts}")
+    assert alerts != bot.NO_DATA_NOGPS
+    assert alerts != bot.ERROR_FETCHING_DATA
\ No newline at end of file
diff --git a/weather_bot.py b/weather_bot.py
index 6390563..223bd38 100644
--- a/weather_bot.py
+++ b/weather_bot.py
@@ -18,7 +18,7 @@ class WeatherBot(MessageProcessor):
 
     def __init__(self, interface: StreamInterface):
         super(WeatherBot, self).__init__(interface)
-        self.trap_list = ["sun", "solar", "hfcond", "tide", "moon", "wxc", "wx"]
+        self.trap_list = ["sun", "solar", "hfcond", "tide", "moon", "wxc", "wx", "alerts"]
         pass
 
     def auto_response(
@@ -42,6 +42,8 @@ def auto_response(
             bot_response = self.get_weather(str(location[0]), str(location[1]), 1)
         elif "wx" in message:
             bot_response = self.get_weather(str(location[0]), str(location[1]))
+        elif "alerts" in message:
+            bot_response = self.get_wx_alerts(str(location[0]), str(location[1]))
 
         return bot_response
 
@@ -64,7 +66,7 @@ def hf_band_conditions(self):
                     + "\n"
                 )
         else:
-            hf_cond += "error fetching"
+            hf_cond = self.ERROR_FETCHING_DATA
         hf_cond = hf_cond[:-1]  # remove the last newline
         return hf_cond
 
@@ -322,6 +324,42 @@ def get_weather(self, lat=0, lon=0, unit=0):
 
         return weather
 
+    def get_wx_alerts(self, lat=0, lon=0):
+        # get weather alerts from NOAA :: NOT IMPLEMENTED YET
+        alerts = ""
+        if float(lat) == 0 and float(lon) == 0:
+            return self.NO_DATA_NOGPS
+
+        # get weather alerts from NOAA
+        alert_url = "https://api.weather.gov/alerts/active.atom?point=" + str(lat) + "," + str(lon)
+        print(f"{log_timestamp()} System: {alert_url}")
+        
+        try:
+            alert_data = requests.get(alert_url, timeout=self.URL_TIMEOUT)
+            if not alert_data.ok:
+                return self.ERROR_FETCHING_DATA
+        except (requests.exceptions.RequestException):
+            return self.ERROR_FETCHING_DATA
+        
+        alerts = ""
+        alertxml = xml.dom.minidom.parseString(alert_data.text)
+        for i in alertxml.getElementsByTagName("entry"):
+            alerts += (
+                i.getElementsByTagName("updated")[0].childNodes[0].nodeValue
+                + ": "
+                + i.getElementsByTagName("title")[0].childNodes[0].nodeValue
+                + "\n"
+            )
+        
+        if alerts == "":
+            alerts = "No weather alerts found"
+
+        # trim off last newline
+        if alerts[-1] == "\n":
+            alerts = alerts[:-1]
+        
+        return alerts
+
     def replace_weather(self, row):
         replacements = {
             "Monday": "Mon ",
@@ -335,6 +373,7 @@ def replace_weather(self, row):
             "Tonight": "Tonight ",
             "Tomorrow": "Tomorrow ",
             "This Afternoon": "Afternoon ",
+            "Overnight": "Overnight ",
             "northwest": "NW",
             "northeast": "NE",
             "southwest": "SW",