Python APIのSynthesizerをclose()可能にする

crates/voicevox_core_python_api/python/voicevox_core/_rust.pyi

@@ -77,6 +77,8 @@ class Synthesizer:
         """
         ...
     def __repr__(self) -> str: ...
+    def __enter__(self) -> "Synthesizer": ...
+    def __exit__(self, exc_type, exc_value, traceback): ...
     @property
     def is_gpu_mode(self) -> bool:
         """ハードウェアアクセラレーションがGPUモードか判定する。
@@ -257,6 +259,7 @@ class Synthesizer:
             疑問文の調整を有効にする。
         """
         ...
+    def close(self) -> None: ...
 
 class UserDict:
     """ユーザー辞書。

crates/voicevox_core_python_api/src/lib.rs

@@ -1,4 +1,4 @@
-use std::sync::Arc;
+use std::{marker::PhantomData, sync::Arc};
 
 mod convert;
 use convert::*;
@@ -9,7 +9,7 @@ use pyo3::{
     exceptions::PyException,
     pyclass, pyfunction, pymethods, pymodule,
     types::{IntoPyDict as _, PyBytes, PyDict, PyList, PyModule},
-    wrap_pyfunction, PyAny, PyObject, PyResult, Python, ToPyObject,
+    wrap_pyfunction, PyAny, PyObject, PyRef, PyResult, PyTypeInfo, Python, ToPyObject,
 };
 use tokio::{runtime::Runtime, sync::Mutex};
 use uuid::Uuid;
@@ -114,7 +114,7 @@ impl OpenJtalk {
 
 #[pyclass]
 struct Synthesizer {
-    synthesizer: Arc<Mutex<voicevox_core::Synthesizer>>,
+    synthesizer: Closable<Arc<Mutex<voicevox_core::Synthesizer>>, Self>,
 }
 
 #[pymethods]
@@ -143,9 +143,10 @@ impl Synthesizer {
                 },
             )
             .await
-            .into_py_result()?;
+            .into_py_result()?
+            .into();
             Ok(Self {
-                synthesizer: Arc::new(Mutex::new(synthesizer)),
+                synthesizer: Closable::new(Arc::new(synthesizer)),
             })
         })
     }
@@ -154,14 +155,29 @@ impl Synthesizer {
         "Synthesizer { .. }"
     }
 
+    fn __enter__(slf: PyRef<'_, Self>) -> PyRef<'_, Self> {
+        slf
+    }
+
+    fn __exit__(
+        &mut self,
+        #[allow(unused_variables)] exc_type: &PyAny,
+        #[allow(unused_variables)] exc_value: &PyAny,
+        #[allow(unused_variables)] traceback: &PyAny,
+    ) -> PyResult<()> {
+        self.close()
+    }
+
     #[getter]
-    fn is_gpu_mode(&self) -> bool {
-        RUNTIME.block_on(self.synthesizer.lock()).is_gpu_mode()
+    fn is_gpu_mode(&self) -> PyResult<bool> {
+        let synthesizer = self.synthesizer.get()?;
+        Ok(RUNTIME.block_on(synthesizer.lock()).is_gpu_mode())
     }
 
     #[getter]
-    fn metas<'py>(&self, py: Python<'py>) -> Vec<&'py PyAny> {
-        to_pydantic_voice_model_meta(RUNTIME.block_on(self.synthesizer.lock()).metas(), py).unwrap()
+    fn metas<'py>(&self, py: Python<'py>) -> PyResult<Vec<&'py PyAny>> {
+        let synthesizer = self.synthesizer.get()?;
+        to_pydantic_voice_model_meta(RUNTIME.block_on(synthesizer.lock()).metas(), py)
     }
 
     fn load_voice_model<'py>(
@@ -170,7 +186,7 @@ impl Synthesizer {
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
         let model: VoiceModel = model.extract()?;
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         pyo3_asyncio::tokio::future_into_py(py, async move {
             synthesizer
                 .lock()
@@ -183,15 +199,15 @@ impl Synthesizer {
 
     fn unload_voice_model(&mut self, voice_model_id: &str) -> PyResult<()> {
         RUNTIME
-            .block_on(self.synthesizer.lock())
+            .block_on(self.synthesizer.get()?.lock())
             .unload_voice_model(&VoiceModelId::new(voice_model_id.to_string()))
             .into_py_result()
     }
 
-    fn is_loaded_voice_model(&self, voice_model_id: &str) -> bool {
-        RUNTIME
-            .block_on(self.synthesizer.lock())
-            .is_loaded_voice_model(&VoiceModelId::new(voice_model_id.to_string()))
+    fn is_loaded_voice_model(&self, voice_model_id: &str) -> PyResult<bool> {
+        Ok(RUNTIME
+            .block_on(self.synthesizer.get()?.lock())
+            .is_loaded_voice_model(&VoiceModelId::new(voice_model_id.to_string())))
     }
 
     #[pyo3(signature=(text,style_id,kana = AudioQueryOptions::default().kana))]
@@ -202,7 +218,7 @@ impl Synthesizer {
         kana: bool,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         let text = text.to_owned();
         pyo3_asyncio::tokio::future_into_py_with_locals(
             py,
@@ -232,7 +248,7 @@ impl Synthesizer {
         kana: bool,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         let text = text.to_owned();
         pyo3_asyncio::tokio::future_into_py_with_locals(
             py,
@@ -267,7 +283,7 @@ impl Synthesizer {
         style_id: u32,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         modify_accent_phrases(
             accent_phrases,
             StyleId::new(style_id),
@@ -282,7 +298,7 @@ impl Synthesizer {
         style_id: u32,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         modify_accent_phrases(
             accent_phrases,
             StyleId::new(style_id),
@@ -297,7 +313,7 @@ impl Synthesizer {
         style_id: u32,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         modify_accent_phrases(
             accent_phrases,
             StyleId::new(style_id),
@@ -314,7 +330,7 @@ impl Synthesizer {
         enable_interrogative_upspeak: bool,
         py: Python<'py>,
     ) -> PyResult<&'py PyAny> {
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         pyo3_asyncio::tokio::future_into_py_with_locals(
             py,
             pyo3_asyncio::tokio::get_current_locals(py)?,
@@ -355,7 +371,7 @@ impl Synthesizer {
             kana,
             enable_interrogative_upspeak,
         };
-        let synthesizer = self.synthesizer.clone();
+        let synthesizer = self.synthesizer.get()?.clone();
         let text = text.to_owned();
         pyo3_asyncio::tokio::future_into_py_with_locals(
             py,
@@ -371,6 +387,52 @@ impl Synthesizer {
             },
         )
     }
+
+    fn close(&mut self) -> PyResult<()> {
+        self.synthesizer.close()
+    }
+}
+
+struct Closable<T, C: PyTypeInfo> {
+    content: MaybeClosed<T>,
+    marker: PhantomData<C>,
+}
+
+enum MaybeClosed<T> {
+    Open(T),
+    Closed,
+}
+
+impl<T, C: PyTypeInfo> Closable<T, C> {
+    fn new(content: T) -> Self {
+        Self {
+            content: MaybeClosed::Open(content),
+            marker: PhantomData,
+        }
+    }
+
+    fn get(&self) -> PyResult<&T> {
+        match &self.content {
+            MaybeClosed::Open(content) => Ok(content),
+            MaybeClosed::Closed => Err(VoicevoxError::new_err(format!(
+                "The `{}` is already closed",
+                C::NAME,
+            ))),
+        }
+    }
+
+    fn close(&mut self) -> PyResult<()> {
+        self.get()?;
+        debug!("Closing a {}", C::NAME);
+        self.content = MaybeClosed::Closed;
+        Ok(())
+    }
+}
+
+impl<T, C: PyTypeInfo> Drop for Closable<T, C> {
+    fn drop(&mut self) {
+        let _ = self.close();
+    }
 }
 
 #[pyfunction]
@@ -451,9 +513,3 @@ impl UserDict {
         Ok(words.into_py_dict(py))
     }
 }
-
-impl Drop for Synthesizer {
-    fn drop(&mut self) {
-        debug!("Destructing a VoicevoxCore");
-    }
-}