From 07f4218bf4ac4d41c4536eeefc9082be8a28f0fe Mon Sep 17 00:00:00 2001 From: "minh.pham2000" Date: Thu, 16 Feb 2023 10:10:15 +0700 Subject: [PATCH] Add co_freevars to code object --- vm/src/builtins/code.rs | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/vm/src/builtins/code.rs b/vm/src/builtins/code.rs index 24cb97007..c672d963d 100644 --- a/vm/src/builtins/code.rs +++ b/vm/src/builtins/code.rs @@ -262,6 +262,17 @@ impl PyRef { vm.ctx.new_tuple(varnames) } + #[pygetset] + pub fn co_freevars(self, vm: &VirtualMachine) -> PyTupleRef { + let names = self.code + .freevars + .deref() + .iter() + .map(|name| name.to_pyobject(vm)) + .collect(); + vm.ctx.new_tuple(names) + } + #[pymethod] pub fn replace(self, args: ReplaceArgs, vm: &VirtualMachine) -> PyResult { let posonlyarg_count = match args.co_posonlyargcount {