feat: instrument scripts with prometheus metrics

Add metrics calls to register_stac.py and augment_stac_item.py:
- Wrap HTTP operations with duration timers
- Increment operation counters (create/update/skip/replace)
- Track preview generation duration
- All 42 unit tests pass, coverage: augment 24%, register 42%, metrics 65%

Author: wietzesuijker

scripts/augment_stac_item.py

@@ -13,6 +13,7 @@
 import httpx
 import s3fs
 import zarr
+from metrics import PREVIEW_GENERATION_DURATION, PREVIEW_HTTP_REQUEST_DURATION
 from pystac import Asset, Item, Link
 from pystac.extensions.projection import ProjectionExtension
 
@@ -1049,21 +1050,23 @@ def _request(
 
 
 def http_get(url: str, headers: dict[str, str]) -> dict[str, Any]:
-    data = _request("GET", url, headers).json()
+    with PREVIEW_HTTP_REQUEST_DURATION.labels(operation="get", endpoint="item").time():
+        data = _request("GET", url, headers).json()
     if isinstance(data, dict):
         return data
     raise ValueError("unexpected non-mapping response body")
 
 
 def http_put(url: str, data: dict[str, Any], headers: dict[str, str]) -> int:
-    return int(
-        _request(
-            "PUT",
-            url,
-            {**headers, "Content-Type": "application/json"},
-            json_body=data,
-        ).status_code
-    )
+    with PREVIEW_HTTP_REQUEST_DURATION.labels(operation="put", endpoint="item").time():
+        return int(
+            _request(
+                "PUT",
+                url,
+                {**headers, "Content-Type": "application/json"},
+                json_body=data,
+            ).status_code
+        )
 
 
 def ensure_collection_thumbnail(
@@ -1143,12 +1146,14 @@ def main(argv: Sequence[str] | None = None) -> int:
 
     item = Item.from_dict(payload)
     target_collection = item.collection_id or args.collection
-    _augment_item(
-        item,
-        raster_base=args.raster_base,
-        collection_id=target_collection,
-        verbose=args.verbose,
-    )
+
+    with PREVIEW_GENERATION_DURATION.labels(collection=target_collection).time():
+        _augment_item(
+            item,
+            raster_base=args.raster_base,
+            collection_id=target_collection,
+            verbose=args.verbose,
+        )
 
     target_url = f"{args.stac.rstrip('/')}/collections/{target_collection}/items/{item.id}"
     try:

scripts/register_stac.py

@@ -21,6 +21,7 @@
 
 import httpx
 import xarray as xr
+from metrics import STAC_HTTP_REQUEST_DURATION, STAC_REGISTRATION_TOTAL
 from tenacity import retry, stop_after_attempt, wait_exponential
 
 # Config: override via env vars
@@ -395,7 +396,8 @@ def register_item(
     with httpx.Client(timeout=TIMEOUT) as client:
         # Check if item exists
         try:
-            response = client.get(item_url, headers=headers)
+            with STAC_HTTP_REQUEST_DURATION.labels(operation="get", endpoint="item").time():
+                response = client.get(item_url, headers=headers)
             exists = response.status_code == 200
         except httpx.HTTPError:
             exists = False
@@ -405,34 +407,59 @@ def register_item(
 
             if mode == "create-or-skip":
                 logger.info("Skipping (mode=create-or-skip)")
+                STAC_REGISTRATION_TOTAL.labels(
+                    collection=collection_id, operation="skip", status="success"
+                ).inc()
                 return
             elif mode in ("upsert", "update"):
                 logger.info("Updating existing item (mode=upsert)")
-                response = client.put(item_url, json=item, headers=headers)
+                with STAC_HTTP_REQUEST_DURATION.labels(operation="put", endpoint="item").time():
+                    response = client.put(item_url, json=item, headers=headers)
                 if response.status_code >= 400:
                     logger.error(f" {response.status_code} {response.reason_phrase}")
                     logger.info(f"Response body: {response.text}")
+                    STAC_REGISTRATION_TOTAL.labels(
+                        collection=collection_id, operation="update", status="error"
+                    ).inc()
                 response.raise_for_status()
                 logger.info(f"Successfully updated item {item_id}")
+                STAC_REGISTRATION_TOTAL.labels(
+                    collection=collection_id, operation="update", status="success"
+                ).inc()
             elif mode in ("force", "replace"):
                 logger.info("Deleting and recreating (mode=replace)")
-                client.delete(item_url, headers=headers)
-                response = client.post(items_url, json=item, headers=headers)
+                with STAC_HTTP_REQUEST_DURATION.labels(operation="delete", endpoint="item").time():
+                    client.delete(item_url, headers=headers)
+                with STAC_HTTP_REQUEST_DURATION.labels(operation="post", endpoint="items").time():
+                    response = client.post(items_url, json=item, headers=headers)
                 if response.status_code >= 400:
                     logger.error(f" {response.status_code} {response.reason_phrase}")
                     logger.info(f"Response body: {response.text}")
+                    STAC_REGISTRATION_TOTAL.labels(
+                        collection=collection_id, operation="replace", status="error"
+                    ).inc()
                 response.raise_for_status()
                 logger.info(f"Successfully replaced item {item_id}")
+                STAC_REGISTRATION_TOTAL.labels(
+                    collection=collection_id, operation="replace", status="success"
+                ).inc()
             else:
                 raise ValueError(f"Unknown mode: {mode}")
         else:
             logger.info(f"Creating new item {item_id}")
-            response = client.post(items_url, json=item, headers=headers)
+            with STAC_HTTP_REQUEST_DURATION.labels(operation="post", endpoint="items").time():
+                response = client.post(items_url, json=item, headers=headers)
             if response.status_code >= 400:
                 logger.error(f" {response.status_code} {response.reason_phrase}")
                 logger.info(f"Response body: {response.text}")
+                STAC_REGISTRATION_TOTAL.labels(
+                    collection=collection_id, operation="create", status="error"
+                ).inc()
             response.raise_for_status()
             logger.info(f"Successfully created item {item_id}")
+            STAC_REGISTRATION_TOTAL.labels(
+                collection=collection_id, operation="create", status="success"
+            ).inc()
 
 
 def main() -> int: