specklia 1.9.21__py3-none-any.whl → 1.9.22__py3-none-any.whl

specklia/chunked_transfer.py

@@ -212,7 +212,7 @@ def deserialise_dataframe(data: bytes) -> Union[DataFrame, GeoDataFrame]:
     """
     try:
         buffer = BytesIO(data)
-        df = read_geofeather(buffer)
+        df = read_geofeather(buffer)  # type: ignore
     except ValueError as e:
         # First attempt to deserialise as a geodataframe. If geo meta is missing, we expect a clear ValueError
         # and we then load as a plain dataframe instead.

specklia/client.py

@@ -4,7 +4,7 @@ from __future__ import annotations
 from datetime import datetime
 import json
 import logging
-from typing import Dict, List, Literal, Optional, Tuple, Union
+from typing import cast, Dict, List, Literal, Optional, Tuple, Union
 import warnings
 
 from dateutil import parser
@@ -15,6 +15,7 @@ from shapely import MultiPolygon, Polygon, to_geojson
 from shapely.geometry import shape
 
 from specklia import chunked_transfer, utilities
+from specklia.utilities import NewPoints
 
 _log = logging.getLogger(__name__)
 
@@ -239,7 +240,7 @@ class Specklia:
         # perform some light deserialisation of sources for backwards compatibility.
         sources = utilities.deserialise_sources(response_dict['sources'])
 
-        return gdf, sources
+        return cast(gpd.GeoDataFrame, gdf), cast(list[dict], sources)
 
     def update_points_in_dataset(
             self: Specklia, _dataset_id: str, _new_points: pd.DataFrame, _source_description: Dict) -> None:
@@ -270,7 +271,7 @@ class Specklia:
         raise NotImplementedError()
 
     def add_points_to_dataset(
-            self: Specklia, dataset_id: str, new_points: List[Dict[str, Union[Dict, gpd.GeoDataFrame]]],
+            self: Specklia, dataset_id: str, new_points: List[NewPoints],
             duplicate_source_behaviour: Literal['error', 'ignore', 'replace', 'merge'] = 'error') -> None:
         """
         Add new data to a dataset.
@@ -309,17 +310,22 @@ class Specklia:
             - 'merge': Append incoming data to existing data, sharing the same source.
         """
         # serialise and upload each dataframe
+        upload_points = []
         for n in new_points:
-            n['chunk_set_uuid'] = chunked_transfer.upload_chunks(
+            chunk_set_uuid = chunked_transfer.upload_chunks(
                 self.server_url, chunked_transfer.split_into_chunks(
                     chunked_transfer.serialise_dataframe(n['gdf'])), _log)
-            del n['gdf']
+            upload_points.append({
+                'source': n['source'],
+                'chunk_set_uuid': chunk_set_uuid
+            })
+            del n
 
         response = requests.post(
             self.server_url + "/ingest",
             json={
                 'dataset_id': dataset_id,
-                'new_points': new_points,
+                'new_points': upload_points,
                 'duplicate_source_behaviour': duplicate_source_behaviour,
             },
             headers={"Authorization": "Bearer " + self.auth_token},
@@ -672,7 +678,7 @@ class Specklia:
                     lambda x: parser.parse(x, ignoretz=True) if x is not None else None)
             if column == 'epsg4326_coverage':
                 datasets_df[column] = gpd.GeoSeries(
-                    datasets_df[column].apply(lambda x: shape(x) if x is not None else None), crs=4326)
+                    datasets_df[column].apply(lambda x: shape(x) if x is not None else None), crs=4326)  # type: ignore
 
         return datasets_df.convert_dtypes()  # convert the rest of the dtypes to pandas' best guest
 

specklia/utilities.py

@@ -1,11 +1,12 @@
 """This file contains client-side utilities provided to make it easier to use Specklia."""
 from datetime import datetime
 import os
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, TypedDict
 
 import geopandas as gpd
 import numpy as np
 import rasterio
+import rasterio.transform
 from shapely.geometry import shape
 
 
@@ -113,3 +114,10 @@ def deserialise_sources(sources: List[Dict]) -> List[Dict]:
         source['max_time'] = datetime.fromisoformat(source['max_time'])
 
     return sources
+
+
+class NewPoints(TypedDict):
+    """List of data points to ingest."""
+
+    gdf: gpd.GeoDataFrame
+    source: dict

{specklia-1.9.21.dist-info → specklia-1.9.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: specklia
-Version: 1.9.21
+Version: 1.9.22
 Summary: Python client for Specklia, a geospatial point cloud database by Earthwave.
 Home-page: https://specklia.earthwave.co.uk/
 Author: Earthwave Ltd

specklia-1.9.22.dist-info/RECORD

@@ -0,0 +1,9 @@
+specklia/__init__.py,sha256=ePVHqq642NocoE8tS0cNTd0B5wJdUB7r3y815oQXD6A,51
+specklia/chunked_transfer.py,sha256=sQGh_0M3DcJw4Uu1Upb7zA9L1syupI9JXVL_94OoNEs,8343
+specklia/client.py,sha256=IlOr3CP_0BJpB0oOvkvHFgtiqmmFynpUriDM3NHUVeU,42088
+specklia/utilities.py,sha256=fs9DOSq-0hdgOlGAnPY_og5QngDcu3essVAupz6ychM,5170
+specklia-1.9.22.dist-info/LICENCE,sha256=kjWTA-TtT_rJtsWuAgWvesvu01BytVXgt_uCbeQgjOg,1061
+specklia-1.9.22.dist-info/METADATA,sha256=D_E5534h_qfU9v4B7djdSEOWw3XheSIt5JHIBmoR24k,3082
+specklia-1.9.22.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+specklia-1.9.22.dist-info/top_level.txt,sha256=XgU53UpAJbqEni5EjJaPdQPYuNx16Geg2I5A9lo1BQw,9
+specklia-1.9.22.dist-info/RECORD,,

specklia-1.9.21.dist-info/RECORD

@@ -1,9 +0,0 @@
-specklia/__init__.py,sha256=ePVHqq642NocoE8tS0cNTd0B5wJdUB7r3y815oQXD6A,51
-specklia/chunked_transfer.py,sha256=Mw0wLaOYzNvVTRE421RdHnsnaDbEpNYgQNGV_kcjuVs,8327
-specklia/client.py,sha256=P4XXzC24OuLK5S6joc7d0zt9Q39-IveAK87EUCF7n3M,41858
-specklia/utilities.py,sha256=0_pgTbcq2RgQhys0-CZ6h5YZJg9ZMPhD_ibGPggFUpE,5018
-specklia-1.9.21.dist-info/LICENCE,sha256=kjWTA-TtT_rJtsWuAgWvesvu01BytVXgt_uCbeQgjOg,1061
-specklia-1.9.21.dist-info/METADATA,sha256=JIadoGfwW8TPR6JqHW-h_N84EdAU9Dn8-Gy1E3M60sQ,3082
-specklia-1.9.21.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-specklia-1.9.21.dist-info/top_level.txt,sha256=XgU53UpAJbqEni5EjJaPdQPYuNx16Geg2I5A9lo1BQw,9
-specklia-1.9.21.dist-info/RECORD,,