Add co_freevars to code object

minhrongcon2000 · minhrongcon2000 · commit 07f4218bf4ac · 2023-02-16T10:10:15.000+07:00
diff --git a/vm/src/builtins/code.rs b/vm/src/builtins/code.rs
@@ -262,6 +262,17 @@ 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 {
