sfeos-tools 0.1.0__tar.gz → 0.2.0__tar.gz

{sfeos_tools-0.1.0/sfeos_tools.egg-info → sfeos_tools-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sfeos-tools
-Version: 0.1.0
+Version: 0.2.0
 Summary: CLI tools for managing stac-fastapi-elasticsearch-opensearch deployments
 Author: CloudFerro S.A.
 Author-email: Jonathan Healy <jon@healy-hypersaptial.dev>
@@ -35,11 +35,12 @@ Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: GIS
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
@@ -47,14 +48,17 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: click>=8.0.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: orjson>=3.0.0
 Provides-Extra: elasticsearch
-Requires-Dist: stac-fastapi-elasticsearch; extra == "elasticsearch"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "elasticsearch"
 Provides-Extra: opensearch
-Requires-Dist: stac-fastapi-opensearch; extra == "opensearch"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "opensearch"
 Provides-Extra: dev
-Requires-Dist: stac-fastapi-elasticsearch; extra == "dev"
-Requires-Dist: stac-fastapi-opensearch; extra == "dev"
+Requires-Dist: stac-fastapi-elasticsearch>=6.6.0; extra == "dev"
+Requires-Dist: stac-fastapi-opensearch>=6.6.0; extra == "dev"
 Requires-Dist: pytest>=6.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
 Requires-Dist: pytest-cov>=2.0; extra == "dev"
 Requires-Dist: black>=21.0; extra == "dev"
 Requires-Dist: isort>=5.0; extra == "dev"
@@ -66,6 +70,22 @@ Dynamic: license-file
 
 CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch) deployments.
 
+<!-- markdownlint-disable MD033 MD041 -->
+
+
+<p align="left">
+  <img src="https://raw.githubusercontent.com/stac-utils/stac-fastapi-elasticsearch-opensearch/refs/heads/main/assets/sfeos.png" width=1000>
+</p>
+
+<!-- **Jump to:** [Project Introduction](#project-introduction---what-is-sfeos) | [Quick Start](#quick-start) | [Table of Contents](#table-of-contents) -->
+
+  [![Downloads](https://static.pepy.tech/badge/sfeos-tools?color=blue)](https://pepy.tech/project/sfeos-tools)
+  [![GitHub contributors](https://img.shields.io/github/contributors/healy-hyperspatial/sfeos-tools?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/graphs/contributors)
+  [![GitHub stars](https://img.shields.io/github/stars/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/stargazers)
+  [![GitHub forks](https://img.shields.io/github/forks/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/network/members)
+   [![PyPI version](https://img.shields.io/pypi/v/sfeos-tools.svg?color=blue)](https://pypi.org/project/sfeos-tools/)
+  [![STAC](https://img.shields.io/badge/STAC-1.1.0-blue.svg)](https://github.com/radiantearth/stac-spec/tree/v1.1.0)
+
 ## Table of Contents
 
 - [Installation](#installation)
@@ -76,6 +96,7 @@ CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.co
 - [Commands](#commands)
   - [add-bbox-shape](#add-bbox-shape)
   - [reindex](#reindex)
+  - [load-data](#load-data)
 - [Development](#development)
 - [License](#license)
 
@@ -165,62 +186,55 @@ Options:
 - `--password`: Database password (default: ES_PASS env var)
 - `--yes`: Skip confirmation prompt
 
-Example:
+Examples:
 ```bash
 # Reindex Elasticsearch with custom host and no SSL
-sfeos-tools reindex --backend elasticsearch --host localhost --port 9200 --no-ssl
+sfeos-tools reindex --backend elasticsearch --host localhost --port 9200 --no-ssl --yes
 
 # Reindex OpenSearch with default settings
-sfeos-tools reindex --backend opensearch
+sfeos-tools reindex --backend opensearch --yes
 ```
 
-# Get help for a specific command
-sfeos-tools add-bbox-shape --help
-```
-
-## Commands
-
-### add-bbox-shape
-
-Add `bbox_shape` field to existing collections for spatial search support.
+### load-data
 
-**Basic usage:**
+Load STAC collections and items from local JSON files into a STAC API instance. This command is useful for:
+- Populating a new STAC API deployment with test data
+- Migrating data between STAC API instances
+- Bulk loading STAC collections and items
 
 ```bash
-# Elasticsearch
-sfeos-tools add-bbox-shape --backend elasticsearch
-
-# OpenSearch
-sfeos-tools add-bbox-shape --backend opensearch
+sfeos-tools load-data --base-url <stac-api-url> [options]
 ```
 
-**Connection options:**
+Options:
+- `--base-url`: Base URL of the STAC API (required)
+- `--collection-id`: ID of the collection to create/update (default: test-collection)
+- `--data-dir`: Directory containing collection.json and feature collection files (default: sample_data/)
+- `--use-bulk`: Use bulk insert method for items (faster for large datasets)
 
-```bash
-# Local Docker Compose (no SSL)
-sfeos-tools add-bbox-shape --backend elasticsearch --no-ssl
-
-# Remote server with SSL
-sfeos-tools add-bbox-shape \
-  --backend elasticsearch \
-  --host db.example.com \
-  --port 9200 \
-  --user admin \
-  --password secret
-
-# Using environment variables
-ES_HOST=my-cluster.cloud.com ES_PORT=9243 ES_USER=elastic ES_PASS=changeme \
-  sfeos-tools add-bbox-shape --backend elasticsearch
-```
+**Data Directory Structure:**
 
-**Available options:**
+Your data directory should contain:
+- `collection.json`: STAC collection definition
+- One or more `.json` files: Feature collections with STAC items
 
-- `--backend`: Database backend (elasticsearch or opensearch) - **required**
-- `--host`: Database host (default: localhost or ES_HOST env var)
-- `--port`: Database port (default: 9200 or ES_PORT env var)
-- `--use-ssl / --no-ssl`: Use SSL connection (default: true or ES_USE_SSL env var)
-- `--user`: Database username (default: ES_USER env var)
-- `--password`: Database password (default: ES_PASS env var)
+Examples:
+```bash
+# Load data from default directory
+sfeos-tools load-data --base-url http://localhost:8080
+
+# Load with custom collection ID and bulk insert
+sfeos-tools load-data \
+  --base-url http://localhost:8080 \
+  --collection-id my-collection \
+  --use-bulk
+
+# Load from custom directory
+sfeos-tools load-data \
+  --base-url http://localhost:8080 \
+  --data-dir /path/to/stac/data \
+  --collection-id production-data
+```
 
 ## Development
 
@@ -241,3 +255,6 @@ pre-commit install
 pre-commit run --all-files
 ```
 
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

{sfeos_tools-0.1.0 → sfeos_tools-0.2.0}/README.md

@@ -2,6 +2,22 @@
 
 CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.com/stac-utils/stac-fastapi-elasticsearch-opensearch) deployments.
 
+<!-- markdownlint-disable MD033 MD041 -->
+
+
+<p align="left">
+  <img src="https://raw.githubusercontent.com/stac-utils/stac-fastapi-elasticsearch-opensearch/refs/heads/main/assets/sfeos.png" width=1000>
+</p>
+
+<!-- **Jump to:** [Project Introduction](#project-introduction---what-is-sfeos) | [Quick Start](#quick-start) | [Table of Contents](#table-of-contents) -->
+
+  [![Downloads](https://static.pepy.tech/badge/sfeos-tools?color=blue)](https://pepy.tech/project/sfeos-tools)
+  [![GitHub contributors](https://img.shields.io/github/contributors/healy-hyperspatial/sfeos-tools?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/graphs/contributors)
+  [![GitHub stars](https://img.shields.io/github/stars/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/stargazers)
+  [![GitHub forks](https://img.shields.io/github/forks/healy-hyperspatial/sfeos-tools.svg?color=blue)](https://github.com/healy-hyperspatial/sfeos-tools/network/members)
+   [![PyPI version](https://img.shields.io/pypi/v/sfeos-tools.svg?color=blue)](https://pypi.org/project/sfeos-tools/)
+  [![STAC](https://img.shields.io/badge/STAC-1.1.0-blue.svg)](https://github.com/radiantearth/stac-spec/tree/v1.1.0)
+
 ## Table of Contents
 
 - [Installation](#installation)
@@ -12,6 +28,7 @@ CLI tools for managing [stac-fastapi-elasticsearch-opensearch](https://github.co
 - [Commands](#commands)
   - [add-bbox-shape](#add-bbox-shape)
   - [reindex](#reindex)
+  - [load-data](#load-data)
 - [Development](#development)
 - [License](#license)
 
@@ -101,62 +118,55 @@ Options:
 - `--password`: Database password (default: ES_PASS env var)
 - `--yes`: Skip confirmation prompt
 
-Example:
+Examples:
 ```bash
 # Reindex Elasticsearch with custom host and no SSL
-sfeos-tools reindex --backend elasticsearch --host localhost --port 9200 --no-ssl
+sfeos-tools reindex --backend elasticsearch --host localhost --port 9200 --no-ssl --yes
 
 # Reindex OpenSearch with default settings
-sfeos-tools reindex --backend opensearch
+sfeos-tools reindex --backend opensearch --yes
 ```
 
-# Get help for a specific command
-sfeos-tools add-bbox-shape --help
-```
-
-## Commands
-
-### add-bbox-shape
-
-Add `bbox_shape` field to existing collections for spatial search support.
+### load-data
 
-**Basic usage:**
+Load STAC collections and items from local JSON files into a STAC API instance. This command is useful for:
+- Populating a new STAC API deployment with test data
+- Migrating data between STAC API instances
+- Bulk loading STAC collections and items
 
 ```bash
-# Elasticsearch
-sfeos-tools add-bbox-shape --backend elasticsearch
-
-# OpenSearch
-sfeos-tools add-bbox-shape --backend opensearch
+sfeos-tools load-data --base-url <stac-api-url> [options]
 ```
 
-**Connection options:**
+Options:
+- `--base-url`: Base URL of the STAC API (required)
+- `--collection-id`: ID of the collection to create/update (default: test-collection)
+- `--data-dir`: Directory containing collection.json and feature collection files (default: sample_data/)
+- `--use-bulk`: Use bulk insert method for items (faster for large datasets)
 
-```bash
-# Local Docker Compose (no SSL)
-sfeos-tools add-bbox-shape --backend elasticsearch --no-ssl
-
-# Remote server with SSL
-sfeos-tools add-bbox-shape \
-  --backend elasticsearch \
-  --host db.example.com \
-  --port 9200 \
-  --user admin \
-  --password secret
-
-# Using environment variables
-ES_HOST=my-cluster.cloud.com ES_PORT=9243 ES_USER=elastic ES_PASS=changeme \
-  sfeos-tools add-bbox-shape --backend elasticsearch
-```
+**Data Directory Structure:**
 
-**Available options:**
+Your data directory should contain:
+- `collection.json`: STAC collection definition
+- One or more `.json` files: Feature collections with STAC items
 
-- `--backend`: Database backend (elasticsearch or opensearch) - **required**
-- `--host`: Database host (default: localhost or ES_HOST env var)
-- `--port`: Database port (default: 9200 or ES_PORT env var)
-- `--use-ssl / --no-ssl`: Use SSL connection (default: true or ES_USE_SSL env var)
-- `--user`: Database username (default: ES_USER env var)
-- `--password`: Database password (default: ES_PASS env var)
+Examples:
+```bash
+# Load data from default directory
+sfeos-tools load-data --base-url http://localhost:8080
+
+# Load with custom collection ID and bulk insert
+sfeos-tools load-data \
+  --base-url http://localhost:8080 \
+  --collection-id my-collection \
+  --use-bulk
+
+# Load from custom directory
+sfeos-tools load-data \
+  --base-url http://localhost:8080 \
+  --data-dir /path/to/stac/data \
+  --collection-id production-data
+```
 
 ## Development
 
@@ -177,3 +187,6 @@ pre-commit install
 pre-commit run --all-files
 ```
 
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

{sfeos_tools-0.1.0 → sfeos_tools-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sfeos-tools"
-version = "0.1.0"
+version = "0.2.0"
 description = "CLI tools for managing stac-fastapi-elasticsearch-opensearch deployments"
 readme = "README.md"
 authors = [
@@ -21,17 +21,20 @@ classifiers = [
     "Intended Audience :: Developers",
     "Intended Audience :: Science/Research",
     "License :: OSI Approved :: MIT License",
-    "Programming Language :: Python :: 3.8",
     "Programming Language :: Python :: 3.9",
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
     "Topic :: Scientific/Engineering",
     "Topic :: Scientific/Engineering :: GIS",
     "Topic :: Software Development :: Libraries :: Python Modules"
 ]
 dependencies = [
     "click>=8.0.0",
+    "httpx>=0.24.0",
+    "orjson>=3.0.0",
 ]
 
 [project.urls]
@@ -42,16 +45,17 @@ Source = "https://github.com/Healy-Hyperspatial/sfeos-tools"
 
 [project.optional-dependencies]
 elasticsearch = [
-    "stac-fastapi-elasticsearch",
+    "stac-fastapi-elasticsearch>=6.6.0",
 ]
 opensearch = [
-    "stac-fastapi-opensearch",
+    "stac-fastapi-opensearch>=6.6.0",
 ]
 dev = [
-    "stac-fastapi-elasticsearch",
-    "stac-fastapi-opensearch",
+    "stac-fastapi-elasticsearch>=6.6.0",
+    "stac-fastapi-opensearch>=6.6.0",
     # Development tools
     "pytest>=6.0",
+    "pytest-asyncio>=0.21.0",
     "pytest-cov>=2.0",
     "black>=21.0",
     "isort>=5.0",
@@ -79,7 +83,13 @@ line_length = 88
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = "test_*.py"
-addopts = "-v --cov=sfeos_tool --cov-report=term-missing"
+addopts = "-v --cov=sfeos_tools --cov-report=term-missing"
+asyncio_mode = "auto"
+filterwarnings = [
+    "ignore::DeprecationWarning:pydantic._internal._config",
+    "ignore::pydantic.warnings.PydanticDeprecatedSince20",
+    "ignore:coroutine.*was never awaited:RuntimeWarning",
+]
 
 [tool.coverage.report]
 exclude_lines = [

sfeos_tools-0.2.0/sfeos_tools/bbox_shape.py

@@ -0,0 +1,114 @@
+"""Bbox shape migration utilities for SFEOS collections."""
+
+import logging
+
+from stac_fastapi.sfeos_helpers.database import add_bbox_shape_to_collection
+from stac_fastapi.sfeos_helpers.mappings import COLLECTIONS_INDEX
+
+logger = logging.getLogger(__name__)
+
+
+async def process_collection_bbox_shape(client, collection_doc, backend):
+    """Process a single collection document to add bbox_shape field.
+
+    Args:
+        client: Elasticsearch/OpenSearch client
+        collection_doc: Collection document from database
+        backend: Backend type ('elasticsearch' or 'opensearch')
+
+    Returns:
+        bool: True if collection was updated, False if no update was needed
+    """
+    collection = collection_doc["_source"]
+    collection_id = collection.get("id", collection_doc["_id"])
+
+    # Use the shared function to add bbox_shape
+    was_added = add_bbox_shape_to_collection(collection)
+
+    if not was_added:
+        return False
+
+    # Update the collection in the database
+    if backend == "elasticsearch":
+        await client.index(
+            index=COLLECTIONS_INDEX,
+            id=collection_id,
+            document=collection,
+            refresh=True,
+        )
+    else:  # opensearch
+        await client.index(
+            index=COLLECTIONS_INDEX,
+            id=collection_id,
+            body=collection,
+            refresh=True,
+        )
+
+    logger.info(f"Collection '{collection_id}': Added bbox_shape field")
+    return True
+
+
+async def run_add_bbox_shape(backend):
+    """Add bbox_shape field to all existing collections.
+
+    Args:
+        backend: Backend type ('elasticsearch' or 'opensearch')
+    """
+    import os
+
+    logger.info(
+        f"Starting migration: Adding bbox_shape to existing collections ({backend})"
+    )
+
+    # Log connection info (showing what will be used by the client)
+    es_host = os.getenv("ES_HOST", "localhost")
+    es_port = os.getenv(
+        "ES_PORT", "9200"
+    )  # Both backends default to 9200 in their config
+    es_use_ssl = os.getenv("ES_USE_SSL", "true")
+    logger.info(f"Connecting to {backend} at {es_host}:{es_port} (SSL: {es_use_ssl})")
+
+    # Create client based on backend
+    if backend == "elasticsearch":
+        from stac_fastapi.elasticsearch.config import AsyncElasticsearchSettings
+
+        settings = AsyncElasticsearchSettings()
+    else:  # opensearch
+        from stac_fastapi.opensearch.config import AsyncOpensearchSettings
+
+        settings = AsyncOpensearchSettings()
+
+    client = settings.create_client
+
+    try:
+        # Get all collections
+        response = await client.search(
+            index=COLLECTIONS_INDEX,
+            body={
+                "query": {"match_all": {}},
+                "size": 10000,
+            },  # Adjust size if you have more collections
+        )
+
+        total_collections = response["hits"]["total"]["value"]
+        logger.info(f"Found {total_collections} collections to process")
+
+        updated_count = 0
+        skipped_count = 0
+
+        for hit in response["hits"]["hits"]:
+            was_updated = await process_collection_bbox_shape(client, hit, backend)
+            if was_updated:
+                updated_count += 1
+            else:
+                skipped_count += 1
+
+        logger.info(
+            f"Migration complete: {updated_count} collections updated, {skipped_count} skipped"
+        )
+
+    except Exception as e:
+        logger.error(f"Migration failed with error: {e}")
+        raise
+    finally:
+        await client.close()

{sfeos_tools-0.1.0 → sfeos_tools-0.2.0}/sfeos_tools/cli.py

@@ -13,123 +13,24 @@ import logging
 import sys
 
 import click
-from stac_fastapi.sfeos_helpers.database import add_bbox_shape_to_collection
-from stac_fastapi.sfeos_helpers.mappings import COLLECTIONS_INDEX
 
+try:
+    from importlib.metadata import version as _get_version
+except ImportError:
+    from importlib_metadata import version as _get_version  # type: ignore[no-redef]
+
+__version__ = _get_version("sfeos-tools")
+
+from .bbox_shape import run_add_bbox_shape
+from .data_loader import load_items
 from .reindex import run as unified_reindex_run
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
-async def process_collection_bbox_shape(client, collection_doc, backend):
-    """Process a single collection document to add bbox_shape field.
-
-    Args:
-        client: Elasticsearch/OpenSearch client
-        collection_doc: Collection document from database
-        backend: Backend type ('elasticsearch' or 'opensearch')
-
-    Returns:
-        bool: True if collection was updated, False if no update was needed
-    """
-    collection = collection_doc["_source"]
-    collection_id = collection.get("id", collection_doc["_id"])
-
-    # Use the shared function to add bbox_shape
-    was_added = add_bbox_shape_to_collection(collection)
-
-    if not was_added:
-        return False
-
-    # Update the collection in the database
-    if backend == "elasticsearch":
-        await client.index(
-            index=COLLECTIONS_INDEX,
-            id=collection_id,
-            document=collection,
-            refresh=True,
-        )
-    else:  # opensearch
-        await client.index(
-            index=COLLECTIONS_INDEX,
-            id=collection_id,
-            body=collection,
-            refresh=True,
-        )
-
-    logger.info(f"Collection '{collection_id}': Added bbox_shape field")
-    return True
-
-
-async def run_add_bbox_shape(backend):
-    """Add bbox_shape field to all existing collections.
-
-    Args:
-        backend: Backend type ('elasticsearch' or 'opensearch')
-    """
-    import os
-
-    logger.info(
-        f"Starting migration: Adding bbox_shape to existing collections ({backend})"
-    )
-
-    # Log connection info (showing what will be used by the client)
-    es_host = os.getenv("ES_HOST", "localhost")
-    es_port = os.getenv(
-        "ES_PORT", "9200"
-    )  # Both backends default to 9200 in their config
-    es_use_ssl = os.getenv("ES_USE_SSL", "true")
-    logger.info(f"Connecting to {backend} at {es_host}:{es_port} (SSL: {es_use_ssl})")
-
-    # Create client based on backend
-    if backend == "elasticsearch":
-        from stac_fastapi.elasticsearch.config import AsyncElasticsearchSettings
-
-        settings = AsyncElasticsearchSettings()
-    else:  # opensearch
-        from stac_fastapi.opensearch.config import AsyncOpensearchSettings
-
-        settings = AsyncOpensearchSettings()
-
-    client = settings.create_client
-
-    try:
-        # Get all collections
-        response = await client.search(
-            index=COLLECTIONS_INDEX,
-            body={
-                "query": {"match_all": {}},
-                "size": 10000,
-            },  # Adjust size if you have more collections
-        )
-
-        total_collections = response["hits"]["total"]["value"]
-        logger.info(f"Found {total_collections} collections to process")
-
-        updated_count = 0
-        skipped_count = 0
-
-        for hit in response["hits"]["hits"]:
-            was_updated = await process_collection_bbox_shape(client, hit, backend)
-            if was_updated:
-                updated_count += 1
-            else:
-                skipped_count += 1
-
-        logger.info(
-            f"Migration complete: {updated_count} collections updated, {skipped_count} skipped"
-        )
-
-    except Exception as e:
-        logger.error(f"Migration failed with error: {e}")
-        raise
-    finally:
-        await client.close()
-
-
 @click.group()
-@click.version_option(version="0.1.0", prog_name="sfeos-tools")
+@click.version_option(version=__version__, prog_name="sfeos-tools")
 def cli():
     """SFEOS Tools - Utilities for managing stac-fastapi-elasticsearch-opensearch deployments."""
     pass
@@ -332,5 +233,47 @@ def reindex(backend, host, port, use_ssl, user, password, yes):
         sys.exit(1)
 
 
+@cli.command("load-data")
+@click.option("--base-url", required=True, help="Base URL of the STAC API")
+@click.option(
+    "--collection-id",
+    default="test-collection",
+    help="ID of the collection to which items are added",
+)
+@click.option("--use-bulk", is_flag=True, help="Use bulk insert method for items")
+@click.option(
+    "--data-dir",
+    type=click.Path(exists=True),
+    default="sample_data/",
+    help="Directory containing collection.json and feature collection file",
+)
+def load_data(base_url: str, collection_id: str, use_bulk: bool, data_dir: str) -> None:
+    """Load STAC items into the database via STAC API.
+
+    This command loads a STAC collection and its items from local JSON files
+    into a STAC API instance. It expects a directory containing:
+    - collection.json: The STAC collection definition
+    - One or more feature collection JSON files with STAC items
+
+    Examples:
+        sfeos-tools load-data --base-url http://localhost:8080
+        sfeos-tools load-data --base-url http://localhost:8080 --collection-id my-collection --use-bulk
+        sfeos-tools load-data --base-url http://localhost:8080 --data-dir /path/to/data
+    """
+    from httpx import Client
+
+    try:
+        with Client(base_url=base_url) as client:
+            load_items(client, collection_id, use_bulk, data_dir)
+        click.echo(click.style("✓ Data loading completed successfully", fg="green"))
+    except KeyboardInterrupt:
+        click.echo(click.style("\n✗ Data loading interrupted by user", fg="yellow"))
+        sys.exit(1)
+    except Exception as e:
+        error_msg = str(e)
+        click.echo(click.style(f"✗ Data loading failed: {error_msg}", fg="red"))
+        sys.exit(1)
+
+
 if __name__ == "__main__":
     cli()