doppy 0.1.1__cp310-abi3-macosx_10_12_x86_64.whl → 0.1.2__cp310-abi3-macosx_10_12_x86_64.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,15 +1,15 @@
-doppy-0.1.1.dist-info/METADATA,sha256=BbbFOBmBi9XwAiIlZy_YoBU9YPPDc1AuMQH1Md3Udy0,1822
-doppy-0.1.1.dist-info/WHEEL,sha256=V2VnYS0Z_2v18J65tisMAzW36GHSI5stkpxW52CkvJE,105
-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=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
-doppy-0.1.1.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.1.2.dist-info/METADATA,sha256=JW3abxUf515cuSyaWu5nalVw70hlsMI0E4BXNzihBwI,1794
+doppy-0.1.2.dist-info/WHEEL,sha256=V2VnYS0Z_2v18J65tisMAzW36GHSI5stkpxW52CkvJE,105
+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=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.1.2.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
 doppy/options.py,sha256=73BDODO4OYHn2qOshhwz6u6G3J1kNd3uj6P0a3V4HBE,205
 doppy/__init__.py,sha256=Z9aEUlbPRWRUAoB8_-djkgrJuS4-6pjem4-mVSB6Z9I,191
 doppy/product/wind.py,sha256=H4IJI1l_pxfbmI8R_RzAZSZzLWKQ8IUpRkq-m3fm3_E,11105
 doppy/product/__init__.py,sha256=lyp88zs7GBk8uUzkj8lxaXPuYwPa52xXeTwf5qwFddU,103
 doppy/product/stare.py,sha256=ehwK1dcNBnuVsjbdmNYNGfNiOMannSZVNHJAs7e4x_M,19705
 doppy/bench.py,sha256=iVNYveMVGGRES2oe3Orsn31jQFCKTXOmxRFuFiJ8_OA,248
-doppy/netcdf.py,sha256=NW5zTW2OY-M4ApMzgxwN7ysmL0LRRWOfyKFPinCw6gg,3403
+doppy/netcdf.py,sha256=8SWBfMdtmstPYGkW6xV7nR2J8jVYZzMHf1kjcbIEkrw,3415
 doppy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 doppy/exceptions.py,sha256=1tljrtzp0McQhB6INFXj4yfLjxj6mXor5jLF9HZjp1A,138
 doppy/defaults.py,sha256=-il4DWU1HPttlGFoVQTPC7R4pDZERoPvt3kw6xwMKrw,38
@@ -23,5 +23,5 @@ doppy/raw/halo_bg.py,sha256=kO03yGlKS-DpMMGHYuy_BuidyeUL38TxT5vMn8H_8lE,4809
 doppy/raw/halo_hpl.py,sha256=mYCE8JWgRbk6CfV4hSKQvChbnE7xXUArpatRez8EsFk,18504
 doppy/raw/windcube.py,sha256=TVqR3SwfUJYHtPJXnA_b6uUGo_ylVkGGEPob63Zcee0,9685
 doppy/__main__.py,sha256=zrKQJVj0k0ypBQCGK65Czt9G9FZ_qx3ussw6Q9VJ14g,346
-doppy/rs.abi3.so,sha256=AfLWMzqAkx6Zzp0u6E1nIinGuw8N85YOME4MuBCyMhw,2716032
-doppy-0.1.1.dist-info/RECORD,,
+doppy/rs.abi3.so,sha256=nyCamOL2y9rgHvK3lkS2f4HySl_aE5repNujABXnzRs,2716128
+doppy-0.1.2.dist-info/RECORD,,