ilæg_nye_punkter.py

import getpass
import math
from itertools import chain
from typing import List

import click
import pandas as pd

import fire.cli
from fire import uuid
from fire.api.model import (
    GeometriObjekt,
    Point,
    Punkt,
    PunktInformation,
    Sag,
    Sagsevent,
    SagseventInfo,
    EventType,
    FikspunktsType,
)

from . import (
    ARKDEF_NYETABLEREDE_PUNKTER,
    bekræft2,
    find_faneblad,
    find_sag,
    find_sagsgang,
    niv,
    normaliser_placeringskoordinat,
    skriv_ark,
)


@niv.command()
@fire.cli.default_options()
@click.argument(
    "projektnavn",
    nargs=1,
    type=str,
)
@click.option(
    "--sagsbehandler",
    default=getpass.getuser(),
    type=str,
    help="Angiv andet brugernavn end den aktuelt indloggede",
)
def ilæg_nye_punkter(projektnavn: str, sagsbehandler: str, **kwargs) -> None:
    """Registrer nyoprettede punkter i databasen"""
    sag = find_sag(projektnavn)
    sagsgang = find_sagsgang(projektnavn)

    fire.cli.print(f"Sags/projekt-navn: {projektnavn}  ({sag.id})")
    fire.cli.print(f"Sagsbehandler:     {sagsbehandler}")

    # Opbyg oversigt over nyetablerede punkter
    nyetablerede = find_faneblad(
        projektnavn, "Nyetablerede punkter", ARKDEF_NYETABLEREDE_PUNKTER
    )
    n = nyetablerede.shape[0]

    if n == 0:
        fire.cli.print("Ingen nyetablerede punkter at registrere")
        return

    landsnummer_pit = fire.cli.firedb.hent_punktinformationtype("IDENT:landsnr")
    beskrivelse_pit = fire.cli.firedb.hent_punktinformationtype("ATTR:beskrivelse")
    h_over_terræn_pit = fire.cli.firedb.hent_punktinformationtype(
        "AFM:højde_over_terræn"
    )
    assert landsnummer_pit is not None, "IDENT:landsnr ikke fundet i database"
    assert beskrivelse_pit is not None, "ATTR:beskrivelse ikke fundet i database"
    assert h_over_terræn_pit is not None, "AFM:højde_over_terræn ikke fundet i database"

    punkter = {}
    fikspunktstyper = []
    punktinfo = []

    # Opret punkter
    fire.cli.print(f"Behandler punkter")
    for i in range(n):
        # Et tomt tekstfelt kan repræsenteres på en del forskellige måder...
        # Punkter udstyret med uuid er allerede registrerede
        if str(nyetablerede.uuid[i]) not in ["", "None", "nan"]:
            continue

        lokation = normaliser_placeringskoordinat(
            nyetablerede["Øst"][i], nyetablerede["Nord"][i]
        )

        # Skab nyt punktobjekt
        punkter[i] = Punkt(
            id=uuid(),
            geometriobjekter=[GeometriObjekt(geometri=Point(lokation))],
        )

        # TODO: Alle punkter er højdefikspunkter indtil videre...
        fikspunktstyper.append(FikspunktsType.HØJDE)

    # sagsevent for punkter
    er = "er" if len(punkter) > 1 else ""
    sagsevent_punkter = Sagsevent(
        sag=sag,
        eventtype=EventType.PUNKT_OPRETTET,
        sagseventinfos=[
            SagseventInfo(
                beskrivelse=f"Oprettelse af punkt{er} ifm. {projektnavn}",
            )
        ],
        punkter=list(punkter.values()),
    )
    fire.cli.firedb.indset_sagsevent(sagsevent_punkter, commit=False)
    fire.cli.firedb.session.flush()  # hvis noget ikke virker får vi fejl her!

    # Generer dokumentation til fanebladet "Sagsgang"
    sagsgangslinje = {
        "Dato": sagsevent_punkter.registreringfra,
        "Hvem": sagsbehandler,
        "Hændelse": "punktoprettelse",
        "Tekst": sagsevent_punkter.sagseventinfos[0].beskrivelse,
        "uuid": sagsevent_punkter.id,
    }
    sagsgang = sagsgang.append(sagsgangslinje, ignore_index=True)

    # Opret punktinfo
    fire.cli.print(f"Behandler punktinformationer")
    for i, punkt in punkter.items():
        # Tilføj punktets højde over terræn som punktinformation, hvis anført
        try:
            ΔH = float(nyetablerede["Højde over terræn"][i])
        except (TypeError, ValueError):
            ΔH = 0
        if math.isnan(ΔH):
            ΔH = 0.0
        if not pd.isna(nyetablerede["Højde over terræn"][i]):
            pi_h = PunktInformation(
                infotype=h_over_terræn_pit,
                punkt=punkt,
                tal=ΔH,
            )
            punktinfo.append(pi_h)

        # Tilføj punktets afmærkning som punktinformation, selv hvis ikke anført
        afm_id = 4999  # AFM:4999 = "ukendt"
        afm_ids = {
            "ukendt": 4999,
            "bolt": 2700,
            "lodret bolt": 2701,
            "skruepløk": 2950,
            "ingen": 5998,
        }

        if not pd.isna(nyetablerede["Afmærkning"][i]):
            # Afmærkningsbeskrivelse
            afm = str(nyetablerede["Afmærkning"][i]).lower()
            # Første ord i afmærkningsbeskrivelsen
            afm_første = afm.split()[0].rstrip(":;,.- ").lstrip("afm:")

            if afm_første.isnumeric():
                afm_id = int(afm_første)
            else:
                afm_id = afm_ids.get(afm, 4999)

            afmærkning_pit = fire.cli.firedb.hent_punktinformationtype(f"AFM:{afm_id}")
            if afmærkning_pit is None:
                afm_id = 4999
                afmærkning_pit = fire.cli.firedb.hent_punktinformationtype("AFM:4999")
            beskrivelse = (
                afmærkning_pit.beskrivelse.replace("-\n", "")
                .replace("\n", " ")
                .rstrip(".")
                .strip()
            )
            nyetablerede.at[i, "Afmærkning"] = f"AFM:{afm_id} - {beskrivelse}"

        if afm_id == 4999:
            fire.cli.print(
                f"ADVARSEL: Nyoprettet punkt index {i} har ingen gyldig afmærkning anført",
                fg="red",
                bg="white",
                bold=True,
            )
        punktinfo.append(PunktInformation(infotype=afmærkning_pit, punkt=punkt))

        # Tilføj punktbeskrivelsen som punktinformation, hvis anført
        if not pd.isna(nyetablerede["Beskrivelse"][i]):
            punktinfo.append(
                PunktInformation(
                    infotype=beskrivelse_pit,
                    punkt=punkt,
                    tekst=nyetablerede["Beskrivelse"][i],
                )
            )

    # Tilknyt landsnumre til punkter
    landsnumre = dict(
        zip(
            punkter.keys(),
            fire.cli.firedb.tilknyt_landsnumre(punkter.values(), fikspunktstyper),
        )
    )
    punktinfo.extend(landsnumre.values())

    # sagsevent for punktinfo
    sagsevent_punktinfo = Sagsevent(
        sag=sag,
        eventtype=EventType.PUNKTINFO_TILFOEJET,
        sagseventinfos=[
            SagseventInfo(
                beskrivelse=f"Oprettelse af punktinfo ifm. {projektnavn}",
            )
        ],
        punktinformationer=punktinfo,
    )
    fire.cli.firedb.indset_sagsevent(sagsevent_punktinfo, commit=False)
    fire.cli.firedb.session.flush()  # hvis noget ikke virker får vi fejl her!

    # Generer dokumentation til fanebladet "Sagsgang"
    sagsgangslinje = {
        "Dato": sagsevent_punktinfo.registreringfra,
        "Hvem": sagsbehandler,
        "Hændelse": "punktinfoindsættelse",
        "Tekst": sagsevent_punktinfo.sagseventinfos[0].beskrivelse,
        "uuid": sagsevent_punktinfo.id,
    }
    sagsgang = sagsgang.append(sagsgangslinje, ignore_index=True)

    # Opdater regneark
    for k in punkter.keys():
        nyetablerede.at[k, "uuid"] = punkter[k].id
        nyetablerede.at[k, "Landsnummer"] = landsnumre[k].tekst

    # Drop numerisk index
    nyetablerede = nyetablerede.reset_index(drop=True)

    # Forbered  resultater til resultatregneark
    resultater = {"Sagsgang": sagsgang, "Nyetablerede punkter": nyetablerede}

    spørgsmål = click.style("Du indsætter nu ", fg="white", bg="red")
    spørgsmål += click.style(
        f"{len(punkter)} punkter ", fg="white", bg="red", bold=True
    )
    spørgsmål += click.style(f"i ", fg="white", bg="red")
    spørgsmål += click.style(f"{fire.cli.firedb.db}", fg="white", bg="red", bold=True)
    spørgsmål += click.style("-databasen - er du sikker?", fg="white", bg="red")
    if bekræft2(spørgsmål):
        # Indsæt rækker i database og skriv resultater til regneark
        fire.cli.firedb.session.commit()
        skriv_ark(projektnavn, resultater)
        fire.cli.print(
            f"Punkter oprettet. Kopiér nu faneblade fra '{projektnavn}-resultat.xlsx' til '{projektnavn}.xlsx'"
        )
    else:
        fire.cli.firedb.session.rollback()