forked from Rust-related/RustPython
96 lines
2.6 KiB
Rust
96 lines
2.6 KiB
Rust
/*! Infamous code object. The python class `code`
|
|
|
|
*/
|
|
|
|
use std::fmt;
|
|
|
|
use crate::bytecode;
|
|
use crate::obj::objtype::PyClassRef;
|
|
use crate::pyobject::{IdProtocol, PyContext, PyObjectRef, PyRef, PyResult, PyValue};
|
|
use crate::vm::VirtualMachine;
|
|
|
|
pub type PyCodeRef = PyRef<PyCode>;
|
|
|
|
pub struct PyCode {
|
|
pub code: bytecode::CodeObject,
|
|
}
|
|
|
|
impl PyCode {
|
|
pub fn new(code: bytecode::CodeObject) -> PyCode {
|
|
PyCode { code }
|
|
}
|
|
}
|
|
|
|
impl fmt::Debug for PyCode {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "code: {:?}", self.code)
|
|
}
|
|
}
|
|
|
|
impl PyValue for PyCode {
|
|
fn class(vm: &VirtualMachine) -> PyClassRef {
|
|
vm.ctx.code_type()
|
|
}
|
|
}
|
|
|
|
impl PyCodeRef {
|
|
fn new(_cls: PyClassRef, vm: &VirtualMachine) -> PyResult {
|
|
Err(vm.new_type_error("Cannot directly create code object".to_string()))
|
|
}
|
|
|
|
fn repr(self, _vm: &VirtualMachine) -> String {
|
|
let code = &self.code;
|
|
format!(
|
|
"<code object {} at 0x{:x} file {:?}, line {}>",
|
|
code.obj_name,
|
|
self.get_id(),
|
|
code.source_path,
|
|
code.first_line_number
|
|
)
|
|
}
|
|
|
|
fn co_argcount(self, _vm: &VirtualMachine) -> usize {
|
|
self.code.arg_names.len()
|
|
}
|
|
|
|
fn co_filename(self, _vm: &VirtualMachine) -> String {
|
|
self.code.source_path.clone()
|
|
}
|
|
|
|
fn co_firstlineno(self, _vm: &VirtualMachine) -> usize {
|
|
self.code.first_line_number
|
|
}
|
|
|
|
fn co_kwonlyargcount(self, _vm: &VirtualMachine) -> usize {
|
|
self.code.kwonlyarg_names.len()
|
|
}
|
|
|
|
fn co_consts(self, vm: &VirtualMachine) -> PyObjectRef {
|
|
let consts = self
|
|
.code
|
|
.get_constants()
|
|
.map(|x| vm.ctx.unwrap_constant(x))
|
|
.collect();
|
|
vm.ctx.new_tuple(consts)
|
|
}
|
|
|
|
fn co_name(self, _vm: &VirtualMachine) -> String {
|
|
self.code.obj_name.clone()
|
|
}
|
|
}
|
|
|
|
pub fn init(context: &PyContext) {
|
|
let code_type = context.code_type.as_object();
|
|
extend_class!(context, code_type, {
|
|
"__new__" => context.new_rustfunc(PyCodeRef::new),
|
|
"__repr__" => context.new_rustfunc(PyCodeRef::repr),
|
|
|
|
"co_argcount" => context.new_property(PyCodeRef::co_argcount),
|
|
"co_consts" => context.new_property(PyCodeRef::co_consts),
|
|
"co_filename" => context.new_property(PyCodeRef::co_filename),
|
|
"co_firstlineno" => context.new_property(PyCodeRef::co_firstlineno),
|
|
"co_kwonlyargcount" => context.new_property(PyCodeRef::co_kwonlyargcount),
|
|
"co_name" => context.new_property(PyCodeRef::co_name),
|
|
});
|
|
}
|