feat: generate a map configuration based on a wms

Author: nicokant

pyproject.toml

@@ -39,7 +39,8 @@ dependencies = [
   "httpx>=0.28.1",
   "setuptools>=65.5.0",
   "backoff>=2.1.0",
-  "pyyaml>=6.0.3"
+  "pyyaml>=6.0.3",
+  "owslib>=0.35.0"
 ]
 description = ""
 license = "GPL-3.0+"

src/datasync/main.py

@@ -4,7 +4,7 @@
 
 import typer
 
-from . import dms, grass, ninagen, nva, pit_registering_salmon, services, ubw
+from . import dms, grass, maps, ninagen, nva, pit_registering_salmon, services, ubw
 
 app = typer.Typer(
     help="Provide subcommands for synchronizing different resources, see subcommands"
@@ -16,6 +16,7 @@
 app.add_typer(pit_registering_salmon.app, name="pit-registering-salmon")
 app.add_typer(grass.app, name="grass-gis")
 app.add_typer(services.app, name="services")
+app.add_typer(maps.app, name="maps")
 
 if __name__ == "__main__":
     app()

src/datasync/maps.py

@@ -0,0 +1,124 @@
+import json
+from pathlib import Path
+from urllib import parse
+
+import typer
+from owslib.wms import WebMapService
+
+from .settings import log
+
+app = typer.Typer()
+
+
+@app.command()
+def wms_to_map(get_capabilities_url: str, output: str = "output.json"):
+    wms = WebMapService(get_capabilities_url)
+    log.info(wms.contents)
+
+    parsed = parse.urlparse(get_capabilities_url)
+    qs = parse.parse_qs(parsed.query)
+    map_param = qs.get("map", [None])[0]
+
+    items = {}
+
+    cn = wms.__class__.__name__
+
+    for key, value in wms.contents.items():
+        log.info(key, value=value)
+
+        req = getattr(wms, f"_{cn}__build_getmap_request")(
+            layers=[key],
+            styles=[""],
+            srs="EPSG:3857",
+            format="image/png",
+            size=(256, 256),
+            transparent=True,
+            bbox=(0, 0, 0, 0),
+        )
+        if map_param:
+            req["map"] = map_param
+
+        del req["bbox"]
+
+        legend = {
+            "service": "WMS",
+            "version": wms.version,
+            "request": "GetLegendGraphic",
+            "layer": key,
+            "style": "",
+            "format": "image/png",
+            "transparent": "true",
+        }
+
+        if map_param:
+            legend["map"] = map_param
+
+        items[key] = {
+            "name": value.title,
+            "type": "layer",
+            "layer": {
+                "type": "raster",
+                "tiles": [
+                    parse.urlunsplit(
+                        (
+                            parsed.scheme,
+                            parsed.netloc,
+                            parsed.path,
+                            parse.urlencode(req, doseq=True) + "&bbox={bbox-epsg-3857}",
+                            "",
+                        )
+                    )
+                ],
+                "legend": {
+                    "type": "image",
+                    "url": parse.urlunsplit(
+                        (
+                            parsed.scheme,
+                            parsed.netloc,
+                            parsed.path,
+                            parse.urlencode(
+                                legend,
+                                doseq=True,
+                            ),
+                            "",
+                        )
+                    ),
+                },
+            },
+        }
+
+    conf = {
+        "id": parsed.hostname,
+        "title": wms.identification.title,
+        "subtitle": "",
+        "description": wms.identification.abstract,
+        "icon": "https://s3-ext-1.nina.no/dms/maps/logosmall-BFfoJdyr.png",
+        "baseMap": "positron",
+        "layerOrder": [],
+        "viewState": {"longitude": 17.8, "latitude": 65.6, "zoom": 2},
+        "expandedItems": [],
+        "items": {
+            "root": {
+                "name": "Layers",
+                "type": "folder",
+                "children": list(items.keys()),
+            },
+            **items,
+        },
+        "config": {
+            "titiler_api_url": "/titiler",
+            "theme": "nina",
+            "language": "en",
+            "footer": {
+                "items": [
+                    "![](https://s3-ext-1.nina.no/dms/maps/logowhite-6HSLYLYZ.png)",
+                    "Norsk institutt for naturforskning - [www.nina.no](https://www.nina.no)",
+                ],
+                "justify": "justify-between",
+                "align": "items-baseline",
+            },
+        },
+    }
+
+    with Path(output).open("w") as f:
+        json.dump(conf, f, indent=4)