audi-app/custom_components/audi_dashboard/historienimport.py

"""Nachträglicher Import vergangener Zeiträume aus dem Home-Assistant-Verlauf
(Einstellungen -> "Daten importieren aus Home Assistant").

WOZU
----
Fahrterkennung, Tankerkennung und Batterieverlauf arbeiten alle nur ab dem
Moment, in dem sie laufen. Alles, was das Fahrzeug gemeldet hat, BEVOR die App
lief (oder während Home Assistant aus war, oder bevor ein Sensor überhaupt
zugeordnet war), taucht in den Beständen der App deshalb nie auf - obwohl der
recorder es längst aufgezeichnet hat. Dieser Import schließt genau diese
Lücke: er liest denselben Verlauf, den die Live-Erkennung sonst in Echtzeit
sieht, und leitet daraus rückwirkend dieselben Datensätze ab.

GRENZE, DIE MAN KENNEN MUSS
---------------------------
Weiter zurück als der recorder aufbewahrt, geht es nicht - was dort gelöscht
ist, ist endgültig weg. Home Assistant löscht standardmäßig nach 10 Tagen;
recorder_snippet.yaml hebt das auf ein Jahr an. Der Import meldet deshalb im
Ergebnis mit, ab wann im gewählten Zeitraum überhaupt Daten vorlagen
(`ab_wann_daten`), damit ein leeres Ergebnis nicht wie ein Fehler aussieht.

DOPPELTE DATENSÄTZE
-------------------
Der Import ist absichtlich mehrfach ausführbar (überlappende Zeiträume,
zweiter Versuch nach einem Abbruch): jede erzeugte Fahrt wird gegen die
bereits vorhandenen geprüft und übersprungen, wenn sich ihr Zeitraum mit einer
bestehenden Fahrt überschneidet - egal ob die live erkannt, von Hand angelegt
oder aus einem früheren Import stammt. Tankvorgänge werden über ein
Zeitfenster (TANK_DUBLETTE_MIN) entdoppelt, Batteriewerte über den Tag als
Schlüssel (dort führt die Ablage min/max ohnehin zusammen, statt Zeilen zu
vervielfachen).

Erzeugte Datensätze tragen `source: "import"` - dieselbe Rolle wie "ha" (live
erkannt), "manual" (von Hand) und "auto" (Tankerkennung), damit später
nachvollziehbar bleibt, woher ein Eintrag stammt.
"""

from __future__ import annotations

import datetime
import logging
from typing import TYPE_CHECKING

from .fahrterkennung import leere_fahrt, pausenzeit_sekunden
from .tankerkennung import leerer_tankvorgang, schwelle_prozent
from .verlauf import Verlaufspunkt, verlauf_lesen, wert_bei, zahl

if TYPE_CHECKING:
    from .koordinator import Koordinator

_LOGGER = logging.getLogger(__name__)

# Zwei Tankvorgänge innerhalb dieser Spanne gelten als derselbe - schützt
# gegen Dubletten, wenn derselbe Zeitraum zweimal importiert wird oder sich
# Import und Live-Erkennung am Rand überschneiden.
TANK_DUBLETTE_MIN = 90

# Fahrten, die kürzer sind, sind Zündung-an-ohne-Fahrt (Radio, Tür öffnen mit
# Zündung, Diagnose) - die Live-Erkennung legt sie zwar an, im Rückblick
# fluten sie den Bestand aber mit Nulleinträgen. Bewusst konservativ.
MINDESTDAUER_S = 60


def als_zeit(wert: object) -> datetime.datetime | None:
    """Akzeptiert, was die Oberfläche schickt: ISO mit oder ohne Zeitzone.

    Ohne Zeitzone gilt die lokale Zeit von Home Assistant - der Nutzer wählt
    im Formular schließlich Ortszeit, keine UTC."""
    if not wert:
        return None
    ts = datetime.datetime.fromisoformat(str(wert))
    if ts.tzinfo is None:
        ts = ts.astimezone()
    return ts.astimezone(datetime.UTC)


def _fahrtfenster(
    zuendung_verlauf: list[Verlaufspunkt], pausenzeit_s: int
) -> list[tuple[datetime.datetime, datetime.datetime]]:
    """Aus dem Zündungsverlauf die Zeiträume, in denen gefahren wurde.

    Zwei Schritte, die zusammen die Pausenregel aus §7.1 nachbilden: erst
    jeden zusammenhängenden "on"-Abschnitt sammeln, dann benachbarte
    Abschnitte verschmelzen, deren Lücke kürzer als die Pausenzeit ist. Genau
    das tut die Live-Erkennung über die abbrechbare Wartezeit, nur eben im
    Nachhinein und ohne Warten."""
    roh: list[tuple[datetime.datetime, datetime.datetime]] = []
    offen: datetime.datetime | None = None
    for ts, wert in zuendung_verlauf:
        an = str(wert).lower() in ("on", "true", "1")
        if an and offen is None:
            offen = ts
        elif not an and offen is not None:
            roh.append((offen, ts))
            offen = None
    # Ein am Ende des Zeitraums noch offener Abschnitt wird verworfen: die
    # Fahrt ist zu diesem Zeitpunkt noch nicht beendet, ihr Ende läge hinter
    # dem gewählten Fenster. Sie beim Fensterende abzuschneiden würde eine
    # Fahrt mit erfundener Endzeit erzeugen.

    if not roh:
        return []

    verschmolzen = [roh[0]]
    for start, ende in roh[1:]:
        vorheriger_start, vorheriges_ende = verschmolzen[-1]
        if (start - vorheriges_ende).total_seconds() < pausenzeit_s:
            verschmolzen[-1] = (vorheriger_start, ende)
        else:
            verschmolzen.append((start, ende))
    return verschmolzen


def _ueberschneidet(
    start: datetime.datetime, ende: datetime.datetime, bestehende: list[dict]
) -> bool:
    """True, wenn sich [start, ende] mit einer bereits erfassten Fahrt
    überschneidet. Verhindert Dubletten beim wiederholten Import."""
    for f in bestehende:
        try:
            f_start = datetime.datetime.fromisoformat(f.get("ts_start"))  # type: ignore[arg-type]
            f_ende = datetime.datetime.fromisoformat(f.get("ts_end"))  # type: ignore[arg-type]
        except (TypeError, ValueError):
            continue
        if f_start.tzinfo is None or f_ende.tzinfo is None:
            continue
        if start < f_ende and f_start < ende:
            return True
    return False


async def _fahrten_importieren(k: Koordinator, verlaeufe: dict) -> dict:
    """Fahrten aus dem Zündungsverlauf, mit Kilometerstand und Start-/
    Zielkoordinaten aus den übrigen Verläufen ergänzt."""
    fenster = _fahrtfenster(verlaeufe["zuendung"], await pausenzeit_sekunden(k))
    if not fenster:
        return {"angelegt": 0, "uebersprungen": 0, "zu_kurz": 0}

    bestehende = await k.ablage.fahrten_lesen()
    km_verlauf = verlaeufe["km"]
    lat_verlauf = verlaeufe["lat"]
    lon_verlauf = verlaeufe["lon"]

    angelegt = uebersprungen = zu_kurz = 0
    neue: list[dict] = []

    for f_start, f_ende in fenster:
        dauer_s = int((f_ende - f_start).total_seconds())
        if dauer_s < MINDESTDAUER_S:
            zu_kurz += 1
            continue
        if _ueberschneidet(f_start, f_ende, bestehende):
            uebersprungen += 1
            continue

        odo_start = wert_bei(km_verlauf, f_start)
        odo_end = wert_bei(km_verlauf, f_ende)
        distanz = None
        if odo_start is not None and odo_end is not None and odo_end >= odo_start:
            distanz = round(odo_end - odo_start, 1)

        durchschnitt = None
        if distanz is not None and dauer_s > 0:
            durchschnitt = round(distanz / (dauer_s / 3600.0), 1)

        fahrt = leere_fahrt(f_start, f_ende, "import")
        fahrt.update({
            "distance_km": distanz,
            "km_quelle": "sensor" if distanz is not None else None,
            "odo_start": odo_start,
            "odo_end": odo_end,
            "avg_speed_kmh": durchschnitt,
            "start_lat": wert_bei(lat_verlauf, f_start),
            "start_lon": wert_bei(lon_verlauf, f_start),
            "end_lat": wert_bei(lat_verlauf, f_ende),
            "end_lon": wert_bei(lon_verlauf, f_ende),
            "status": "vollständig" if distanz is not None else "offen",
        })
        neue.append(fahrt)
        bestehende.append(fahrt)
        angelegt += 1

    await k.ablage.fahrten_ergaenzen(neue)
    return {"angelegt": angelegt, "uebersprungen": uebersprungen, "zu_kurz": zu_kurz}


async def _tankvorgaenge_importieren(k: Koordinator, verlaeufe: dict) -> dict:
    """Tankvorgänge aus dem Füllstandsverlauf - dieselbe Tiefststand-Logik wie
    in der Live-Erkennung: jeder Anstieg über die Schwelle gegen den zuletzt
    gesehenen Tiefststand ist ein Tankvorgang, nicht jeder Anstieg gegen den
    unmittelbar vorherigen Wert."""
    verlauf = verlaeufe["tank"]
    if not verlauf:
        return {"angelegt": 0, "uebersprungen": 0}

    schwelle = schwelle_prozent(await k.ablage.profil_lesen())
    km_verlauf = verlaeufe["km"]
    bestehende = await k.ablage.tankvorgaenge_lesen()
    fenster_s = TANK_DUBLETTE_MIN * 60

    bekannte_zeiten: list[datetime.datetime] = []
    for t in bestehende:
        try:
            ts = datetime.datetime.fromisoformat(t.get("ts"))  # type: ignore[arg-type]
        except (TypeError, ValueError):
            continue
        if ts.tzinfo is not None:
            bekannte_zeiten.append(ts)

    angelegt = uebersprungen = 0
    neue: list[dict] = []
    tiefststand: float | None = None

    for ts, wert in verlauf:
        aktuell = zahl(wert)
        if aktuell is None:
            continue
        if tiefststand is None or aktuell <= tiefststand:
            tiefststand = aktuell
            continue
        if aktuell - tiefststand < schwelle:
            continue

        if any(abs((bekannt - ts).total_seconds()) < fenster_s for bekannt in bekannte_zeiten):
            uebersprungen += 1
            tiefststand = aktuell
            continue

        odometer_km = wert_bei(km_verlauf, ts)
        tankvorgang = leerer_tankvorgang(ts.isoformat(), "import")
        tankvorgang["odometer_km"] = odometer_km
        tankvorgang["distance_km"] = _distanz_zum_vorherigen(
            ts.isoformat(), odometer_km, neue + bestehende
        )
        neue.append(tankvorgang)
        bekannte_zeiten.append(ts)
        angelegt += 1
        tiefststand = aktuell

    await k.ablage.tankvorgaenge_ergaenzen(neue)
    return {"angelegt": angelegt, "uebersprungen": uebersprungen}


def _distanz_zum_vorherigen(
    eigene_ts: str, odometer_km: float | None, kandidaten: list[dict]
) -> float | None:
    """"Gefahren seit der letzten Tankung" heißt: seit der letzten Tankung VOR
    dieser hier - nicht seit der zeitlich jüngsten überhaupt.

    Beim Import eines vergangenen Zeitraums liegen im Bestand regelmäßig
    bereits neuere Tankvorgänge; die als Bezug zu nehmen ergäbe eine negative
    Strecke (und damit, nach der Prüfung unten, gar keine)."""
    if odometer_km is None:
        return None
    vorheriger: tuple[str, float] | None = None
    for t in kandidaten:
        t_ts = t.get("ts") or ""
        if t.get("odometer_km") is None or not t_ts or t_ts >= eigene_ts:
            continue
        if vorheriger is None or t_ts > vorheriger[0]:
            vorheriger = (t_ts, t["odometer_km"])
    if vorheriger is None:
        return None
    distanz = round(odometer_km - vorheriger[1], 1)
    return None if distanz < 0 else distanz


async def _batterie_importieren(k: Koordinator, verlaeufe: dict) -> dict:
    """Tagesminimum/-maximum der 12V-Spannung je Tag des Zeitraums.

    Die Ablage führt bestehende und neue Werte pro Tag zusammen (min bleibt
    min, max bleibt max), deshalb braucht es hier keine eigene
    Dubletten-Prüfung: ein zweiter Import desselben Zeitraums verändert die
    Einträge nicht mehr."""
    verlauf = verlaeufe["batterie"]
    if not verlauf:
        return {"tage": 0}

    tage: dict[str, dict] = {}
    for ts, wert in verlauf:
        spannung = zahl(wert)
        if spannung is None:
            continue
        tag = ts.date().isoformat()
        eintrag = tage.get(tag)
        if eintrag is None:
            tage[tag] = {"min": spannung, "min_ts": ts, "max": spannung, "max_ts": ts}
            continue
        if spannung < eintrag["min"]:
            eintrag["min"] = spannung
            eintrag["min_ts"] = ts
        if spannung > eintrag["max"]:
            eintrag["max"] = spannung
            eintrag["max_ts"] = ts

    for tag in sorted(tage):
        werte = tage[tag]
        await k.ablage.batterieverlauf_tageswert_aktualisieren(
            tag, werte["min_ts"].isoformat(), werte["min"]
        )
        await k.ablage.batterieverlauf_tageswert_aktualisieren(
            tag, werte["max_ts"].isoformat(), werte["max"]
        )

    return {"tage": len(tage)}


async def importieren(k: Koordinator, start: object, ende: object) -> None:
    """Liest den Verlauf im gewählten Zeitraum und leitet daraus Fahrten,
    Tankvorgänge und Batteriewerte ab.

    Der Fortschritt geht über die Entität sensor.audi_dashboard_import_status
    an die Oberfläche - ein Dienstaufruf kehrt sofort zurück und kann kein
    Ergebnis mitbringen."""
    await k.ablage.ordner_sicherstellen()

    try:
        von = als_zeit(start)
        bis = als_zeit(ende)
    except ValueError as fehler:
        _LOGGER.error("Import mit unlesbarem Zeitraum aufgerufen (%s)", fehler)
        await k.import_status_veroeffentlichen("fehler", {"meldung": "Zeitraum nicht lesbar"})
        return

    if von is None or bis is None or von >= bis:
        _LOGGER.warning("Import ohne gültigen Zeitraum aufgerufen")
        await k.import_status_veroeffentlichen("fehler", {
            "meldung": "Bitte einen Zeitraum wählen, dessen Ende nach dem Start liegt."
        })
        return

    await k.import_status_veroeffentlichen(
        "laeuft", {"von": von.isoformat(), "bis": bis.isoformat()}
    )
    _LOGGER.info("Import gestartet für %s bis %s", von.isoformat(), bis.isoformat())

    werte = k.zuordnung.werte
    verlaeufe = {
        "zuendung": await verlauf_lesen(k.hass, werte.ZUENDUNG_SENSOR, von, bis),
        "km": await verlauf_lesen(k.hass, werte.KM_SENSOR, von, bis),
        "tank": await verlauf_lesen(k.hass, werte.TANK_SENSOR, von, bis),
        "batterie": await verlauf_lesen(k.hass, werte.BATTERIE_SENSOR, von, bis),
        "lat": await verlauf_lesen(k.hass, werte.STANDORT_LAT_SENSOR, von, bis),
        "lon": await verlauf_lesen(k.hass, werte.STANDORT_LON_SENSOR, von, bis),
    }

    # Frühester Zeitpunkt, zu dem im gewählten Fenster überhaupt etwas
    # aufgezeichnet war - damit ein leeres Ergebnis erklärbar wird
    # ("recorder reicht nur bis ...") statt wie ein Fehler auszusehen.
    frueheste = None
    for punkte in verlaeufe.values():
        if punkte and (frueheste is None or punkte[0][0] < frueheste):
            frueheste = punkte[0][0]

    fahrten = await _fahrten_importieren(k, verlaeufe)
    tank = await _tankvorgaenge_importieren(k, verlaeufe)
    batterie = await _batterie_importieren(k, verlaeufe)

    await k.fahrten_veroeffentlichen()
    await k.tankvorgaenge_veroeffentlichen()
    await k.batterieverlauf_veroeffentlichen()

    ergebnis = {
        "von": von.isoformat(),
        "bis": bis.isoformat(),
        "ab_wann_daten": frueheste.isoformat() if frueheste else None,
        "fahrten_angelegt": fahrten["angelegt"],
        "fahrten_uebersprungen": fahrten["uebersprungen"],
        "fahrten_zu_kurz": fahrten["zu_kurz"],
        "tankvorgaenge_angelegt": tank["angelegt"],
        "tankvorgaenge_uebersprungen": tank["uebersprungen"],
        "batterie_tage": batterie["tage"],
    }
    await k.import_status_veroeffentlichen("fertig", ergebnis)
    _LOGGER.info("Import abgeschlossen - %s", ergebnis)