import redis
import json

REDIS = redis.Redis(host="localhost", port=6379, decode_responses=True)

KEY = "echoweather:active_alerts"


def save_discord_message(alert_id: str, message_id: int):
    data = REDIS.hget(KEY, alert_id)
    if not data:
        return

    alert = json.loads(data)
    alert["discord_message_id"] = message_id

    REDIS.hset(KEY, alert_id, json.dumps(alert))


def get_alert(alert_id: str):
    data = REDIS.hget(KEY, alert_id)
    return json.loads(data) if data else None