implement ToPyObject for JsonValue (#15)

.github/workflows/ci.yml

@@ -21,6 +21,11 @@ jobs:
     steps:
     - uses: actions/checkout@v3
 
+    - name: set up python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.11'
+
     - uses: dtolnay/rust-toolchain@master
       with:
         toolchain: ${{ matrix.rust-version }}
@@ -33,7 +38,7 @@ jobs:
 
     - run: rustup component add llvm-tools-preview
 
-    - run: cargo test --test main
+    - run: cargo test -F python
       env:
         RUST_BACKTRACE: 1
         RUSTFLAGS: '-C instrument-coverage'

.gitignore

@@ -4,3 +4,4 @@
 /profiling/
 /scratch/
 /*.bm
+/env*/

.rustfmt.toml

@@ -1,2 +1 @@
-imports_granularity = "Module"
 max_width = 120

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "jiter"
-version = "0.0.1"
+version = "0.0.2"
 edition = "2021"
 description = "Iterable JSON parser"
 readme = "README.md"
@@ -18,11 +18,16 @@ num-bigint = "0.4.3"
 num-traits = "0.2.16"
 ahash = "0.8.0"
 smallvec = "1.11.0"
+pyo3 = { version = "0.19.2", features = ["num-bigint"], optional = true }
+
+[features]
+python = ["dep:pyo3"]
 
 [dev-dependencies]
 paste = "1.0.7"
 serde_json = {version = "1.0.87", features = ["preserve_order", "arbitrary_precision"]}
 serde = "1.0.147"
+pyo3 = { version = "0.19.2", features = ["num-bigint", "auto-initialize"] }
 
 [profile.bench]
 lto = true

src/value.rs

@@ -20,6 +20,28 @@ pub enum JsonValue {
     Object(Box<LazyIndexMap<String, JsonValue>>),
 }
 
+#[cfg(feature = "python")]
+impl pyo3::ToPyObject for JsonValue {
+    fn to_object(&self, py: pyo3::Python<'_>) -> pyo3::PyObject {
+        match self {
+            Self::Null => py.None(),
+            Self::Bool(b) => b.to_object(py),
+            Self::Int(i) => i.to_object(py),
+            Self::BigInt(b) => b.to_object(py),
+            Self::Float(f) => f.to_object(py),
+            Self::String(s) => s.to_object(py),
+            Self::Array(v) => pyo3::types::PyList::new(py, v.iter().map(|v| v.to_object(py))).to_object(py),
+            Self::Object(o) => {
+                let dict = pyo3::types::PyDict::new(py);
+                for (k, v) in o.iter() {
+                    dict.set_item(k, v.to_object(py)).unwrap();
+                }
+                dict.to_object(py)
+            }
+        }
+    }
+}
+
 impl JsonValue {
     pub fn parse(data: &[u8]) -> Result<Self, JsonValueError> {
         let mut parser = Parser::new(data);