update method

Author: victorteokw

src/dynamic/mod.rs

@@ -16,7 +16,7 @@ use crate::object::model::teo_model_object_to_py_any;
 use crate::object::py_any_to_teo_object;
 use crate::object::value::{teo_value_to_py_any, py_any_to_teo_value};
 use crate::result::{IntoPyResultWithGil, IntoTeoPathResult};
-use crate::utils::await_coroutine_if_needed::await_coroutine_if_needed_async_value;
+use crate::utils::await_coroutine_if_needed::await_coroutine_if_needed_value_with_locals;
 use crate::utils::check_py_dict::check_py_dict;
 
 use self::model_ctx_wrapper::ModelCtxWrapper;
@@ -532,7 +532,7 @@ fn synthesize_direct_dynamic_nodejs_classes_for_namespace(py: Python<'_>, namesp
                         let coroutine_or_value = shared_argument.call1(py, (ctx_python,)).into_teo_path_result()?;
                         Ok::<PyObject, teo::prelude::path::Error>(coroutine_or_value)
                     })?;
-                    Ok(await_coroutine_if_needed_async_value(user_retval, main_thread_locals).await.into_teo_path_result()?)
+                    Ok(await_coroutine_if_needed_value_with_locals(user_retval, main_thread_locals).await.into_teo_path_result()?)
                 }).await.into_py_result_with_gil()?;
                 Python::with_gil(|py| {
                     Ok::<PyObject, PyErr>(retval.into_py(py))    

src/handler/group.rs

@@ -5,7 +5,7 @@ use crate::dynamic::py_ctx_object_from_teo_transaction_ctx;
 use crate::object::value::teo_value_to_py_any;
 use crate::request::request::Request;
 use crate::response::Response;
-use crate::utils::await_coroutine_if_needed::await_coroutine_if_needed_async_value;
+use crate::utils::await_coroutine_if_needed::await_coroutine_if_needed_value_with_locals;
 use crate::utils::check_callable::check_callable;
 use crate::result::IntoTeoPathResult;
 
@@ -31,7 +31,7 @@ impl HandlerGroup {
                 let result = callback_owned.call1(py, (request, body, py_ctx))?;
                 Ok::<PyObject, PyErr>(result)
             }).into_teo_path_result()?;
-            let awaited_result = await_coroutine_if_needed_async_value(result, main_thread_locals).await.into_teo_path_result()?;
+            let awaited_result = await_coroutine_if_needed_value_with_locals(result, main_thread_locals).await.into_teo_path_result()?;
             Python::with_gil(|py| {
                 let response: Response = awaited_result.extract(py).into_teo_path_result()?;
                 Ok(response.teo_response.clone())

src/namespace/namespace.rs

@@ -1,7 +1,7 @@
 use pyo3::{pyclass, pymethods, types::PyCFunction, IntoPy, Py, PyErr, PyObject, PyResult, Python};
 use teo::prelude::{handler::Group as TeoHandlerGroup, model::Field as TeoField, model::Property as TeoProperty, model::Relation as TeoRelation, request, Enum as TeoEnum, Member as TeoEnumMember, Middleware, Model as TeoModel, Namespace as TeoNamespace, Next};
 
-use crate::{utils::{check_callable::check_callable, await_coroutine_if_needed::await_coroutine_if_needed_async_value}, object::{arguments::teo_args_to_py_args, value::teo_value_to_py_any}, model::{model::Model, field::field::Field, relation::relation::Relation, property::property::Property}, result::{IntoPyResultWithGil, IntoTeoPathResult, IntoTeoResult}, r#enum::{r#enum::Enum, member::member::EnumMember}, request::{Request, RequestCtx}, dynamic::py_ctx_object_from_teo_transaction_ctx, response::Response, handler::group::HandlerGroup};
+use crate::{utils::{check_callable::check_callable, await_coroutine_if_needed::await_coroutine_if_needed_value_with_locals}, object::{arguments::teo_args_to_py_args, value::teo_value_to_py_any}, model::{model::Model, field::field::Field, relation::relation::Relation, property::property::Property}, result::{IntoPyResultWithGil, IntoTeoPathResult, IntoTeoResult}, r#enum::{r#enum::Enum, member::member::EnumMember}, request::{Request, RequestCtx}, dynamic::py_ctx_object_from_teo_transaction_ctx, response::Response, handler::group::HandlerGroup};
 
 #[pyclass]
 pub struct Namespace {
@@ -165,7 +165,7 @@ impl Namespace {
                 let result = callback_owned.call1(py, (request, body, py_ctx))?;
                 Ok::<PyObject, PyErr>(result)
             }).into_teo_path_result()?;
-            let awaited_result = await_coroutine_if_needed_async_value(result, main_thread_locals).await.into_teo_path_result()?;
+            let awaited_result = await_coroutine_if_needed_value_with_locals(result, main_thread_locals).await.into_teo_path_result()?;
             Python::with_gil(|py| {
                 let response: Response = awaited_result.extract(py).into_teo_path_result()?;
                 Ok(response.teo_response.clone())
@@ -231,7 +231,7 @@ impl Namespace {
                         let coroutine = shared_result_function.call1(py, (py_ctx, py_next)).into_teo_result()?;
                         Ok::<PyObject, teo::prelude::Error>(coroutine.into_py(py))
                     })?;
-                    let result = await_coroutine_if_needed_async_value(coroutine, main_thread_locals).await.into_teo_result()?;
+                    let result = await_coroutine_if_needed_value_with_locals(coroutine, main_thread_locals).await.into_teo_result()?;
                     Python::with_gil(|py| {
                         let response: Response = result.extract(py).into_teo_result()?;
                         Ok(response.teo_response)    

src/utils/await_coroutine_if_needed.rs

@@ -2,7 +2,7 @@ use pyo3::{IntoPy, PyAny, PyErr, PyObject, PyResult, Python};
 use pyo3_asyncio::{into_future_with_locals, TaskLocals};
 use crate::utils::is_coroutine::is_coroutine;
 
-pub async fn await_coroutine_if_needed_async_value(transformed_py: PyObject, main_thread_locals: &TaskLocals) -> PyResult<PyObject> {
+pub async fn await_coroutine_if_needed_value_with_locals(transformed_py: PyObject, main_thread_locals: &TaskLocals) -> PyResult<PyObject> {
     let is_coroutine_bool = Python::with_gil(|py| {
         is_coroutine(transformed_py.as_ref(py))
     })?;
@@ -19,7 +19,7 @@ pub async fn await_coroutine_if_needed_async_value(transformed_py: PyObject, mai
     }
 }
 
-pub async fn await_coroutine_if_needed_async(transformed_py: &PyAny) -> PyResult<PyObject> {
+pub async fn await_coroutine_if_needed(transformed_py: &PyAny) -> PyResult<PyObject> {
     if is_coroutine(transformed_py)? {
         let f = pyo3_asyncio::tokio::into_future(transformed_py)?;
         f.await