Started adding expicit array types

Signed-off-by: chandr-andr (Kiselev Aleksandr) <chandr@chandr.net>

Author: chandr-andr

python/psqlpy/_internal/extra_types.pyi

@@ -251,3 +251,13 @@ class PyCircle:
         ### Parameters:
         - `value`: any valid structure with int/float numbers.
         """
+
+
+class BoolArray:
+    """Represent circle field in PostgreSQL and Circle in Rust."""
+
+    def __init__(
+        self: Self,
+        value: typing.Any,
+    ) -> None:
+        """Create new instance."""

python/psqlpy/extra_types.py

@@ -18,6 +18,7 @@
     PyText,
     PyVarChar,
     SmallInt,
+    BoolArray,
 )
 
 __all__ = [
@@ -40,4 +41,5 @@
     "PyLine",
     "PyLineSegment",
     "PyCircle",
+    "BoolArray",
 ]

src/extra_types.rs

@@ -1,5 +1,6 @@
-use std::str::FromStr;
+use std::{net::IpAddr, str::FromStr};
 
+use chrono::{DateTime, FixedOffset, NaiveDate, NaiveDateTime, NaiveTime};
 use geo_types::{Line as LineSegment, LineString, Point, Rect};
 use macaddr::{MacAddr6, MacAddr8};
 use pyo3::{
@@ -12,8 +13,13 @@ use serde_json::Value;
 use crate::{
     additional_types::{Circle, Line},
     exceptions::rust_errors::RustPSQLDriverPyResult,
-    value_converter::{build_flat_geo_coords, build_geo_coords, build_serde_value},
+    value_converter::{
+        build_flat_geo_coords, build_geo_coords, build_serde_value, InternalSerdeValue,
+        InternalUuid, PythonDTO,
+    },
 };
+use pyo3::exceptions::PyStopIteration;
+use pyo3::prelude::*;
 
 macro_rules! build_python_type {
     ($st_name:ident, $rust_type:ty) => {
@@ -290,6 +296,67 @@ impl PyCircle {
     }
 }
 
+macro_rules! build_array_type {
+    ($st_name:ident, $rust_type:ty, $single_rust_type:ty) => {
+        #[pyclass(sequence)]
+        #[derive(Clone)]
+        pub struct $st_name {
+            inner: $rust_type,
+            index: usize,
+        }
+
+        #[pymethods]
+        impl $st_name {
+            #[new]
+            #[must_use]
+            pub fn new_class(inner: $rust_type) -> Self {
+                Self { inner, index: 0 }
+            }
+
+            fn __iter__(slf: PyRef<'_, Self>) -> PyRef<'_, Self> {
+                slf
+            }
+
+            fn __next__(mut slf: PyRefMut<'_, Self>) -> Option<$single_rust_type> {
+                let index = slf.index;
+                slf.index += 1;
+                if let Some(element) = slf.inner.get(index) {
+                    return Some(*element);
+                }
+                None
+            }
+        }
+
+        impl $st_name {
+            #[must_use]
+            pub fn inner(&self) -> $rust_type {
+                self.inner.clone()
+            }
+        }
+    };
+}
+
+build_array_type!(BoolArray, Vec<bool>, bool);
+// build_array_type!(BoolArray, Vec<bool>);
+// build_array_type!(UUIDArray, Vec<InternalUuid>);
+// build_array_type!(VarCharArray, Vec<String>);
+// build_array_type!(TextArray, Vec<String>);
+// build_array_type!(Int16Array, Vec<i16>);
+// build_array_type!(Int32Array, Vec<i32>);
+// build_array_type!(Int64Array, Vec<i64>);
+// build_array_type!(Flaot32Array, Vec<f32>);
+// build_array_type!(Flaot64Array, Vec<f64>);
+// build_array_type!(MoneyArray, Vec<i64>);
+// build_array_type!(IpAddressArray, Vec<IpAddr>);
+// build_array_type!(JSONBArray, Vec<InternalSerdeValue>);
+// build_array_type!(JSONArray, Vec<InternalSerdeValue>);
+// build_array_type!(DateArray, Vec<NaiveDate>);
+// build_array_type!(TimeArray, Vec<NaiveTime>);
+// build_array_type!(DateTimeArray, Vec<NaiveDateTime>);
+// build_array_type!(DateTimeTZArray, Vec<DateTime<FixedOffset>>);
+// build_array_type!(MacAddr6Array, Vec<PyMacAddr6>);
+// build_array_type!(MacAddr8Array, Vec<PyMacAddr8>);
+
 #[allow(clippy::module_name_repetitions)]
 #[allow(clippy::missing_errors_doc)]
 pub fn extra_types_module(_py: Python<'_>, pymod: &Bound<'_, PyModule>) -> PyResult<()> {
@@ -312,5 +379,7 @@ pub fn extra_types_module(_py: Python<'_>, pymod: &Bound<'_, PyModule>) -> PyRes
     pymod.add_class::<PyLine>()?;
     pymod.add_class::<PyLineSegment>()?;
     pymod.add_class::<PyCircle>()?;
+
+    pymod.add_class::<BoolArray>()?;
     Ok(())
 }

src/value_converter.rs

@@ -14,9 +14,10 @@ use pyo3::{
     sync::GILOnceCell,
     types::{
         PyAnyMethods, PyBool, PyBytes, PyDate, PyDateTime, PyDict, PyDictMethods, PyFloat, PyInt,
-        PyList, PyListMethods, PySequence, PySet, PyString, PyTime, PyTuple, PyType, PyTypeMethods,
+        PyIterator, PyList, PyListMethods, PySequence, PySet, PyString, PyTime, PyTuple, PyType,
+        PyTypeMethods,
     },
-    Bound, IntoPy, Py, PyAny, PyObject, PyResult, Python, ToPyObject,
+    Bound, FromPyObject, IntoPy, Py, PyAny, PyObject, PyResult, Python, ToPyObject,
 };
 use tokio_postgres::{
     types::{to_sql_checked, Type},
@@ -30,8 +31,8 @@ use crate::{
     },
     exceptions::rust_errors::{RustPSQLDriverError, RustPSQLDriverPyResult},
     extra_types::{
-        BigInt, Float32, Float64, Integer, Money, PyBox, PyCircle, PyCustomType, PyJSON, PyJSONB,
-        PyLine, PyLineSegment, PyMacAddr6, PyMacAddr8, PyPath, PyPoint, PyText, PyVarChar,
+        BigInt, BoolArray, Float32, Float64, Integer, Money, PyBox, PyCircle, PyCustomType, PyJSON,
+        PyJSONB, PyLine, PyLineSegment, PyMacAddr6, PyMacAddr8, PyPath, PyPoint, PyText, PyVarChar,
         SmallInt,
     },
 };
@@ -57,7 +58,10 @@ fn get_decimal_cls(py: Python<'_>) -> PyResult<&Bound<'_, PyType>> {
 ///
 /// We use custom struct because we need to implement external traits
 /// to it.
-struct InternalUuid(Uuid);
+#[derive(Clone)]
+pub struct InternalUuid(Uuid);
+
+impl<'a> FromPyObject<'a> for InternalUuid {}
 
 impl ToPyObject for InternalUuid {
     fn to_object(&self, py: Python<'_>) -> PyObject {
@@ -82,7 +86,10 @@ impl<'a> FromSql<'a> for InternalUuid {
 ///
 /// We use custom struct because we need to implement external traits
 /// to it.
-struct InternalSerdeValue(Value);
+#[derive(Clone)]
+pub struct InternalSerdeValue(Value);
+
+impl<'a> FromPyObject<'a> for InternalSerdeValue {}
 
 impl ToPyObject for InternalSerdeValue {
     fn to_object(&self, py: Python<'_>) -> PyObject {
@@ -786,6 +793,9 @@ pub fn py_to_rust(parameter: &pyo3::Bound<'_, PyAny>) -> RustPSQLDriverPyResult<
         }
     }
 
+    let a = parameter.downcast::<PyIterator>();
+    println!("{:?}", a.iter());
+
     Err(RustPSQLDriverError::PyToRustValueConversionError(format!(
         "Can not covert you type {parameter} into inner one",
     )))