Support set nodes in simple_eval

sebastinas · sebastinas · commit 99c601a7ce88 · 2020-10-11T11:28:29.000+02:00
Merges changes from Python 3.9's ast.literal_eval.
diff --git a/bpython/simpleeval.py b/bpython/simpleeval.py
@@ -81,7 +81,7 @@ def simple_eval(node_or_string, namespace=None):
 
     The string or node provided may only consist of:
     * the following Python literal structures: strings, numbers, tuples,
-        lists, and dicts
+        lists, dicts, sets (Python 3.9+)
     * variable names causing lookups in the passed in namespace or builtins
     * getitem calls using the [] syntax on objects of the types above
 
@@ -113,6 +113,8 @@ def _convert(node):
                 (_convert(k), _convert(v))
                 for k, v in zip(node.keys, node.values)
             )
+        elif sys.version_info[:2] >= (3, 9) and isinstance(node, ast.Set):
+            return set(map(_convert, node.elts))
 
         # this is a deviation from literal_eval: we allow non-literals
         elif isinstance(node, _name_type_nodes):
