Function bytes_from_object

crates/vm/src/byte.rs:6–18

(vm: &VirtualMachine, obj: &PyObject)

use num_traits::ToPrimitive;

pub fn bytes_from_object(vm: &VirtualMachine, obj: &PyObject) -> PyResult<Vec<u8>> {
    if let Ok(elements) = obj.try_bytes_like(vm, |bytes| bytes.to_vec()) {
        return Ok(elements);
    }

    if !obj.fast_isinstance(vm.ctx.types.str_type)
        && let Ok(elements) = vm.map_iterable_object(obj, |x| value_from_object(vm, &x))
    {
        return elements;
    }

    Err(vm.new_type_error("can assign only bytes, buffers, or iterables of ints in range(0, 256)"))
}

pub fn value_from_object(vm: &VirtualMachine, obj: &PyObject) -> PyResult<u8> {
    obj.try_index(vm)?