rust/hg-cpython/src/revlog.rs
changeset 51228 5807e3a8865e
parent 51226 1b23aaf5eb7b
child 51229 3551f2a1c963
--- a/rust/hg-cpython/src/revlog.rs	Tue Oct 31 17:34:31 2023 +0100
+++ b/rust/hg-cpython/src/revlog.rs	Tue Oct 31 17:36:59 2023 +0100
@@ -675,7 +675,9 @@
                         .expect("default header is broken")
                         .unwrap(),
                 )
-                .unwrap(),
+                .map_err(|e| {
+                    revlog_error_with_msg(py, e.to_string().as_bytes())
+                })?,
             ),
             RefCell::new(None),
             RefCell::new(None),
@@ -1052,6 +1054,21 @@
     }
 }
 
+fn revlog_error_with_msg(py: Python, msg: &[u8]) -> PyErr {
+    match py
+        .import("mercurial.error")
+        .and_then(|m| m.get(py, "RevlogError"))
+    {
+        Err(e) => e,
+        Ok(cls) => PyErr::from_instance(
+            py,
+            cls.call(py, (PyBytes::new(py, msg),), None)
+                .ok()
+                .into_py_object(py),
+        ),
+    }
+}
+
 fn graph_error(py: Python, _err: hg::GraphError) -> PyErr {
     // ParentOutOfRange is currently the only alternative
     // in `hg::GraphError`. The C index always raises this simple ValueError.