Ruff format fixes

Author: lukasbindreiter

tilebox-datasets/tests/query/time_interval.py

@@ -49,7 +49,7 @@ def datetime_scalar_for_datetime(draw: DrawFn, dt: datetime) -> DatetimeScalar:
     """
     understood_formats = [
         lambda dt: dt,  # converting a datetime to a datetime scalar should be a no-op
-        lambda dt: pd.to_datetime(dt),  # pandas Timestamp objects are also supported
+        lambda dt: pd.to_datetime(dt),  # noqa: PLW0108 # pandas Timestamp objects are also supported
         lambda dt: pd.to_datetime(dt).to_datetime64(),  # and so are numpy datetime64 objects
         lambda dt: datetime_to_us(dt) * 10**3,  # timestamp in nanoseconds
         # as well as strings in various formats

tilebox-datasets/tilebox/datasets/data/data_access.py

@@ -62,8 +62,8 @@ class SpatialFilter:
     def from_message(cls, filter_message: data_access_pb2.SpatialFilter) -> "SpatialFilter":
         return SpatialFilter(
             geometry=from_wkb(filter_message.geometry.wkb),
-            mode=_filter_mode_int_to_enum.get(filter_message.mode, None),
-            coordinate_system=_coordinate_system_int_to_enum.get(filter_message.coordinate_system, None),
+            mode=_filter_mode_int_to_enum.get(filter_message.mode),
+            coordinate_system=_coordinate_system_int_to_enum.get(filter_message.coordinate_system),
         )
 
     def to_message(self) -> data_access_pb2.SpatialFilter:
@@ -91,10 +91,10 @@ def parse(cls, spatial_filter_like: SpatialFilterLike) -> "SpatialFilter":
         if isinstance(spatial_filter_like, dict):
             mode = spatial_filter_like.get("mode", None)
             if isinstance(mode, str):
-                mode = _filter_modes_from_string.get(mode.lower(), None)
+                mode = _filter_modes_from_string.get(mode.lower())
             coordinate_system = spatial_filter_like.get("coordinate_system", None)
             if isinstance(coordinate_system, str):
-                coordinate_system = _coordinate_systems_from_string.get(coordinate_system.lower(), None)
+                coordinate_system = _coordinate_systems_from_string.get(coordinate_system.lower())
             return SpatialFilter(
                 geometry=spatial_filter_like["geometry"], mode=mode, coordinate_system=coordinate_system
             )

tilebox-datasets/tilebox/datasets/data/datasets.py

@@ -136,7 +136,7 @@ class DatasetType:
     @classmethod
     def from_message(cls, dataset_type: dataset_type_pb2.DatasetType) -> "DatasetType":
         return cls(
-            kind=_dataset_kind_int_to_enum.get(dataset_type.kind, None),
+            kind=_dataset_kind_int_to_enum.get(dataset_type.kind),
             fields=[Field.from_message(f) for f in dataset_type.fields],
         )
 

tilebox-storage/tilebox/storage/aio.py

@@ -209,7 +209,7 @@ async def writer(chunk: bytes) -> None:
         if progress is not None:
             progress.close()
 
-        output_dir.mkdir(parents=True, exist_ok=True)
+        output_dir.mkdir(parents=True, exist_ok=True)  # noqa: ASYNC240
         output_file = output_dir / url.rsplit("/", 1)[-1]
         shutil.move(download_file, output_file)
         return output_file

tilebox-storage/tilebox/storage/granule.py

@@ -51,7 +51,7 @@ def from_data(cls, dataset: "xr.Dataset | ASFStorageGranule") -> "ASFStorageGran
 
 
 def _asf_download_urls(granule_name: str) -> StorageURLs:
-    platform = granule_name.split("_")[0]
+    platform = granule_name.split("_", maxsplit=1)[0]
     file_name = granule_name
     processing_level = "L0"
     quicklook = None
@@ -100,7 +100,7 @@ def _thumbnail_relative_to_eodata_location(thumbnail_url: str, location: str) ->
         "preview/thumbnail.png"
     """
 
-    url_path = thumbnail_url.split("?path=")[-1]
+    url_path = thumbnail_url.rsplit("?path=", maxsplit=1)[-1]
     url_path = url_path.removeprefix("/")
     location = location.removeprefix("/eodata/")
     return str(ObjectPath(url_path).relative_to(location))