sfeos-helpers 6.10.0__tar.gz → 6.10.1__tar.gz

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfeos_helpers
-Version: 6.10.0
+Version: 6.10.1
 Summary: Helper library for the Elasticsearch and Opensearch stac-fastapi backends.
 Project-URL: Homepage, https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch
 License: MIT
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Requires-Python: >=3.11
-Requires-Dist: stac-fastapi-core==6.10.0
+Requires-Dist: stac-fastapi-core==6.10.1
 Description-Content-Type: text/markdown
 
 # sfeos-helpers

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/pyproject.toml

@@ -29,7 +29,7 @@ keywords = [
 ]
 dynamic = ["version"]
 dependencies = [
-    "stac-fastapi.core==6.10.0",
+    "stac-fastapi.core==6.10.1",
 ]
 
 [project.urls]

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/database/__init__.py

@@ -60,7 +60,14 @@ from .query import (
     apply_intersects_filter_shared,
     populate_sort_shared,
 )
-from .utils import add_bbox_shape_to_collection, get_bool_env, validate_refresh
+from .utils import (
+    ItemAlreadyExistsError,
+    add_bbox_shape_to_collection,
+    check_item_exists_in_alias,
+    check_item_exists_in_alias_sync,
+    get_bool_env,
+    validate_refresh,
+)
 
 __all__ = [
     # Catalog operations
@@ -90,6 +97,10 @@ __all__ = [
     "validate_refresh",
     "get_bool_env",
     "add_bbox_shape_to_collection",
+    "check_item_exists_in_alias",
+    "check_item_exists_in_alias_sync",
+    # Errors
+    "ItemAlreadyExistsError",
     # Datetime utilities
     "return_date",
     "extract_date",

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/database/index.py

@@ -165,7 +165,8 @@ def filter_indexes_by_datetime(
             datetime_alias if use_datetime else start_datetime_alias
         )
 
-        filtered_indexes.append(primary_datetime_alias)
+        if primary_datetime_alias is not None:
+            filtered_indexes.append(primary_datetime_alias)
 
     return filtered_indexes
 

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/database/utils.py

@@ -14,10 +14,73 @@ from stac_fastapi.extensions.core.transaction.request import (
     PatchRemove,
 )
 from stac_fastapi.sfeos_helpers.models.patch import ElasticPath, ESCommandSet
+from stac_fastapi.types.errors import ConflictError
 
 logger = logging.getLogger(__name__)
 
 
+class ItemAlreadyExistsError(ConflictError):
+    """Error raised when attempting to create an item that already exists.
+
+    Attributes:
+        item_id: The ID of the item that already exists.
+        collection_id: The ID of the collection containing the item.
+    """
+
+    def __init__(self, item_id: str, collection_id: str):
+        """Initialize the error with item and collection IDs."""
+        self.item_id = item_id
+        self.collection_id = collection_id
+        message = f"Item {item_id} in collection {collection_id} already exists"
+        super().__init__(message)
+
+
+async def check_item_exists_in_alias(client: Any, alias: str, doc_id: str) -> bool:
+    """Check if an item exists across all indexes for an alias.
+
+    Args:
+        client: The async Elasticsearch/OpenSearch client.
+        alias: The index alias to search against.
+        doc_id: The document ID to check for existence.
+
+    Returns:
+        bool: True if the item exists in any index under the alias, False otherwise.
+    """
+    resp = await client.search(
+        index=alias,
+        body={
+            "query": {"ids": {"values": [doc_id]}},
+            "_source": False,
+        },
+        size=0,
+        terminate_after=1,
+    )
+    return bool(resp["hits"]["total"]["value"])
+
+
+def check_item_exists_in_alias_sync(client: Any, alias: str, doc_id: str) -> bool:
+    """Check if an item exists across all indexes for an alias (sync).
+
+    Args:
+        client: The sync Elasticsearch/OpenSearch client.
+        alias: The index alias to search against.
+        doc_id: The document ID to check for existence.
+
+    Returns:
+        bool: True if the item exists in any index under the alias, False otherwise.
+    """
+    resp = client.search(
+        index=alias,
+        body={
+            "query": {"ids": {"values": [doc_id]}},
+            "_source": False,
+        },
+        size=0,
+        terminate_after=1,
+    )
+    return bool(resp["hits"]["total"]["value"])
+
+
 def add_bbox_shape_to_collection(collection: Dict[str, Any]) -> bool:
     """Add bbox_shape field to a collection document for spatial queries.
 

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/search_engine/inserters.py

@@ -257,8 +257,10 @@ class DatetimeIndexInserter(BaseIndexInserter):
                     extract_date(latest_item["_source"]["properties"]["datetime"])
                 ),
                 end_datetime=str(
-                    extract_date(latest_item["_source"]["properties"]["end_datetime"])
-                ),
+                    extract_first_date_from_index(aliases_dict["end_datetime"])
+                )
+                if aliases_dict.get("end_datetime")
+                else None,
             )
 
             await self.datetime_manager.handle_oversized_index(

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/search_engine/selection/selectors.py

@@ -1,5 +1,4 @@
 """Async index selectors with datetime-based filtering."""
-
 from typing import Any, Dict, List, Optional, Tuple
 
 from stac_fastapi.core.utilities import get_bool_env

{sfeos_helpers-6.10.0 → sfeos_helpers-6.10.1}/stac_fastapi/sfeos_helpers/version.py

@@ -1,2 +1,2 @@
 """library version."""
-__version__ = "6.10.0"
+__version__ = "6.10.1"