First variant for custom encoder

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

Author: chandr-andr

python/psqlpy/_internal/extra_types.pyi

@@ -123,3 +123,6 @@ class PyMacAddr8:
         ### Parameters:
         - `value`: value for MACADDR8 field.
         """
+
+class PyCustomType:
+    def __init__(self, value: bytes) -> None: ...

python/psqlpy/extra_types.py

@@ -1,6 +1,7 @@
 from ._internal.extra_types import (
     BigInt,
     Integer,
+    PyCustomType,
     PyJSON,
     PyJSONB,
     PyMacAddr6,
@@ -22,4 +23,5 @@
     "PyMacAddr8",
     "PyVarChar",
     "PyText",
+    "PyCustomType",
 ]

src/extra_types.rs

@@ -190,6 +190,27 @@ macro_rules! build_macaddr_type {
 build_macaddr_type!(PyMacAddr6, MacAddr6);
 build_macaddr_type!(PyMacAddr8, MacAddr8);
 
+#[pyclass]
+#[derive(Clone, Debug)]
+pub struct PyCustomType {
+    inner: Vec<u8>,
+}
+
+impl PyCustomType {
+    #[must_use]
+    pub fn inner(&self) -> Vec<u8> {
+        self.inner.clone()
+    }
+}
+
+#[pymethods]
+impl PyCustomType {
+    #[new]
+    fn new_class(type_bytes: Vec<u8>) -> Self {
+        PyCustomType { inner: type_bytes }
+    }
+}
+
 #[allow(clippy::module_name_repetitions)]
 #[allow(clippy::missing_errors_doc)]
 pub fn extra_types_module(_py: Python<'_>, pymod: &Bound<'_, PyModule>) -> PyResult<()> {
@@ -203,5 +224,6 @@ pub fn extra_types_module(_py: Python<'_>, pymod: &Bound<'_, PyModule>) -> PyRes
     pymod.add_class::<PyJSON>()?;
     pymod.add_class::<PyMacAddr6>()?;
     pymod.add_class::<PyMacAddr8>()?;
+    pymod.add_class::<PyCustomType>()?;
     Ok(())
 }

src/value_converter.rs

@@ -23,8 +23,8 @@ use crate::{
     additional_types::{RustMacAddr6, RustMacAddr8},
     exceptions::rust_errors::{RustPSQLDriverError, RustPSQLDriverPyResult},
     extra_types::{
-        BigInt, Integer, PyJSON, PyJSONB, PyMacAddr6, PyMacAddr8, PyText, PyUUID, PyVarChar,
-        SmallInt,
+        BigInt, Integer, PyCustomType, PyJSON, PyJSONB, PyMacAddr6, PyMacAddr8, PyText, PyUUID,
+        PyVarChar, SmallInt,
     },
 };
 
@@ -62,6 +62,7 @@ pub enum PythonDTO {
     PyJson(Value),
     PyMacAddr6(MacAddr6),
     PyMacAddr8(MacAddr8),
+    PyCustomType(Vec<u8>),
 }
 
 impl PythonDTO {
@@ -174,6 +175,9 @@ impl ToSql for PythonDTO {
 
         match self {
             PythonDTO::PyNone => {}
+            PythonDTO::PyCustomType(some_bytes) => {
+                <&[u8] as ToSql>::to_sql(&some_bytes.as_slice(), ty, out)?;
+            }
             PythonDTO::PyBytes(pybytes) => {
                 <Vec<u8> as ToSql>::to_sql(pybytes, ty, out)?;
             }
@@ -284,6 +288,12 @@ pub fn py_to_rust(parameter: &pyo3::Bound<'_, PyAny>) -> RustPSQLDriverPyResult<
         return Ok(PythonDTO::PyNone);
     }
 
+    if parameter.is_instance_of::<PyCustomType>() {
+        return Ok(PythonDTO::PyCustomType(
+            parameter.extract::<PyCustomType>()?.inner(),
+        ));
+    }
+
     if parameter.is_instance_of::<PyBool>() {
         return Ok(PythonDTO::PyBool(parameter.extract::<bool>()?));
     }