Merge pull request #7 from bitcraze/rik/wait_disconnect

Add wait_disconnect method to Crazyflie class for graceful disconnection

Author: gemenerik

cflib2/_rust.pyi

@@ -324,6 +324,12 @@ class Crazyflie:
         r"""
         Disconnect from the Crazyflie
         """
+    async def wait_disconnect(self) -> builtins.str:
+        r"""
+        Wait for the Crazyflie to be disconnected
+
+        Returns a human-readable string describing the reason for disconnection.
+        """
     def commander(self) -> Commander:
         r"""
         Get the commander subsystem

rust/src/crazyflie.rs

@@ -134,6 +134,18 @@ impl Crazyflie {
         })
     }
 
+    /// Wait for the Crazyflie to be disconnected
+    ///
+    /// Returns a human-readable string describing the reason for disconnection.
+    #[gen_stub(override_return_type(type_repr = "collections.abc.Coroutine[typing.Any, typing.Any, builtins.str]"))]
+    fn wait_disconnect<'py>(&self, py: Python<'py>) -> PyResult<Bound<'py, PyAny>> {
+        let inner = self.inner.clone();
+        pyo3_async_runtimes::tokio::future_into_py(py, async move {
+            let reason = inner.wait_disconnect().await;
+            Ok(reason)
+        })
+    }
+
     /// Get the commander subsystem
     fn commander(&self) -> Commander {
         Commander {