Как мне демонтировать PyCodeObject, используя PyO3? - PullRequest
Новая
       74

Как мне демонтировать PyCodeObject, используя PyO3?

1 голос
/ 22 марта 2020

Я читаю файлы .py c, и мне нужно иметь возможность разбирать объекты кода. Когда я пытаюсь уменьшить значение unmarshalled PyAny до PyCodeObject, я получаю следующее сообщение об ошибке: 

error[E0277]: the trait bound `pyo3::ffi::code::PyCodeObject: pyo3::type_object::PyTypeInfo` is not satisfied
   --> src/lib.rs:179:47
    |
179 |         let code = *(loads(py, &code_buffer)?.downcast::<PyCodeObject>()?);
    |                                               ^^^^^^^^ the trait `pyo3::type_object::PyTypeInfo` is not implemented for `pyo3::ffi::code::PyCodeObject`
    |
    = note: required because of the requirements on the impl of `for<'py> pyo3::conversion::PyTryFrom<'py>` for `pyo3::ffi::code::PyCodeObject`

error[E0277]: the trait bound `pyo3::ffi::code::PyCodeObject: pyo3::instance::PyNativeType` is not satisfied
   --> src/lib.rs:179:47
    |
179 |         let code = *(loads(py, &code_buffer)?.downcast::<PyCodeObject>()?);
    |                                               ^^^^^^^^ the trait `pyo3::instance::PyNativeType` is not implemented for `pyo3::ffi::code::PyCodeObject`
    |
    = note: required because of the requirements on the impl of `for<'py> pyo3::conversion::PyTryFrom<'py>` for `pyo3::ffi::code::PyCodeObject`

Как правильно это сделать?

MCVE 

use pyo3::{ffi::PyCodeObject, marshal::loads, Python};

fn main() {
    let gil_guard = Python::acquire_gil();
    let py = gil_guard.python();
    let code_buffer = &include_bytes!("__pycache__/test.cpython-37.pyc")[16..];
    let code = *(loads(py, &code_buffer)
        .unwrap()
        .downcast::<PyCodeObject>()
        .unwrap());
}

Чтобы создать тестовый файл:

  1. Создание файла .py
  2. Импорт модуля в Python (например, python(3) -c 'import ...')
  3. В папке __pycache__ должен быть файл .py c
  4. Заменить путь в коде при вызове на include_bytes! фактическим путем

Информация о версии

  • Ржавчина 2018
  • Ржавчина c 1.43.0-nightly (564758c4 c 2020-03-08)
  • машина go 1.43.0-nightly (bda50510d 2020-03-02)
  • CPython 3.7.3
  • PyO3 0.9.1

1 Ответ

0 голосов
/ 23 марта 2020

Думаю, я понял, как это сделать: 

let code_ptr = loads(py, &code_buffer)?.as_ptr() as *mut PyCodeObject;
// This should be valid, since PyCodeObject is Copy, as long as the refcount is positive
let code = unsafe { *code_ptr };
...