itf-py 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

itf_py/__init__.py

@@ -7,4 +7,11 @@ from .itf import ITFState
 __version__ = "0.1.0"
 __all__ = [
     "ITFState",
+    "ITFTrace",
+    "value_to_json",
+    "value_from_json",
+    "state_to_json",
+    "state_from_json",
+    "trace_to_json",
+    "trace_from_json",
 ]

itf_py/itf.py

@@ -1,7 +1,10 @@
+from collections import namedtuple
 from dataclasses import dataclass
-from types import SimpleNamespace
 from typing import Any, Dict, List, Optional
 
+from frozendict import frozendict
+from frozenlist import FrozenList
+
 
 @dataclass
 class ITFState:
@@ -39,13 +42,17 @@ def value_from_json(val: Any) -> Any:
         elif "#set" in val:
             return frozenset(value_from_json(v) for v in val["#set"])
         elif "#map" in val:
-            return {value_from_json(k): value_from_json(v) for (k, v) in val["#map"]}
+            d = {value_from_json(k): value_from_json(v) for (k, v) in val["#map"]}
+            return frozendict(d)  # immutable dictionary
         elif "#unserializable" in val:
             return ITFUnserializable(value=val["#unserializable"])
         else:
-            return SimpleNamespace(**{k: value_from_json(v) for k, v in val.items()})
+            tup_type = namedtuple("ITFRecord", val.keys())  # type: ignore
+            return tup_type(**{k: value_from_json(v) for k, v in val.items()})
     elif isinstance(val, list):
-        return [value_from_json(v) for v in val]
+        lst = FrozenList([value_from_json(v) for v in val])
+        lst.freeze()  # make it immutable
+        return lst
     else:
         return val  # int, str, bool
 
@@ -56,16 +63,19 @@ def value_to_json(val: Any) -> Any:
         return val
     elif isinstance(val, int):
         return {"#bigint": str(val)}
-    elif isinstance(val, tuple):
+    elif isinstance(val, tuple) and not hasattr(val, "_fields"):
         return {"#tup": [value_to_json(v) for v in val]}
     elif isinstance(val, frozenset):
         return {"#set": [value_to_json(v) for v in val]}
     elif isinstance(val, dict):
         return {"#map": [[value_to_json(k), value_to_json(v)] for k, v in val.items()]}
-    elif isinstance(val, list):
+    elif isinstance(val, list) or isinstance(val, FrozenList):
         return [value_to_json(v) for v in val]
     elif hasattr(val, "__dict__"):
         return {k: value_to_json(v) for k, v in val.__dict__.items()}
+    elif isinstance(val, tuple) and hasattr(val, "_fields"):
+        # namedtuple
+        return {k: value_to_json(v) for k, v in val._asdict().items()}  # type: ignore
     elif isinstance(val, str):
         return val
     else:

{itf_py-0.1.1.dist-info → itf_py-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: itf-py
-Version: 0.1.1
+Version: 0.2.0
 Summary: Python library to parse and emit Apalache ITF traces
 Keywords: itf,trace,parser,tlaplus,apalache,quint
 Author: Igor Konnov
@@ -9,6 +9,8 @@ Requires-Python: >=3.12,<4.0
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: frozendict (>=2.4.6,<3.0.0)
+Requires-Dist: frozenlist (>=1.7.0,<2.0.0)
 Project-URL: Homepage, https://github.com/konnov/itf-py
 Description-Content-Type: text/markdown
 

itf_py-0.2.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+itf_py/__init__.py,sha256=zRXHiq7vDwHKepUc9RKs9oA57QBRKOZQRthHT3u0k5M,290
+itf_py/itf.py,sha256=tiZR3g3-HRoDNk7eH26M7hxVyK3tMVGXhxsUxw-PCrQ,3934
+itf_py-0.2.0.dist-info/METADATA,sha256=2E76gdEJiWOHZkmtwL_a_IiJSEPymqrxQWdNlx-8evo,1071
+itf_py-0.2.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+itf_py-0.2.0.dist-info/RECORD,,

itf_py-0.1.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-itf_py/__init__.py,sha256=TxSVie2ZFBXBjC11rLydd61ddyRgUjnCFo31CCk69aE,142
-itf_py/itf.py,sha256=OGTElw_qBsh0fngjw3PjGF-Q2dRXMwt6o-YT2xvZ2Pw,3438
-itf_py-0.1.1.dist-info/METADATA,sha256=ZUFtFuzvNuEZ-NRfvuRF48ZIY9rQxMj6fjiqYc8Ndjg,985
-itf_py-0.1.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-itf_py-0.1.1.dist-info/RECORD,,