fastexcel 0.20.2

A fast excel reader for Rust and Python
Documentation 
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use arrow_schema::{DataType as ArrowDataType, TimeUnit};
use pyo3::{Borrowed, Bound, FromPyObject, IntoPyObject, PyAny, PyErr, Python, types::PyString};

use crate::{
    error::{FastExcelErrorKind, py_errors::IntoPyResult},
    types::dtype::{DType, DTypeCoercion, DTypeMap, DTypes},
};

impl<'py> IntoPyObject<'py> for DType {
    type Target = PyString;

    type Output = Bound<'py, Self::Target>;

    type Error = std::convert::Infallible;

    fn into_pyobject(self, py: Python<'py>) -> Result<Self::Output, Self::Error> {
        self.to_string().into_pyobject(py)
    }
}

impl<'py> IntoPyObject<'py> for &DType {
    type Target = PyString;

    type Output = Bound<'py, Self::Target>;

    type Error = std::convert::Infallible;

    fn into_pyobject(self, py: Python<'py>) -> Result<Self::Output, Self::Error> {
        self.to_string().into_pyobject(py)
    }
}

impl<'a, 'py> FromPyObject<'a, 'py> for DType {
    type Error = PyErr;
    fn extract(py_dtype: Borrowed<'a, 'py, PyAny>) -> Result<Self, Self::Error> {
        if let Ok(dtype_pystr) = py_dtype.extract::<String>() {
            dtype_pystr.parse()
        } else {
            Err(FastExcelErrorKind::InvalidParameters(format!(
                "{py_dtype:?} cannot be converted to str"
            ))
            .into())
        }
        .into_pyresult()
    }
}

impl<'a, 'py> FromPyObject<'a, 'py> for DTypes {
    type Error = PyErr;
    fn extract(py_dtypes: Borrowed<'a, 'py, PyAny>) -> Result<Self, Self::Error> {
        if let Ok(py_dtypes_str) = py_dtypes.extract::<String>() {
            py_dtypes_str.parse()
        } else {
            Ok(DTypes::Map(py_dtypes.extract::<DTypeMap>()?))
        }
        .into_pyresult()
    }
}

impl From<&DType> for ArrowDataType {
    fn from(dtype: &DType) -> Self {
        match dtype {
            DType::Null => ArrowDataType::Null,
            DType::Int => ArrowDataType::Int64,
            DType::Float => ArrowDataType::Float64,
            DType::String => ArrowDataType::Utf8,
            DType::Bool => ArrowDataType::Boolean,
            DType::DateTime => ArrowDataType::Timestamp(TimeUnit::Millisecond, None),
            DType::Date => ArrowDataType::Date32,
            DType::Duration => ArrowDataType::Duration(TimeUnit::Millisecond),
        }
    }
}

impl<'a, 'py> FromPyObject<'a, 'py> for DTypeCoercion {
    type Error = PyErr;
    fn extract(py_dtype_coercion: Borrowed<'a, 'py, PyAny>) -> Result<Self, Self::Error> {
        if let Ok(dtype_coercion_pystr) = py_dtype_coercion.extract::<String>() {
            dtype_coercion_pystr.parse()
        } else {
            Err(FastExcelErrorKind::InvalidParameters(format!(
                "{py_dtype_coercion:?} cannot be converted to str"
            ))
            .into())
        }
        .into_pyresult()
    }
}