minarrow-pyo3 0.2.1

// Copyright 2025 Peter Garfield Bower
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

//! # Error Module for minarrow-pyo3
//!
//! Provides error types and conversions between MinArrow errors and Python exceptions.

use pyo3::exceptions::{PyRuntimeError, PyTypeError, PyValueError};
use pyo3::prelude::*;
use thiserror::Error;

/// Error type for minarrow-pyo3 operations.
#[derive(Error, Debug)]
pub enum PyMinarrowError {
    /// Error from MinArrow core library.
    #[error("MinArrow error: {0}")]
    Minarrow(#[from] minarrow::enums::error::MinarrowError),

    /// Error during FFI operations.
    #[error("FFI error: {0}")]
    Ffi(String),

    /// Type conversion error.
    #[error("Type error: {0}")]
    Type(String),

    /// Unsupported Arrow type.
    #[error("Unsupported type: {0}")]
    UnsupportedType(String),

    /// PyArrow import error.
    #[error("PyArrow error: {0}")]
    PyArrow(String),
}

impl From<PyMinarrowError> for PyErr {
    fn from(err: PyMinarrowError) -> PyErr {
        match err {
            PyMinarrowError::Type(msg) => PyTypeError::new_err(msg),
            PyMinarrowError::UnsupportedType(msg) => PyTypeError::new_err(msg),
            PyMinarrowError::Ffi(msg) => PyRuntimeError::new_err(msg),
            PyMinarrowError::PyArrow(msg) => PyValueError::new_err(msg),
            PyMinarrowError::Minarrow(e) => {
                // Map MinArrow errors to appropriate Python exceptions
                match e {
                    minarrow::enums::error::MinarrowError::TypeError { .. } => {
                        PyTypeError::new_err(e.to_string())
                    }
                    minarrow::enums::error::MinarrowError::IncompatibleTypeError { .. } => {
                        PyTypeError::new_err(e.to_string())
                    }
                    minarrow::enums::error::MinarrowError::IndexError(_) => {
                        pyo3::exceptions::PyIndexError::new_err(e.to_string())
                    }
                    minarrow::enums::error::MinarrowError::Overflow { .. } => {
                        pyo3::exceptions::PyOverflowError::new_err(e.to_string())
                    }
                    _ => PyRuntimeError::new_err(e.to_string()),
                }
            }
        }
    }
}

/// Result type alias for minarrow-pyo3 operations.
pub type PyMinarrowResult<T> = Result<T, PyMinarrowError>;