From 07f4218bf4ac4d41c4536eeefc9082be8a28f0fe Mon Sep 17 00:00:00 2001 From: "minh.pham2000" Date: Thu, 16 Feb 2023 10:10:15 +0700 Subject: [PATCH 1/2] 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 24cb9700775..c672d963d38 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 { From 81fc6f16cf3f31c3ada464871e87cde8d4f77111 Mon Sep 17 00:00:00 2001 From: "minh.pham2000" Date: Thu, 16 Feb 2023 10:19:56 +0700 Subject: [PATCH 2/2] Fix code linting --- vm/src/builtins/code.rs | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/vm/src/builtins/code.rs b/vm/src/builtins/code.rs index c672d963d38..97c32dbe075 100644 --- a/vm/src/builtins/code.rs +++ b/vm/src/builtins/code.rs @@ -264,12 +264,13 @@ impl PyRef { #[pygetset] pub fn co_freevars(self, vm: &VirtualMachine) -> PyTupleRef { - let names = self.code - .freevars - .deref() - .iter() - .map(|name| name.to_pyobject(vm)) - .collect(); + let names = self + .code + .freevars + .deref() + .iter() + .map(|name| name.to_pyobject(vm)) + .collect(); vm.ctx.new_tuple(names) }