eodash_catalog 0.0.16__tar.gz → 0.0.17__tar.gz

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/.bumpversion.cfg

@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.0.16
+current_version = 0.0.17
 commit = True
 tag = True
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\-(?P<release>[a-z]+)\.(?P<build>\d+))?

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: eodash_catalog
-Version: 0.0.16
+Version: 0.0.17
 Summary: This package is intended to help create a compatible STAC catalog for the eodash dashboard client. It supports configuration of multiple endpoint types for information extraction.
 Project-URL: Documentation, https://github.com/eodash/eodash_catalog#readme
 Project-URL: Issues, https://github.com/eodash/eodash_catalog/issues

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/ruff.toml

@@ -5,4 +5,4 @@ target-version = "py312"
 
 [lint]
 select = ["E", "F", "UP", "B", "SIM", "I", "W", "C90", "ASYNC", "A", "C4", "PERF", "RUF"]
-ignore = ["B019", "C901"]
+ignore = ["B019", "C901", "UP017"]

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/src/eodash_catalog/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Daniel Santillan <daniel.santillan@eox.at>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.0.16"
+__version__ = "0.0.17"

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/src/eodash_catalog/endpoints.py

@@ -4,7 +4,7 @@ import os
 import sys
 import uuid
 from collections.abc import Callable
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from itertools import groupby
 from operator import itemgetter
 
@@ -27,6 +27,7 @@ from eodash_catalog.utils import (
     create_geojson_from_bbox,
     create_geojson_point,
     generate_veda_cog_link,
+    replace_with_env_variables,
     retrieveExtentFromWMSWMTS,
 )
 
@@ -293,7 +294,7 @@ def handle_collection_only(
         catalog, collection_config["Name"], collection_config, catalog_config, endpoint_config
     )
     times = get_collection_times_from_config(endpoint_config)
-    if len(times) > 0 and not endpoint_config.get("Disable_Items"):
+    if len(times) > 0:
         for t in times:
             item = Item(
                 id=t,
@@ -473,7 +474,7 @@ def handle_SH_endpoint(
     catalog: Catalog,
     options: Options,
 ) -> Collection:
-    token = get_SH_token()
+    token = get_SH_token(endpoint_config)
     headers = {"Authorization": f"Bearer {token}"}
     endpoint_config["EndPoint"] = "https://services.sentinel-hub.com/api/v1/catalog/1.0.0/"
     # Overwrite collection id with type, such as ZARR or BYOC
@@ -509,8 +510,21 @@ def handle_WMS_endpoint(
             version=endpoint_config.get("Version", "1.1.1"),
             wmts=wmts,
         )
+    # optionally filter time results
+    if query := endpoint_config.get("Query"):
+        datetime_query = [times[0], times[-1]]
+        if start := query.get("Start"):
+            datetime_query[0] = parser.isoparse(start).replace(tzinfo=timezone.utc)
+        if end := query.get("End"):
+            datetime_query[1] = parser.isoparse(end).replace(tzinfo=timezone.utc)
+        # filter times based on query Start/End
+        times = [
+            datetime_str
+            for datetime_str in times
+            if datetime_query[0] <= parser.isoparse(datetime_str) < datetime_query[1]
+        ]
     # Create an item per time to allow visualization in stac clients
-    if len(times) > 0 and not endpoint_config.get("Disable_Items"):
+    if len(times) > 0:
         for t in times:
             item = Item(
                 id=t,
@@ -567,6 +581,10 @@ def add_visualization_info(
         instanceId = os.getenv("SH_INSTANCE_ID")
         if "InstanceId" in endpoint_config:
             instanceId = endpoint_config["InstanceId"]
+        if env_id := endpoint_config.get("CustomSHEnvId"):
+            # special handling for custom environment
+            # (will take SH_INSTANCE_ID_{env_id}) as ENV VAR
+            instanceId = os.getenv(f"SH_INSTANCE_ID_{env_id}")
         extra_fields: dict[str, list[str] | dict[str, str]] = {
             "wms:layers": [endpoint_config["LayerId"]],
             "role": ["data"],
@@ -604,10 +622,13 @@ def add_visualization_info(
         media_type = "image/jpeg"
         if "MediaType" in endpoint_config:
             media_type = endpoint_config["MediaType"]
+        endpoint_url = endpoint_config["EndPoint"]
+        # custom replacing of all ENV VARS present as template in URL as {VAR}
+        endpoint_url = replace_with_env_variables(endpoint_url)
         stac_object.add_link(
             Link(
                 rel="wms",
-                target=endpoint_config["EndPoint"],
+                target=endpoint_url,
                 media_type=media_type,
                 title=collection_config["Name"],
                 extra_fields=extra_fields,

eodash_catalog-0.0.17/src/eodash_catalog/sh_endpoint.py

@@ -0,0 +1,30 @@
+import os
+
+from oauthlib.oauth2 import BackendApplicationClient
+from requests_oauthlib import OAuth2Session
+
+SH_TOKEN_URL = "https://services.sentinel-hub.com/oauth/token"
+_token_cache: dict[str, str] = {}
+
+
+def get_SH_token(endpoint_config: dict) -> str:
+    # Your client credentials
+    client_id = os.getenv("SH_CLIENT_ID", "")
+    client_secret = os.getenv("SH_CLIENT_SECRET", "")
+    if env_id := endpoint_config.get("CustomSHEnvId"):
+        client_id = os.getenv(f"SH_CLIENT_ID_{env_id}", "")
+        client_secret = os.getenv(f"SH_CLIENT_SECRET_{env_id}", "")
+    if client_id in _token_cache:
+        return _token_cache[client_id]
+    # Create a session
+    client = BackendApplicationClient(client_id=client_id)
+    oauth = OAuth2Session(client=client)
+    # Get token for the session
+    token = oauth.fetch_token(
+        token_url=SH_TOKEN_URL,
+        client_secret=client_secret,
+    )
+    access_token = token["access_token"]
+    _token_cache[client_id] = access_token
+
+    return access_token

{eodash_catalog-0.0.16 → eodash_catalog-0.0.17}/src/eodash_catalog/utils.py

@@ -1,3 +1,4 @@
+import os
 import re
 import threading
 import uuid
@@ -252,3 +253,18 @@ def add_single_item_if_collection_empty(collection: Collection) -> None:
             end_datetime=datetime.now(),
         )
         collection.add_item(item)
+
+
+def replace_with_env_variables(s: str) -> str:
+    # Define the regex pattern to find text within curly brackets
+    pattern = r"\{(\w+)\}"
+
+    # Define the replacement function
+    def replacer(match):
+        # Extract the variable name from the match
+        var_name = match.group(1)
+        # Get the environment variable value, if it doesn't exist, keep the original placeholder
+        return os.getenv(var_name, match.group(0))
+
+    # Use re.sub with the replacement function
+    return re.sub(pattern, replacer, s)

eodash_catalog-0.0.16/src/eodash_catalog/sh_endpoint.py

@@ -1,22 +0,0 @@
-import os
-
-from oauthlib.oauth2 import BackendApplicationClient
-from requests_oauthlib import OAuth2Session
-
-SH_TOKEN_URL = "https://services.sentinel-hub.com/oauth/token"
-
-
-def get_SH_token() -> str:
-    # Your client credentials
-    client_id = os.getenv("SH_CLIENT_ID")
-    client_secret = os.getenv("SH_CLIENT_SECRET")
-    # Create a session
-    client = BackendApplicationClient(client_id=client_id)
-    oauth = OAuth2Session(client=client)
-    # Get token for the session
-    token = oauth.fetch_token(
-        token_url=SH_TOKEN_URL,
-        client_secret=client_secret,
-    )
-
-    return token["access_token"]