Skip to content

Commit b681e85

Browse files
committed
update imports
1 parent dc4b1d0 commit b681e85

File tree

3 files changed

+23
-25
lines changed

3 files changed

+23
-25
lines changed

stac_fastapi/core/stac_fastapi/core/core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -866,7 +866,7 @@ def preprocess_item(
866866
@overrides
867867
def bulk_item_insert(
868868
self, items: Items, chunk_size: Optional[int] = None, **kwargs
869-
) -> int:
869+
) -> str:
870870
"""Perform bulk insertion of items.
871871
872872
Args:
@@ -875,7 +875,7 @@ def bulk_item_insert(
875875
**kwargs: Additional keyword arguments
876876
877877
Returns:
878-
int: Number of items successfully added
878+
str: Message indicating number of items successfully added
879879
880880
Raises:
881881
BulkInsertError: If any items fail insertion
@@ -896,7 +896,7 @@ def bulk_item_insert(
896896
refresh=kwargs.get("refresh", False),
897897
raise_errors=True,
898898
)
899-
return success_count
899+
return "sucessfully added {} items".format(success_count)
900900

901901

902902
_DEFAULT_QUERYABLES: Dict[str, Dict[str, Any]] = {

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
mk_actions,
2929
mk_item_id,
3030
)
31-
from stac_fastapi.core.exceptions import BulkInsertError
3231
from stac_fastapi.core.extensions import filter
3332
from stac_fastapi.core.serializers import CollectionSerializer, ItemSerializer
3433
from stac_fastapi.core.utilities import MAX_LIMIT, bbox2polygon
@@ -956,16 +955,16 @@ def _handle_bulk_result(
956955
"""
957956
success_count, error_list = result
958957

959-
if raise_errors and error_list:
960-
error_messages = [
961-
f"Item {e['index']['_id']}: {e['index']['error']['reason']}"
962-
for e in error_list
963-
]
964-
raise BulkInsertError(
965-
f"Bulk operation failed with {len(error_list)} errors",
966-
success_count=success_count,
967-
errors=error_messages,
968-
)
958+
# if raise_errors and error_list:
959+
# error_messages = [
960+
# f"Item {e['index']['_id']}: {e['index']['error']['reason']}"
961+
# for e in error_list
962+
# ]
963+
# raise BulkInsertError(
964+
# f"Bulk operation failed with {len(error_list)} errors",
965+
# success_count=success_count,
966+
# errors=error_messages,
967+
# )
969968

970969
return success_count, error_list
971970

stac_fastapi/opensearch/stac_fastapi/opensearch/database_logic.py

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@
3131
mk_actions,
3232
mk_item_id,
3333
)
34-
from stac_fastapi.core.exceptions import BulkInsertError
3534
from stac_fastapi.core.extensions import filter
3635
from stac_fastapi.core.utilities import MAX_LIMIT, bbox2polygon
3736
from stac_fastapi.opensearch.config import (
@@ -987,16 +986,16 @@ def _handle_bulk_result(
987986
"""
988987
success_count, error_list = result
989988

990-
if raise_errors and error_list:
991-
error_messages = [
992-
f"Item {e['index']['_id']}: {e['index']['error']['reason']}"
993-
for e in error_list
994-
]
995-
raise BulkInsertError(
996-
f"Bulk operation completed with {len(error_list)} errors",
997-
success_count=success_count,
998-
errors=error_messages,
999-
)
989+
# if raise_errors and error_list:
990+
# error_messages = [
991+
# f"Item {e['index']['_id']}: {e['index']['error']['reason']}"
992+
# for e in error_list
993+
# ]
994+
# raise BulkInsertError(
995+
# f"Bulk operation completed with {len(error_list)} errors",
996+
# success_count=success_count,
997+
# errors=error_messages,
998+
# )
1000999

10011000
return success_count, error_list
10021001

0 commit comments

Comments
 (0)