Added support for just array in JSONB column

chandr-andr · chandr-andr · commit 0989b7ded0d7 · 2025-08-13T14:27:46.000+02:00
diff --git a/python/tests/test_value_converter.py b/python/tests/test_value_converter.py
@@ -178,6 +178,11 @@ async def test_as_class(
             JSONB([{"array": "json"}, {"one more": "test"}]),
             [{"array": "json"}, {"one more": "test"}],
         ),
+        (
+            "JSONB",
+            JSONB([1, "1", 1.0]),
+            [1, "1", 1.0],
+        ),
         (
             "JSON",
             {
@@ -194,6 +199,11 @@ async def test_as_class(
             JSON([{"array": "json"}, {"one more": "test"}]),
             [{"array": "json"}, {"one more": "test"}],
         ),
+        (
+            "JSON",
+            JSON([1, "1", 1.0]),
+            [1, "1", 1.0],
+        ),
         (
             "MACADDR",
             MacAddr6("08:00:2b:01:02:03"),
diff --git a/src/value_converter/models/serde_value.rs b/src/value_converter/models/serde_value.rs
@@ -67,16 +67,12 @@ fn serde_value_from_list(_gil: Python<'_>, bind_value: &Bound<'_, PyAny>) -> PSQ
     let mut result_vec: Vec<Value> = Vec::with_capacity(py_list.len());
 
     for item in py_list.iter() {
-        if item.is_instance_of::<PyDict>() {
-            let python_dto = from_python_untyped(&item)?;
-            result_vec.push(python_dto.to_serde_value()?);
-        } else if item.is_instance_of::<PyList>() {
+        if item.is_instance_of::<PyList>() {
             let serde_value = build_serde_value(&item)?;
             result_vec.push(serde_value);
         } else {
-            return Err(RustPSQLDriverError::PyToRustValueConversionError(
-                "Items in JSON array must be dicts or lists.".to_string(),
-            ));
+            let python_dto = from_python_untyped(&item)?;
+            result_vec.push(python_dto.to_serde_value()?);
         }
     }
 
@@ -120,7 +116,7 @@ pub fn build_serde_value(value: &Bound<'_, PyAny>) -> PSQLPyResult<Value> {
             return serde_value_from_dict(value);
         } else {
             return Err(RustPSQLDriverError::PyToRustValueConversionError(
-                "PyJSON must be dict value.".to_string(),
+                "PyJSON must be dict or list value.".to_string(),
             ));
         }
     })
