Merge pull request RustPython#2115 from minoring/support-__all__

Support __all__

Author: youknowone

Lib/_codecs.py

@@ -46,7 +46,7 @@
            'latin_1_encode', 'mbcs_decode', 'readbuffer_encode', 'escape_encode',
            'utf_8_decode', 'raw_unicode_escape_decode', 'utf_7_decode',
            'unicode_escape_encode', 'latin_1_decode', 'utf_16_decode',
-           'unicode_escape_decode', 'ascii_decode', 'charmap_encode',
+           'unicode_escape_decode', 'ascii_decode', 'charmap_encode', 'charmap_build',
            'unicode_internal_encode', 'unicode_internal_decode', 'utf_16_ex_decode',
            'escape_decode', 'charmap_decode', 'utf_7_encode', 'mbcs_encode',
            'ascii_encode', 'utf_16_encode', 'raw_unicode_escape_encode', 'utf_8_encode',

Lib/asyncio/tasks.py

@@ -4,6 +4,7 @@
            'FIRST_COMPLETED', 'FIRST_EXCEPTION', 'ALL_COMPLETED',
            'wait', 'wait_for', 'as_completed', 'sleep', 'async',
            'gather', 'shield', 'ensure_future', 'run_coroutine_threadsafe',
+           'all_tasks'
            ]
 
 import concurrent.futures

vm/src/frame.rs

@@ -19,7 +19,7 @@ use crate::obj::objgenerator::PyGenerator;
 use crate::obj::objiter;
 use crate::obj::objlist;
 use crate::obj::objslice::PySlice;
-use crate::obj::objstr::{self, PyString};
+use crate::obj::objstr::{self, PyString, PyStringRef};
 use crate::obj::objtraceback::PyTraceback;
 use crate::obj::objtuple::PyTuple;
 use crate::obj::objtype::{self, PyClassRef};
@@ -797,10 +797,21 @@ impl ExecutingFrame<'_> {
 
         // Grab all the names from the module and put them in the context
         if let Some(dict) = module.dict() {
+            let filter_pred: Box<dyn Fn(&str) -> bool> =
+                if let Ok(all) = dict.get_item("__all__", vm) {
+                    let all: Vec<PyStringRef> = vm.extract_elements(&all)?;
+                    let all: Vec<String> = all
+                        .into_iter()
+                        .map(|name| name.as_ref().to_owned())
+                        .collect();
+                    Box::new(move |name| all.contains(&name.to_owned()))
+                } else {
+                    Box::new(|name| !name.starts_with('_'))
+                };
             for (k, v) in &dict {
-                let k = vm.to_str(&k)?;
-                let k = k.borrow_value();
-                if !k.starts_with('_') {
+                let k = PyStringRef::try_from_object(vm, k)?;
+                let k = k.as_ref();
+                if filter_pred(k) {
                     self.scope.store_name(&vm, k, v);
                 }
             }