rusqlite 和 pyo3 PyResult 处理错误

编程问答 2022-05-31

问题描述

我正在尝试在 pyo3 的帮助下打开并写入我将从 python 调用的 Rust 库中的数据库。如果发生错误，我想引发可以在调用 Python 进程中捕获的异常，但我在终止执行和引发错误时遇到困难。

use rusqlite::{Connection};
use rusqlite::NO_ParaMS;
use pyo3::{Python,wrap_pyfunction};
use pyo3::exceptions::PyIOError;

#[pyfunction]
fn do_something(_py: Python) -> PyResult<u32> {
    match Connection::open("database.sql") {
        Ok(t) => conn = t,Err(e) => {
            let gil = Python::acquire_gil();
            let py = gil.python();
            let error_message = format!("Unable to open database! {}",e.to_string());
            PyIOError::new_err(error_message).restore(py)
        }
    };
    
    match conn.execute(
        "create table if not exists cats (
            id              INTEGER PRIMARY KEY,name            TEXT NOT NULL,)",NO_ParaMS,) {
        Ok(_t) => (),e.to_string());
            PyIOError::new_err(error_message).restore(py)
        }
    }
    Ok(0)

据我了解，通过在 restore 对象上调用 PyIOError 函数，会引发错误，但是，我一定是误解了，因为编译器似乎认为 {{1 }} 未初始化：

conn

这里什么是合适的方法？

解决方法

首先，您的 Ok(t) = conn = t 失败了，因为您还没有定义 conn。所以在 match 之前添加 let conn;。或者，您也可以将匹配结果分配给 conn。

其次，您仍然需要返回一个 Err。

#[pyfunction]
fn do_something(_py: Python) -> PyResult<u32> {
    let conn = match Connection::open("database.sql") {
        Ok(t) => t,Err(e) => {
            let gil = Python::acquire_gil();
            let py = gil.python();
            let error_message = format!("Unable to open database! {}",e.to_string());
            PyIOError::new_err(error_message).restore(py);
            return Err(PyErr::fetch(py));
        }
    };

    match conn.execute(
        "create table if not exists cats (
            id              INTEGER PRIMARY KEY,name            TEXT NOT NULL,)",NO_PARAMS,) {
        Ok(_t) => (),e.to_string());
            PyIOError::new_err(error_message).restore(py);
            return Err(PyErr::fetch(py));
        }
    }
    Ok(0)
}

我已经有一段时间没有使用 PyO3。但除非我没记错，否则您可以删除 restore() 也只需返回 Err 并让 PyO3 处理其余的。

#[pyfunction]
fn do_something(_py: Python) -> PyResult<u32> {
    let conn = match Connection::open("database.sql") {
        Ok(t) => t,Err(e) => {
            let error_message = format!("Unable to open database! {}",e.to_string());
            return Err(PyIOError::new_err(error_message));
        }
    };

    match conn.execute(
        "create table if not exists cats (
            id              INTEGER PRIMARY KEY,e.to_string());
            return Err(PyIOError::new_err(error_message));
        }
    }
    Ok(0)
}
pyo3rusqliterust