Merge pull request #4509 from minhrongcon2000/fix/add_co_freevars

DimitrisJim · web-flow · commit dcda3e43b923 · 2023-02-16T09:24:23.000+02:00
Add co_freevars to code object
diff --git a/vm/src/builtins/code.rs b/vm/src/builtins/code.rs
@@ -262,6 +262,18 @@ impl PyRef<PyCode> {
         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<PyCode> {
         let posonlyarg_count = match args.co_posonlyargcount {
