toml-rs 0.0.9__pp311-pypy311_pp73-win_amd64.whl → 0.1.0__pp311-pypy311_pp73-win_amd64.whl

toml_rs/__init__.py

@@ -1,15 +1,19 @@
 __all__ = (
     "TOMLDecodeError",
+    "TOMLEncodeError",
     "__version__",
+    "dump",
+    "dumps",
     "load",
     "loads",
 )
 
 from collections.abc import Callable
-from typing import Any, BinaryIO
+from pathlib import Path
+from typing import Any, BinaryIO, TextIO
 
 from ._toml_rs import (
-    _load,
+    _dumps,
     _loads,
     _version,
 )
@@ -18,7 +22,13 @@ __version__: str = _version
 
 
 def load(fp: BinaryIO, /, *, parse_float: Callable[[str], Any] = float) -> dict[str, Any]:
-    return _load(fp, parse_float=parse_float)
+    _bytes = fp.read()
+    try:
+        _str = _bytes.decode()
+    except AttributeError:
+        msg = "File must be opened in binary mode, e.g. use `open('foo.toml', 'rb')`"
+        raise TypeError(msg) from None
+    return loads(_str, parse_float=parse_float)
 
 
 def loads(s: str, /, *, parse_float: Callable[[str], Any] = float) -> dict[str, Any]:
@@ -27,6 +37,33 @@ def loads(s: str, /, *, parse_float: Callable[[str], Any] = float) -> dict[str,
     return _loads(s, parse_float=parse_float)
 
 
+def dump(
+        obj: Any,
+        /,
+        file: str | Path | TextIO,
+        inline_tables: set[str] | None = None,
+        *,
+        pretty: bool = False,
+) -> int:
+    _str = _dumps(obj, inline_tables=inline_tables, pretty=pretty)
+    if isinstance(file, str):
+        file = Path(file)
+    if isinstance(file, Path):
+        return file.write_text(_str, encoding="utf-8")
+    else:
+        return file.write(_str)
+
+
+def dumps(
+        obj: Any,
+        /,
+        inline_tables: set[str] | None = None,
+        *,
+        pretty: bool = False,
+) -> str:
+    return _dumps(obj, inline_tables=inline_tables, pretty=pretty)
+
+
 class TOMLDecodeError(ValueError):
     def __init__(self, msg: str, doc: str, pos: int, *args: Any):
         msg = msg.rstrip()
@@ -41,3 +78,10 @@ class TOMLDecodeError(ValueError):
         self.pos = pos
         self.colno = colno
         self.lineno = lineno
+
+
+class TOMLEncodeError(TypeError):
+    def __init__(self, msg: str, *args: Any):
+        msg = msg.rstrip()
+        super().__init__(msg)
+        self.msg = msg

{toml_rs-0.0.9.dist-info → toml_rs-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toml-rs
-Version: 0.0.9
+Version: 0.1.0
 Classifier: Typing :: Typed
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python
@@ -31,7 +31,7 @@ Project-URL: Bug Tracker, https://github.com/lava-sh/toml-rs/issues
 
 * The fastest TOML parser in Python (see [benchmarks](https://github.com/lava-sh/toml-rs/tree/main/benchmark))
 
-* Drop-in compatibility with most [`tomllib`](https://docs.python.org/3/library/tomllib.html) use cases ([see below](#differences-with-tomllib))
+* Drop-in compatibility with most [`tomllib`](https://docs.python.org/3/library/tomllib.html) use cases (see [below](#differences-with-tomllib))
 
 ## Installation
 ```bash
@@ -44,10 +44,10 @@ uv pip install toml-rs
 
 ## Examples
 ```python
+import tomllib
 from pprint import pprint
 
 import toml_rs
-import tomllib
 
 toml = """\
 title = "TOML Example"
@@ -73,13 +73,14 @@ role = "backend"
 
 tomllib_loads = tomllib.loads(toml)
 toml_rs_loads = toml_rs.loads(toml)
+toml_rs_dumps = toml_rs.dumps(toml_rs_loads)
 
 assert tomllib_loads == toml_rs_loads
 
-print("tomllib:")
-pprint(tomllib_loads)
-print("toml_rs:")
+print("toml_rs.loads:")
 pprint(toml_rs_loads)
+print("toml_rs.dumps:")
+print(toml_rs_dumps)
 ```
 
 ## Differences with [`tomllib`](https://docs.python.org/3/library/tomllib.html)
@@ -136,4 +137,22 @@ print(toml_rs.loads(t))
 # 1 | x = 999_999_999_999_999_999_999_999
 #   |     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 # invalid type: integer `999999999999999999999999` as i128, expected any valid TOML value
+```
+
+3. Supports serialization (`toml_rs.dumps` and `toml_rs.dump`)
+
+```python
+from pathlib import Path
+
+import toml_rs
+
+data = {
+    "title": "TOML Example",
+    "owner": {"name": "Alice", "age": 30},
+}
+
+print(toml_rs.dumps(data))
+
+toml_rs.dump(data, Path("example.toml"))  
+# or `toml_rs.dump(data, "example.toml")`
 ```

toml_rs-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+toml_rs-0.1.0.dist-info/METADATA,sha256=FsQ8OXhhxugTkjFoRmEhUw5_VN8tLrs_-JoHGyaosM4,4048
+toml_rs-0.1.0.dist-info/WHEEL,sha256=XvltcVfFjkSizTjc8NGh5IW8xXp3Lufh9SqWYuaiv-c,103
+toml_rs-0.1.0.dist-info/licenses/UNLICENSE,sha256=pO9SJPWzhoS7yVUh8EuWMlwkAsA9o-yzXiMtgqmWKC8,1233
+toml_rs/__init__.py,sha256=Iveymfrcmv6oMs_ixXXsD26L0Qy8T_Ab1bj5kt3M3M8,2250
+toml_rs/_toml_rs.pypy311-pp73-win_amd64.pyd,sha256=SnPQTX3AYEn9A3Ohg4xbX4vq3hyksIGbSQNXyLGQAE4,659456
+toml_rs/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+toml_rs-0.1.0.dist-info/RECORD,,

toml_rs-0.0.9.dist-info/RECORD

@@ -1,7 +0,0 @@
-toml_rs-0.0.9.dist-info/METADATA,sha256=G6BDV0sZXc7ntdvx7w6of9g4n16LOGj6dNnEPPVMlC0,3655
-toml_rs-0.0.9.dist-info/WHEEL,sha256=XvltcVfFjkSizTjc8NGh5IW8xXp3Lufh9SqWYuaiv-c,103
-toml_rs-0.0.9.dist-info/licenses/UNLICENSE,sha256=pO9SJPWzhoS7yVUh8EuWMlwkAsA9o-yzXiMtgqmWKC8,1233
-toml_rs/__init__.py,sha256=NIiAfcSc1whF7NT0jfypUpEACbx4_yfP1PATASZiQi8,1113
-toml_rs/_toml_rs.pypy311-pp73-win_amd64.pyd,sha256=BhyrGWAWtYVqeBak-NGjUf5rxoZ-IZWLuiG7rmYp8gI,536576
-toml_rs/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-toml_rs-0.0.9.dist-info/RECORD,,