Define classes for all the ast types.

Author: cthulahoops

vm/src/stdlib/ast.rs

@@ -458,10 +458,54 @@ pub fn make_module(vm: &VirtualMachine) -> PyObjectRef {
     let ast_base = py_class!(ctx, "_ast.AST", ctx.object(), {});
     py_module!(vm, "ast", {
         "parse" => ctx.new_rustfunc(ast_parse),
-        "Module" => py_class!(ctx, "_ast.Module", ast_base.clone(), {}),
-        "FunctionDef" => py_class!(ctx, "_ast.FunctionDef", ast_base.clone(), {}),
-        "ClassDef" => py_class!(ctx, "_ast.ClassDef", ast_base.clone(), {}),
+        "AST" => ast_base.clone(),
+        // TODO: There's got to be a better way!
+        "arg" => py_class!(ctx, "_ast.arg", ast_base.clone(), {}),
+        "arguments" => py_class!(ctx, "_ast.arguments", ast_base.clone(), {}),
+        "Assert" => py_class!(ctx, "_ast.Assert", ast_base.clone(), {}),
+        "Attribute" => py_class!(ctx, "_ast.Attribute", ast_base.clone(), {}),
+        "BinOp" => py_class!(ctx, "_ast.BinOp", ast_base.clone(), {}),
+        "BoolOp" => py_class!(ctx, "_ast.BoolOp", ast_base.clone(), {}),
+        "Break" => py_class!(ctx, "_ast.Break", ast_base.clone(), {}),
+        "Bytes" => py_class!(ctx, "_ast.Bytes", ast_base.clone(), {}),
         "Call" => py_class!(ctx, "_ast.Call", ast_base.clone(), {}),
-        "AST" => ast_base,
+        "ClassDef" => py_class!(ctx, "_ast.ClassDef", ast_base.clone(), {}),
+        "Compare" => py_class!(ctx, "_ast.Compare", ast_base.clone(), {}),
+        "comprehension" => py_class!(ctx, "_ast.comprehension", ast_base.clone(), {}),
+        "Continue" => py_class!(ctx, "_ast.Continue", ast_base.clone(), {}),
+        "Delete" => py_class!(ctx, "_ast.Delete", ast_base.clone(), {}),
+        "Dict" => py_class!(ctx, "_ast.Dict", ast_base.clone(), {}),
+        "DictComp" => py_class!(ctx, "_ast.DictComp", ast_base.clone(), {}),
+        "Ellipsis" => py_class!(ctx, "_ast.Ellipsis", ast_base.clone(), {}),
+        "Expr" => py_class!(ctx, "_ast.Expr", ast_base.clone(), {}),
+        "For" => py_class!(ctx, "_ast.For", ast_base.clone(), {}),
+        "FormattedValue" => py_class!(ctx, "_ast.FormattedValue", ast_base.clone(), {}),
+        "FunctionDef" => py_class!(ctx, "_ast.FunctionDef", ast_base.clone(), {}),
+        "GeneratorExp" => py_class!(ctx, "_ast.GeneratorExp", ast_base.clone(), {}),
+        "Identifier" => py_class!(ctx, "_ast.Identifier", ast_base.clone(), {}),
+        "If" => py_class!(ctx, "_ast.If", ast_base.clone(), {}),
+        "IfExp" => py_class!(ctx, "_ast.IfExp", ast_base.clone(), {}),
+        "JoinedStr" => py_class!(ctx, "_ast.JoinedStr", ast_base.clone(), {}),
+        "Lambda" => py_class!(ctx, "_ast.Lambda", ast_base.clone(), {}),
+        "List" => py_class!(ctx, "_ast.List", ast_base.clone(), {}),
+        "ListComp" => py_class!(ctx, "_ast.ListComp", ast_base.clone(), {}),
+        "Module" => py_class!(ctx, "_ast.Module", ast_base.clone(), {}),
+        "NameConstant" => py_class!(ctx, "_ast.NameConstant", ast_base.clone(), {}),
+        "NameConstant" => py_class!(ctx, "_ast.NameConstant", ast_base.clone(), {}),
+        "NameConstant" => py_class!(ctx, "_ast.NameConstant", ast_base.clone(), {}),
+        "Num" => py_class!(ctx, "_ast.Num", ast_base.clone(), {}),
+        "Pass" => py_class!(ctx, "_ast.Pass", ast_base.clone(), {}),
+        "Return" => py_class!(ctx, "_ast.Return", ast_base.clone(), {}),
+        "Set" => py_class!(ctx, "_ast.Set", ast_base.clone(), {}),
+        "SetComp" => py_class!(ctx, "_ast.SetComp", ast_base.clone(), {}),
+        "Starred" => py_class!(ctx, "_ast.Starred", ast_base.clone(), {}),
+        "Starred" => py_class!(ctx, "_ast.Starred", ast_base.clone(), {}),
+        "Str" => py_class!(ctx, "_ast.Str", ast_base.clone(), {}),
+        "Subscript" => py_class!(ctx, "_ast.Subscript", ast_base.clone(), {}),
+        "Tuple" => py_class!(ctx, "_ast.Tuple", ast_base.clone(), {}),
+        "UnaryOp" => py_class!(ctx, "_ast.UnaryOp", ast_base.clone(), {}),
+        "While" => py_class!(ctx, "_ast.While", ast_base.clone(), {}),
+        "Yield" => py_class!(ctx, "_ast.Yield", ast_base.clone(), {}),
+        "YieldFrom" => py_class!(ctx, "_ast.YieldFrom", ast_base.clone(), {}),
     })
 }