spothole/providers/alert/hamsat.py

from datetime import datetime

import pytz

from core.enums import AlertType
from data.alert import Alert
from data.sig_ref import SIGRef
from providers.alert.http_alert_provider import HTTPAlertProvider


class Hamsat(HTTPAlertProvider):
    """Alert provider for Hamsat (hams.at)"""

    POLL_INTERVAL_SEC = 1800
    ALERTS_URL = "https://hams.at/api/alerts"

    def __init__(self, provider_config):
        super().__init__("Hamsat", 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()["data"]:
            # Convert to our alert format
            alert = Alert(
                source=self.name,
                source_id=source_alert["id"],
                dx_calls=[source_alert["callsign"].upper()],
                freqs_modes=f"{source_alert['mhz']!s} {source_alert['mhz_direction']}, {source_alert['mode']}",
                comment=source_alert["comment"],
                # Fudge a SIG ref to provide the remaining bits of data we need: the satellite and the operator's grid
                sig_refs=[
                    SIGRef(
                        sig="AMSAT",
                        id=f"{source_alert['satellite']['name']} from {source_alert['grids'][0]}",
                    )
                ],
                start_time=datetime.strptime(source_alert["aos_at"], "%Y-%m-%dT%H:%M:%SZ")
                .replace(tzinfo=pytz.UTC)
                .timestamp(),
                end_time=datetime.strptime(source_alert["los_at"], "%Y-%m-%dT%H:%M:%SZ")
                .replace(tzinfo=pytz.UTC)
                .timestamp(),
                alert_type=AlertType.SATELLITE,
            )

            # Add to our list
            new_alerts.append(alert)
        return new_alerts