--- a/rust/hg-cpython/src/update.rs Fri Nov 08 17:08:11 2024 +0100
+++ b/rust/hg-cpython/src/update.rs Tue Nov 12 12:52:13 2024 +0100
@@ -15,7 +15,7 @@
use crate::{
exceptions::FallbackError,
- utils::{hgerror_to_pyerr, repo_from_path},
+ utils::{hgerror_to_pyerr, repo_from_path, with_sigint_wrapper},
};
pub fn update_from_null_fast_path(
@@ -27,10 +27,12 @@
log::trace!("Using update from null fastpath");
let repo = repo_from_path(py, repo_path)?;
let progress: &dyn Progress = &HgProgressBar::new("updating");
- hgerror_to_pyerr(
- py,
- update_from_null(&repo, to.into(), progress, num_cpus),
- )
+
+ let res = with_sigint_wrapper(py, || {
+ update_from_null(&repo, to.into(), progress, num_cpus)
+ })?;
+
+ hgerror_to_pyerr(py, res)
}
pub fn init_module(py: Python, package: &str) -> PyResult<PyModule> {