Starting to implement alerts #17

alertproviders/pota.py

@@ -0,0 +1,39 @@
+from datetime import datetime
+
+import pytz
+
+from alertproviders.http_alert_provider import HTTPAlertProvider
+from data.alert import Alert
+
+
+# Alert provider for Parks on the Air
+class POTA(HTTPAlertProvider):
+    POLL_INTERVAL_SEC = 3600
+    ALERTS_URL = "https://api.pota.app/activation"
+
+    def __init__(self, provider_config):
+        super().__init__(provider_config, self.ALERTS_URL, self.POLL_INTERVAL_SEC)
+
+    def http_response_to_alerts(self, http_response):
+        new_alerts = []
+        # Iterate through source data
+        for source_alert in http_response.json():
+            # Convert to our alert format
+            alert = Alert(source=self.name,
+                          source_id=source_alert["scheduledActivitiesId"],
+                          dx_call=source_alert["activator"].upper(),
+                          freqs_modes=source_alert["frequencies"],
+                          comment=source_alert["comments"],
+                          sig="POTA",
+                          sig_refs=[source_alert["reference"]],
+                          sig_refs_names=[source_alert["name"]],
+                          icon="tree",
+                          start_time=datetime.strptime(source_alert["startDate"] + source_alert["startTime"],
+                                                       "%Y-%m-%d%H:%M").replace(tzinfo=pytz.UTC).timestamp(),
+                          end_time=datetime.strptime(source_alert["endDate"] + source_alert["endTime"],
+                                                     "%Y-%m-%d%H:%M").replace(tzinfo=pytz.UTC).timestamp())
+
+            # Add to our list. Don't worry about de-duping, removing old alerts etc. at this point; other code will do
+            # that for us.
+            new_alerts.append(alert)
+        return new_alerts