Rust-related/RustPython
1
0
Fork 1
mirror of https://github.com/RustPython/RustPython.git synced 2026-06-17 01:51:39 +09:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1787 from youknowone/posonly

Browse Source 
PEP570 - positional only parameter support
This commit is contained in:
Noah
2020-03-13 17:19:21 -05:00
committed by GitHub
parent 1fd1c19cdf b4f0ced701
commit 75eabfdc2c
8 changed files with 1893 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
vm/src/obj/objcode.rs
View File

@@ -7,11 +7,12 @@ use std::ops::Deref;
use super::objtype::PyClassRef;
use crate::bytecode;
use crate::pyobject::{IdProtocol, PyContext, PyObjectRef, PyRef, PyResult, PyValue};
use crate::pyobject::{IdProtocol, PyClassImpl, PyContext, PyObjectRef, PyRef, PyResult, PyValue};
use crate::vm::VirtualMachine;
pub type PyCodeRef = PyRef<PyCode>;
#[pyclass]
pub struct PyCode {
pub code: bytecode::CodeObject,
}
@@ -41,12 +42,15 @@ impl PyValue for PyCode {
}
}
#[pyimpl]
impl PyCodeRef {
#[pyslot]
#[allow(clippy::new_ret_no_self)]
fn new(_cls: PyClassRef, vm: &VirtualMachine) -> PyResult {
Err(vm.new_type_error("Cannot directly create code object".to_owned()))
}
#[pymethod(magic)]
fn repr(self) -> String {
let code = &self.code;
format!(
@@ -58,22 +62,32 @@ impl PyCodeRef {
)
}
#[pyproperty]
fn co_posonlyargcount(self) -> usize {
self.code.posonlyarg_count
}
#[pyproperty]
fn co_argcount(self) -> usize {
self.code.arg_names.len()
}
#[pyproperty]
fn co_filename(self) -> String {
self.code.source_path.clone()
}
#[pyproperty]
fn co_firstlineno(self) -> usize {
self.code.first_line_number
}
#[pyproperty]
fn co_kwonlyargcount(self) -> usize {
self.code.kwonlyarg_names.len()
}
#[pyproperty]
fn co_consts(self, vm: &VirtualMachine) -> PyObjectRef {
let consts = self
.code
@@ -83,26 +97,17 @@ impl PyCodeRef {
vm.ctx.new_tuple(consts)
}
#[pyproperty]
fn co_name(self) -> String {
self.code.obj_name.clone()
}
#[pyproperty]
fn co_flags(self) -> u8 {
self.code.flags.bits()
}
}
pub fn init(ctx: &PyContext) {
extend_class!(ctx, &ctx.types.code_type, {
(slot new) => PyCodeRef::new,
"__repr__" => ctx.new_method(PyCodeRef::repr),
"co_argcount" => ctx.new_readonly_getset("co_argcount", PyCodeRef::co_argcount),
"co_consts" => ctx.new_readonly_getset("co_consts", PyCodeRef::co_consts),
"co_filename" => ctx.new_readonly_getset("co_filename", PyCodeRef::co_filename),
"co_firstlineno" => ctx.new_readonly_getset("co_firstlineno", PyCodeRef::co_firstlineno),
"co_kwonlyargcount" => ctx.new_readonly_getset("co_kwonlyargcount", PyCodeRef::co_kwonlyargcount),
"co_name" => ctx.new_readonly_getset("co_name", PyCodeRef::co_name),
"co_flags" => ctx.new_readonly_getset("co_flags", PyCodeRef::co_flags),
});
PyCodeRef::extend_class(ctx, &ctx.types.code_type);
}