kutil 0.0.6

Kutil utilities collection
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

use {
    pyo3::prelude::*,
    std::{
        borrow::*,
        io::{self, Read},
        str::*,
    },
};

#[allow(unused_imports)]
pub use {pyo3_filelike::PyBinaryFile, pyo3_filelike::PyTextFile};

//
// ReadableAny
//

/// Readable [PyAny].
pub enum ReadableAny<'this> {
    /// Bytes.
    Bytes(&'this [u8]),

    /// String.
    String(&'this str),

    /// File-like.
    FileLike(PyBinaryFile),
}

impl<'this> ReadableAny<'this> {
    /// Into bytes.
    pub fn into_bytes(self) -> io::Result<Cow<'this, [u8]>> {
        Ok(match self {
            Self::Bytes(bytes) => bytes.into(),
            Self::String(string) => string.as_bytes().into(),
            Self::FileLike(mut file_like) => {
                let mut buffer = Vec::default();
                file_like.read_to_end(&mut buffer)?;
                buffer.into()
            }
        })
    }

    /// Into string.
    pub fn into_string(self) -> io::Result<Cow<'this, str>> {
        Ok(match self {
            Self::Bytes(bytes) => from_utf8(bytes).map_err(io::Error::other)?.into(),
            Self::String(string) => string.into(),
            Self::FileLike(file_like) => io::read_to_string(file_like)?.into(),
        })
    }
}

impl<'bound, 'py> From<&'bound Bound<'py, PyAny>> for ReadableAny<'bound> {
    fn from(any: &'bound Bound<'py, PyAny>) -> Self {
        if let Ok(bytes) = any.extract() {
            Self::Bytes(bytes)
        } else if let Ok(string) = any.extract() {
            Self::String(string)
        } else {
            // Bound::clone() is a cheap clone_ref
            Self::FileLike(PyBinaryFile::from(any.clone()))
        }
    }
}