use std::collections::HashMap;
use pyo3::prelude::*;
use example_client::SchemaClient;
use example_common::models::schemas::v1::*;
use example_common::models::*;
use crate::codegen::error::{PyExampleError, PyExampleResult};
#[pyclass(name = "SchemaClient", module = "example._core.schemas")]
pub struct PySchemaClient {
pub(crate) client: SchemaClient,
}
#[pymethods]
impl PySchemaClient {
#[pyo3(signature = (view))]
pub fn get(
&self,
py: Python,
view: PyGetSchemaRequestView,
) -> PyExampleResult<PySchema> {
let request = self.client.get(view.into());
#[allow(clippy::let_unit_value)]
let result = crate::codegen::runtime::block_on(py, request.into_future())??;
Ok(PySchema::from(result))
}
#[pyo3(signature = (schema = None))]
pub fn update(
&self,
py: Python,
schema: ::core::option::Option<PySchema>,
) -> PyExampleResult<PySchema> {
let request = self
.client
.update()
.with_schema(schema.map(::core::convert::Into::into));
#[allow(clippy::let_unit_value)]
let result = crate::codegen::runtime::block_on(py, request.into_future())??;
Ok(PySchema::from(result))
}
pub fn delete(&self, py: Python) -> PyExampleResult<()> {
let request = self.client.delete();
#[allow(clippy::let_unit_value)]
let result = crate::codegen::runtime::block_on(py, request.into_future())??;
Ok(result)
}
}
impl PySchemaClient {
pub fn new(client: SchemaClient) -> Self {
Self { client }
}
}