terrakio-core 0.4.96__py3-none-any.whl → 0.4.97__py3-none-any.whl

terrakio_core/__init__.py

@@ -5,7 +5,7 @@ Terrakio Core
 Core components for Terrakio API clients.
 """
 
-__version__ = "0.4.96"
+__version__ = "0.4.97"
 
 from .async_client import AsyncClient
 from .sync_client import SyncClient as Client

terrakio_core/convenience_functions/zonal_stats.py

@@ -513,6 +513,35 @@ async def zonal_stats(
         job_name = await client.mass_stats.track_job([mass_stats_id])
         job_name = job_name[mass_stats_id]["name"]
         cloud_files_object = cloud_object(job_id = mass_stats_id, job_name = job_name, client = client)
+        
+        # Attach id column behavior to cloud object via a wrapper method
+        async def _head_with_id(n = 5):
+            result_gdf = await cloud_files_object._head_async(n)
+            if id_column is not None and id_column in gdf.columns:
+                geometry_to_id = {geom.wkb: id_val for geom, id_val in zip(gdf.geometry, gdf[id_column])}
+                if hasattr(result_gdf.index, 'names') and 'geometry' in result_gdf.index.names:
+                    if isinstance(result_gdf.index, pd.MultiIndex):
+                        geometry_index = result_gdf.index.get_level_values('geometry')
+                    else:
+                        geometry_index = result_gdf.index
+                    id_values = [geometry_to_id.get(geom.wkb) for geom in geometry_index]
+                    result_gdf[id_column] = id_values
+                    result_gdf = result_gdf.reset_index()
+                    if 'time' in result_gdf.columns:
+                        result_gdf = result_gdf.set_index([id_column, 'geometry', 'time'])
+                    else:
+                        result_gdf = result_gdf.set_index([id_column, 'geometry'])
+                else:
+                    id_values = [geometry_to_id.get(geom.wkb) for geom in result_gdf['geometry']]
+                    result_gdf[id_column] = id_values
+                    if 'time' in result_gdf.columns:
+                        result_gdf = result_gdf.set_index([id_column, 'geometry', 'time'])
+                    else:
+                        result_gdf = result_gdf.set_index([id_column, 'geometry'])
+            return result_gdf
+
+        # Monkey-patch a convenience method without modifying original class contract
+        cloud_files_object.head_with_id = lambda n=5: asyncio.run(_head_with_id(n))
         return cloud_files_object
     
     quries = []
@@ -539,5 +568,33 @@ async def zonal_stats(
             "is_cloud_backed": False,
         } 
         gdf_with_datasets = expand_on_variables_and_time(gdf_with_datasets)
+        
+        # If an id_column is provided, attach it to the result and include in the index
+        if id_column is not None and id_column in gdf.columns:
+            # Build a mapping from input geometries to id values (use WKB for robust equality)
+            geometry_to_id = {geom.wkb: id_val for geom, id_val in zip(gdf.geometry, gdf[id_column])}
+
+            # Determine geometry values in the result (index may be geometry or (geometry, time))
+            if hasattr(gdf_with_datasets.index, 'names') and 'geometry' in gdf_with_datasets.index.names:
+                if isinstance(gdf_with_datasets.index, pd.MultiIndex):
+                    geometry_index = gdf_with_datasets.index.get_level_values('geometry')
+                else:
+                    geometry_index = gdf_with_datasets.index
+                id_values = [geometry_to_id.get(geom.wkb) for geom in geometry_index]
+                gdf_with_datasets[id_column] = id_values
+                # Reset index to control index composition precisely, then set to desired levels
+                gdf_with_datasets = gdf_with_datasets.reset_index()
+                if 'time' in gdf_with_datasets.columns:
+                    gdf_with_datasets = gdf_with_datasets.set_index([id_column, 'geometry', 'time'])
+                else:
+                    gdf_with_datasets = gdf_with_datasets.set_index([id_column, 'geometry'])
+            else:
+                # geometry exists as a column
+                id_values = [geometry_to_id.get(geom.wkb) for geom in gdf_with_datasets['geometry']]
+                gdf_with_datasets[id_column] = id_values
+                if 'time' in gdf_with_datasets.columns:
+                    gdf_with_datasets = gdf_with_datasets.set_index([id_column, 'geometry', 'time'])
+                else:
+                    gdf_with_datasets = gdf_with_datasets.set_index([id_column, 'geometry'])
     return gdf_with_datasets
 

terrakio_core/endpoints/dataset_management.py

@@ -69,7 +69,7 @@ class DatasetManagement:
             name: Name of the dataset (required)
             collection: Dataset collection (default: 'terrakio-datasets')
             products: List of products
-            dates_iso8601: List of dates
+            dates_iso8601: List of dates (will be automatically sorted chronologically)
             bucket: Storage bucket
             path: Storage path
             data_type: Data type
@@ -142,7 +142,7 @@ class DatasetManagement:
             append: Whether to append data or replace (default: True)
             collection: Dataset collection (default: 'terrakio-datasets')
             products: List of products
-            dates_iso8601: List of dates
+            dates_iso8601: List of dates (will be automatically sorted chronologically)
             bucket: Storage bucket
             path: Storage path
             data_type: Data type
@@ -162,6 +162,10 @@ class DatasetManagement:
         Raises:
             APIError: If the API request fails
         """
+        # Sort dates_iso8601 chronologically if provided
+        if dates_iso8601 is not None:
+            dates_iso8601 = sorted(dates_iso8601)
+
         params = {"collection": collection, "append": str(append).lower()}
         payload = {"name": name}
         param_mapping = {
@@ -215,7 +219,7 @@ class DatasetManagement:
             append: Whether to append data or replace (default: True)
             collection: Dataset collection (default: 'terrakio-datasets')
             products: List of products
-            dates_iso8601: List of dates
+            dates_iso8601: List of dates (will be automatically sorted chronologically)
             bucket: Storage bucket
             path: Storage path
             data_type: Data type
@@ -236,6 +240,10 @@ class DatasetManagement:
         Raises:
             APIError: If the API request fails
         """
+        # Sort dates_iso8601 chronologically if provided
+        if dates_iso8601 is not None:
+            dates_iso8601 = sorted(dates_iso8601)
+
         params = {"collection": collection, "append": str(append).lower()}
         payload = {"name": name}
         param_mapping = {
@@ -289,7 +297,7 @@ class DatasetManagement:
             name: Name of the dataset (required)
             collection: Dataset collection (default: 'terrakio-datasets')
             products: List of products
-            dates_iso8601: List of dates
+            dates_iso8601: List of dates (will be automatically sorted chronologically)
             bucket: Storage bucket
             path: Storage path
             data_type: Data type

{terrakio_core-0.4.96.dist-info → terrakio_core-0.4.97.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: terrakio-core
-Version: 0.4.96
+Version: 0.4.97
 Summary: Core package for the terrakio-python-api
 Requires-Python: >=3.11
 Requires-Dist: aiofiles>=24.1.0

{terrakio_core-0.4.96.dist-info → terrakio_core-0.4.97.dist-info}/RECORD

@@ -1,4 +1,4 @@
-terrakio_core/__init__.py,sha256=5fmKNGnXc1Yt5XJTaXPIfcmP2XVJFWPeH6apnD8xs5w,274
+terrakio_core/__init__.py,sha256=HTlEcbJMNK-hrdq99CSMO6xU-A4UTVRBi3fVglJFSRQ,274
 terrakio_core/accessors.py,sha256=UZIi9y4RpBxouSmKuwuNYLIYqDxD8BH-GnUzwJuc1JI,47570
 terrakio_core/async_client.py,sha256=txdSsX3IwqtHlcS86u6N6vjV0-PIiermxNOIjEMQ3Yg,14950
 terrakio_core/client.py,sha256=VXP7BtJWIfpPPZR7_yNdSTcGwNgTwhb7KorusqkQrzk,5603
@@ -7,9 +7,9 @@ terrakio_core/exceptions.py,sha256=4qnpOM1gOxsNIXDXY4qwY1d3I4Myhp7HBh7b2D0SVrU,5
 terrakio_core/sync_client.py,sha256=jbG2sMnbR3QPvhAxQX2dBWeX_6f-Qx_MFSRLLpvfRh4,14604
 terrakio_core/convenience_functions/create_dataset_file.py,sha256=RDTAQnKUigyczv3EKhKrs34VMDZDCgL4iz0bge1d9e4,4774
 terrakio_core/convenience_functions/geoquries.py,sha256=7E3drOD5ffNk2-rKLbwKsNp3_Berq-S1lQk5wwHSuAo,3786
-terrakio_core/convenience_functions/zonal_stats.py,sha256=5YBy9V1QK0iovuLffuQY3M2n3NXpY8l__n-6F8vEuOY,20149
+terrakio_core/convenience_functions/zonal_stats.py,sha256=Sg_T3_85acMPvZkDxBf3fMTmNXnEfKnjVCEB7SKT4Fc,23807
 terrakio_core/endpoints/auth.py,sha256=FdLsPScPIBo-Gxl6ZnE-46cp2molggAJtL72LssN3fg,6049
-terrakio_core/endpoints/dataset_management.py,sha256=BUm8IIlW_Q45vDiQp16CiJGeSLheI8uWRVRQtMdhaNk,13161
+terrakio_core/endpoints/dataset_management.py,sha256=D2foX8DGbSXQ4vYLRt0Es3j96a_qfd920Ct3uN3dd7Y,13641
 terrakio_core/endpoints/group_management.py,sha256=VFl3jakjQa9OPi351D3DZvLU9M7fHdfjCzGhmyJsx3U,6309
 terrakio_core/endpoints/mass_stats.py,sha256=Vb6Tf8kKf5Hlch4ddsrQnfayfiK6z7NSjO8D0pop4p8,25699
 terrakio_core/endpoints/model_management.py,sha256=LH_gHPrqYA-_45KWpDBRcFbwHgm-Kg0zk1ealy7P_C0,52379
@@ -18,6 +18,6 @@ terrakio_core/endpoints/user_management.py,sha256=WlFr3EfK8iI6DfkpMuYLHZUPk2n7_D
 terrakio_core/helper/bounded_taskgroup.py,sha256=wiTH10jhKZgrsgrFUNG6gig8bFkUEPHkGRT2XY7Rgmo,677
 terrakio_core/helper/decorators.py,sha256=L6om7wmWNgCei3Wy5U0aZ-70OzsCwclkjIf7SfQuhCg,2289
 terrakio_core/helper/tiles.py,sha256=lcLCO6KiP05lCI9vngo3zCZJ6Z9C0pUxHSQS4H58EHc,2699
-terrakio_core-0.4.96.dist-info/METADATA,sha256=vX7L1UsQc_JDco4PWgnQgIdUehRq5CGFMftVJGz4pFs,1151
-terrakio_core-0.4.96.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-terrakio_core-0.4.96.dist-info/RECORD,,
+terrakio_core-0.4.97.dist-info/METADATA,sha256=wxkjwA2YX_vXeBnpenFaVjoglb7Muu84Oe5UdY5-jyM,1151
+terrakio_core-0.4.97.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+terrakio_core-0.4.97.dist-info/RECORD,,