foamlib 0.3.8__py3-none-any.whl → 0.3.9__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.9"
 
 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.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.3.8
+Version: 0.3.9
 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

{foamlib-0.3.8.dist-info → foamlib-0.3.9.dist-info}/RECORD

@@ -1,15 +1,15 @@
-foamlib/__init__.py,sha256=4Iybtfu_RUem_6XkwHEYhTShzrq7Zb2IiBhP-Vv7FeA,431
-foamlib/_cases.py,sha256=vq30uymy0lcYvkdEj_v94YuRMqUhgnw_mGxSCgNbdFU,21142
+foamlib/__init__.py,sha256=VA5OODQTc1hOiKx4Hg0kYE_r9Q9num_-mR-wMHHhR1c,431
+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=1GO9jKXoJc45_GfrneCNlR1nSnT8HqYDY4VsRFdHwEQ,1912
+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.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,,
+foamlib-0.3.9.dist-info/LICENSE.txt,sha256=5Dte9TUnLZzPRs4NQzl-Jc2-Ljd-t_v0ZR5Ng5r0UsY,35131
+foamlib-0.3.9.dist-info/METADATA,sha256=FmZVopicBSscMw3JI3D6dnpLLDfcHLn4BQkrhklp8Sw,5242
+foamlib-0.3.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+foamlib-0.3.9.dist-info/top_level.txt,sha256=ZdVYtetXGwPwyfL-WhlhbTFQGAwKX5P_gXxtH9JYFPI,8
+foamlib-0.3.9.dist-info/RECORD,,