doppy 0.1.1__cp310-abi3-win_amd64.whl → 0.1.2__cp310-abi3-win_amd64.whl

doppy/netcdf.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-from pathlib import Path
 from typing import Literal, TypeAlias
 
 import netCDF4
@@ -11,13 +10,17 @@ NetCDFDataType: TypeAlias = Literal["f4", "f8", "i4", "i8", "u4", "u8"]
 
 
 class Dataset:
-    def __init__(self) -> None:
-        self.nc = netCDF4.Dataset("inmemory.nc", mode="w", memory=1028)
+    def __init__(self, filename: str) -> None:
+        self.nc = netCDF4.Dataset(filename, mode="w")
 
     def add_dimension(self, dim: str) -> Dataset:
         self.nc.createDimension(dim, None)
         return self
 
+    def add_atribute(self, key: str, val: str) -> Dataset:
+        setattr(self.nc, key, val)
+        return self
+
     def add_time(
         self,
         name: str,
@@ -28,7 +31,7 @@ class Dataset:
         long_name: str | None = None,
     ) -> Dataset:
         time, units, calendar = _convert_time(data)
-        var = self.nc.createVariable(name, dtype, dimensions)
+        var = self.nc.createVariable(name, dtype, dimensions, compression="zlib")
         var.units = units
         var.calendar = calendar
         var[:] = time
@@ -49,7 +52,10 @@ class Dataset:
         long_name: str | None = None,
         mask: npt.NDArray[np.bool_] | None = None,
     ) -> Dataset:
-        var = self.nc.createVariable(name, dtype, dimensions)
+        fill_value = netCDF4.default_fillvals[dtype] if mask is not None else None
+        var = self.nc.createVariable(
+            name, dtype, dimensions, fill_value=fill_value, compression="zlib"
+        )
         var.units = units
         if mask is not None:
             var[:] = np.ma.masked_array(data, mask)  # type: ignore
@@ -80,16 +86,8 @@ class Dataset:
             var.long_name = long_name
         return self
 
-    def close(self) -> memoryview:
-        buf = self.nc.close()
-        if isinstance(buf, memoryview):
-            return buf
-        raise TypeError
-
-    def write(self, path: str | Path) -> None:
-        buf = self.nc.close()
-        with Path(path).open("wb") as f:
-            f.write(buf)
+    def close(self) -> None:
+        self.nc.close()
 
 
 def _convert_time(

{doppy-0.1.1.dist-info → doppy-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: doppy
-Version: 0.1.1
+Version: 0.1.2
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
@@ -22,13 +22,12 @@ Requires-Dist: ruff ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'
 Requires-Dist: types-requests ; extra == 'dev'
 Requires-Dist: py-spy ; extra == 'dev'
-Requires-Dist: maturin ; extra == 'dev'
+Requires-Dist: maturin ==1.4 ; extra == 'dev'
 Requires-Dist: release-version ; extra == 'dev'
 Requires-Dist: pre-commit ; extra == 'dev'
 Provides-Extra: dev
 License-File: LICENSE
 License-File: LICENSE
-Summary: Doppler lidar processing
 Author: Niko Leskinen <niko.leskinen@fmi.fi>
 Author-email: Niko Leskinen <niko.leskinen@fmi.fi>
 Requires-Python: >=3.10

{doppy-0.1.1.dist-info → doppy-0.1.2.dist-info}/RECORD

@@ -1,8 +1,8 @@
-doppy-0.1.1.dist-info/METADATA,sha256=G8dGAVytWwjcSmP0SnN4ULMlNOT5lJIABaL-CUa8ODs,1826
-doppy-0.1.1.dist-info/WHEEL,sha256=Q2npN4Wz3hcQ5uc7KmG7fh3MaY55ijiBnwI7N8UUMDM,95
-doppy-0.1.1.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
-doppy-0.1.1.dist-info/license_files/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
-doppy-0.1.1.dist-info/license_files/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
+doppy-0.1.2.dist-info/METADATA,sha256=x1OqOHI6PDqSZOdytPtIABdhJw60RcOhlxf_hmlnmfQ,1798
+doppy-0.1.2.dist-info/WHEEL,sha256=Q2npN4Wz3hcQ5uc7KmG7fh3MaY55ijiBnwI7N8UUMDM,95
+doppy-0.1.2.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.1.2.dist-info/license_files/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
+doppy-0.1.2.dist-info/license_files/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
 doppy/bench.py,sha256=fLN2iS5mmoYH4qZjD80Vl1h9lp3C-KDfhj9fteWRPtM,260
 doppy/data/api.py,sha256=yzniozSzs8N1sJe5ht4Y5SwyZn06RlGSkiwm0GvBTUE,1787
 doppy/data/cache.py,sha256=VNPB3XsWGwY2bNXBs1r_sEWF4qBq_U7sJSlSmt1Rxm8,1033
@@ -10,7 +10,7 @@ doppy/data/exceptions.py,sha256=JOyekvUO-Ew4ZVezf3_IxZOrPN0IksfUILd8R2YcSts,95
 doppy/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 doppy/defaults.py,sha256=nhxcZcFd2jyDVbq0azwtekEJEjiz8k21MZmXFlSXAjo,40
 doppy/exceptions.py,sha256=YNEyz4r0ObzZHZ9re83K3wZlR2CRI1GyhH0vvFGasgQ,148
-doppy/netcdf.py,sha256=rvcm_6RnmTIrwqDgIJnPyZvO2rcRkBpbQODEBOA5yN8,3516
+doppy/netcdf.py,sha256=RxJhYv6JEtNF7G7gEX5dcCL4sv5F4-TdDt2uKrkhHjs,3526
 doppy/options.py,sha256=uyIKM_G2GtbmV6Gve8o13eIShQqUwsnYZ41mhX2ypGE,218
 doppy/product/stare.py,sha256=GBw5P16smTLtTdNtLTucb4NP2nUcW2WEj0sXysYSvZY,20314
 doppy/product/wind.py,sha256=WNPPMY71HKIpnj3T96jqWNcScYT5Fa3vB7PKnm4FP_I,11452
@@ -23,5 +23,5 @@ doppy/raw/windcube.py,sha256=aHAKN1tgyPNpCVBY4co9FwM6vMLu4swqVoFFhpDT33M,9939
 doppy/raw/__init__.py,sha256=5szGvPn_KS7PViMPQ1umGrbd9jkJzQUZpTiST1z2MvQ,200
 doppy/__init__.py,sha256=Af7_8p3oN1nTqS9fo0mVKVuiKf5CAEK69uQa32CSFBA,197
 doppy/__main__.py,sha256=38hIWWfanILuBBGorQiAaleSC4qYJoIxuzVBkxf7Dng,371
-doppy/rs.pyd,sha256=okIYMy5c7W8kZiWwDAETLeBh1CJrCR15hHxeKM2IDjs,2088960
-doppy-0.1.1.dist-info/RECORD,,
+doppy/rs.pyd,sha256=DwzH6MeOJpfYtoNoeDwCmdVBkQaJA9BKe4DysjY14nk,2088960
+doppy-0.1.2.dist-info/RECORD,,