sz-orm-python 0.1.0

Python bindings for sz-orm-core (PyO3): Model, QueryBuilder, Pool, Transaction
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

//! DbError → PyErr 映射

use pyo3::exceptions::{PyConnectionError, PyRuntimeError, PyValueError};
use pyo3::prelude::*;
use sz_orm_core::DbError;

#[pyclass(name = "DbError")]
#[derive(Clone)]
pub struct PyDbError {
    pub message: String,
    pub code: String,
}

#[pymethods]
impl PyDbError {
    #[new]
    fn new(message: String) -> Self {
        Self {
            message,
            code: "DB000".to_string(),
        }
    }

    #[getter]
    fn message(&self) -> &str {
        &self.message
    }

    #[getter]
    fn code(&self) -> &str {
        &self.code
    }

    fn __str__(&self) -> String {
        format!("[{}] {}", self.code, self.message)
    }
}

#[allow(dead_code)]
pub fn db_error_to_pyerr(err: DbError) -> PyErr {
    let code = err.error_code().to_string();
    let msg = err.to_string();
    match &err {
        DbError::ConnectionRefused(_) | DbError::ConnectionError(_) => {
            PyConnectionError::new_err(format!("[{}] {}", code, msg))
        }
        DbError::InvalidInput(_) | DbError::Validation(_) => {
            PyValueError::new_err(format!("[{}] {}", code, msg))
        }
        _ => PyRuntimeError::new_err(format!("[{}] {}", code, msg)),
    }
}