wire nsv-rust for dumps/loads

.github/workflows/tests.yml

@@ -2,7 +2,6 @@ name: Tests
 
 on:
   push:
-    branches: [ master, claude/* ]
   pull_request:
     branches: [ master ]
 

MANIFEST.in

@@ -1,3 +1,5 @@
 include README.md
 include LICENSE
-recursive-include nsv *.py
+recursive-include nsv *.py
+include rust/Cargo.toml
+recursive-include rust/src *

nsv/core.py

@@ -35,3 +35,11 @@ def dumps(data: Iterable[Iterable[str]]) -> str:
             lines.append(Writer.escape(cell))
         lines.append('')
     return ''.join(f'{line}\n' for line in lines)
+
+# Try to use fast Rust implementation if available
+try:
+    from ._rust import loads as _loads_rs, dumps as _dumps_rs
+    loads = _loads_rs
+    dumps = _dumps_rs
+except ImportError:
+    pass

pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools>=61.0", "setuptools-rust>=1.0"]
+build-backend = "setuptools.build_meta"

rust/Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "_rust"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+name = "_rust"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.20", features = ["extension-module"] }
+nsv = "0.0.12"

rust/src/lib.rs

@@ -0,0 +1,35 @@
+use pyo3::prelude::*;
+use pyo3::types::PyList;
+
+/// Parse NSV string into a list of lists of str
+#[pyfunction]
+fn loads(py: Python, s: &str) -> PyResult<PyObject> {
+    let data = nsv::decode(s);
+
+    // Convert Vec<Vec<String>> to Python list of lists
+    let result = PyList::empty(py);
+    for row in data {
+        let py_row = PyList::empty(py);
+        for cell in row {
+            py_row.append(cell)?;
+        }
+        result.append(py_row)?;
+    }
+
+    Ok(result.into())
+}
+
+/// Serialize data to NSV string
+#[pyfunction]
+fn dumps(data: Vec<Vec<String>>) -> PyResult<String> {
+    Ok(nsv::encode(&data))
+}
+
+/// A Python module implemented in Rust.
+#[pymodule]
+fn _rust(_py: Python, m: &PyModule) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(loads, m)?)?;
+    m.add_function(wrap_pyfunction!(dumps, m)?)?;
+    Ok(())
+}
+