datacosmos 0.0.7__py3-none-any.whl → 0.0.8__py3-none-any.whl

datacosmos/config/config.py

@@ -29,7 +29,6 @@ class Config(BaseSettings):
     authentication: Optional[AuthenticationConfig] = None
     stac: Optional[URL] = None
     datacosmos_cloud_storage: Optional[URL] = None
-    mission_id: int = 0
 
     DEFAULT_AUTH_TYPE: ClassVar[str] = "m2m"
     DEFAULT_AUTH_TOKEN_URL: ClassVar[str] = "https://login.open-cosmos.com/oauth/token"

datacosmos/datacosmos_client.py

@@ -1,8 +1,4 @@
-"""DatacosmosClient handles authenticated interactions with the Datacosmos API.
-
-Automatically manages token refreshing and provides HTTP convenience
-methods.
-"""
+"""Client to interact with the Datacosmos API with authentication and request handling."""
 
 from datetime import datetime, timedelta, timezone
 from typing import Any, Optional
@@ -19,23 +15,57 @@ from datacosmos.exceptions.datacosmos_exception import DatacosmosException
 class DatacosmosClient:
     """Client to interact with the Datacosmos API with authentication and request handling."""
 
-    def __init__(self, config: Optional[Config] = None):
+    def __init__(
+        self,
+        config: Optional[Config | Any] = None,
+        http_session: Optional[requests.Session | OAuth2Session] = None,
+    ):
         """Initialize the DatacosmosClient.
 
         Args:
-            config (Optional[Config]): Configuration object.
+            config (Optional[Config]): Configuration object (only needed when SDK creates its own session).
+            http_session (Optional[requests.Session]): Pre-authenticated session.
         """
-        if config:
-            self.config = config
-        else:
+        if http_session is not None:
+            self._http_client = http_session
+            self._owns_session = False
+            if isinstance(http_session, OAuth2Session):
+                token_data = http_session.token
+            elif isinstance(http_session, requests.Session):
+                auth_header = http_session.headers.get("Authorization", "")
+                if not auth_header.startswith("Bearer "):
+                    raise DatacosmosException(
+                        "Injected requests.Session must include a 'Bearer' token in its headers"
+                    )
+                token_data = {"access_token": auth_header.split(" ", 1)[1]}
+            else:
+                raise DatacosmosException(
+                    f"Unsupported session type: {type(http_session)}"
+                )
             try:
-                self.config = Config.from_yaml()
-            except ValueError:
-                self.config = Config.from_env()
+                self.token = token_data.get("access_token")
+                self.token_expiry = token_data.get("expires_at") or token_data.get(
+                    "expires_in"
+                )
+            except Exception:
+                raise DatacosmosException(
+                    "Failed to extract token from injected session"
+                )
 
-        self.token = None
-        self.token_expiry = None
-        self._http_client = self._authenticate_and_initialize_client()
+            self.config = config
+        else:
+            if config:
+                self.config = config
+            else:
+                try:
+                    self.config = Config.from_yaml()
+                except ValueError:
+                    self.config = Config.from_env()
+
+            self._owns_session = True
+            self.token = None
+            self.token_expiry = None
+            self._http_client = self._authenticate_and_initialize_client()
 
     def _authenticate_and_initialize_client(self) -> requests.Session:
         """Authenticate and initialize the HTTP client with a valid token."""
@@ -68,8 +98,10 @@ class DatacosmosClient:
             ) from e
 
     def _refresh_token_if_needed(self):
-        """Refresh the token if it has expired."""
-        if not self.token or self.token_expiry <= datetime.now(timezone.utc):
+        """Refresh the token if it has expired (only if SDK created it)."""
+        if self._owns_session and (
+            not self.token or self.token_expiry <= datetime.now(timezone.utc)
+        ):
             self._http_client = self._authenticate_and_initialize_client()
 
     def request(

datacosmos/uploader/dataclasses/upload_path.py

@@ -8,7 +8,6 @@ import structlog
 
 from datacosmos.stac.enums.processing_level import ProcessingLevel
 from datacosmos.stac.item.models.datacosmos_item import DatacosmosItem
-from datacosmos.utils.missions import get_mission_id
 
 logger = structlog.get_logger()
 
@@ -35,11 +34,6 @@ class UploadPath:
         cls, item: DatacosmosItem, mission: str, item_path: str
     ) -> "Path":
         """Create a Path instance from a DatacosmosItem and a path."""
-        for asset in item.assets.values():
-            if mission == "":
-                mission = cls._get_mission_name(asset.href)
-            else:
-                break
         dt = datetime.strptime(item.properties["datetime"], "%Y-%m-%dT%H:%M:%SZ")
         path = UploadPath(
             mission=mission,
@@ -67,27 +61,3 @@ class UploadPath:
             id=parts[5],
             path="/".join(parts[6:]),
         )
-
-    @classmethod
-    def _get_mission_name(cls, href: str) -> str:
-        mission = ""
-        # bruteforce mission name from asset path
-        # traverse the path and check if any part is a mission name (generates a mission id)
-        href_parts = href.split("/")
-        for idx, part in enumerate(href_parts):
-            try:
-                # when an id is found, then the mission name is valid
-                get_mission_id(
-                    part, "test"
-                )  # using test as it is more wide and anything on prod should exists on test
-            except KeyError:
-                continue
-            # validate the mission name by checking if the path is correct
-            # using the same logic as the __str__ method
-            mission = part.lower()
-            h = "/".join(["full", *href_parts[idx:]])
-            p = UploadPath.from_path("/".join([mission, *href_parts[idx + 1 :]]))
-            if str(p) != h:
-                raise ValueError(f"Could not find mission name in asset path {href}")
-            break
-        return mission

datacosmos/uploader/datacosmos_uploader.py

@@ -9,7 +9,6 @@ from datacosmos.datacosmos_client import DatacosmosClient
 from datacosmos.stac.item.item_client import ItemClient
 from datacosmos.stac.item.models.datacosmos_item import DatacosmosItem
 from datacosmos.uploader.dataclasses.upload_path import UploadPath
-from datacosmos.utils.missions import get_mission_name
 
 
 class DatacosmosUploader:
@@ -17,14 +16,10 @@ class DatacosmosUploader:
 
     def __init__(self, client: DatacosmosClient):
         """Initialize the uploader with DatacosmosClient."""
-        mission_id = client.config.mission_id
-        environment = client.config.environment
+        self.environment = client.config.environment
 
         self.datacosmos_client = client
         self.item_client = ItemClient(client)
-        self.mission_name = (
-            get_mission_name(mission_id, environment) if mission_id != 0 else ""
-        )
         self.base_url = client.config.datacosmos_cloud_storage.as_domain_url()
 
     def upload_and_register_item(self, item_json_file_path: str) -> None:
@@ -92,9 +87,9 @@ class DatacosmosUploader:
         except Exception:  # nosec
             pass  # Ignore if item doesn't exist
 
-    def _get_upload_path(self, item: DatacosmosItem) -> str:
+    def _get_upload_path(self, item: DatacosmosItem, mission_name: str = "") -> str:
         """Constructs the storage upload path based on the item and mission name."""
-        return UploadPath.from_item_path(item, self.mission_name, "")
+        return UploadPath.from_item_path(item, mission_name, "")
 
     def _update_item_assets(self, item: DatacosmosItem) -> None:
         """Updates the item's assets with uploaded file URLs."""

{datacosmos-0.0.7.dist-info → datacosmos-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datacosmos
-Version: 0.0.7
+Version: 0.0.8
 Summary: A library for interacting with DataCosmos from Python code
 Author-email: Open Cosmos <support@open-cosmos.com>
 Classifier: Programming Language :: Python :: 3

{datacosmos-0.0.7.dist-info → datacosmos-0.0.8.dist-info}/RECORD

@@ -1,7 +1,7 @@
 datacosmos/__init__.py,sha256=dVHKpbz5FVtfoJAWHRdsUENG6H-vs4UrkuwnIvOGJr4,66
-datacosmos/datacosmos_client.py,sha256=Uage-PRVRrDWy5frefcR2FClDsdpbFBS4JQC3OCupJU,4978
+datacosmos/datacosmos_client.py,sha256=3BurTz1fPk1Dzp8B5xt5gZZrFiqk1AT5oaqKeYmXPec,6517
 datacosmos/config/__init__.py,sha256=KCsaTb9-ZgFui1GM8wZFIPLJy0D0O8l8Z1Sv3NRD9UM,140
-datacosmos/config/config.py,sha256=xSzri8sr_hRK1JlG0mxuBW5dLQ4ExL00uk3-77Z4wHo,7243
+datacosmos/config/config.py,sha256=qj9e68enxX61Iw8lqEXUUCNJ2O7vy-1j9nAoBIB-4Ao,7219
 datacosmos/config/models/__init__.py,sha256=r3lThPkyKjBjUZXRNscFzOrmn_-m_i9DvG3RePfCFYc,41
 datacosmos/config/models/authentication_config.py,sha256=01Q90-yupbJ5orYDtatZIm9EaL7roQ-oUMoZfFMRzIM,499
 datacosmos/config/models/local_user_account_authentication_config.py,sha256=8WApn720MBXMKQa6w7bCd7Z37GRmYR-I7mBUgUI20lQ,701
@@ -32,20 +32,18 @@ datacosmos/stac/item/models/eo_band.py,sha256=YC3Scn_wFhIo51pIVcJeuJienF7JGWoEv3
 datacosmos/stac/item/models/item_update.py,sha256=_CpjQn9SsfedfuxlHSiGeptqY4M-p15t9YX__mBRueI,2088
 datacosmos/stac/item/models/raster_band.py,sha256=CoEVs-YyPE5Fse0He9DdOs4dGZpzfCsCuVzOcdXa_UM,354
 datacosmos/uploader/__init__.py,sha256=ZtfCVJ_pWKKh2F1r_NArnbG3_JtpcEiXcA_tmSwSKmQ,128
-datacosmos/uploader/datacosmos_uploader.py,sha256=QFFzR9Z2KFu_G5EcmvEn251IiwbPAfZSrOYZ_vC3NSg,4393
+datacosmos/uploader/datacosmos_uploader.py,sha256=FDBUSZ9mpieXi3dms3RM17aIrnRuAKd0rFkD7Jwl1pY,4195
 datacosmos/uploader/dataclasses/__init__.py,sha256=IjcyA8Vod-z1_Gi1FMZhK58Owman0foL25Hs0YtkYYs,43
-datacosmos/uploader/dataclasses/upload_path.py,sha256=X8zkfw3_FO9qTiKHu-nL_uDmQJYfaov6e4Y2-f-opaU,3204
+datacosmos/uploader/dataclasses/upload_path.py,sha256=5QadynHxkJrnOk1lyPtLyiVAHdzBshEuhjA9hwVF0NI,1903
 datacosmos/utils/__init__.py,sha256=XQbAnoqJrPpnSpEzAbjh84yqYWw8cBM8mNp8ynTG-54,50
-datacosmos/utils/constants.py,sha256=f7pOqCpdXk7WFGoaTyuCpr65jb-TtfhoVGuYTz3_T6Y,272
-datacosmos/utils/missions.py,sha256=7GOnrjxB8V11C_Jr3HHI4vpXifgkOSeirNjIDx17C58,940
 datacosmos/utils/url.py,sha256=iQwZr6mYRoePqUZg-k3KQSV9o2wju5ZuCa5WS_GyJo4,2114
 datacosmos/utils/http_response/__init__.py,sha256=BvOWwC5coYqq_kFn8gIw5m54TLpdfJKlW9vgRkfhXiA,33
 datacosmos/utils/http_response/check_api_response.py,sha256=dKWW01jn2_lWV0xpOBABhEP42CFSsx9dP0iSxykbN54,1186
 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.7.dist-info/licenses/LICENSE.md,sha256=vpbRI-UUbZVQfr3VG_CXt9HpRnL1b5kt8uTVbirxeyI,1486
-datacosmos-0.0.7.dist-info/METADATA,sha256=tCAZlxKlDV8qI9pjgvdREZBffzkwJ4Ko0CMiK9oe2Bg,896
-datacosmos-0.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datacosmos-0.0.7.dist-info/top_level.txt,sha256=ueobs5CNeyDbPMgXPcVV0d0yNdm8CvGtDT3CaksRVtA,11
-datacosmos-0.0.7.dist-info/RECORD,,
+datacosmos-0.0.8.dist-info/licenses/LICENSE.md,sha256=vpbRI-UUbZVQfr3VG_CXt9HpRnL1b5kt8uTVbirxeyI,1486
+datacosmos-0.0.8.dist-info/METADATA,sha256=tK8nQIG9_5zJ-gCQPQB7S2q6z7HabdgYdpPLfkinP9g,896
+datacosmos-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datacosmos-0.0.8.dist-info/top_level.txt,sha256=ueobs5CNeyDbPMgXPcVV0d0yNdm8CvGtDT3CaksRVtA,11
+datacosmos-0.0.8.dist-info/RECORD,,

datacosmos/utils/constants.py

@@ -1,16 +0,0 @@
-"""Package for storing constants."""
-
-TEST_MISSION_NAMES = {
-    55: "MENUT",
-    56: "PHISAT-2",
-    57: "HAMMER",
-    63: "MANTIS",
-    64: "PLATERO",
-}
-PROD_MISSION_NAMES = {
-    23: "MENUT",
-    29: "MANTIS",
-    35: "PHISAT-2",
-    37: "PLATERO",
-    48: "HAMMER",
-}

datacosmos/utils/missions.py

@@ -1,27 +0,0 @@
-"""Package for storing mission specific information."""
-
-from datacosmos.utils.constants import PROD_MISSION_NAMES, TEST_MISSION_NAMES
-
-
-def get_mission_name(mission: int, env: str) -> str:
-    """Get the mission name from the mission number."""
-    if env == "test" or env == "local":
-        return TEST_MISSION_NAMES[mission]
-    elif env == "prod":
-        return PROD_MISSION_NAMES[mission]
-    else:
-        raise ValueError(f"Unsupported environment: {env}")
-
-
-def get_mission_id(mission_name: str, env: str) -> int:
-    """Get the mission number from the mission name."""
-    if env == "test" or env == "local":
-        return {v.upper(): k for k, v in TEST_MISSION_NAMES.items()}[
-            mission_name.upper()
-        ]
-    elif env == "prod":
-        return {v.upper(): k for k, v in PROD_MISSION_NAMES.items()}[
-            mission_name.upper()
-        ]
-    else:
-        raise ValueError(f"Unsupported environment: {env}")