datacosmos 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

datacosmos/stac/collection/collection_client.py

@@ -6,6 +6,7 @@ from pystac import Collection, Extent, SpatialExtent, TemporalExtent
 from pystac.utils import str_to_datetime
 
 from datacosmos.datacosmos_client import DatacosmosClient
+from datacosmos.exceptions.datacosmos_exception import DatacosmosException
 from datacosmos.stac.collection.models.collection_update import CollectionUpdate
 from datacosmos.utils.http_response.check_api_response import check_api_response
 
@@ -145,5 +146,8 @@ class CollectionClient:
         """
         try:
             return next_href.split("?")[1].split("=")[-1]
-        except (IndexError, AttributeError):
-            raise InvalidRequest(f"Failed to parse pagination token from {next_href}")
+        except (IndexError, AttributeError) as e:
+            raise DatacosmosException(
+                f"Failed to parse pagination token from {next_href}",
+                response=e.response,
+            ) from e

datacosmos/stac/collection/models/collection_update.py

@@ -2,6 +2,7 @@
 
 Allows partial updates where only the provided fields are modified.
 """
+
 from typing import Any, Dict, List, Optional
 
 from pydantic import BaseModel, Field

datacosmos/stac/constants/__init__.py

@@ -0,0 +1 @@
+"""Constants for STAC."""

datacosmos/stac/constants/satellite_name_mapping.py

@@ -0,0 +1,20 @@
+"""Satellite name mapping."""
+
+SATELLITE_NAME_MAPPING = {
+    "GEOSAT-2": "2014-033D",
+    "SUPERVIEW-1-01": "2016-083A",
+    "SUPERVIEW-1-02": "2016-083B",
+    "SUPERVIEW-1-03": "2018-002A",
+    "SUPERVIEW-1-04": "2018-002B",
+    "MANTIS": "2023-174B",
+    "MENUT": "2023-001B",
+    "HAMMER": "2024-043BC",
+    "HAMMER-EM": "COSPAR-HAMMER-EM-TBD",
+    "Alisio": "2023-185M",
+    "Platero": "2023-174G",
+    "PHISAT-2": "2024-149C",
+    "PHISAT-2 EM": "COSPAR-PHISAT2-EM-TBD",
+    "Sentinel-2A": "2015-028A",
+    "Sentinel-2B": "2017-013A",
+    "Sentinel-2C": "2024-157A",
+}

datacosmos/stac/enums/processing_level.py

@@ -0,0 +1,15 @@
+"""Level enum class."""
+
+from enum import Enum
+
+
+class ProcessingLevel(Enum):
+    """Enum class for the processing levels of the data."""
+
+    L0 = "L0"
+    L1A = "L1A"
+    L2A = "L2A"
+    L1B = "L1B"
+    L1C = "L1C"
+    L1D = "L1D"
+    L3 = "L3"

datacosmos/stac/enums/product_type.py

@@ -0,0 +1,11 @@
+"""Product type enum class."""
+
+from enum import Enum
+
+
+class ProductType(str, Enum):
+    """Different product types."""
+
+    SATELLITE = "Satellite"
+    VECTOR = "Vector"
+    INSIGHT = "Insight"

datacosmos/stac/enums/season.py

@@ -0,0 +1,14 @@
+"""Season enum class."""
+
+from enum import Enum
+
+
+class Season(str, Enum):
+    """Different Open Cosmos seasons."""
+
+    SUMMER = "Summer"
+    WINTER = "Winter"
+    AUTUMN = "Autumn"
+    SPRING = "Spring"
+    RAINY = "Rainy"
+    DRY = "Dry"

datacosmos/stac/item/item_client.py

@@ -9,6 +9,9 @@ from pystac import Item
 
 from datacosmos.datacosmos_client import DatacosmosClient
 from datacosmos.exceptions.datacosmos_exception import DatacosmosException
+from datacosmos.stac.item.models.catalog_search_parameters import (
+    CatalogSearchParameters,
+)
 from datacosmos.stac.item.models.datacosmos_item import DatacosmosItem
 from datacosmos.stac.item.models.item_update import ItemUpdate
 from datacosmos.stac.item.models.search_parameters import SearchParameters
@@ -59,17 +62,20 @@ class ItemClient:
 
         return self.search_items(parameters)
 
-    def search_items(self, parameters: SearchParameters) -> Generator[Item, None, None]:
+    def search_items(
+        self, parameters: CatalogSearchParameters, project_id: str
+    ) -> Generator[Item, None, None]:
         """Query the STAC catalog using the POST endpoint with filtering and pagination.
 
         Args:
-            parameters (SearchParameters): The search parameters.
+            parameters (CatalogSearchParameters): The search parameters.
 
         Yields:
             Item: Parsed STAC item.
         """
         url = self.base_url.with_suffix("/search")
-        body = parameters.model_dump(by_alias=True, exclude_none=True)
+        parameters_query = parameters.to_query()
+        body = {"project": project_id, "limit": 50, "query": parameters_query}
         return self._paginate_items(url, body)
 
     def create_item(self, collection_id: str, item: Item | DatacosmosItem) -> None:

datacosmos/stac/item/models/catalog_search_parameters.py

@@ -0,0 +1,132 @@
+"""Query parameters for catalog search."""
+
+from datetime import datetime, timedelta
+from typing import Any, List, Optional
+
+from pydantic import BaseModel, field_validator, model_validator
+
+from datacosmos.stac.constants.satellite_name_mapping import SATELLITE_NAME_MAPPING
+from datacosmos.stac.enums.processing_level import ProcessingLevel
+from datacosmos.stac.enums.product_type import ProductType
+from datacosmos.stac.enums.season import Season
+
+
+class CatalogSearchParameters(BaseModel):
+    """Query parameters for catalog search."""
+
+    start_date: Optional[str] = None
+    end_date: Optional[str] = None
+    seasons: Optional[List[Season]] = None
+    satellite: Optional[List[str]] = None
+    product_type: Optional[List[ProductType]] = None
+    processing_level: Optional[List[ProcessingLevel]] = None
+
+    # --- Field Validators ---
+
+    @field_validator("seasons", mode="before")
+    @classmethod
+    def parse_seasons(cls, value):
+        """Parses seasons values into a list of Season object."""
+        if value is None:
+            return None
+        return [Season(v) if not isinstance(v, Season) else v for v in value]
+
+    @field_validator("product_type", mode="before")
+    @classmethod
+    def parse_product_types(cls, value):
+        """Parses product types values into a list of ProductType object."""
+        if value is None:
+            return None
+        return [ProductType(v) if not isinstance(v, ProductType) else v for v in value]
+
+    @field_validator("processing_level", mode="before")
+    @classmethod
+    def parse_processing_levels(cls, value):
+        """Parses processing levels values into a list of ProcessingLevel object."""
+        if value is None:
+            return None
+        return [
+            ProcessingLevel(v) if not isinstance(v, ProcessingLevel) else v
+            for v in value
+        ]
+
+    @field_validator("start_date", mode="before")
+    @classmethod
+    def parse_start_date(cls, value: Any) -> Optional[str]:
+        """Validations on start_date."""
+        if value is None:
+            return None
+        try:
+            dt = datetime.strptime(value, "%m/%d/%Y")
+            if dt < datetime(2015, 5, 15):
+                raise ValueError("Date must be 5/15/2015 or later.")
+            return dt.isoformat() + "Z"
+        except ValueError:
+            raise ValueError(
+                "Invalid start_date format. Use mm/dd/yyyy (e.g., 05/15/2024)"
+            )
+
+    @field_validator("end_date", mode="before")
+    @classmethod
+    def parse_end_date(cls, value: Any) -> Optional[str]:
+        """Validations on end_date."""
+        if value is None:
+            return None
+        try:
+            dt = datetime.strptime(value, "%m/%d/%Y")
+            if dt < datetime(2015, 5, 15):
+                raise ValueError("Date must be 5/15/2015 or later.")
+            dt = dt + timedelta(days=1) - timedelta(milliseconds=1)
+            return dt.isoformat() + "Z"
+        except ValueError:
+            raise ValueError(
+                "Invalid end_date format. Use mm/dd/yyyy (e.g., 05/15/2024)"
+            )
+
+    # --- Model Validator ---
+
+    @model_validator(mode="after")
+    def validate_date_range(self) -> "CatalogSearchParameters":
+        """Checks if end_date is after the start_date."""
+        if self.start_date and self.end_date:
+            start_dt = datetime.fromisoformat(self.start_date.rstrip("Z"))
+            end_dt = datetime.fromisoformat(self.end_date.rstrip("Z"))
+            if start_dt > end_dt:
+                raise ValueError("end_date cannot be before start_date.")
+        return self
+
+    # --- Query Mapper ---
+
+    def to_query(self) -> dict:
+        """Map user-friendly input to STAC query structure."""
+        query = {}
+
+        if self.start_date or self.end_date:
+            query["datetime"] = {"gte": self.start_date, "lte": self.end_date}
+
+        if self.seasons:
+            query["opencosmos:season"] = {
+                "in": [seasons.value for seasons in self.seasons]
+            }
+
+        if self.product_type:
+            query["opencosmos:product_type"] = {
+                "in": [product_type.value for product_type in self.product_type]
+            }
+
+        if self.processing_level:
+            query["processing:level"] = {
+                "in": [
+                    processing_level.value for processing_level in self.processing_level
+                ]
+            }
+
+        if self.satellite:
+            cospars = [
+                SATELLITE_NAME_MAPPING[ui]
+                for ui in self.satellite
+                if ui in SATELLITE_NAME_MAPPING
+            ]
+            query["sat:platform_international_designator"] = {"in": cospars}
+
+        return query

datacosmos/stac/item/models/datacosmos_item.py

@@ -4,7 +4,7 @@ from datetime import datetime
 
 from pydantic import BaseModel
 
-from datacosmos.stac.enums.level import Level
+from datacosmos.stac.enums.processing_level import ProcessingLevel
 from datacosmos.stac.item.models.asset import Asset
 
 
@@ -36,9 +36,9 @@ class DatacosmosItem(BaseModel):
         return datetime.strptime(self.properties["datetime"], "%Y-%m-%dT%H:%M:%SZ")
 
     @property
-    def level(self) -> Level:
+    def level(self) -> ProcessingLevel:
         """Get the processing level of the Datacosmos item."""
-        return Level(self.properties["processing:level"].lower())
+        return ProcessingLevel(self.properties["processing:level"].lower())
 
     @property
     def sat_int_designator(self) -> str:

{datacosmos-0.0.3.dist-info → datacosmos-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datacosmos
-Version: 0.0.3
+Version: 0.0.4
 Summary: A library for interacting with DataCosmos from Python code
 Author-email: Open Cosmos <support@open-cosmos.com>
 Classifier: Programming Language :: Python :: 3
@@ -13,6 +13,7 @@ Requires-Dist: oauthlib==3.2.0
 Requires-Dist: requests-oauthlib==1.3.1
 Requires-Dist: pydantic==2.10.6
 Requires-Dist: pystac==1.12.1
+Requires-Dist: pyyaml==6.0.2
 Provides-Extra: dev
 Requires-Dist: black==22.3.0; extra == "dev"
 Requires-Dist: ruff==0.9.5; extra == "dev"

{datacosmos-0.0.3.dist-info → datacosmos-0.0.4.dist-info}/RECORD

@@ -10,16 +10,21 @@ datacosmos/exceptions/datacosmos_exception.py,sha256=rKjJvQDvCEbxXWWccxB5GI_sth6
 datacosmos/stac/__init__.py,sha256=B4x_Mr4X7TzQoYtRC-VzI4W-fEON5WUOaz8cWJbk3Fc,214
 datacosmos/stac/stac_client.py,sha256=Cz_p96RmAgWX8t7Sye4OJRanQpCLihKStvfEw7IgYZc,472
 datacosmos/stac/collection/__init__.py,sha256=VQMLnsU3sER5kh4YxHrHP7XCA3DG1y0n9yoSmvycOY0,212
-datacosmos/stac/collection/collection_client.py,sha256=XTO2s309-cktJosvnwnFFXHDVmJc4vjvbEsZjpsCDmY,5904
+datacosmos/stac/collection/collection_client.py,sha256=-Nn3yqL4mQS05YAMd0IUmv03hdHKYBtVG2_EqoaAQWc,6064
 datacosmos/stac/collection/models/__init__.py,sha256=TQaihUS_CM9Eaekm4SbzFTNfv7BmabHv3Z-f37Py5Qs,40
-datacosmos/stac/collection/models/collection_update.py,sha256=Tqmfg4H4UQj5jsgy1dpKJCR59NSfWeiCSi9y8CY8-Cg,1656
+datacosmos/stac/collection/models/collection_update.py,sha256=XC6-29nLz1VGWMxYAw7r1OuL8PdJ3b2oI-RPvnM-XXI,1657
+datacosmos/stac/constants/__init__.py,sha256=dDRSsF7CKqNF44yIlNdE-PD1sp0Q5mhTEPT7hHIK7YE,26
+datacosmos/stac/constants/satellite_name_mapping.py,sha256=EJqNdO9uW5B-sIeDF72AjnW7va5BM9mm4oNwijtl51w,575
 datacosmos/stac/enums/__init__.py,sha256=GUEL2xGtdjsrszrxivs0X6daxkaZs2JsTu2JoBtsvB4,22
-datacosmos/stac/enums/level.py,sha256=dqrkSRtoutMTWatGyRRUz3uNKVlNXn3qa_ubXNbw618,237
+datacosmos/stac/enums/processing_level.py,sha256=5gHG-0kG5rCUxmXYwF3t94ALKk6zUqguOdyTL-jwgps,247
+datacosmos/stac/enums/product_type.py,sha256=7lL0unJ1hxevW8Pepn9rmydUUWIORu2x4MEtp6rSFbA,196
+datacosmos/stac/enums/season.py,sha256=QvUzXBYtPEfixhlbV0SAw2u_HK3tRFEnHKshJyIatdg,241
 datacosmos/stac/item/__init__.py,sha256=lRuD_yp-JxoLqBA23q0XMkCNImf4T-X3BJnSw9u_3Yk,200
-datacosmos/stac/item/item_client.py,sha256=E6zHf3ANzVXd5Di_u05mLen5-EOKTdCs0VKcXXJ6lUc,6826
+datacosmos/stac/item/item_client.py,sha256=ib848Pb2j6njvbx97vFFw7AWeKyBnBlK-05D3pFmIdU,7027
 datacosmos/stac/item/models/__init__.py,sha256=bcOrOcIxGxGBrRVIyQVxSM3C3Xj_qzxIHgQeWo6f7Q8,34
 datacosmos/stac/item/models/asset.py,sha256=mvg_fenYCGOTMGwXXpK2nyqBk5RMsUYxl6KhQTWW_b0,631
-datacosmos/stac/item/models/datacosmos_item.py,sha256=jHuOkNvbVXUHdpplClPnA5mR4mcrfYQNm51EgQaVtNk,1704
+datacosmos/stac/item/models/catalog_search_parameters.py,sha256=VKMBnaTYjpn_zAM6wdk3P69ZoGBcqdzPcTGBlYPFvVk,4704
+datacosmos/stac/item/models/datacosmos_item.py,sha256=AImz0GRxrpZfIETdzzNfaKX35wpr39Q4f4u0z6r8eys,1745
 datacosmos/stac/item/models/eo_band.py,sha256=YC3Scn_wFhIo51pIVcJeuJienF7JGWoEv39JngDM6rI,309
 datacosmos/stac/item/models/item_update.py,sha256=_CpjQn9SsfedfuxlHSiGeptqY4M-p15t9YX__mBRueI,2088
 datacosmos/stac/item/models/raster_band.py,sha256=CoEVs-YyPE5Fse0He9DdOs4dGZpzfCsCuVzOcdXa_UM,354
@@ -37,8 +42,8 @@ datacosmos/utils/http_response/check_api_response.py,sha256=dKWW01jn2_lWV0xpOBAB
 datacosmos/utils/http_response/models/__init__.py,sha256=Wj8YT6dqw7rAz_rctllxo5Or_vv8DwopvQvBzwCTvpw,45
 datacosmos/utils/http_response/models/datacosmos_error.py,sha256=Uqi2uM98nJPeCbM7zngV6vHSk97jEAb_nkdDEeUjiQM,740
 datacosmos/utils/http_response/models/datacosmos_response.py,sha256=oV4n-sue7K1wwiIQeHpxdNU8vxeqF3okVPE2rydw5W0,336
-datacosmos-0.0.3.dist-info/licenses/LICENSE.md,sha256=vpbRI-UUbZVQfr3VG_CXt9HpRnL1b5kt8uTVbirxeyI,1486
-datacosmos-0.0.3.dist-info/METADATA,sha256=ejuFFFnmdaInVXwXQtO6m35BQdTyotYYMLDVT-oWec4,843
-datacosmos-0.0.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-datacosmos-0.0.3.dist-info/top_level.txt,sha256=Iu5b533Fmdfz0rFKTnuBPjSUOQL2lEkTfHxsokP72s4,18
-datacosmos-0.0.3.dist-info/RECORD,,
+datacosmos-0.0.4.dist-info/licenses/LICENSE.md,sha256=vpbRI-UUbZVQfr3VG_CXt9HpRnL1b5kt8uTVbirxeyI,1486
+datacosmos-0.0.4.dist-info/METADATA,sha256=8yM44qsv1vBDxutLlUte916Z6zgPVT0mnyR1Bz2drJ8,872
+datacosmos-0.0.4.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+datacosmos-0.0.4.dist-info/top_level.txt,sha256=Iu5b533Fmdfz0rFKTnuBPjSUOQL2lEkTfHxsokP72s4,18
+datacosmos-0.0.4.dist-info/RECORD,,

{datacosmos-0.0.3.dist-info → datacosmos-0.0.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

datacosmos/stac/enums/level.py

@@ -1,15 +0,0 @@
-"""Level enum class."""
-
-from enum import Enum
-
-
-class Level(Enum):
-    """Enum class for the processing levels of the data."""
-
-    L0 = "l0"
-    L1A = "l1a"
-    L2A = "l2a"
-    L1B = "l1b"
-    L1C = "l1c"
-    L1D = "l1d"
-    L3 = "l3"