foamlib 0.3.8__py3-none-any.whl → 0.3.10__py3-none-any.whl

foamlib/__init__.py

@@ -1,6 +1,6 @@
 """A Python interface for interacting with OpenFOAM."""
 
-__version__ = "0.3.8"
+__version__ = "0.3.10"
 
 from ._cases import AsyncFoamCase, FoamCase, FoamCaseBase
 from ._files import FoamDict, FoamFieldFile, FoamFile

foamlib/_cases.py

@@ -56,23 +56,29 @@ class FoamCaseBase(Sequence["FoamCaseBase.TimeDirectory"]):
             return self.path.name
 
         def __getitem__(self, key: str) -> FoamFieldFile:
-            try:
+            if (self.path / key).is_file():
                 return FoamFieldFile(self.path / key)
-            except FileNotFoundError as e:
-                raise KeyError(key) from e
+            elif (self.path / f"{key}.gz").is_file():
+                return FoamFieldFile(self.path / f"{key}.gz")
+            else:
+                raise KeyError(key)
 
         def __contains__(self, obj: object) -> bool:
             if isinstance(obj, FoamFieldFile):
                 return obj.path.parent == self.path
             elif isinstance(obj, str):
-                return (self.path / obj).is_file()
+                return (self.path / obj).is_file() or (
+                    self.path / f"{obj}.gz"
+                ).is_file()
             else:
                 return False
 
         def __iter__(self) -> Iterator[FoamFieldFile]:
             for p in self.path.iterdir():
-                if p.is_file():
-                    yield FoamFieldFile(p.name)
+                if p.is_file() and (
+                    p.suffix != ".gz" or not p.with_suffix("").is_file()
+                ):
+                    yield FoamFieldFile(p)
 
         def __len__(self) -> int:
             return len(list(iter(self)))
@@ -114,7 +120,7 @@ class FoamCaseBase(Sequence["FoamCaseBase.TimeDirectory"]):
         self, index: Union[int, slice, float, str]
     ) -> Union["FoamCaseBase.TimeDirectory", Sequence["FoamCaseBase.TimeDirectory"]]:
         if isinstance(index, str):
-            return FoamCaseBase.TimeDirectory(self.path / str(index))
+            return FoamCaseBase.TimeDirectory(self.path / index)
         elif isinstance(index, float):
             for time in self._times:
                 if time.time == index:

foamlib/_files/_io.py

@@ -1,3 +1,4 @@
+import gzip
 import sys
 from copy import deepcopy
 from pathlib import Path
@@ -47,6 +48,10 @@ class FoamFileIO:
     def _read(self) -> Tuple[bytes, Parsed]:
         if not self.__defer_io:
             contents = self.path.read_bytes()
+
+            if self.path.suffix == ".gz":
+                contents = gzip.decompress(contents)
+
             if contents != self.__contents:
                 self.__contents = contents
                 self.__parsed = None
@@ -63,6 +68,9 @@ class FoamFileIO:
         self.__contents = contents
         self.__parsed = None
         if not self.__defer_io:
+            if self.path.suffix == ".gz":
+                contents = gzip.compress(contents)
+
             self.path.write_bytes(contents)
             self.__dirty = False
         else:

{foamlib-0.3.8.dist-info → foamlib-0.3.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.3.8
+Version: 0.3.10
 Summary: A Python interface for interacting with OpenFOAM
 Author-email: "Gabriel S. Gerlero" <ggerlero@cimec.unl.edu.ar>
 Project-URL: Homepage, https://github.com/gerlero/foamlib
@@ -28,22 +28,28 @@ License-File: LICENSE.txt
 Requires-Dist: aioshutil <2,>=1
 Requires-Dist: pyparsing <4,>=3
 Requires-Dist: typing-extensions <5,>=4 ; python_version < "3.11"
+Provides-Extra: dev
+Requires-Dist: foamlib[numpy] ; extra == 'dev'
+Requires-Dist: foamlib[lint] ; extra == 'dev'
+Requires-Dist: foamlib[test] ; extra == 'dev'
+Requires-Dist: foamlib[typing] ; extra == 'dev'
+Requires-Dist: foamlib[docs] ; extra == 'dev'
 Provides-Extra: docs
+Requires-Dist: foamlib[numpy] ; extra == 'docs'
 Requires-Dist: sphinx <8,>=7 ; extra == 'docs'
 Requires-Dist: sphinx-rtd-theme ; extra == 'docs'
-Requires-Dist: numpy <2,>=1 ; extra == 'docs'
 Provides-Extra: lint
 Requires-Dist: ruff ; extra == 'lint'
+Provides-Extra: numpy
+Requires-Dist: numpy <2,>=1 ; extra == 'numpy'
 Provides-Extra: test
+Requires-Dist: foamlib[numpy] ; extra == 'test'
 Requires-Dist: pytest <9,>=7 ; extra == 'test'
 Requires-Dist: pytest-asyncio <0.24,>=0.21 ; extra == 'test'
 Requires-Dist: pytest-cov ; extra == 'test'
-Requires-Dist: numpy <2,>=1 ; extra == 'test'
 Provides-Extra: typing
+Requires-Dist: foamlib[test] ; extra == 'typing'
 Requires-Dist: mypy <2,>=1 ; extra == 'typing'
-Requires-Dist: pytest <9,>=7 ; extra == 'typing'
-Requires-Dist: pytest-asyncio <0.24,>=0.21 ; extra == 'typing'
-Requires-Dist: numpy <2,>=1 ; extra == 'typing'
 
 [<img alt="foamlib" src="https://github.com/gerlero/foamlib/raw/main/logo.png" height="50">](https://github.com/gerlero/foamlib)
 

foamlib-0.3.10.dist-info/RECORD

@@ -0,0 +1,15 @@
+foamlib/__init__.py,sha256=OJLrmY7uGGxH91O7KRnS7BR1s254VtAqPYaVzI2CikA,432
+foamlib/_cases.py,sha256=4b9CCDAuKi5Dp8DUksIbaMsp0Yi0xv5QeZLVAUrKRh8,21417
+foamlib/_util.py,sha256=vL03aAzpWdZyYIhe2WTxHiz9b4lnttVnRuzqUmZvXXk,3047
+foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+foamlib/_files/__init__.py,sha256=vDkPj8u8bX_I_m2YfeKvXBgwg8D1ufyFCfHGHKN3JPQ,140
+foamlib/_files/_base.py,sha256=YA5a-i5HZuA3JslCD6r-DwZzpSA8r42dqSXef286Ako,2050
+foamlib/_files/_files.py,sha256=4rZb3HMJADmJBGmRpChiujJNX1UIxhgHI5YPEmEOWvE,10784
+foamlib/_files/_io.py,sha256=pGYMoLI5Dz2Y90EB1CtklZiffpDTGuQbbbkyqRe0JAo,2115
+foamlib/_files/_parsing.py,sha256=sWUeF9xiM6AriQjljl3jEpvoFRMmaS1h3bT3b0oor3U,7279
+foamlib/_files/_serialization.py,sha256=rmZW42Pyn1DIJbGMZqFG_lZ5DcLhPmfUgH24qsu2rIk,3400
+foamlib-0.3.10.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
+foamlib-0.3.10.dist-info/METADATA,sha256=tmwwFbw9VKAIMBeJ-W8nP6iEfKXFxIRYmW57yDVvyLc,5458
+foamlib-0.3.10.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+foamlib-0.3.10.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
+foamlib-0.3.10.dist-info/RECORD,,

foamlib-0.3.8.dist-info/RECORD

@@ -1,15 +0,0 @@
-foamlib/__init__.py,sha256=4Iybtfu_RUem_6XkwHEYhTShzrq7Zb2IiBhP-Vv7FeA,431
-foamlib/_cases.py,sha256=vq30uymy0lcYvkdEj_v94YuRMqUhgnw_mGxSCgNbdFU,21142
-foamlib/_util.py,sha256=vL03aAzpWdZyYIhe2WTxHiz9b4lnttVnRuzqUmZvXXk,3047
-foamlib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-foamlib/_files/__init__.py,sha256=vDkPj8u8bX_I_m2YfeKvXBgwg8D1ufyFCfHGHKN3JPQ,140
-foamlib/_files/_base.py,sha256=YA5a-i5HZuA3JslCD6r-DwZzpSA8r42dqSXef286Ako,2050
-foamlib/_files/_files.py,sha256=4rZb3HMJADmJBGmRpChiujJNX1UIxhgHI5YPEmEOWvE,10784
-foamlib/_files/_io.py,sha256=1GO9jKXoJc45_GfrneCNlR1nSnT8HqYDY4VsRFdHwEQ,1912
-foamlib/_files/_parsing.py,sha256=sWUeF9xiM6AriQjljl3jEpvoFRMmaS1h3bT3b0oor3U,7279
-foamlib/_files/_serialization.py,sha256=rmZW42Pyn1DIJbGMZqFG_lZ5DcLhPmfUgH24qsu2rIk,3400
-foamlib-0.3.8.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
-foamlib-0.3.8.dist-info/METADATA,sha256=m-CbGI_cHcps-ZkkyspgchzDnv3Zk9ML9hANZN2coIY,5242
-foamlib-0.3.8.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-foamlib-0.3.8.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
-foamlib-0.3.8.dist-info/RECORD,,