Commit008c364

committed

Add support for __prepare__.

1 parent4be2f24 commit008c364Copy full SHA for 008c364

File tree

+25

-4

lines changed

+25

-4

lines changed

Lines changed: 7 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,9 @@ class MC(type):`
`2`	`2`	`classes= []`
`3`	`3`	`count=0`
`4`	`4`
	`5`	`+def__prepare__(name,bases):`
	`6`	`+return {'prepared':True}`
	`7`	`+`
`5`	`8`	`def__new__(cls,name,bases,namespace):`
`6`	`9`	`MC.classes.append(name)`
`7`	`10`	`returntype.__new__(cls,name,bases,namespace)`
`@@ -16,10 +19,13 @@ def __new__(cls, count):`
`16`	`19`	`self.count=count`
`17`	`20`	`returnself`
`18`	`21`
`19`		`-classD(object,metaclass=MC):`
	`22`	`+classD(metaclass=MC):`
`20`	`23`	`pass`
`21`	`24`
`22`	`25`	`assertMC==type(C)`
`23`	`26`	`assertC==type(C())`
`24`	`27`	`assertMC.classes== ['C','D']`
`25`	`28`	`assertC().count==2`
	`29`	`+`
	`30`	`+assertC.prepared`
	`31`	`+assertD.prepared`

Lines changed: 13 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -640,7 +640,19 @@ pub fn builtin_build_class_(vm: &mut VirtualMachine, mut args: PyFuncArgs) -> Py`
`640`	`640`	`let metaclass = args.get_kwarg("metaclass", vm.get_type());`
`641`	`641`
`642`	`642`	`bases.push(vm.context().object());`
`643`		`-let namespace = vm.new_dict();`
	`643`	`+let bases = vm.context().new_tuple(bases);`
	`644`	`+`
	`645`	`+// Prepare uses full __getattribute__ resolution chain.`
	`646`	`+let prepare_name = vm.new_str("__prepare__".to_string());`
	`647`	`+let prepare = vm.get_attribute(metaclass.clone(), prepare_name)?;`
	`648`	`+let namespace = vm.invoke(`
	`649`	`+ prepare,`
	`650`	`+PyFuncArgs{`
	`651`	`+args:vec![name_arg.clone(), bases.clone()],`
	`652`	`+kwargs:vec![],`
	`653`	`+},`
	`654`	`+)?;`
	`655`	`+`
`644`	`656`	`&vm.invoke(`
`645`	`657`	`function,`
`646`	`658`	`PyFuncArgs{`
`@@ -649,8 +661,6 @@ pub fn builtin_build_class_(vm: &mut VirtualMachine, mut args: PyFuncArgs) -> Py`
`649`	`661`	`},`
`650`	`662`	`);`
`651`	`663`
`652`		`-let bases = vm.context().new_tuple(bases);`
`653`		`-`
`654`	`664`	`// Special case: __new__ must be looked up on the metaclass, not the meta-metaclass as`
`655`	`665`	`// per vm.call(metaclass, "__new__", ...)`
`656`	`666`	`let new = metaclass.get_attr("__new__").unwrap();`

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,7 @@ pub fn init(context: &PyContext) {`
`28`	`28`	`type_type.set_attr("__mro__", context.new_member_descriptor(type_mro));`
`29`	`29`	`type_type.set_attr("__class__", context.new_member_descriptor(type_new));`
`30`	`30`	`type_type.set_attr("__repr__", context.new_rustfunc(type_repr));`
	`31`	`+ type_type.set_attr("__prepare__", context.new_rustfunc(type_prepare));`
`31`	`32`	`}`
`32`	`33`
`33`	`34`	`fntype_mro(vm:&mutVirtualMachine,args:PyFuncArgs) ->PyResult{`
`@@ -300,6 +301,10 @@ fn type_repr(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {`
`300`	`301`	`Ok(vm.new_str(format!("<class '{}'>", type_name)))`
`301`	`302`	`}`
`302`	`303`
	`304`	`+fntype_prepare(vm:&mutVirtualMachine,_args:PyFuncArgs) ->PyResult{`
	`305`	`+Ok(vm.new_dict())`
	`306`	`+}`
	`307`	`+`
`303`	`308`	`#[cfg(test)]`
`304`	`309`	`mod tests{`
`305`	`310`	`usesuper::{linearise_mro, new};`

Comments

(0)