update_config.py

#!/usr/bin/env python3

import os
from pathlib import Path
from urllib.parse import urljoin

import hydra
import nvdb
import requests
from omegaconf import OmegaConf, open_dict


@hydra.main(version_base=None, config_path="conf", config_name="config")
def main(cfg):
    pygeoapi_config = Path(os.environ["PYGEOAPI_CONFIG"])

    OmegaConf.set_struct(cfg, True)
    with open_dict(cfg):
        object_types = requests.get(
            urljoin(nvdb.URL, "/vegobjekttyper"), timeout=10
        ).json()
        for object_type in object_types:
            # https://nvdb.atlas.vegvesen.no/docs/data-i-nvdb/data-vi-ikke-publiserer/
            if object_type["id"] >= 1000:
                continue  # internal object
            if object_type["sensitiv"]:
                continue  # authentication required
            if object_type["id"] == 562:
                continue  # test object
            if object_type["id"] == 793:
                continue  # documentation
            resource = {
                "type": "collection",
                "title": f"[NVDB] {object_type['navn']}",
                "description": (
                    f"{object_type.get('beskrivelse')}\n"
                    f"VegObjekter ID: {object_type['id']}"
                ).lstrip(),
                "keywords": ["NVDB", "VegObjekter"],
                "links": [],
                "extents": {
                    "spatial": {
                        "bbox": [-180, -90, 180, 90],
                        "crs": "http://www.opengis.net/def/crs/OGC/1.3/CRS84",
                    },
                },
                "providers": [
                    {
                        "type": "feature",
                        "name": "nvdb.vegobjekter.VegObjekter",
                        "data": "/dev/null",
                        "options": {"obj_id": object_type["id"]},
                    }
                ],
            }
            temporal_end = object_type.get("objektliste_dato")
            if temporal_end:
                resource["temporal"] = {"end": temporal_end}
            layer_name = nvdb.normalize("nvdb_" + object_type["kortnavn"])
            cfg["resources"][layer_name] = resource

    with pygeoapi_config.open("w") as config_file:
        config_file.write(OmegaConf.to_yaml(cfg))


if __name__ == "__main__":
    main()