anybytes 0.20.2

A small library abstracting over bytes owning types in an extensible way.
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

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * Copyright (c) Jan-Paul Bultmann
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

use pyo3::{ffi, prelude::*, PyResult};
use std::os::raw::c_int;

use crate::Bytes;

/// Python wrapper around [`Bytes`].
#[pyclass(name = "Bytes")]
pub struct PyAnyBytes {
    bytes: Bytes,
}

#[pymethods]
impl PyAnyBytes {
    /// Exposes the bytes to Python's buffer protocol.
    ///
    /// # Safety
    /// This follows the semantics of the CPython `__getbuffer__` hook and is
    /// therefore unsafe.
    unsafe fn __getbuffer__(
        slf: PyRefMut<Self>,
        view: *mut ffi::Py_buffer,
        flags: c_int,
    ) -> PyResult<()> {
        let bytes = slf.bytes.as_slice();
        let ret = ffi::PyBuffer_FillInfo(
            view,
            slf.as_ptr() as *mut _,
            bytes.as_ptr() as *mut _,
            bytes.len().try_into().unwrap(),
            1, // read only
            flags,
        );
        if ret == -1 {
            return Err(PyErr::fetch(slf.py()));
        }
        Ok(())
    }
}

impl PyAnyBytes {
    /// Wrap a [`Bytes`] instance for Python exposure.
    pub fn new(bytes: Bytes) -> Self {
        Self { bytes }
    }
}