foamlib 0.2.6__tar.gz → 0.2.7__tar.gz

{foamlib-0.2.6 → foamlib-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: foamlib
-Version: 0.2.6
+Version: 0.2.7
 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.2.6 → foamlib-0.2.7}/foamlib/__init__.py

@@ -1,6 +1,6 @@
 """A Python interface for interacting with OpenFOAM."""
 
-__version__ = "0.2.6"
+__version__ = "0.2.7"
 
 from ._cases import AsyncFoamCase, FoamCase, FoamCaseBase
 from ._dictionaries import FoamDictionaryBase, FoamFieldFile, FoamFile

{foamlib-0.2.6 → foamlib-0.2.7}/foamlib/_dictionaries/_base.py

@@ -4,9 +4,15 @@ from dataclasses import dataclass
 from typing import Dict, NamedTuple, Optional, Union
 
 if sys.version_info >= (3, 9):
-    from collections.abc import Sequence
+    from collections.abc import Mapping, Sequence
 else:
-    from typing import Sequence
+    from typing import Mapping, Sequence
+
+try:
+    import numpy as np
+    from numpy.typing import NDArray
+except ModuleNotFoundError:
+    pass
 
 
 class FoamDictionaryBase:
@@ -39,9 +45,13 @@ class FoamDictionaryBase:
     A value that can be stored in an OpenFOAM dictionary.
     """
 
-    _Dict = Dict[str, Union["FoamDictionaryBase.Value", "_Dict"]]
+    _Dict = Dict[str, Union["Value", "_Dict"]]
 
     @abstractmethod
     def as_dict(self) -> _Dict:
         """Return a nested dict representation of the dictionary."""
         raise NotImplementedError
+
+    _SetValue = Union[Value, "NDArray[np.generic]"]
+
+    _SetMapping = Mapping[str, Union["_SetValue", "_SetMapping"]]

{foamlib-0.2.6 → foamlib-0.2.7}/foamlib/_dictionaries/_files.py

@@ -1,9 +1,11 @@
 import sys
 from pathlib import Path
+from types import TracebackType
 from typing import (
     Any,
     Optional,
     Tuple,
+    Type,
     Union,
     cast,
 )
@@ -48,7 +50,12 @@ class _FoamFileBase:
         self.__defer_io += 1
         return self
 
-    def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
+    def __exit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_val: Optional[BaseException],
+        exc_tb: Optional[TracebackType],
+    ) -> None:
         self.__defer_io -= 1
         if self.__defer_io == 0 and self.__dirty:
             assert self.__contents is not None
@@ -124,7 +131,11 @@ class FoamFile(
                 assume_dimensions=assume_dimensions,
             )
 
-        def __setitem__(self, keyword: str, value: Any) -> None:
+        def __setitem__(
+            self,
+            keyword: str,
+            value: Union["FoamFile._SetValue", "FoamFile._SetMapping"],
+        ) -> None:
             self._setitem(keyword, value)
 
         def __delitem__(self, keyword: str) -> None:
@@ -180,7 +191,7 @@ class FoamFile(
     def _setitem(
         self,
         keywords: Union[str, Tuple[str, ...]],
-        value: Any,
+        value: Union["FoamFile._SetValue", "FoamFile._SetMapping"],
         *,
         assume_field: bool = False,
         assume_dimensions: bool = False,
@@ -210,7 +221,11 @@ class FoamFile(
                 f"{contents[:start]}\n{serialize_entry(keywords[-1], value, assume_field=assume_field, assume_dimensions=assume_dimensions)}\n{contents[end:]}"
             )
 
-    def __setitem__(self, keywords: Union[str, Tuple[str, ...]], value: Any) -> None:
+    def __setitem__(
+        self,
+        keywords: Union[str, Tuple[str, ...]],
+        value: Union["FoamFile._SetValue", "FoamFile._SetMapping"],
+    ) -> None:
         self._setitem(keywords, value)
 
     def __delitem__(self, keywords: Union[str, Tuple[str, ...]]) -> None:
@@ -274,7 +289,11 @@ class FoamFieldFile(FoamFile):
     class BoundaryDictionary(FoamFile.Dictionary):
         """An OpenFOAM dictionary representing a boundary condition as a mutable mapping."""
 
-        def __setitem__(self, key: str, value: Any) -> None:
+        def __setitem__(
+            self,
+            key: str,
+            value: Union[FoamFile._SetValue, FoamFile._SetMapping],
+        ) -> None:
             if key == "value":
                 self._setitem(key, value, assume_field=True)
             else:

{foamlib-0.2.6 → foamlib-0.2.7}/foamlib/_dictionaries/_serialization.py

@@ -1,6 +1,6 @@
 import sys
 from contextlib import suppress
-from typing import Any
+from typing import Union
 
 if sys.version_info >= (3, 9):
     from collections.abc import Mapping
@@ -11,7 +11,9 @@ from .._util import is_sequence
 from ._base import FoamDictionaryBase
 
 
-def _serialize_switch(value: Any) -> str:
+def _serialize_switch(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if value is True:
         return "yes"
     elif value is False:
@@ -20,14 +22,18 @@ def _serialize_switch(value: Any) -> str:
         raise TypeError(f"Not a bool: {type(value)}")
 
 
-def _serialize_list(value: Any) -> str:
+def _serialize_list(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if is_sequence(value):
         return f"({' '.join(_serialize_value(v) for v in value)})"
     else:
         raise TypeError(f"Not a valid sequence: {type(value)}")
 
 
-def _serialize_field(value: Any) -> str:
+def _serialize_field(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if is_sequence(value):
         try:
             s = _serialize_list(value)
@@ -54,14 +60,18 @@ def _serialize_field(value: Any) -> str:
         return f"uniform {value}"
 
 
-def _serialize_dimensions(value: Any) -> str:
+def _serialize_dimensions(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if is_sequence(value) and len(value) == 7:
         return f"[{' '.join(str(v) for v in value)}]"
     else:
         raise TypeError(f"Not a valid dimension set: {type(value)}")
 
 
-def _serialize_dimensioned(value: Any) -> str:
+def _serialize_dimensioned(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if isinstance(value, FoamDictionaryBase.Dimensioned):
         if value.name is not None:
             return f"{value.name} {_serialize_dimensions(value.dimensions)} {_serialize_value(value.value)}"
@@ -72,7 +82,10 @@ def _serialize_dimensioned(value: Any) -> str:
 
 
 def _serialize_value(
-    value: Any, *, assume_field: bool = False, assume_dimensions: bool = False
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+    *,
+    assume_field: bool = False,
+    assume_dimensions: bool = False,
 ) -> str:
     if isinstance(value, FoamDictionaryBase.DimensionSet) or assume_dimensions:
         with suppress(TypeError):
@@ -94,7 +107,9 @@ def _serialize_value(
     return str(value)
 
 
-def _serialize_dictionary(value: Any) -> str:
+def _serialize_dictionary(
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
+) -> str:
     if isinstance(value, Mapping):
         return "\n".join(serialize_entry(k, v) for k, v in value.items())
     else:
@@ -103,7 +118,7 @@ def _serialize_dictionary(value: Any) -> str:
 
 def serialize_entry(
     keyword: str,
-    value: Any,
+    value: Union[FoamDictionaryBase._SetValue, FoamDictionaryBase._SetMapping],
     *,
     assume_field: bool = False,
     assume_dimensions: bool = False,

{foamlib-0.2.6 → foamlib-0.2.7}/foamlib.egg-info/PKG-INFO

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