blob: 2603b812f08688336f4ebe950ada0bc8c25e7610 [file] [log] [blame]
Thomas Wouterscf297e42007-02-23 15:07:44 +00001/* File automatically generated by Parser/asdl_c.py. */
2
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003#include "Python.h"
4#include "Python-ast.h"
5
Neal Norwitz207c9f32008-03-31 04:42:11 +00006static PyTypeObject AST_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +00007static PyTypeObject *mod_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00008static PyObject* ast2obj_mod(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +00009static PyTypeObject *Module_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020010_Py_IDENTIFIER(body);
Neal Norwitz53d960c2006-02-28 22:47:29 +000011static char *Module_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000012 "body",
13};
Neal Norwitz53d960c2006-02-28 22:47:29 +000014static PyTypeObject *Interactive_type;
15static char *Interactive_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000016 "body",
17};
Neal Norwitz53d960c2006-02-28 22:47:29 +000018static PyTypeObject *Expression_type;
19static char *Expression_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000020 "body",
21};
Neal Norwitz53d960c2006-02-28 22:47:29 +000022static PyTypeObject *Suite_type;
23static char *Suite_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000024 "body",
25};
Neal Norwitz53d960c2006-02-28 22:47:29 +000026static PyTypeObject *stmt_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020027_Py_IDENTIFIER(lineno);
28_Py_IDENTIFIER(col_offset);
Neal Norwitz53d960c2006-02-28 22:47:29 +000029static char *stmt_attributes[] = {
Martin v. Löwis577b5b92006-02-27 15:23:19 +000030 "lineno",
Martin v. Löwis49c5da12006-03-01 22:49:05 +000031 "col_offset",
Martin v. Löwis577b5b92006-02-27 15:23:19 +000032};
Martin v. Löwisbd260da2006-02-26 19:42:26 +000033static PyObject* ast2obj_stmt(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +000034static PyTypeObject *FunctionDef_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020035_Py_IDENTIFIER(name);
36_Py_IDENTIFIER(args);
37_Py_IDENTIFIER(decorator_list);
38_Py_IDENTIFIER(returns);
Neal Norwitz53d960c2006-02-28 22:47:29 +000039static char *FunctionDef_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000040 "name",
41 "args",
42 "body",
Guido van Rossumd59da4b2007-05-22 18:11:13 +000043 "decorator_list",
Neal Norwitzc1505362006-12-28 06:47:50 +000044 "returns",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000045};
Neal Norwitz53d960c2006-02-28 22:47:29 +000046static PyTypeObject *ClassDef_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020047_Py_IDENTIFIER(bases);
48_Py_IDENTIFIER(keywords);
49_Py_IDENTIFIER(starargs);
50_Py_IDENTIFIER(kwargs);
Neal Norwitz53d960c2006-02-28 22:47:29 +000051static char *ClassDef_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000052 "name",
53 "bases",
Guido van Rossum52cc1d82007-03-18 15:41:51 +000054 "keywords",
55 "starargs",
56 "kwargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000057 "body",
Guido van Rossumd59da4b2007-05-22 18:11:13 +000058 "decorator_list",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000059};
Neal Norwitz53d960c2006-02-28 22:47:29 +000060static PyTypeObject *Return_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020061_Py_IDENTIFIER(value);
Neal Norwitz53d960c2006-02-28 22:47:29 +000062static char *Return_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000063 "value",
64};
Neal Norwitz53d960c2006-02-28 22:47:29 +000065static PyTypeObject *Delete_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020066_Py_IDENTIFIER(targets);
Neal Norwitz53d960c2006-02-28 22:47:29 +000067static char *Delete_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000068 "targets",
69};
Neal Norwitz53d960c2006-02-28 22:47:29 +000070static PyTypeObject *Assign_type;
71static char *Assign_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000072 "targets",
73 "value",
74};
Neal Norwitz53d960c2006-02-28 22:47:29 +000075static PyTypeObject *AugAssign_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020076_Py_IDENTIFIER(target);
77_Py_IDENTIFIER(op);
Neal Norwitz53d960c2006-02-28 22:47:29 +000078static char *AugAssign_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000079 "target",
80 "op",
81 "value",
82};
Neal Norwitz53d960c2006-02-28 22:47:29 +000083static PyTypeObject *For_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020084_Py_IDENTIFIER(iter);
85_Py_IDENTIFIER(orelse);
Neal Norwitz53d960c2006-02-28 22:47:29 +000086static char *For_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000087 "target",
88 "iter",
89 "body",
90 "orelse",
91};
Neal Norwitz53d960c2006-02-28 22:47:29 +000092static PyTypeObject *While_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020093_Py_IDENTIFIER(test);
Neal Norwitz53d960c2006-02-28 22:47:29 +000094static char *While_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000095 "test",
96 "body",
97 "orelse",
98};
Neal Norwitz53d960c2006-02-28 22:47:29 +000099static PyTypeObject *If_type;
100static char *If_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000101 "test",
102 "body",
103 "orelse",
104};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000105static PyTypeObject *With_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200106_Py_IDENTIFIER(items);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000107static char *With_fields[]={
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500108 "items",
Guido van Rossumc2e20742006-02-27 22:32:47 +0000109 "body",
110};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000111static PyTypeObject *Raise_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200112_Py_IDENTIFIER(exc);
113_Py_IDENTIFIER(cause);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000114static char *Raise_fields[]={
Collin Winter828f04a2007-08-31 00:04:24 +0000115 "exc",
116 "cause",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000117};
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500118static PyTypeObject *Try_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200119_Py_IDENTIFIER(handlers);
120_Py_IDENTIFIER(finalbody);
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500121static char *Try_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000122 "body",
123 "handlers",
124 "orelse",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000125 "finalbody",
126};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000127static PyTypeObject *Assert_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200128_Py_IDENTIFIER(msg);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000129static char *Assert_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000130 "test",
131 "msg",
132};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000133static PyTypeObject *Import_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200134_Py_IDENTIFIER(names);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000135static char *Import_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000136 "names",
137};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000138static PyTypeObject *ImportFrom_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200139_Py_IDENTIFIER(module);
140_Py_IDENTIFIER(level);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000141static char *ImportFrom_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000142 "module",
143 "names",
Thomas Woutersf7f438b2006-02-28 16:09:29 +0000144 "level",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000145};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000146static PyTypeObject *Global_type;
147static char *Global_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000148 "names",
149};
Jeremy Hylton81e95022007-02-27 06:50:52 +0000150static PyTypeObject *Nonlocal_type;
151static char *Nonlocal_fields[]={
152 "names",
153};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000154static PyTypeObject *Expr_type;
155static char *Expr_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000156 "value",
157};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000158static PyTypeObject *Pass_type;
159static PyTypeObject *Break_type;
160static PyTypeObject *Continue_type;
161static PyTypeObject *expr_type;
162static char *expr_attributes[] = {
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000163 "lineno",
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000164 "col_offset",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000165};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000166static PyObject* ast2obj_expr(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000167static PyTypeObject *BoolOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200168_Py_IDENTIFIER(values);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000169static char *BoolOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000170 "op",
171 "values",
172};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000173static PyTypeObject *BinOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200174_Py_IDENTIFIER(left);
175_Py_IDENTIFIER(right);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000176static char *BinOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000177 "left",
178 "op",
179 "right",
180};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000181static PyTypeObject *UnaryOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200182_Py_IDENTIFIER(operand);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000183static char *UnaryOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000184 "op",
185 "operand",
186};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000187static PyTypeObject *Lambda_type;
188static char *Lambda_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000189 "args",
190 "body",
191};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000192static PyTypeObject *IfExp_type;
193static char *IfExp_fields[]={
Thomas Woutersdca3b9c2006-02-27 00:24:13 +0000194 "test",
195 "body",
196 "orelse",
197};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000198static PyTypeObject *Dict_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200199_Py_IDENTIFIER(keys);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000200static char *Dict_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000201 "keys",
202 "values",
203};
Guido van Rossum86e58e22006-08-28 15:27:34 +0000204static PyTypeObject *Set_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200205_Py_IDENTIFIER(elts);
Guido van Rossum86e58e22006-08-28 15:27:34 +0000206static char *Set_fields[]={
207 "elts",
208};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000209static PyTypeObject *ListComp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200210_Py_IDENTIFIER(elt);
211_Py_IDENTIFIER(generators);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000212static char *ListComp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000213 "elt",
214 "generators",
215};
Nick Coghlan650f0d02007-04-15 12:05:43 +0000216static PyTypeObject *SetComp_type;
217static char *SetComp_fields[]={
218 "elt",
219 "generators",
220};
Guido van Rossum992d4a32007-07-11 13:09:30 +0000221static PyTypeObject *DictComp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200222_Py_IDENTIFIER(key);
Guido van Rossum992d4a32007-07-11 13:09:30 +0000223static char *DictComp_fields[]={
224 "key",
225 "value",
226 "generators",
227};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000228static PyTypeObject *GeneratorExp_type;
229static char *GeneratorExp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000230 "elt",
231 "generators",
232};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000233static PyTypeObject *Yield_type;
234static char *Yield_fields[]={
Benjamin Peterson527c6222012-01-14 08:58:23 -0500235 "value",
236};
237static PyTypeObject *YieldFrom_type;
238static char *YieldFrom_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000239 "value",
240};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000241static PyTypeObject *Compare_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200242_Py_IDENTIFIER(ops);
243_Py_IDENTIFIER(comparators);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000244static char *Compare_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000245 "left",
246 "ops",
247 "comparators",
248};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000249static PyTypeObject *Call_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200250_Py_IDENTIFIER(func);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000251static char *Call_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000252 "func",
253 "args",
254 "keywords",
255 "starargs",
256 "kwargs",
257};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000258static PyTypeObject *Num_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200259_Py_IDENTIFIER(n);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000260static char *Num_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000261 "n",
262};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000263static PyTypeObject *Str_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200264_Py_IDENTIFIER(s);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000265static char *Str_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000266 "s",
267};
Thomas Wouters00e41de2007-02-23 19:56:57 +0000268static PyTypeObject *Bytes_type;
269static char *Bytes_fields[]={
270 "s",
271};
Georg Brandl52318d62006-09-06 07:06:08 +0000272static PyTypeObject *Ellipsis_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +0000273static PyTypeObject *Attribute_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200274_Py_IDENTIFIER(attr);
275_Py_IDENTIFIER(ctx);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000276static char *Attribute_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000277 "value",
278 "attr",
279 "ctx",
280};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000281static PyTypeObject *Subscript_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200282_Py_IDENTIFIER(slice);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000283static char *Subscript_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000284 "value",
285 "slice",
286 "ctx",
287};
Guido van Rossum0368b722007-05-11 16:50:42 +0000288static PyTypeObject *Starred_type;
289static char *Starred_fields[]={
290 "value",
291 "ctx",
292};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000293static PyTypeObject *Name_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200294_Py_IDENTIFIER(id);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000295static char *Name_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000296 "id",
297 "ctx",
298};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000299static PyTypeObject *List_type;
300static char *List_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000301 "elts",
302 "ctx",
303};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000304static PyTypeObject *Tuple_type;
305static char *Tuple_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000306 "elts",
307 "ctx",
308};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000309static PyTypeObject *expr_context_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000310static PyObject *Load_singleton, *Store_singleton, *Del_singleton,
311*AugLoad_singleton, *AugStore_singleton, *Param_singleton;
312static PyObject* ast2obj_expr_context(expr_context_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000313static PyTypeObject *Load_type;
314static PyTypeObject *Store_type;
315static PyTypeObject *Del_type;
316static PyTypeObject *AugLoad_type;
317static PyTypeObject *AugStore_type;
318static PyTypeObject *Param_type;
319static PyTypeObject *slice_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000320static PyObject* ast2obj_slice(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000321static PyTypeObject *Slice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200322_Py_IDENTIFIER(lower);
323_Py_IDENTIFIER(upper);
324_Py_IDENTIFIER(step);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000325static char *Slice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000326 "lower",
327 "upper",
328 "step",
329};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000330static PyTypeObject *ExtSlice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200331_Py_IDENTIFIER(dims);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000332static char *ExtSlice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000333 "dims",
334};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000335static PyTypeObject *Index_type;
336static char *Index_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000337 "value",
338};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000339static PyTypeObject *boolop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000340static PyObject *And_singleton, *Or_singleton;
341static PyObject* ast2obj_boolop(boolop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000342static PyTypeObject *And_type;
343static PyTypeObject *Or_type;
344static PyTypeObject *operator_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000345static PyObject *Add_singleton, *Sub_singleton, *Mult_singleton,
346*Div_singleton, *Mod_singleton, *Pow_singleton, *LShift_singleton,
347*RShift_singleton, *BitOr_singleton, *BitXor_singleton, *BitAnd_singleton,
348*FloorDiv_singleton;
349static PyObject* ast2obj_operator(operator_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000350static PyTypeObject *Add_type;
351static PyTypeObject *Sub_type;
352static PyTypeObject *Mult_type;
353static PyTypeObject *Div_type;
354static PyTypeObject *Mod_type;
355static PyTypeObject *Pow_type;
356static PyTypeObject *LShift_type;
357static PyTypeObject *RShift_type;
358static PyTypeObject *BitOr_type;
359static PyTypeObject *BitXor_type;
360static PyTypeObject *BitAnd_type;
361static PyTypeObject *FloorDiv_type;
362static PyTypeObject *unaryop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000363static PyObject *Invert_singleton, *Not_singleton, *UAdd_singleton,
364*USub_singleton;
365static PyObject* ast2obj_unaryop(unaryop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000366static PyTypeObject *Invert_type;
367static PyTypeObject *Not_type;
368static PyTypeObject *UAdd_type;
369static PyTypeObject *USub_type;
370static PyTypeObject *cmpop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000371static PyObject *Eq_singleton, *NotEq_singleton, *Lt_singleton, *LtE_singleton,
372*Gt_singleton, *GtE_singleton, *Is_singleton, *IsNot_singleton, *In_singleton,
373*NotIn_singleton;
374static PyObject* ast2obj_cmpop(cmpop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000375static PyTypeObject *Eq_type;
376static PyTypeObject *NotEq_type;
377static PyTypeObject *Lt_type;
378static PyTypeObject *LtE_type;
379static PyTypeObject *Gt_type;
380static PyTypeObject *GtE_type;
381static PyTypeObject *Is_type;
382static PyTypeObject *IsNot_type;
383static PyTypeObject *In_type;
384static PyTypeObject *NotIn_type;
385static PyTypeObject *comprehension_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000386static PyObject* ast2obj_comprehension(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200387_Py_IDENTIFIER(ifs);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000388static char *comprehension_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000389 "target",
390 "iter",
391 "ifs",
392};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000393static PyTypeObject *excepthandler_type;
Neal Norwitzad74aa82008-03-31 05:14:30 +0000394static char *excepthandler_attributes[] = {
395 "lineno",
396 "col_offset",
397};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000398static PyObject* ast2obj_excepthandler(void*);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000399static PyTypeObject *ExceptHandler_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200400_Py_IDENTIFIER(type);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000401static char *ExceptHandler_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000402 "type",
403 "name",
404 "body",
405};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000406static PyTypeObject *arguments_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000407static PyObject* ast2obj_arguments(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200408_Py_IDENTIFIER(vararg);
409_Py_IDENTIFIER(varargannotation);
410_Py_IDENTIFIER(kwonlyargs);
411_Py_IDENTIFIER(kwarg);
412_Py_IDENTIFIER(kwargannotation);
413_Py_IDENTIFIER(defaults);
414_Py_IDENTIFIER(kw_defaults);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000415static char *arguments_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000416 "args",
417 "vararg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000418 "varargannotation",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000419 "kwonlyargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000420 "kwarg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000421 "kwargannotation",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000422 "defaults",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000423 "kw_defaults",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000424};
Neal Norwitzc1505362006-12-28 06:47:50 +0000425static PyTypeObject *arg_type;
426static PyObject* ast2obj_arg(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200427_Py_IDENTIFIER(arg);
428_Py_IDENTIFIER(annotation);
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000429static char *arg_fields[]={
Neal Norwitzc1505362006-12-28 06:47:50 +0000430 "arg",
431 "annotation",
432};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000433static PyTypeObject *keyword_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000434static PyObject* ast2obj_keyword(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000435static char *keyword_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000436 "arg",
437 "value",
438};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000439static PyTypeObject *alias_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000440static PyObject* ast2obj_alias(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200441_Py_IDENTIFIER(asname);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000442static char *alias_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000443 "name",
444 "asname",
445};
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500446static PyTypeObject *withitem_type;
447static PyObject* ast2obj_withitem(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200448_Py_IDENTIFIER(context_expr);
449_Py_IDENTIFIER(optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500450static char *withitem_fields[]={
451 "context_expr",
452 "optional_vars",
453};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000454
455
Neal Norwitz207c9f32008-03-31 04:42:11 +0000456static int
457ast_type_init(PyObject *self, PyObject *args, PyObject *kw)
458{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200459 _Py_IDENTIFIER(_fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000460 Py_ssize_t i, numfields = 0;
461 int res = -1;
462 PyObject *key, *value, *fields;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200463 fields = _PyObject_GetAttrId((PyObject*)Py_TYPE(self), &PyId__fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000464 if (!fields)
465 PyErr_Clear();
466 if (fields) {
467 numfields = PySequence_Size(fields);
468 if (numfields == -1)
469 goto cleanup;
470 }
471 res = 0; /* if no error occurs, this stays 0 to the end */
472 if (PyTuple_GET_SIZE(args) > 0) {
473 if (numfields != PyTuple_GET_SIZE(args)) {
474 PyErr_Format(PyExc_TypeError, "%.400s constructor takes %s"
Amaury Forgeot d'Arc245c70b2008-09-10 22:24:24 +0000475 "%zd positional argument%s",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000476 Py_TYPE(self)->tp_name,
477 numfields == 0 ? "" : "either 0 or ",
478 numfields, numfields == 1 ? "" : "s");
479 res = -1;
480 goto cleanup;
481 }
482 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
483 /* cannot be reached when fields is NULL */
484 PyObject *name = PySequence_GetItem(fields, i);
485 if (!name) {
486 res = -1;
487 goto cleanup;
488 }
489 res = PyObject_SetAttr(self, name, PyTuple_GET_ITEM(args, i));
490 Py_DECREF(name);
491 if (res < 0)
492 goto cleanup;
493 }
494 }
495 if (kw) {
496 i = 0; /* needed by PyDict_Next */
497 while (PyDict_Next(kw, &i, &key, &value)) {
498 res = PyObject_SetAttr(self, key, value);
499 if (res < 0)
500 goto cleanup;
501 }
502 }
503 cleanup:
504 Py_XDECREF(fields);
505 return res;
506}
507
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000508/* Pickling support */
509static PyObject *
510ast_type_reduce(PyObject *self, PyObject *unused)
511{
512 PyObject *res;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200513 _Py_IDENTIFIER(__dict__);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200514 PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__);
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000515 if (dict == NULL) {
516 if (PyErr_ExceptionMatches(PyExc_AttributeError))
517 PyErr_Clear();
518 else
519 return NULL;
520 }
521 if (dict) {
522 res = Py_BuildValue("O()O", Py_TYPE(self), dict);
523 Py_DECREF(dict);
524 return res;
525 }
526 return Py_BuildValue("O()", Py_TYPE(self));
527}
528
529static PyMethodDef ast_type_methods[] = {
530 {"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
531 {NULL}
532};
533
Neal Norwitz207c9f32008-03-31 04:42:11 +0000534static PyTypeObject AST_type = {
535 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000536 "_ast.AST",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000537 sizeof(PyObject),
538 0,
539 0, /* tp_dealloc */
540 0, /* tp_print */
541 0, /* tp_getattr */
542 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000543 0, /* tp_reserved */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000544 0, /* tp_repr */
545 0, /* tp_as_number */
546 0, /* tp_as_sequence */
547 0, /* tp_as_mapping */
548 0, /* tp_hash */
549 0, /* tp_call */
550 0, /* tp_str */
551 PyObject_GenericGetAttr, /* tp_getattro */
552 PyObject_GenericSetAttr, /* tp_setattro */
553 0, /* tp_as_buffer */
554 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
555 0, /* tp_doc */
556 0, /* tp_traverse */
557 0, /* tp_clear */
558 0, /* tp_richcompare */
559 0, /* tp_weaklistoffset */
560 0, /* tp_iter */
561 0, /* tp_iternext */
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000562 ast_type_methods, /* tp_methods */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000563 0, /* tp_members */
564 0, /* tp_getset */
565 0, /* tp_base */
566 0, /* tp_dict */
567 0, /* tp_descr_get */
568 0, /* tp_descr_set */
569 0, /* tp_dictoffset */
570 (initproc)ast_type_init, /* tp_init */
571 PyType_GenericAlloc, /* tp_alloc */
572 PyType_GenericNew, /* tp_new */
573 PyObject_Del, /* tp_free */
574};
575
576
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000577static PyTypeObject* make_type(char *type, PyTypeObject* base, char**fields, int num_fields)
578{
579 PyObject *fnames, *result;
580 int i;
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000581 fnames = PyTuple_New(num_fields);
582 if (!fnames) return NULL;
583 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000584 PyObject *field = PyUnicode_FromString(fields[i]);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000585 if (!field) {
586 Py_DECREF(fnames);
587 return NULL;
588 }
589 PyTuple_SET_ITEM(fnames, i, field);
590 }
Victor Stinner7eeb5b52010-06-07 19:57:46 +0000591 result = PyObject_CallFunction((PyObject*)&PyType_Type, "s(O){sOss}",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000592 type, base, "_fields", fnames, "__module__", "_ast");
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000593 Py_DECREF(fnames);
594 return (PyTypeObject*)result;
595}
596
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000597static int add_attributes(PyTypeObject* type, char**attrs, int num_fields)
598{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000599 int i, result;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200600 _Py_IDENTIFIER(_attributes);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000601 PyObject *s, *l = PyTuple_New(num_fields);
Benjamin Peterson3e5cd1d2010-06-27 21:45:24 +0000602 if (!l)
603 return 0;
604 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000605 s = PyUnicode_FromString(attrs[i]);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000606 if (!s) {
607 Py_DECREF(l);
608 return 0;
609 }
Neal Norwitz207c9f32008-03-31 04:42:11 +0000610 PyTuple_SET_ITEM(l, i, s);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000611 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200612 result = _PyObject_SetAttrId((PyObject*)type, &PyId__attributes, l) >= 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000613 Py_DECREF(l);
614 return result;
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000615}
616
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000617/* Conversion AST -> Python */
618
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000619static PyObject* ast2obj_list(asdl_seq *seq, PyObject* (*func)(void*))
620{
621 int i, n = asdl_seq_LEN(seq);
622 PyObject *result = PyList_New(n);
623 PyObject *value;
624 if (!result)
625 return NULL;
626 for (i = 0; i < n; i++) {
627 value = func(asdl_seq_GET(seq, i));
628 if (!value) {
629 Py_DECREF(result);
630 return NULL;
631 }
632 PyList_SET_ITEM(result, i, value);
633 }
634 return result;
635}
636
637static PyObject* ast2obj_object(void *o)
638{
639 if (!o)
640 o = Py_None;
641 Py_INCREF((PyObject*)o);
642 return (PyObject*)o;
643}
644#define ast2obj_identifier ast2obj_object
645#define ast2obj_string ast2obj_object
Benjamin Petersone2498412011-08-09 16:08:39 -0500646#define ast2obj_bytes ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000647
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000648static PyObject* ast2obj_int(long b)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000649{
Christian Heimes217cfd12007-12-02 14:31:20 +0000650 return PyLong_FromLong(b);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000651}
652
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000653/* Conversion Python -> AST */
654
655static int obj2ast_object(PyObject* obj, PyObject** out, PyArena* arena)
656{
657 if (obj == Py_None)
658 obj = NULL;
659 if (obj)
660 PyArena_AddPyObject(arena, obj);
661 Py_XINCREF(obj);
662 *out = obj;
663 return 0;
664}
665
Benjamin Peterson180e6352011-07-22 11:09:07 -0500666static int obj2ast_identifier(PyObject* obj, PyObject** out, PyArena* arena)
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500667{
Benjamin Peterson180e6352011-07-22 11:09:07 -0500668 if (!PyUnicode_CheckExact(obj) && obj != Py_None) {
669 PyErr_SetString(PyExc_TypeError, "AST identifier must be of type str");
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500670 return 1;
671 }
672 return obj2ast_object(obj, out, arena);
673}
674
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500675static int obj2ast_string(PyObject* obj, PyObject** out, PyArena* arena)
676{
Benjamin Peterson0224d4e2011-08-31 22:13:03 -0400677 if (!PyUnicode_CheckExact(obj) && !PyBytes_CheckExact(obj)) {
Benjamin Peterson180e6352011-07-22 11:09:07 -0500678 PyErr_SetString(PyExc_TypeError, "AST string must be of type str");
679 return 1;
680 }
681 return obj2ast_object(obj, out, arena);
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500682}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000683
Benjamin Petersone2498412011-08-09 16:08:39 -0500684static int obj2ast_bytes(PyObject* obj, PyObject** out, PyArena* arena)
685{
686 if (!PyBytes_CheckExact(obj)) {
687 PyErr_SetString(PyExc_TypeError, "AST bytes must be of type bytes");
688 return 1;
689 }
690 return obj2ast_object(obj, out, arena);
691}
692
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000693static int obj2ast_int(PyObject* obj, int* out, PyArena* arena)
694{
695 int i;
696 if (!PyLong_Check(obj)) {
Amaury Forgeot d'Arc5e8f8102011-11-22 21:52:30 +0100697 PyErr_Format(PyExc_ValueError, "invalid integer value: %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000698 return 1;
699 }
700
701 i = (int)PyLong_AsLong(obj);
702 if (i == -1 && PyErr_Occurred())
703 return 1;
704 *out = i;
705 return 0;
706}
707
Benjamin Peterson1a6e0d02008-10-25 15:49:17 +0000708static int add_ast_fields(void)
Benjamin Petersonce825f12008-10-24 23:11:02 +0000709{
710 PyObject *empty_tuple, *d;
711 if (PyType_Ready(&AST_type) < 0)
712 return -1;
713 d = AST_type.tp_dict;
714 empty_tuple = PyTuple_New(0);
715 if (!empty_tuple ||
716 PyDict_SetItemString(d, "_fields", empty_tuple) < 0 ||
717 PyDict_SetItemString(d, "_attributes", empty_tuple) < 0) {
718 Py_XDECREF(empty_tuple);
719 return -1;
720 }
721 Py_DECREF(empty_tuple);
722 return 0;
723}
724
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000725
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000726static int init_types(void)
727{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000728 static int initialized;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000729 if (initialized) return 1;
Benjamin Petersonce825f12008-10-24 23:11:02 +0000730 if (add_ast_fields() < 0) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000731 mod_type = make_type("mod", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000732 if (!mod_type) return 0;
733 if (!add_attributes(mod_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000734 Module_type = make_type("Module", mod_type, Module_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000735 if (!Module_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000736 Interactive_type = make_type("Interactive", mod_type,
737 Interactive_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000738 if (!Interactive_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000739 Expression_type = make_type("Expression", mod_type, Expression_fields,
740 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000741 if (!Expression_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000742 Suite_type = make_type("Suite", mod_type, Suite_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000743 if (!Suite_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000744 stmt_type = make_type("stmt", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000745 if (!stmt_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000746 if (!add_attributes(stmt_type, stmt_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000747 FunctionDef_type = make_type("FunctionDef", stmt_type,
Neal Norwitzc1505362006-12-28 06:47:50 +0000748 FunctionDef_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000749 if (!FunctionDef_type) return 0;
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000750 ClassDef_type = make_type("ClassDef", stmt_type, ClassDef_fields, 7);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000751 if (!ClassDef_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000752 Return_type = make_type("Return", stmt_type, Return_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000753 if (!Return_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000754 Delete_type = make_type("Delete", stmt_type, Delete_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000755 if (!Delete_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000756 Assign_type = make_type("Assign", stmt_type, Assign_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000757 if (!Assign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000758 AugAssign_type = make_type("AugAssign", stmt_type, AugAssign_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000759 if (!AugAssign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000760 For_type = make_type("For", stmt_type, For_fields, 4);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000761 if (!For_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000762 While_type = make_type("While", stmt_type, While_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000763 if (!While_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000764 If_type = make_type("If", stmt_type, If_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000765 if (!If_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500766 With_type = make_type("With", stmt_type, With_fields, 2);
Guido van Rossumc2e20742006-02-27 22:32:47 +0000767 if (!With_type) return 0;
Collin Winter828f04a2007-08-31 00:04:24 +0000768 Raise_type = make_type("Raise", stmt_type, Raise_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000769 if (!Raise_type) return 0;
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500770 Try_type = make_type("Try", stmt_type, Try_fields, 4);
771 if (!Try_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000772 Assert_type = make_type("Assert", stmt_type, Assert_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000773 if (!Assert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000774 Import_type = make_type("Import", stmt_type, Import_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000775 if (!Import_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000776 ImportFrom_type = make_type("ImportFrom", stmt_type, ImportFrom_fields,
Thomas Woutersf7f438b2006-02-28 16:09:29 +0000777 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000778 if (!ImportFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000779 Global_type = make_type("Global", stmt_type, Global_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000780 if (!Global_type) return 0;
Jeremy Hylton81e95022007-02-27 06:50:52 +0000781 Nonlocal_type = make_type("Nonlocal", stmt_type, Nonlocal_fields, 1);
782 if (!Nonlocal_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000783 Expr_type = make_type("Expr", stmt_type, Expr_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000784 if (!Expr_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000785 Pass_type = make_type("Pass", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000786 if (!Pass_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000787 Break_type = make_type("Break", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000788 if (!Break_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000789 Continue_type = make_type("Continue", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000790 if (!Continue_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000791 expr_type = make_type("expr", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000792 if (!expr_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000793 if (!add_attributes(expr_type, expr_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000794 BoolOp_type = make_type("BoolOp", expr_type, BoolOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000795 if (!BoolOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000796 BinOp_type = make_type("BinOp", expr_type, BinOp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000797 if (!BinOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000798 UnaryOp_type = make_type("UnaryOp", expr_type, UnaryOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000799 if (!UnaryOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000800 Lambda_type = make_type("Lambda", expr_type, Lambda_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000801 if (!Lambda_type) return 0;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +0000802 IfExp_type = make_type("IfExp", expr_type, IfExp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000803 if (!IfExp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000804 Dict_type = make_type("Dict", expr_type, Dict_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000805 if (!Dict_type) return 0;
Guido van Rossum86e58e22006-08-28 15:27:34 +0000806 Set_type = make_type("Set", expr_type, Set_fields, 1);
807 if (!Set_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000808 ListComp_type = make_type("ListComp", expr_type, ListComp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000809 if (!ListComp_type) return 0;
Nick Coghlan650f0d02007-04-15 12:05:43 +0000810 SetComp_type = make_type("SetComp", expr_type, SetComp_fields, 2);
811 if (!SetComp_type) return 0;
Guido van Rossum992d4a32007-07-11 13:09:30 +0000812 DictComp_type = make_type("DictComp", expr_type, DictComp_fields, 3);
813 if (!DictComp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000814 GeneratorExp_type = make_type("GeneratorExp", expr_type,
815 GeneratorExp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000816 if (!GeneratorExp_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500817 Yield_type = make_type("Yield", expr_type, Yield_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000818 if (!Yield_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500819 YieldFrom_type = make_type("YieldFrom", expr_type, YieldFrom_fields, 1);
820 if (!YieldFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000821 Compare_type = make_type("Compare", expr_type, Compare_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000822 if (!Compare_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000823 Call_type = make_type("Call", expr_type, Call_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000824 if (!Call_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000825 Num_type = make_type("Num", expr_type, Num_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000826 if (!Num_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000827 Str_type = make_type("Str", expr_type, Str_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000828 if (!Str_type) return 0;
Thomas Wouters00e41de2007-02-23 19:56:57 +0000829 Bytes_type = make_type("Bytes", expr_type, Bytes_fields, 1);
830 if (!Bytes_type) return 0;
Georg Brandl52318d62006-09-06 07:06:08 +0000831 Ellipsis_type = make_type("Ellipsis", expr_type, NULL, 0);
832 if (!Ellipsis_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000833 Attribute_type = make_type("Attribute", expr_type, Attribute_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000834 if (!Attribute_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000835 Subscript_type = make_type("Subscript", expr_type, Subscript_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000836 if (!Subscript_type) return 0;
Guido van Rossum0368b722007-05-11 16:50:42 +0000837 Starred_type = make_type("Starred", expr_type, Starred_fields, 2);
838 if (!Starred_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000839 Name_type = make_type("Name", expr_type, Name_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000840 if (!Name_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000841 List_type = make_type("List", expr_type, List_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000842 if (!List_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000843 Tuple_type = make_type("Tuple", expr_type, Tuple_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000844 if (!Tuple_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000845 expr_context_type = make_type("expr_context", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000846 if (!expr_context_type) return 0;
847 if (!add_attributes(expr_context_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000848 Load_type = make_type("Load", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000849 if (!Load_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000850 Load_singleton = PyType_GenericNew(Load_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000851 if (!Load_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000852 Store_type = make_type("Store", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000853 if (!Store_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000854 Store_singleton = PyType_GenericNew(Store_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000855 if (!Store_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000856 Del_type = make_type("Del", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000857 if (!Del_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000858 Del_singleton = PyType_GenericNew(Del_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000859 if (!Del_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000860 AugLoad_type = make_type("AugLoad", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000861 if (!AugLoad_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000862 AugLoad_singleton = PyType_GenericNew(AugLoad_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000863 if (!AugLoad_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000864 AugStore_type = make_type("AugStore", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000865 if (!AugStore_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000866 AugStore_singleton = PyType_GenericNew(AugStore_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000867 if (!AugStore_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000868 Param_type = make_type("Param", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000869 if (!Param_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000870 Param_singleton = PyType_GenericNew(Param_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000871 if (!Param_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000872 slice_type = make_type("slice", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000873 if (!slice_type) return 0;
874 if (!add_attributes(slice_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000875 Slice_type = make_type("Slice", slice_type, Slice_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000876 if (!Slice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000877 ExtSlice_type = make_type("ExtSlice", slice_type, ExtSlice_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000878 if (!ExtSlice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000879 Index_type = make_type("Index", slice_type, Index_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000880 if (!Index_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000881 boolop_type = make_type("boolop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000882 if (!boolop_type) return 0;
883 if (!add_attributes(boolop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000884 And_type = make_type("And", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000885 if (!And_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000886 And_singleton = PyType_GenericNew(And_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000887 if (!And_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000888 Or_type = make_type("Or", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000889 if (!Or_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000890 Or_singleton = PyType_GenericNew(Or_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000891 if (!Or_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000892 operator_type = make_type("operator", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000893 if (!operator_type) return 0;
894 if (!add_attributes(operator_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000895 Add_type = make_type("Add", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000896 if (!Add_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000897 Add_singleton = PyType_GenericNew(Add_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000898 if (!Add_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000899 Sub_type = make_type("Sub", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000900 if (!Sub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000901 Sub_singleton = PyType_GenericNew(Sub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000902 if (!Sub_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000903 Mult_type = make_type("Mult", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000904 if (!Mult_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000905 Mult_singleton = PyType_GenericNew(Mult_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000906 if (!Mult_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000907 Div_type = make_type("Div", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000908 if (!Div_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000909 Div_singleton = PyType_GenericNew(Div_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000910 if (!Div_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000911 Mod_type = make_type("Mod", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000912 if (!Mod_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000913 Mod_singleton = PyType_GenericNew(Mod_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000914 if (!Mod_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000915 Pow_type = make_type("Pow", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000916 if (!Pow_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000917 Pow_singleton = PyType_GenericNew(Pow_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000918 if (!Pow_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000919 LShift_type = make_type("LShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000920 if (!LShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000921 LShift_singleton = PyType_GenericNew(LShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000922 if (!LShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000923 RShift_type = make_type("RShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000924 if (!RShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000925 RShift_singleton = PyType_GenericNew(RShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000926 if (!RShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000927 BitOr_type = make_type("BitOr", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000928 if (!BitOr_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000929 BitOr_singleton = PyType_GenericNew(BitOr_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000930 if (!BitOr_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000931 BitXor_type = make_type("BitXor", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000932 if (!BitXor_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000933 BitXor_singleton = PyType_GenericNew(BitXor_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000934 if (!BitXor_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000935 BitAnd_type = make_type("BitAnd", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000936 if (!BitAnd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000937 BitAnd_singleton = PyType_GenericNew(BitAnd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000938 if (!BitAnd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000939 FloorDiv_type = make_type("FloorDiv", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000940 if (!FloorDiv_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000941 FloorDiv_singleton = PyType_GenericNew(FloorDiv_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000942 if (!FloorDiv_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000943 unaryop_type = make_type("unaryop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000944 if (!unaryop_type) return 0;
945 if (!add_attributes(unaryop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000946 Invert_type = make_type("Invert", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000947 if (!Invert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000948 Invert_singleton = PyType_GenericNew(Invert_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000949 if (!Invert_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000950 Not_type = make_type("Not", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000951 if (!Not_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000952 Not_singleton = PyType_GenericNew(Not_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000953 if (!Not_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000954 UAdd_type = make_type("UAdd", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000955 if (!UAdd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000956 UAdd_singleton = PyType_GenericNew(UAdd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000957 if (!UAdd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000958 USub_type = make_type("USub", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000959 if (!USub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000960 USub_singleton = PyType_GenericNew(USub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000961 if (!USub_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000962 cmpop_type = make_type("cmpop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000963 if (!cmpop_type) return 0;
964 if (!add_attributes(cmpop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000965 Eq_type = make_type("Eq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000966 if (!Eq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000967 Eq_singleton = PyType_GenericNew(Eq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000968 if (!Eq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000969 NotEq_type = make_type("NotEq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000970 if (!NotEq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000971 NotEq_singleton = PyType_GenericNew(NotEq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000972 if (!NotEq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000973 Lt_type = make_type("Lt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000974 if (!Lt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000975 Lt_singleton = PyType_GenericNew(Lt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000976 if (!Lt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000977 LtE_type = make_type("LtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000978 if (!LtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000979 LtE_singleton = PyType_GenericNew(LtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000980 if (!LtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000981 Gt_type = make_type("Gt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000982 if (!Gt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000983 Gt_singleton = PyType_GenericNew(Gt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000984 if (!Gt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000985 GtE_type = make_type("GtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000986 if (!GtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000987 GtE_singleton = PyType_GenericNew(GtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000988 if (!GtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000989 Is_type = make_type("Is", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000990 if (!Is_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000991 Is_singleton = PyType_GenericNew(Is_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000992 if (!Is_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000993 IsNot_type = make_type("IsNot", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000994 if (!IsNot_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000995 IsNot_singleton = PyType_GenericNew(IsNot_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000996 if (!IsNot_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000997 In_type = make_type("In", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000998 if (!In_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000999 In_singleton = PyType_GenericNew(In_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001000 if (!In_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001001 NotIn_type = make_type("NotIn", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001002 if (!NotIn_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001003 NotIn_singleton = PyType_GenericNew(NotIn_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001004 if (!NotIn_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001005 comprehension_type = make_type("comprehension", &AST_type,
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001006 comprehension_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001007 if (!comprehension_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001008 excepthandler_type = make_type("excepthandler", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001009 if (!excepthandler_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001010 if (!add_attributes(excepthandler_type, excepthandler_attributes, 2))
1011 return 0;
1012 ExceptHandler_type = make_type("ExceptHandler", excepthandler_type,
1013 ExceptHandler_fields, 3);
1014 if (!ExceptHandler_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001015 arguments_type = make_type("arguments", &AST_type, arguments_fields, 8);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001016 if (!arguments_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001017 arg_type = make_type("arg", &AST_type, arg_fields, 2);
Neal Norwitzc1505362006-12-28 06:47:50 +00001018 if (!arg_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001019 keyword_type = make_type("keyword", &AST_type, keyword_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001020 if (!keyword_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001021 alias_type = make_type("alias", &AST_type, alias_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001022 if (!alias_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001023 withitem_type = make_type("withitem", &AST_type, withitem_fields, 2);
1024 if (!withitem_type) return 0;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001025 initialized = 1;
1026 return 1;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001027}
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001028
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001029static int obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena);
1030static int obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena);
1031static int obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena);
1032static int obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena*
1033 arena);
1034static int obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena);
1035static int obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena);
1036static int obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena);
1037static int obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena);
1038static int obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena);
1039static int obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena*
1040 arena);
1041static int obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena*
1042 arena);
1043static int obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena);
1044static int obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena);
1045static int obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena);
1046static int obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001047static int obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001048
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001049mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001050Module(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001051{
1052 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001053 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001054 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001055 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001056 p->kind = Module_kind;
1057 p->v.Module.body = body;
1058 return p;
1059}
1060
1061mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001062Interactive(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001063{
1064 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001065 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001066 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001067 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001068 p->kind = Interactive_kind;
1069 p->v.Interactive.body = body;
1070 return p;
1071}
1072
1073mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001074Expression(expr_ty body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001075{
1076 mod_ty p;
1077 if (!body) {
1078 PyErr_SetString(PyExc_ValueError,
1079 "field body is required for Expression");
1080 return NULL;
1081 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001082 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001083 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001084 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001085 p->kind = Expression_kind;
1086 p->v.Expression.body = body;
1087 return p;
1088}
1089
1090mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001091Suite(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001092{
1093 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001094 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001095 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001096 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001097 p->kind = Suite_kind;
1098 p->v.Suite.body = body;
1099 return p;
1100}
1101
1102stmt_ty
1103FunctionDef(identifier name, arguments_ty args, asdl_seq * body, asdl_seq *
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001104 decorator_list, expr_ty returns, int lineno, int col_offset,
1105 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001106{
1107 stmt_ty p;
1108 if (!name) {
1109 PyErr_SetString(PyExc_ValueError,
1110 "field name is required for FunctionDef");
1111 return NULL;
1112 }
1113 if (!args) {
1114 PyErr_SetString(PyExc_ValueError,
1115 "field args is required for FunctionDef");
1116 return NULL;
1117 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001118 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001119 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001120 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001121 p->kind = FunctionDef_kind;
1122 p->v.FunctionDef.name = name;
1123 p->v.FunctionDef.args = args;
1124 p->v.FunctionDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001125 p->v.FunctionDef.decorator_list = decorator_list;
Neal Norwitzc1505362006-12-28 06:47:50 +00001126 p->v.FunctionDef.returns = returns;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001127 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001128 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001129 return p;
1130}
1131
1132stmt_ty
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001133ClassDef(identifier name, asdl_seq * bases, asdl_seq * keywords, expr_ty
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001134 starargs, expr_ty kwargs, asdl_seq * body, asdl_seq * decorator_list,
1135 int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001136{
1137 stmt_ty p;
1138 if (!name) {
1139 PyErr_SetString(PyExc_ValueError,
1140 "field name is required for ClassDef");
1141 return NULL;
1142 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001143 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001144 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001145 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001146 p->kind = ClassDef_kind;
1147 p->v.ClassDef.name = name;
1148 p->v.ClassDef.bases = bases;
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001149 p->v.ClassDef.keywords = keywords;
1150 p->v.ClassDef.starargs = starargs;
1151 p->v.ClassDef.kwargs = kwargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001152 p->v.ClassDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001153 p->v.ClassDef.decorator_list = decorator_list;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001154 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001155 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001156 return p;
1157}
1158
1159stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001160Return(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001161{
1162 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001163 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001164 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001165 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001166 p->kind = Return_kind;
1167 p->v.Return.value = value;
1168 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001169 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001170 return p;
1171}
1172
1173stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001174Delete(asdl_seq * targets, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001175{
1176 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001177 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001178 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001179 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001180 p->kind = Delete_kind;
1181 p->v.Delete.targets = targets;
1182 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001183 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001184 return p;
1185}
1186
1187stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001188Assign(asdl_seq * targets, expr_ty value, int lineno, int col_offset, PyArena
1189 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001190{
1191 stmt_ty p;
1192 if (!value) {
1193 PyErr_SetString(PyExc_ValueError,
1194 "field value is required for Assign");
1195 return NULL;
1196 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001197 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001198 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001199 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001200 p->kind = Assign_kind;
1201 p->v.Assign.targets = targets;
1202 p->v.Assign.value = value;
1203 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001204 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001205 return p;
1206}
1207
1208stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001209AugAssign(expr_ty target, operator_ty op, expr_ty value, int lineno, int
1210 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001211{
1212 stmt_ty p;
1213 if (!target) {
1214 PyErr_SetString(PyExc_ValueError,
1215 "field target is required for AugAssign");
1216 return NULL;
1217 }
1218 if (!op) {
1219 PyErr_SetString(PyExc_ValueError,
1220 "field op is required for AugAssign");
1221 return NULL;
1222 }
1223 if (!value) {
1224 PyErr_SetString(PyExc_ValueError,
1225 "field value is required for AugAssign");
1226 return NULL;
1227 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001228 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001229 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001230 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001231 p->kind = AugAssign_kind;
1232 p->v.AugAssign.target = target;
1233 p->v.AugAssign.op = op;
1234 p->v.AugAssign.value = value;
1235 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001236 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001237 return p;
1238}
1239
1240stmt_ty
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001241For(expr_ty target, expr_ty iter, asdl_seq * body, asdl_seq * orelse, int
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001242 lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001243{
1244 stmt_ty p;
1245 if (!target) {
1246 PyErr_SetString(PyExc_ValueError,
1247 "field target is required for For");
1248 return NULL;
1249 }
1250 if (!iter) {
1251 PyErr_SetString(PyExc_ValueError,
1252 "field iter is required for For");
1253 return NULL;
1254 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001255 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001256 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001257 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001258 p->kind = For_kind;
1259 p->v.For.target = target;
1260 p->v.For.iter = iter;
1261 p->v.For.body = body;
1262 p->v.For.orelse = orelse;
1263 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001264 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001265 return p;
1266}
1267
1268stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001269While(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1270 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001271{
1272 stmt_ty p;
1273 if (!test) {
1274 PyErr_SetString(PyExc_ValueError,
1275 "field test is required for While");
1276 return NULL;
1277 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001278 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001279 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001280 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001281 p->kind = While_kind;
1282 p->v.While.test = test;
1283 p->v.While.body = body;
1284 p->v.While.orelse = orelse;
1285 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001286 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001287 return p;
1288}
1289
1290stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001291If(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1292 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001293{
1294 stmt_ty p;
1295 if (!test) {
1296 PyErr_SetString(PyExc_ValueError,
1297 "field test is required for If");
1298 return NULL;
1299 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001300 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001301 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001302 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001303 p->kind = If_kind;
1304 p->v.If.test = test;
1305 p->v.If.body = body;
1306 p->v.If.orelse = orelse;
1307 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001308 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001309 return p;
1310}
1311
1312stmt_ty
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001313With(asdl_seq * items, asdl_seq * body, int lineno, int col_offset, PyArena
1314 *arena)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001315{
1316 stmt_ty p;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001317 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001318 if (!p)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001319 return NULL;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001320 p->kind = With_kind;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001321 p->v.With.items = items;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001322 p->v.With.body = body;
1323 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001324 p->col_offset = col_offset;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001325 return p;
1326}
1327
1328stmt_ty
Collin Winter828f04a2007-08-31 00:04:24 +00001329Raise(expr_ty exc, expr_ty cause, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001330{
1331 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001332 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001333 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001334 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001335 p->kind = Raise_kind;
Collin Winter828f04a2007-08-31 00:04:24 +00001336 p->v.Raise.exc = exc;
1337 p->v.Raise.cause = cause;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001338 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001339 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001340 return p;
1341}
1342
1343stmt_ty
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001344Try(asdl_seq * body, asdl_seq * handlers, asdl_seq * orelse, asdl_seq *
1345 finalbody, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001346{
1347 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001348 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001349 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001350 return NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001351 p->kind = Try_kind;
1352 p->v.Try.body = body;
1353 p->v.Try.handlers = handlers;
1354 p->v.Try.orelse = orelse;
1355 p->v.Try.finalbody = finalbody;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001356 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001357 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001358 return p;
1359}
1360
1361stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001362Assert(expr_ty test, expr_ty msg, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001363{
1364 stmt_ty p;
1365 if (!test) {
1366 PyErr_SetString(PyExc_ValueError,
1367 "field test is required for Assert");
1368 return NULL;
1369 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001370 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001371 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001372 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001373 p->kind = Assert_kind;
1374 p->v.Assert.test = test;
1375 p->v.Assert.msg = msg;
1376 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001377 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001378 return p;
1379}
1380
1381stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001382Import(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001383{
1384 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001385 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001386 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001387 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001388 p->kind = Import_kind;
1389 p->v.Import.names = names;
1390 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001391 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001392 return p;
1393}
1394
1395stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001396ImportFrom(identifier module, asdl_seq * names, int level, int lineno, int
1397 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001398{
1399 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001400 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001401 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001402 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001403 p->kind = ImportFrom_kind;
1404 p->v.ImportFrom.module = module;
1405 p->v.ImportFrom.names = names;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001406 p->v.ImportFrom.level = level;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001407 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001408 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001409 return p;
1410}
1411
1412stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001413Global(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001414{
1415 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001416 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001417 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001418 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001419 p->kind = Global_kind;
1420 p->v.Global.names = names;
1421 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001422 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001423 return p;
1424}
1425
1426stmt_ty
Jeremy Hylton81e95022007-02-27 06:50:52 +00001427Nonlocal(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
1428{
1429 stmt_ty p;
1430 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
1431 if (!p)
1432 return NULL;
1433 p->kind = Nonlocal_kind;
1434 p->v.Nonlocal.names = names;
1435 p->lineno = lineno;
1436 p->col_offset = col_offset;
1437 return p;
1438}
1439
1440stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001441Expr(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001442{
1443 stmt_ty p;
1444 if (!value) {
1445 PyErr_SetString(PyExc_ValueError,
1446 "field value is required for Expr");
1447 return NULL;
1448 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001449 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001450 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001451 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001452 p->kind = Expr_kind;
1453 p->v.Expr.value = value;
1454 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001455 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001456 return p;
1457}
1458
1459stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001460Pass(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001461{
1462 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001463 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001464 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001465 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001466 p->kind = Pass_kind;
1467 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001468 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001469 return p;
1470}
1471
1472stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001473Break(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001474{
1475 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001476 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001477 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001478 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001479 p->kind = Break_kind;
1480 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001481 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001482 return p;
1483}
1484
1485stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001486Continue(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001487{
1488 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001489 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001490 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001491 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001492 p->kind = Continue_kind;
1493 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001494 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001495 return p;
1496}
1497
1498expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001499BoolOp(boolop_ty op, asdl_seq * values, int lineno, int col_offset, PyArena
1500 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001501{
1502 expr_ty p;
1503 if (!op) {
1504 PyErr_SetString(PyExc_ValueError,
1505 "field op is required for BoolOp");
1506 return NULL;
1507 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001508 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001509 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001510 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001511 p->kind = BoolOp_kind;
1512 p->v.BoolOp.op = op;
1513 p->v.BoolOp.values = values;
1514 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001515 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001516 return p;
1517}
1518
1519expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001520BinOp(expr_ty left, operator_ty op, expr_ty right, int lineno, int col_offset,
1521 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001522{
1523 expr_ty p;
1524 if (!left) {
1525 PyErr_SetString(PyExc_ValueError,
1526 "field left is required for BinOp");
1527 return NULL;
1528 }
1529 if (!op) {
1530 PyErr_SetString(PyExc_ValueError,
1531 "field op is required for BinOp");
1532 return NULL;
1533 }
1534 if (!right) {
1535 PyErr_SetString(PyExc_ValueError,
1536 "field right is required for BinOp");
1537 return NULL;
1538 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001539 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001540 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001541 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001542 p->kind = BinOp_kind;
1543 p->v.BinOp.left = left;
1544 p->v.BinOp.op = op;
1545 p->v.BinOp.right = right;
1546 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001547 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001548 return p;
1549}
1550
1551expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001552UnaryOp(unaryop_ty op, expr_ty operand, int lineno, int col_offset, PyArena
1553 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001554{
1555 expr_ty p;
1556 if (!op) {
1557 PyErr_SetString(PyExc_ValueError,
1558 "field op is required for UnaryOp");
1559 return NULL;
1560 }
1561 if (!operand) {
1562 PyErr_SetString(PyExc_ValueError,
1563 "field operand is required for UnaryOp");
1564 return NULL;
1565 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001566 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001567 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001568 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001569 p->kind = UnaryOp_kind;
1570 p->v.UnaryOp.op = op;
1571 p->v.UnaryOp.operand = operand;
1572 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001573 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001574 return p;
1575}
1576
1577expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001578Lambda(arguments_ty args, expr_ty body, int lineno, int col_offset, PyArena
1579 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001580{
1581 expr_ty p;
1582 if (!args) {
1583 PyErr_SetString(PyExc_ValueError,
1584 "field args is required for Lambda");
1585 return NULL;
1586 }
1587 if (!body) {
1588 PyErr_SetString(PyExc_ValueError,
1589 "field body is required for Lambda");
1590 return NULL;
1591 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001592 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001593 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001594 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001595 p->kind = Lambda_kind;
1596 p->v.Lambda.args = args;
1597 p->v.Lambda.body = body;
1598 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001599 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001600 return p;
1601}
1602
1603expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001604IfExp(expr_ty test, expr_ty body, expr_ty orelse, int lineno, int col_offset,
1605 PyArena *arena)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001606{
1607 expr_ty p;
1608 if (!test) {
1609 PyErr_SetString(PyExc_ValueError,
1610 "field test is required for IfExp");
1611 return NULL;
1612 }
1613 if (!body) {
1614 PyErr_SetString(PyExc_ValueError,
1615 "field body is required for IfExp");
1616 return NULL;
1617 }
1618 if (!orelse) {
1619 PyErr_SetString(PyExc_ValueError,
1620 "field orelse is required for IfExp");
1621 return NULL;
1622 }
1623 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001624 if (!p)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001625 return NULL;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001626 p->kind = IfExp_kind;
1627 p->v.IfExp.test = test;
1628 p->v.IfExp.body = body;
1629 p->v.IfExp.orelse = orelse;
1630 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001631 p->col_offset = col_offset;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001632 return p;
1633}
1634
1635expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001636Dict(asdl_seq * keys, asdl_seq * values, int lineno, int col_offset, PyArena
1637 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001638{
1639 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001640 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001641 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001642 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001643 p->kind = Dict_kind;
1644 p->v.Dict.keys = keys;
1645 p->v.Dict.values = values;
1646 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001647 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001648 return p;
1649}
1650
1651expr_ty
Guido van Rossum86e58e22006-08-28 15:27:34 +00001652Set(asdl_seq * elts, int lineno, int col_offset, PyArena *arena)
1653{
1654 expr_ty p;
1655 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001656 if (!p)
Guido van Rossum86e58e22006-08-28 15:27:34 +00001657 return NULL;
Guido van Rossum86e58e22006-08-28 15:27:34 +00001658 p->kind = Set_kind;
1659 p->v.Set.elts = elts;
1660 p->lineno = lineno;
1661 p->col_offset = col_offset;
1662 return p;
1663}
1664
1665expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001666ListComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1667 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001668{
1669 expr_ty p;
1670 if (!elt) {
1671 PyErr_SetString(PyExc_ValueError,
1672 "field elt is required for ListComp");
1673 return NULL;
1674 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001675 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001676 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001677 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001678 p->kind = ListComp_kind;
1679 p->v.ListComp.elt = elt;
1680 p->v.ListComp.generators = generators;
1681 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001682 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001683 return p;
1684}
1685
1686expr_ty
Nick Coghlan650f0d02007-04-15 12:05:43 +00001687SetComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset, PyArena
1688 *arena)
1689{
1690 expr_ty p;
1691 if (!elt) {
1692 PyErr_SetString(PyExc_ValueError,
1693 "field elt is required for SetComp");
1694 return NULL;
1695 }
1696 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1697 if (!p)
1698 return NULL;
1699 p->kind = SetComp_kind;
1700 p->v.SetComp.elt = elt;
1701 p->v.SetComp.generators = generators;
1702 p->lineno = lineno;
1703 p->col_offset = col_offset;
1704 return p;
1705}
1706
1707expr_ty
Guido van Rossum992d4a32007-07-11 13:09:30 +00001708DictComp(expr_ty key, expr_ty value, asdl_seq * generators, int lineno, int
1709 col_offset, PyArena *arena)
1710{
1711 expr_ty p;
1712 if (!key) {
1713 PyErr_SetString(PyExc_ValueError,
1714 "field key is required for DictComp");
1715 return NULL;
1716 }
1717 if (!value) {
1718 PyErr_SetString(PyExc_ValueError,
1719 "field value is required for DictComp");
1720 return NULL;
1721 }
1722 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1723 if (!p)
1724 return NULL;
1725 p->kind = DictComp_kind;
1726 p->v.DictComp.key = key;
1727 p->v.DictComp.value = value;
1728 p->v.DictComp.generators = generators;
1729 p->lineno = lineno;
1730 p->col_offset = col_offset;
1731 return p;
1732}
1733
1734expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001735GeneratorExp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1736 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001737{
1738 expr_ty p;
1739 if (!elt) {
1740 PyErr_SetString(PyExc_ValueError,
1741 "field elt is required for GeneratorExp");
1742 return NULL;
1743 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001744 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001745 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001746 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001747 p->kind = GeneratorExp_kind;
1748 p->v.GeneratorExp.elt = elt;
1749 p->v.GeneratorExp.generators = generators;
1750 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001751 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001752 return p;
1753}
1754
1755expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001756Yield(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001757{
1758 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001759 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001760 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001761 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001762 p->kind = Yield_kind;
1763 p->v.Yield.value = value;
1764 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001765 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001766 return p;
1767}
1768
1769expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001770YieldFrom(expr_ty value, int lineno, int col_offset, PyArena *arena)
1771{
1772 expr_ty p;
1773 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1774 if (!p)
1775 return NULL;
1776 p->kind = YieldFrom_kind;
1777 p->v.YieldFrom.value = value;
1778 p->lineno = lineno;
1779 p->col_offset = col_offset;
1780 return p;
1781}
1782
1783expr_ty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001784Compare(expr_ty left, asdl_int_seq * ops, asdl_seq * comparators, int lineno,
1785 int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001786{
1787 expr_ty p;
1788 if (!left) {
1789 PyErr_SetString(PyExc_ValueError,
1790 "field left is required for Compare");
1791 return NULL;
1792 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001793 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001794 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001795 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001796 p->kind = Compare_kind;
1797 p->v.Compare.left = left;
1798 p->v.Compare.ops = ops;
1799 p->v.Compare.comparators = comparators;
1800 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001801 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001802 return p;
1803}
1804
1805expr_ty
1806Call(expr_ty func, asdl_seq * args, asdl_seq * keywords, expr_ty starargs,
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001807 expr_ty kwargs, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001808{
1809 expr_ty p;
1810 if (!func) {
1811 PyErr_SetString(PyExc_ValueError,
1812 "field func is required for Call");
1813 return NULL;
1814 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001815 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001816 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001817 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001818 p->kind = Call_kind;
1819 p->v.Call.func = func;
1820 p->v.Call.args = args;
1821 p->v.Call.keywords = keywords;
1822 p->v.Call.starargs = starargs;
1823 p->v.Call.kwargs = kwargs;
1824 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001825 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001826 return p;
1827}
1828
1829expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001830Num(object n, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001831{
1832 expr_ty p;
1833 if (!n) {
1834 PyErr_SetString(PyExc_ValueError,
1835 "field n is required for Num");
1836 return NULL;
1837 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001838 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001839 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001840 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001841 p->kind = Num_kind;
1842 p->v.Num.n = n;
1843 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001844 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001845 return p;
1846}
1847
1848expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001849Str(string s, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001850{
1851 expr_ty p;
1852 if (!s) {
1853 PyErr_SetString(PyExc_ValueError,
1854 "field s is required for Str");
1855 return NULL;
1856 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001857 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001858 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001859 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001860 p->kind = Str_kind;
1861 p->v.Str.s = s;
1862 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001863 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001864 return p;
1865}
1866
1867expr_ty
Benjamin Petersone2498412011-08-09 16:08:39 -05001868Bytes(bytes s, int lineno, int col_offset, PyArena *arena)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001869{
1870 expr_ty p;
1871 if (!s) {
1872 PyErr_SetString(PyExc_ValueError,
1873 "field s is required for Bytes");
1874 return NULL;
1875 }
1876 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001877 if (!p)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001878 return NULL;
Thomas Wouters00e41de2007-02-23 19:56:57 +00001879 p->kind = Bytes_kind;
1880 p->v.Bytes.s = s;
1881 p->lineno = lineno;
1882 p->col_offset = col_offset;
1883 return p;
1884}
1885
1886expr_ty
Georg Brandl52318d62006-09-06 07:06:08 +00001887Ellipsis(int lineno, int col_offset, PyArena *arena)
1888{
1889 expr_ty p;
1890 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001891 if (!p)
Georg Brandl52318d62006-09-06 07:06:08 +00001892 return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00001893 p->kind = Ellipsis_kind;
1894 p->lineno = lineno;
1895 p->col_offset = col_offset;
1896 return p;
1897}
1898
1899expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001900Attribute(expr_ty value, identifier attr, expr_context_ty ctx, int lineno, int
1901 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001902{
1903 expr_ty p;
1904 if (!value) {
1905 PyErr_SetString(PyExc_ValueError,
1906 "field value is required for Attribute");
1907 return NULL;
1908 }
1909 if (!attr) {
1910 PyErr_SetString(PyExc_ValueError,
1911 "field attr is required for Attribute");
1912 return NULL;
1913 }
1914 if (!ctx) {
1915 PyErr_SetString(PyExc_ValueError,
1916 "field ctx is required for Attribute");
1917 return NULL;
1918 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001919 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001920 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001921 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001922 p->kind = Attribute_kind;
1923 p->v.Attribute.value = value;
1924 p->v.Attribute.attr = attr;
1925 p->v.Attribute.ctx = ctx;
1926 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001927 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001928 return p;
1929}
1930
1931expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001932Subscript(expr_ty value, slice_ty slice, expr_context_ty ctx, int lineno, int
1933 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001934{
1935 expr_ty p;
1936 if (!value) {
1937 PyErr_SetString(PyExc_ValueError,
1938 "field value is required for Subscript");
1939 return NULL;
1940 }
1941 if (!slice) {
1942 PyErr_SetString(PyExc_ValueError,
1943 "field slice is required for Subscript");
1944 return NULL;
1945 }
1946 if (!ctx) {
1947 PyErr_SetString(PyExc_ValueError,
1948 "field ctx is required for Subscript");
1949 return NULL;
1950 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001951 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001952 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001953 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001954 p->kind = Subscript_kind;
1955 p->v.Subscript.value = value;
1956 p->v.Subscript.slice = slice;
1957 p->v.Subscript.ctx = ctx;
1958 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001959 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001960 return p;
1961}
1962
1963expr_ty
Guido van Rossum0368b722007-05-11 16:50:42 +00001964Starred(expr_ty value, expr_context_ty ctx, int lineno, int col_offset, PyArena
1965 *arena)
1966{
1967 expr_ty p;
1968 if (!value) {
1969 PyErr_SetString(PyExc_ValueError,
1970 "field value is required for Starred");
1971 return NULL;
1972 }
1973 if (!ctx) {
1974 PyErr_SetString(PyExc_ValueError,
1975 "field ctx is required for Starred");
1976 return NULL;
1977 }
1978 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1979 if (!p)
1980 return NULL;
1981 p->kind = Starred_kind;
1982 p->v.Starred.value = value;
1983 p->v.Starred.ctx = ctx;
1984 p->lineno = lineno;
1985 p->col_offset = col_offset;
1986 return p;
1987}
1988
1989expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001990Name(identifier id, expr_context_ty ctx, int lineno, int col_offset, PyArena
1991 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001992{
1993 expr_ty p;
1994 if (!id) {
1995 PyErr_SetString(PyExc_ValueError,
1996 "field id is required for Name");
1997 return NULL;
1998 }
1999 if (!ctx) {
2000 PyErr_SetString(PyExc_ValueError,
2001 "field ctx is required for Name");
2002 return NULL;
2003 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002004 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002005 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002006 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002007 p->kind = Name_kind;
2008 p->v.Name.id = id;
2009 p->v.Name.ctx = ctx;
2010 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002011 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002012 return p;
2013}
2014
2015expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002016List(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2017 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002018{
2019 expr_ty p;
2020 if (!ctx) {
2021 PyErr_SetString(PyExc_ValueError,
2022 "field ctx is required for List");
2023 return NULL;
2024 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002025 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002026 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002027 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002028 p->kind = List_kind;
2029 p->v.List.elts = elts;
2030 p->v.List.ctx = ctx;
2031 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002032 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002033 return p;
2034}
2035
2036expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002037Tuple(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2038 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002039{
2040 expr_ty p;
2041 if (!ctx) {
2042 PyErr_SetString(PyExc_ValueError,
2043 "field ctx is required for Tuple");
2044 return NULL;
2045 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002046 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002047 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002048 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002049 p->kind = Tuple_kind;
2050 p->v.Tuple.elts = elts;
2051 p->v.Tuple.ctx = ctx;
2052 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002053 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002054 return p;
2055}
2056
2057slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002058Slice(expr_ty lower, expr_ty upper, expr_ty step, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002059{
2060 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002061 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002062 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002063 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002064 p->kind = Slice_kind;
2065 p->v.Slice.lower = lower;
2066 p->v.Slice.upper = upper;
2067 p->v.Slice.step = step;
2068 return p;
2069}
2070
2071slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002072ExtSlice(asdl_seq * dims, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002073{
2074 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002075 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002076 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002077 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002078 p->kind = ExtSlice_kind;
2079 p->v.ExtSlice.dims = dims;
2080 return p;
2081}
2082
2083slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002084Index(expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002085{
2086 slice_ty p;
2087 if (!value) {
2088 PyErr_SetString(PyExc_ValueError,
2089 "field value is required for Index");
2090 return NULL;
2091 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002092 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002093 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002094 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002095 p->kind = Index_kind;
2096 p->v.Index.value = value;
2097 return p;
2098}
2099
2100comprehension_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002101comprehension(expr_ty target, expr_ty iter, asdl_seq * ifs, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002102{
2103 comprehension_ty p;
2104 if (!target) {
2105 PyErr_SetString(PyExc_ValueError,
2106 "field target is required for comprehension");
2107 return NULL;
2108 }
2109 if (!iter) {
2110 PyErr_SetString(PyExc_ValueError,
2111 "field iter is required for comprehension");
2112 return NULL;
2113 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002114 p = (comprehension_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002115 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002116 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002117 p->target = target;
2118 p->iter = iter;
2119 p->ifs = ifs;
2120 return p;
2121}
2122
2123excepthandler_ty
Neal Norwitzad74aa82008-03-31 05:14:30 +00002124ExceptHandler(expr_ty type, identifier name, asdl_seq * body, int lineno, int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002125 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002126{
2127 excepthandler_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002128 p = (excepthandler_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002129 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002130 return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00002131 p->kind = ExceptHandler_kind;
2132 p->v.ExceptHandler.type = type;
2133 p->v.ExceptHandler.name = name;
2134 p->v.ExceptHandler.body = body;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002135 p->lineno = lineno;
2136 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002137 return p;
2138}
2139
2140arguments_ty
Neal Norwitzc1505362006-12-28 06:47:50 +00002141arguments(asdl_seq * args, identifier vararg, expr_ty varargannotation,
2142 asdl_seq * kwonlyargs, identifier kwarg, expr_ty kwargannotation,
2143 asdl_seq * defaults, asdl_seq * kw_defaults, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002144{
2145 arguments_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002146 p = (arguments_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002147 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002148 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002149 p->args = args;
2150 p->vararg = vararg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002151 p->varargannotation = varargannotation;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002152 p->kwonlyargs = kwonlyargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002153 p->kwarg = kwarg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002154 p->kwargannotation = kwargannotation;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002155 p->defaults = defaults;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002156 p->kw_defaults = kw_defaults;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002157 return p;
2158}
2159
Neal Norwitzc1505362006-12-28 06:47:50 +00002160arg_ty
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002161arg(identifier arg, expr_ty annotation, PyArena *arena)
Neal Norwitzc1505362006-12-28 06:47:50 +00002162{
2163 arg_ty p;
2164 if (!arg) {
2165 PyErr_SetString(PyExc_ValueError,
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002166 "field arg is required for arg");
Neal Norwitzc1505362006-12-28 06:47:50 +00002167 return NULL;
2168 }
2169 p = (arg_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002170 if (!p)
Neal Norwitzc1505362006-12-28 06:47:50 +00002171 return NULL;
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002172 p->arg = arg;
2173 p->annotation = annotation;
Neal Norwitzc1505362006-12-28 06:47:50 +00002174 return p;
2175}
2176
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002177keyword_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002178keyword(identifier arg, expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002179{
2180 keyword_ty p;
2181 if (!arg) {
2182 PyErr_SetString(PyExc_ValueError,
2183 "field arg is required for keyword");
2184 return NULL;
2185 }
2186 if (!value) {
2187 PyErr_SetString(PyExc_ValueError,
2188 "field value is required for keyword");
2189 return NULL;
2190 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002191 p = (keyword_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002192 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002193 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002194 p->arg = arg;
2195 p->value = value;
2196 return p;
2197}
2198
2199alias_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002200alias(identifier name, identifier asname, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002201{
2202 alias_ty p;
2203 if (!name) {
2204 PyErr_SetString(PyExc_ValueError,
2205 "field name is required for alias");
2206 return NULL;
2207 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002208 p = (alias_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002209 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002210 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002211 p->name = name;
2212 p->asname = asname;
2213 return p;
2214}
2215
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002216withitem_ty
2217withitem(expr_ty context_expr, expr_ty optional_vars, PyArena *arena)
2218{
2219 withitem_ty p;
2220 if (!context_expr) {
2221 PyErr_SetString(PyExc_ValueError,
2222 "field context_expr is required for withitem");
2223 return NULL;
2224 }
2225 p = (withitem_ty)PyArena_Malloc(arena, sizeof(*p));
2226 if (!p)
2227 return NULL;
2228 p->context_expr = context_expr;
2229 p->optional_vars = optional_vars;
2230 return p;
2231}
2232
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002233
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002234PyObject*
2235ast2obj_mod(void* _o)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002236{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002237 mod_ty o = (mod_ty)_o;
2238 PyObject *result = NULL, *value = NULL;
2239 if (!o) {
2240 Py_INCREF(Py_None);
2241 return Py_None;
2242 }
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002243
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002244 switch (o->kind) {
2245 case Module_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002246 result = PyType_GenericNew(Module_type, NULL, NULL);
2247 if (!result) goto failed;
2248 value = ast2obj_list(o->v.Module.body, ast2obj_stmt);
2249 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002250 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002251 goto failed;
2252 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002253 break;
2254 case Interactive_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002255 result = PyType_GenericNew(Interactive_type, NULL, NULL);
2256 if (!result) goto failed;
2257 value = ast2obj_list(o->v.Interactive.body, ast2obj_stmt);
2258 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002259 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002260 goto failed;
2261 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002262 break;
2263 case Expression_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002264 result = PyType_GenericNew(Expression_type, NULL, NULL);
2265 if (!result) goto failed;
2266 value = ast2obj_expr(o->v.Expression.body);
2267 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002268 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002269 goto failed;
2270 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002271 break;
2272 case Suite_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002273 result = PyType_GenericNew(Suite_type, NULL, NULL);
2274 if (!result) goto failed;
2275 value = ast2obj_list(o->v.Suite.body, ast2obj_stmt);
2276 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002277 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002278 goto failed;
2279 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002280 break;
2281 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002282 return result;
2283failed:
2284 Py_XDECREF(value);
2285 Py_XDECREF(result);
2286 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002287}
2288
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002289PyObject*
2290ast2obj_stmt(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002291{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002292 stmt_ty o = (stmt_ty)_o;
2293 PyObject *result = NULL, *value = NULL;
2294 if (!o) {
2295 Py_INCREF(Py_None);
2296 return Py_None;
2297 }
2298
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002299 switch (o->kind) {
2300 case FunctionDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002301 result = PyType_GenericNew(FunctionDef_type, NULL, NULL);
2302 if (!result) goto failed;
2303 value = ast2obj_identifier(o->v.FunctionDef.name);
2304 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002305 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002306 goto failed;
2307 Py_DECREF(value);
2308 value = ast2obj_arguments(o->v.FunctionDef.args);
2309 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002310 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002311 goto failed;
2312 Py_DECREF(value);
2313 value = ast2obj_list(o->v.FunctionDef.body, ast2obj_stmt);
2314 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002315 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002316 goto failed;
2317 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002318 value = ast2obj_list(o->v.FunctionDef.decorator_list,
2319 ast2obj_expr);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002320 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002321 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002322 -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002323 goto failed;
2324 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00002325 value = ast2obj_expr(o->v.FunctionDef.returns);
2326 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002327 if (_PyObject_SetAttrId(result, &PyId_returns, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00002328 goto failed;
2329 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002330 break;
2331 case ClassDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002332 result = PyType_GenericNew(ClassDef_type, NULL, NULL);
2333 if (!result) goto failed;
2334 value = ast2obj_identifier(o->v.ClassDef.name);
2335 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002336 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002337 goto failed;
2338 Py_DECREF(value);
2339 value = ast2obj_list(o->v.ClassDef.bases, ast2obj_expr);
2340 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002341 if (_PyObject_SetAttrId(result, &PyId_bases, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002342 goto failed;
2343 Py_DECREF(value);
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002344 value = ast2obj_list(o->v.ClassDef.keywords, ast2obj_keyword);
2345 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002346 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002347 goto failed;
2348 Py_DECREF(value);
2349 value = ast2obj_expr(o->v.ClassDef.starargs);
2350 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002351 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002352 goto failed;
2353 Py_DECREF(value);
2354 value = ast2obj_expr(o->v.ClassDef.kwargs);
2355 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002356 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002357 goto failed;
2358 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002359 value = ast2obj_list(o->v.ClassDef.body, ast2obj_stmt);
2360 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002361 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002362 goto failed;
2363 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002364 value = ast2obj_list(o->v.ClassDef.decorator_list,
2365 ast2obj_expr);
2366 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002367 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002368 -1)
2369 goto failed;
2370 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002371 break;
2372 case Return_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002373 result = PyType_GenericNew(Return_type, NULL, NULL);
2374 if (!result) goto failed;
2375 value = ast2obj_expr(o->v.Return.value);
2376 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002377 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002378 goto failed;
2379 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002380 break;
2381 case Delete_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002382 result = PyType_GenericNew(Delete_type, NULL, NULL);
2383 if (!result) goto failed;
2384 value = ast2obj_list(o->v.Delete.targets, ast2obj_expr);
2385 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002386 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002387 goto failed;
2388 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002389 break;
2390 case Assign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002391 result = PyType_GenericNew(Assign_type, NULL, NULL);
2392 if (!result) goto failed;
2393 value = ast2obj_list(o->v.Assign.targets, ast2obj_expr);
2394 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002395 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002396 goto failed;
2397 Py_DECREF(value);
2398 value = ast2obj_expr(o->v.Assign.value);
2399 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002400 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002401 goto failed;
2402 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002403 break;
2404 case AugAssign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002405 result = PyType_GenericNew(AugAssign_type, NULL, NULL);
2406 if (!result) goto failed;
2407 value = ast2obj_expr(o->v.AugAssign.target);
2408 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002409 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002410 goto failed;
2411 Py_DECREF(value);
2412 value = ast2obj_operator(o->v.AugAssign.op);
2413 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002414 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002415 goto failed;
2416 Py_DECREF(value);
2417 value = ast2obj_expr(o->v.AugAssign.value);
2418 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002419 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002420 goto failed;
2421 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002422 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002423 case For_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002424 result = PyType_GenericNew(For_type, NULL, NULL);
2425 if (!result) goto failed;
2426 value = ast2obj_expr(o->v.For.target);
2427 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002428 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002429 goto failed;
2430 Py_DECREF(value);
2431 value = ast2obj_expr(o->v.For.iter);
2432 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002433 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002434 goto failed;
2435 Py_DECREF(value);
2436 value = ast2obj_list(o->v.For.body, ast2obj_stmt);
2437 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002438 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002439 goto failed;
2440 Py_DECREF(value);
2441 value = ast2obj_list(o->v.For.orelse, ast2obj_stmt);
2442 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002443 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002444 goto failed;
2445 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002446 break;
2447 case While_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002448 result = PyType_GenericNew(While_type, NULL, NULL);
2449 if (!result) goto failed;
2450 value = ast2obj_expr(o->v.While.test);
2451 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002452 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002453 goto failed;
2454 Py_DECREF(value);
2455 value = ast2obj_list(o->v.While.body, ast2obj_stmt);
2456 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002457 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002458 goto failed;
2459 Py_DECREF(value);
2460 value = ast2obj_list(o->v.While.orelse, ast2obj_stmt);
2461 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002462 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002463 goto failed;
2464 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002465 break;
2466 case If_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002467 result = PyType_GenericNew(If_type, NULL, NULL);
2468 if (!result) goto failed;
2469 value = ast2obj_expr(o->v.If.test);
2470 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002471 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002472 goto failed;
2473 Py_DECREF(value);
2474 value = ast2obj_list(o->v.If.body, ast2obj_stmt);
2475 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002476 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002477 goto failed;
2478 Py_DECREF(value);
2479 value = ast2obj_list(o->v.If.orelse, ast2obj_stmt);
2480 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002481 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002482 goto failed;
2483 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002484 break;
Guido van Rossumc2e20742006-02-27 22:32:47 +00002485 case With_kind:
2486 result = PyType_GenericNew(With_type, NULL, NULL);
2487 if (!result) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002488 value = ast2obj_list(o->v.With.items, ast2obj_withitem);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002489 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002490 if (_PyObject_SetAttrId(result, &PyId_items, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002491 goto failed;
2492 Py_DECREF(value);
2493 value = ast2obj_list(o->v.With.body, ast2obj_stmt);
2494 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002495 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002496 goto failed;
2497 Py_DECREF(value);
2498 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002499 case Raise_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002500 result = PyType_GenericNew(Raise_type, NULL, NULL);
2501 if (!result) goto failed;
Collin Winter828f04a2007-08-31 00:04:24 +00002502 value = ast2obj_expr(o->v.Raise.exc);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002503 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002504 if (_PyObject_SetAttrId(result, &PyId_exc, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002505 goto failed;
2506 Py_DECREF(value);
Collin Winter828f04a2007-08-31 00:04:24 +00002507 value = ast2obj_expr(o->v.Raise.cause);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002508 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002509 if (_PyObject_SetAttrId(result, &PyId_cause, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002510 goto failed;
2511 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002512 break;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002513 case Try_kind:
2514 result = PyType_GenericNew(Try_type, NULL, NULL);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002515 if (!result) goto failed;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002516 value = ast2obj_list(o->v.Try.body, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002517 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002518 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002519 goto failed;
2520 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002521 value = ast2obj_list(o->v.Try.handlers, ast2obj_excepthandler);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002522 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002523 if (_PyObject_SetAttrId(result, &PyId_handlers, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002524 goto failed;
2525 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002526 value = ast2obj_list(o->v.Try.orelse, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002527 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002528 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002529 goto failed;
2530 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002531 value = ast2obj_list(o->v.Try.finalbody, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002532 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002533 if (_PyObject_SetAttrId(result, &PyId_finalbody, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002534 goto failed;
2535 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002536 break;
2537 case Assert_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002538 result = PyType_GenericNew(Assert_type, NULL, NULL);
2539 if (!result) goto failed;
2540 value = ast2obj_expr(o->v.Assert.test);
2541 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002542 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002543 goto failed;
2544 Py_DECREF(value);
2545 value = ast2obj_expr(o->v.Assert.msg);
2546 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002547 if (_PyObject_SetAttrId(result, &PyId_msg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002548 goto failed;
2549 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002550 break;
2551 case Import_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002552 result = PyType_GenericNew(Import_type, NULL, NULL);
2553 if (!result) goto failed;
2554 value = ast2obj_list(o->v.Import.names, ast2obj_alias);
2555 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002556 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002557 goto failed;
2558 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002559 break;
2560 case ImportFrom_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002561 result = PyType_GenericNew(ImportFrom_type, NULL, NULL);
2562 if (!result) goto failed;
2563 value = ast2obj_identifier(o->v.ImportFrom.module);
2564 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002565 if (_PyObject_SetAttrId(result, &PyId_module, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002566 goto failed;
2567 Py_DECREF(value);
2568 value = ast2obj_list(o->v.ImportFrom.names, ast2obj_alias);
2569 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002570 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002571 goto failed;
2572 Py_DECREF(value);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002573 value = ast2obj_int(o->v.ImportFrom.level);
2574 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002575 if (_PyObject_SetAttrId(result, &PyId_level, value) == -1)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002576 goto failed;
2577 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002578 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002579 case Global_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002580 result = PyType_GenericNew(Global_type, NULL, NULL);
2581 if (!result) goto failed;
2582 value = ast2obj_list(o->v.Global.names, ast2obj_identifier);
2583 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002584 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002585 goto failed;
2586 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002587 break;
Jeremy Hylton81e95022007-02-27 06:50:52 +00002588 case Nonlocal_kind:
2589 result = PyType_GenericNew(Nonlocal_type, NULL, NULL);
2590 if (!result) goto failed;
2591 value = ast2obj_list(o->v.Nonlocal.names, ast2obj_identifier);
2592 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002593 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Jeremy Hylton81e95022007-02-27 06:50:52 +00002594 goto failed;
2595 Py_DECREF(value);
2596 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002597 case Expr_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002598 result = PyType_GenericNew(Expr_type, NULL, NULL);
2599 if (!result) goto failed;
2600 value = ast2obj_expr(o->v.Expr.value);
2601 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002602 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002603 goto failed;
2604 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002605 break;
2606 case Pass_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002607 result = PyType_GenericNew(Pass_type, NULL, NULL);
2608 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002609 break;
2610 case Break_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002611 result = PyType_GenericNew(Break_type, NULL, NULL);
2612 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002613 break;
2614 case Continue_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002615 result = PyType_GenericNew(Continue_type, NULL, NULL);
2616 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002617 break;
2618 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002619 value = ast2obj_int(o->lineno);
2620 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002621 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002622 goto failed;
2623 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002624 value = ast2obj_int(o->col_offset);
2625 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002626 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002627 goto failed;
2628 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002629 return result;
2630failed:
2631 Py_XDECREF(value);
2632 Py_XDECREF(result);
2633 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002634}
2635
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002636PyObject*
2637ast2obj_expr(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002638{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002639 expr_ty o = (expr_ty)_o;
2640 PyObject *result = NULL, *value = NULL;
2641 if (!o) {
2642 Py_INCREF(Py_None);
2643 return Py_None;
2644 }
2645
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002646 switch (o->kind) {
2647 case BoolOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002648 result = PyType_GenericNew(BoolOp_type, NULL, NULL);
2649 if (!result) goto failed;
2650 value = ast2obj_boolop(o->v.BoolOp.op);
2651 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002652 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002653 goto failed;
2654 Py_DECREF(value);
2655 value = ast2obj_list(o->v.BoolOp.values, ast2obj_expr);
2656 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002657 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002658 goto failed;
2659 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002660 break;
2661 case BinOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002662 result = PyType_GenericNew(BinOp_type, NULL, NULL);
2663 if (!result) goto failed;
2664 value = ast2obj_expr(o->v.BinOp.left);
2665 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002666 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002667 goto failed;
2668 Py_DECREF(value);
2669 value = ast2obj_operator(o->v.BinOp.op);
2670 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002671 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002672 goto failed;
2673 Py_DECREF(value);
2674 value = ast2obj_expr(o->v.BinOp.right);
2675 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002676 if (_PyObject_SetAttrId(result, &PyId_right, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002677 goto failed;
2678 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002679 break;
2680 case UnaryOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002681 result = PyType_GenericNew(UnaryOp_type, NULL, NULL);
2682 if (!result) goto failed;
2683 value = ast2obj_unaryop(o->v.UnaryOp.op);
2684 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002685 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002686 goto failed;
2687 Py_DECREF(value);
2688 value = ast2obj_expr(o->v.UnaryOp.operand);
2689 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002690 if (_PyObject_SetAttrId(result, &PyId_operand, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002691 goto failed;
2692 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002693 break;
2694 case Lambda_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002695 result = PyType_GenericNew(Lambda_type, NULL, NULL);
2696 if (!result) goto failed;
2697 value = ast2obj_arguments(o->v.Lambda.args);
2698 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002699 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002700 goto failed;
2701 Py_DECREF(value);
2702 value = ast2obj_expr(o->v.Lambda.body);
2703 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002704 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002705 goto failed;
2706 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002707 break;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002708 case IfExp_kind:
2709 result = PyType_GenericNew(IfExp_type, NULL, NULL);
2710 if (!result) goto failed;
2711 value = ast2obj_expr(o->v.IfExp.test);
2712 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002713 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002714 goto failed;
2715 Py_DECREF(value);
2716 value = ast2obj_expr(o->v.IfExp.body);
2717 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002718 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002719 goto failed;
2720 Py_DECREF(value);
2721 value = ast2obj_expr(o->v.IfExp.orelse);
2722 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002723 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002724 goto failed;
2725 Py_DECREF(value);
2726 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002727 case Dict_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002728 result = PyType_GenericNew(Dict_type, NULL, NULL);
2729 if (!result) goto failed;
2730 value = ast2obj_list(o->v.Dict.keys, ast2obj_expr);
2731 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002732 if (_PyObject_SetAttrId(result, &PyId_keys, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002733 goto failed;
2734 Py_DECREF(value);
2735 value = ast2obj_list(o->v.Dict.values, ast2obj_expr);
2736 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002737 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002738 goto failed;
2739 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002740 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002741 case Set_kind:
2742 result = PyType_GenericNew(Set_type, NULL, NULL);
2743 if (!result) goto failed;
2744 value = ast2obj_list(o->v.Set.elts, ast2obj_expr);
2745 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002746 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Guido van Rossum86e58e22006-08-28 15:27:34 +00002747 goto failed;
2748 Py_DECREF(value);
2749 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002750 case ListComp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002751 result = PyType_GenericNew(ListComp_type, NULL, NULL);
2752 if (!result) goto failed;
2753 value = ast2obj_expr(o->v.ListComp.elt);
2754 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002755 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002756 goto failed;
2757 Py_DECREF(value);
2758 value = ast2obj_list(o->v.ListComp.generators,
2759 ast2obj_comprehension);
2760 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002761 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002762 goto failed;
2763 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002764 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00002765 case SetComp_kind:
2766 result = PyType_GenericNew(SetComp_type, NULL, NULL);
2767 if (!result) goto failed;
2768 value = ast2obj_expr(o->v.SetComp.elt);
2769 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002770 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002771 goto failed;
2772 Py_DECREF(value);
2773 value = ast2obj_list(o->v.SetComp.generators,
2774 ast2obj_comprehension);
2775 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002776 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002777 goto failed;
2778 Py_DECREF(value);
2779 break;
Guido van Rossum992d4a32007-07-11 13:09:30 +00002780 case DictComp_kind:
2781 result = PyType_GenericNew(DictComp_type, NULL, NULL);
2782 if (!result) goto failed;
2783 value = ast2obj_expr(o->v.DictComp.key);
2784 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002785 if (_PyObject_SetAttrId(result, &PyId_key, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002786 goto failed;
2787 Py_DECREF(value);
2788 value = ast2obj_expr(o->v.DictComp.value);
2789 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002790 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002791 goto failed;
2792 Py_DECREF(value);
2793 value = ast2obj_list(o->v.DictComp.generators,
2794 ast2obj_comprehension);
2795 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002796 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002797 goto failed;
2798 Py_DECREF(value);
2799 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002800 case GeneratorExp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002801 result = PyType_GenericNew(GeneratorExp_type, NULL, NULL);
2802 if (!result) goto failed;
2803 value = ast2obj_expr(o->v.GeneratorExp.elt);
2804 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002805 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002806 goto failed;
2807 Py_DECREF(value);
2808 value = ast2obj_list(o->v.GeneratorExp.generators,
2809 ast2obj_comprehension);
2810 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002811 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002812 goto failed;
2813 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002814 break;
2815 case Yield_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002816 result = PyType_GenericNew(Yield_type, NULL, NULL);
2817 if (!result) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002818 value = ast2obj_expr(o->v.Yield.value);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002819 if (!value) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002820 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002821 goto failed;
2822 Py_DECREF(value);
Benjamin Peterson527c6222012-01-14 08:58:23 -05002823 break;
2824 case YieldFrom_kind:
2825 result = PyType_GenericNew(YieldFrom_type, NULL, NULL);
2826 if (!result) goto failed;
2827 value = ast2obj_expr(o->v.YieldFrom.value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002828 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002829 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002830 goto failed;
2831 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002832 break;
2833 case Compare_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002834 result = PyType_GenericNew(Compare_type, NULL, NULL);
2835 if (!result) goto failed;
2836 value = ast2obj_expr(o->v.Compare.left);
2837 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002838 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002839 goto failed;
2840 Py_DECREF(value);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002841 {
2842 int i, n = asdl_seq_LEN(o->v.Compare.ops);
2843 value = PyList_New(n);
2844 if (!value) goto failed;
2845 for(i = 0; i < n; i++)
2846 PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(o->v.Compare.ops, i)));
2847 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002848 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002849 if (_PyObject_SetAttrId(result, &PyId_ops, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002850 goto failed;
2851 Py_DECREF(value);
2852 value = ast2obj_list(o->v.Compare.comparators, ast2obj_expr);
2853 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002854 if (_PyObject_SetAttrId(result, &PyId_comparators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002855 goto failed;
2856 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002857 break;
2858 case Call_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002859 result = PyType_GenericNew(Call_type, NULL, NULL);
2860 if (!result) goto failed;
2861 value = ast2obj_expr(o->v.Call.func);
2862 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002863 if (_PyObject_SetAttrId(result, &PyId_func, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002864 goto failed;
2865 Py_DECREF(value);
2866 value = ast2obj_list(o->v.Call.args, ast2obj_expr);
2867 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002868 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002869 goto failed;
2870 Py_DECREF(value);
2871 value = ast2obj_list(o->v.Call.keywords, ast2obj_keyword);
2872 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002873 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002874 goto failed;
2875 Py_DECREF(value);
2876 value = ast2obj_expr(o->v.Call.starargs);
2877 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002878 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002879 goto failed;
2880 Py_DECREF(value);
2881 value = ast2obj_expr(o->v.Call.kwargs);
2882 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002883 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002884 goto failed;
2885 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002886 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002887 case Num_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002888 result = PyType_GenericNew(Num_type, NULL, NULL);
2889 if (!result) goto failed;
2890 value = ast2obj_object(o->v.Num.n);
2891 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002892 if (_PyObject_SetAttrId(result, &PyId_n, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002893 goto failed;
2894 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002895 break;
2896 case Str_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002897 result = PyType_GenericNew(Str_type, NULL, NULL);
2898 if (!result) goto failed;
2899 value = ast2obj_string(o->v.Str.s);
2900 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002901 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002902 goto failed;
2903 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002904 break;
Thomas Wouters00e41de2007-02-23 19:56:57 +00002905 case Bytes_kind:
2906 result = PyType_GenericNew(Bytes_type, NULL, NULL);
2907 if (!result) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05002908 value = ast2obj_bytes(o->v.Bytes.s);
Thomas Wouters00e41de2007-02-23 19:56:57 +00002909 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002910 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Thomas Wouters00e41de2007-02-23 19:56:57 +00002911 goto failed;
2912 Py_DECREF(value);
2913 break;
Georg Brandl52318d62006-09-06 07:06:08 +00002914 case Ellipsis_kind:
2915 result = PyType_GenericNew(Ellipsis_type, NULL, NULL);
2916 if (!result) goto failed;
2917 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002918 case Attribute_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002919 result = PyType_GenericNew(Attribute_type, NULL, NULL);
2920 if (!result) goto failed;
2921 value = ast2obj_expr(o->v.Attribute.value);
2922 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002923 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002924 goto failed;
2925 Py_DECREF(value);
2926 value = ast2obj_identifier(o->v.Attribute.attr);
2927 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002928 if (_PyObject_SetAttrId(result, &PyId_attr, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002929 goto failed;
2930 Py_DECREF(value);
2931 value = ast2obj_expr_context(o->v.Attribute.ctx);
2932 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002933 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002934 goto failed;
2935 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002936 break;
2937 case Subscript_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002938 result = PyType_GenericNew(Subscript_type, NULL, NULL);
2939 if (!result) goto failed;
2940 value = ast2obj_expr(o->v.Subscript.value);
2941 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002942 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002943 goto failed;
2944 Py_DECREF(value);
2945 value = ast2obj_slice(o->v.Subscript.slice);
2946 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002947 if (_PyObject_SetAttrId(result, &PyId_slice, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002948 goto failed;
2949 Py_DECREF(value);
2950 value = ast2obj_expr_context(o->v.Subscript.ctx);
2951 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002952 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002953 goto failed;
2954 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002955 break;
Guido van Rossum0368b722007-05-11 16:50:42 +00002956 case Starred_kind:
2957 result = PyType_GenericNew(Starred_type, NULL, NULL);
2958 if (!result) goto failed;
2959 value = ast2obj_expr(o->v.Starred.value);
2960 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002961 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002962 goto failed;
2963 Py_DECREF(value);
2964 value = ast2obj_expr_context(o->v.Starred.ctx);
2965 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002966 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002967 goto failed;
2968 Py_DECREF(value);
2969 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002970 case Name_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002971 result = PyType_GenericNew(Name_type, NULL, NULL);
2972 if (!result) goto failed;
2973 value = ast2obj_identifier(o->v.Name.id);
2974 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002975 if (_PyObject_SetAttrId(result, &PyId_id, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002976 goto failed;
2977 Py_DECREF(value);
2978 value = ast2obj_expr_context(o->v.Name.ctx);
2979 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002980 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002981 goto failed;
2982 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002983 break;
2984 case List_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002985 result = PyType_GenericNew(List_type, NULL, NULL);
2986 if (!result) goto failed;
2987 value = ast2obj_list(o->v.List.elts, ast2obj_expr);
2988 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002989 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002990 goto failed;
2991 Py_DECREF(value);
2992 value = ast2obj_expr_context(o->v.List.ctx);
2993 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002994 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002995 goto failed;
2996 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002997 break;
2998 case Tuple_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002999 result = PyType_GenericNew(Tuple_type, NULL, NULL);
3000 if (!result) goto failed;
3001 value = ast2obj_list(o->v.Tuple.elts, ast2obj_expr);
3002 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003003 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003004 goto failed;
3005 Py_DECREF(value);
3006 value = ast2obj_expr_context(o->v.Tuple.ctx);
3007 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003008 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003009 goto failed;
3010 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003011 break;
3012 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00003013 value = ast2obj_int(o->lineno);
3014 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003015 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003016 goto failed;
3017 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00003018 value = ast2obj_int(o->col_offset);
3019 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003020 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003021 goto failed;
3022 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003023 return result;
3024failed:
3025 Py_XDECREF(value);
3026 Py_XDECREF(result);
3027 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003028}
3029
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003030PyObject* ast2obj_expr_context(expr_context_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003031{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003032 switch(o) {
3033 case Load:
3034 Py_INCREF(Load_singleton);
3035 return Load_singleton;
3036 case Store:
3037 Py_INCREF(Store_singleton);
3038 return Store_singleton;
3039 case Del:
3040 Py_INCREF(Del_singleton);
3041 return Del_singleton;
3042 case AugLoad:
3043 Py_INCREF(AugLoad_singleton);
3044 return AugLoad_singleton;
3045 case AugStore:
3046 Py_INCREF(AugStore_singleton);
3047 return AugStore_singleton;
3048 case Param:
3049 Py_INCREF(Param_singleton);
3050 return Param_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003051 default:
3052 /* should never happen, but just in case ... */
3053 PyErr_Format(PyExc_SystemError, "unknown expr_context found");
3054 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003055 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003056}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003057PyObject*
3058ast2obj_slice(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003059{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003060 slice_ty o = (slice_ty)_o;
3061 PyObject *result = NULL, *value = NULL;
3062 if (!o) {
3063 Py_INCREF(Py_None);
3064 return Py_None;
3065 }
3066
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003067 switch (o->kind) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003068 case Slice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003069 result = PyType_GenericNew(Slice_type, NULL, NULL);
3070 if (!result) goto failed;
3071 value = ast2obj_expr(o->v.Slice.lower);
3072 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003073 if (_PyObject_SetAttrId(result, &PyId_lower, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003074 goto failed;
3075 Py_DECREF(value);
3076 value = ast2obj_expr(o->v.Slice.upper);
3077 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003078 if (_PyObject_SetAttrId(result, &PyId_upper, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003079 goto failed;
3080 Py_DECREF(value);
3081 value = ast2obj_expr(o->v.Slice.step);
3082 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003083 if (_PyObject_SetAttrId(result, &PyId_step, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003084 goto failed;
3085 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003086 break;
3087 case ExtSlice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003088 result = PyType_GenericNew(ExtSlice_type, NULL, NULL);
3089 if (!result) goto failed;
3090 value = ast2obj_list(o->v.ExtSlice.dims, ast2obj_slice);
3091 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003092 if (_PyObject_SetAttrId(result, &PyId_dims, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003093 goto failed;
3094 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003095 break;
3096 case Index_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003097 result = PyType_GenericNew(Index_type, NULL, NULL);
3098 if (!result) goto failed;
3099 value = ast2obj_expr(o->v.Index.value);
3100 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003101 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003102 goto failed;
3103 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003104 break;
3105 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003106 return result;
3107failed:
3108 Py_XDECREF(value);
3109 Py_XDECREF(result);
3110 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003111}
3112
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003113PyObject* ast2obj_boolop(boolop_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003114{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003115 switch(o) {
3116 case And:
3117 Py_INCREF(And_singleton);
3118 return And_singleton;
3119 case Or:
3120 Py_INCREF(Or_singleton);
3121 return Or_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003122 default:
3123 /* should never happen, but just in case ... */
3124 PyErr_Format(PyExc_SystemError, "unknown boolop found");
3125 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003126 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003127}
3128PyObject* ast2obj_operator(operator_ty o)
3129{
3130 switch(o) {
3131 case Add:
3132 Py_INCREF(Add_singleton);
3133 return Add_singleton;
3134 case Sub:
3135 Py_INCREF(Sub_singleton);
3136 return Sub_singleton;
3137 case Mult:
3138 Py_INCREF(Mult_singleton);
3139 return Mult_singleton;
3140 case Div:
3141 Py_INCREF(Div_singleton);
3142 return Div_singleton;
3143 case Mod:
3144 Py_INCREF(Mod_singleton);
3145 return Mod_singleton;
3146 case Pow:
3147 Py_INCREF(Pow_singleton);
3148 return Pow_singleton;
3149 case LShift:
3150 Py_INCREF(LShift_singleton);
3151 return LShift_singleton;
3152 case RShift:
3153 Py_INCREF(RShift_singleton);
3154 return RShift_singleton;
3155 case BitOr:
3156 Py_INCREF(BitOr_singleton);
3157 return BitOr_singleton;
3158 case BitXor:
3159 Py_INCREF(BitXor_singleton);
3160 return BitXor_singleton;
3161 case BitAnd:
3162 Py_INCREF(BitAnd_singleton);
3163 return BitAnd_singleton;
3164 case FloorDiv:
3165 Py_INCREF(FloorDiv_singleton);
3166 return FloorDiv_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003167 default:
3168 /* should never happen, but just in case ... */
3169 PyErr_Format(PyExc_SystemError, "unknown operator found");
3170 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003171 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003172}
3173PyObject* ast2obj_unaryop(unaryop_ty o)
3174{
3175 switch(o) {
3176 case Invert:
3177 Py_INCREF(Invert_singleton);
3178 return Invert_singleton;
3179 case Not:
3180 Py_INCREF(Not_singleton);
3181 return Not_singleton;
3182 case UAdd:
3183 Py_INCREF(UAdd_singleton);
3184 return UAdd_singleton;
3185 case USub:
3186 Py_INCREF(USub_singleton);
3187 return USub_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003188 default:
3189 /* should never happen, but just in case ... */
3190 PyErr_Format(PyExc_SystemError, "unknown unaryop found");
3191 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003192 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003193}
3194PyObject* ast2obj_cmpop(cmpop_ty o)
3195{
3196 switch(o) {
3197 case Eq:
3198 Py_INCREF(Eq_singleton);
3199 return Eq_singleton;
3200 case NotEq:
3201 Py_INCREF(NotEq_singleton);
3202 return NotEq_singleton;
3203 case Lt:
3204 Py_INCREF(Lt_singleton);
3205 return Lt_singleton;
3206 case LtE:
3207 Py_INCREF(LtE_singleton);
3208 return LtE_singleton;
3209 case Gt:
3210 Py_INCREF(Gt_singleton);
3211 return Gt_singleton;
3212 case GtE:
3213 Py_INCREF(GtE_singleton);
3214 return GtE_singleton;
3215 case Is:
3216 Py_INCREF(Is_singleton);
3217 return Is_singleton;
3218 case IsNot:
3219 Py_INCREF(IsNot_singleton);
3220 return IsNot_singleton;
3221 case In:
3222 Py_INCREF(In_singleton);
3223 return In_singleton;
3224 case NotIn:
3225 Py_INCREF(NotIn_singleton);
3226 return NotIn_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003227 default:
3228 /* should never happen, but just in case ... */
3229 PyErr_Format(PyExc_SystemError, "unknown cmpop found");
3230 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003231 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003232}
3233PyObject*
3234ast2obj_comprehension(void* _o)
3235{
3236 comprehension_ty o = (comprehension_ty)_o;
3237 PyObject *result = NULL, *value = NULL;
3238 if (!o) {
3239 Py_INCREF(Py_None);
3240 return Py_None;
3241 }
3242
3243 result = PyType_GenericNew(comprehension_type, NULL, NULL);
3244 if (!result) return NULL;
3245 value = ast2obj_expr(o->target);
3246 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003247 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003248 goto failed;
3249 Py_DECREF(value);
3250 value = ast2obj_expr(o->iter);
3251 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003252 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003253 goto failed;
3254 Py_DECREF(value);
3255 value = ast2obj_list(o->ifs, ast2obj_expr);
3256 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003257 if (_PyObject_SetAttrId(result, &PyId_ifs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003258 goto failed;
3259 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003260 return result;
3261failed:
3262 Py_XDECREF(value);
3263 Py_XDECREF(result);
3264 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003265}
3266
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003267PyObject*
3268ast2obj_excepthandler(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003269{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003270 excepthandler_ty o = (excepthandler_ty)_o;
3271 PyObject *result = NULL, *value = NULL;
3272 if (!o) {
3273 Py_INCREF(Py_None);
3274 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003275 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003276
Neal Norwitzad74aa82008-03-31 05:14:30 +00003277 switch (o->kind) {
3278 case ExceptHandler_kind:
3279 result = PyType_GenericNew(ExceptHandler_type, NULL, NULL);
3280 if (!result) goto failed;
3281 value = ast2obj_expr(o->v.ExceptHandler.type);
3282 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003283 if (_PyObject_SetAttrId(result, &PyId_type, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003284 goto failed;
3285 Py_DECREF(value);
3286 value = ast2obj_identifier(o->v.ExceptHandler.name);
3287 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003288 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003289 goto failed;
3290 Py_DECREF(value);
3291 value = ast2obj_list(o->v.ExceptHandler.body, ast2obj_stmt);
3292 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003293 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003294 goto failed;
3295 Py_DECREF(value);
3296 break;
3297 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003298 value = ast2obj_int(o->lineno);
3299 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003300 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003301 goto failed;
3302 Py_DECREF(value);
3303 value = ast2obj_int(o->col_offset);
3304 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003305 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003306 goto failed;
3307 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003308 return result;
3309failed:
3310 Py_XDECREF(value);
3311 Py_XDECREF(result);
3312 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003313}
3314
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003315PyObject*
3316ast2obj_arguments(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003317{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003318 arguments_ty o = (arguments_ty)_o;
3319 PyObject *result = NULL, *value = NULL;
3320 if (!o) {
3321 Py_INCREF(Py_None);
3322 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003323 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003324
3325 result = PyType_GenericNew(arguments_type, NULL, NULL);
3326 if (!result) return NULL;
Neal Norwitzc1505362006-12-28 06:47:50 +00003327 value = ast2obj_list(o->args, ast2obj_arg);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003328 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003329 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003330 goto failed;
3331 Py_DECREF(value);
3332 value = ast2obj_identifier(o->vararg);
3333 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003334 if (_PyObject_SetAttrId(result, &PyId_vararg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003335 goto failed;
3336 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003337 value = ast2obj_expr(o->varargannotation);
3338 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003339 if (_PyObject_SetAttrId(result, &PyId_varargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003340 goto failed;
3341 Py_DECREF(value);
3342 value = ast2obj_list(o->kwonlyargs, ast2obj_arg);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003343 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003344 if (_PyObject_SetAttrId(result, &PyId_kwonlyargs, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003345 goto failed;
3346 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003347 value = ast2obj_identifier(o->kwarg);
3348 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003349 if (_PyObject_SetAttrId(result, &PyId_kwarg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003350 goto failed;
3351 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003352 value = ast2obj_expr(o->kwargannotation);
3353 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003354 if (_PyObject_SetAttrId(result, &PyId_kwargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003355 goto failed;
3356 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003357 value = ast2obj_list(o->defaults, ast2obj_expr);
3358 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003359 if (_PyObject_SetAttrId(result, &PyId_defaults, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003360 goto failed;
3361 Py_DECREF(value);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003362 value = ast2obj_list(o->kw_defaults, ast2obj_expr);
3363 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003364 if (_PyObject_SetAttrId(result, &PyId_kw_defaults, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003365 goto failed;
3366 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003367 return result;
3368failed:
3369 Py_XDECREF(value);
3370 Py_XDECREF(result);
3371 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003372}
3373
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003374PyObject*
Neal Norwitzc1505362006-12-28 06:47:50 +00003375ast2obj_arg(void* _o)
3376{
3377 arg_ty o = (arg_ty)_o;
3378 PyObject *result = NULL, *value = NULL;
3379 if (!o) {
3380 Py_INCREF(Py_None);
3381 return Py_None;
3382 }
3383
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003384 result = PyType_GenericNew(arg_type, NULL, NULL);
3385 if (!result) return NULL;
3386 value = ast2obj_identifier(o->arg);
3387 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003388 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003389 goto failed;
3390 Py_DECREF(value);
3391 value = ast2obj_expr(o->annotation);
3392 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003393 if (_PyObject_SetAttrId(result, &PyId_annotation, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003394 goto failed;
3395 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003396 return result;
3397failed:
3398 Py_XDECREF(value);
3399 Py_XDECREF(result);
3400 return NULL;
3401}
3402
3403PyObject*
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003404ast2obj_keyword(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003405{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003406 keyword_ty o = (keyword_ty)_o;
3407 PyObject *result = NULL, *value = NULL;
3408 if (!o) {
3409 Py_INCREF(Py_None);
3410 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003411 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003412
3413 result = PyType_GenericNew(keyword_type, NULL, NULL);
3414 if (!result) return NULL;
3415 value = ast2obj_identifier(o->arg);
3416 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003417 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003418 goto failed;
3419 Py_DECREF(value);
3420 value = ast2obj_expr(o->value);
3421 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003422 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003423 goto failed;
3424 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003425 return result;
3426failed:
3427 Py_XDECREF(value);
3428 Py_XDECREF(result);
3429 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003430}
3431
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003432PyObject*
3433ast2obj_alias(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003434{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003435 alias_ty o = (alias_ty)_o;
3436 PyObject *result = NULL, *value = NULL;
3437 if (!o) {
3438 Py_INCREF(Py_None);
3439 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003440 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003441
3442 result = PyType_GenericNew(alias_type, NULL, NULL);
3443 if (!result) return NULL;
3444 value = ast2obj_identifier(o->name);
3445 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003446 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003447 goto failed;
3448 Py_DECREF(value);
3449 value = ast2obj_identifier(o->asname);
3450 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003451 if (_PyObject_SetAttrId(result, &PyId_asname, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003452 goto failed;
3453 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003454 return result;
3455failed:
3456 Py_XDECREF(value);
3457 Py_XDECREF(result);
3458 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003459}
3460
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003461PyObject*
3462ast2obj_withitem(void* _o)
3463{
3464 withitem_ty o = (withitem_ty)_o;
3465 PyObject *result = NULL, *value = NULL;
3466 if (!o) {
3467 Py_INCREF(Py_None);
3468 return Py_None;
3469 }
3470
3471 result = PyType_GenericNew(withitem_type, NULL, NULL);
3472 if (!result) return NULL;
3473 value = ast2obj_expr(o->context_expr);
3474 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003475 if (_PyObject_SetAttrId(result, &PyId_context_expr, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003476 goto failed;
3477 Py_DECREF(value);
3478 value = ast2obj_expr(o->optional_vars);
3479 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003480 if (_PyObject_SetAttrId(result, &PyId_optional_vars, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003481 goto failed;
3482 Py_DECREF(value);
3483 return result;
3484failed:
3485 Py_XDECREF(value);
3486 Py_XDECREF(result);
3487 return NULL;
3488}
3489
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003490
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003491int
3492obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena)
3493{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003494 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003495
Benjamin Petersond8f65972010-11-20 04:31:07 +00003496 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003497
3498 if (obj == Py_None) {
3499 *out = NULL;
3500 return 0;
3501 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003502 isinstance = PyObject_IsInstance(obj, (PyObject*)Module_type);
3503 if (isinstance == -1) {
3504 return 1;
3505 }
3506 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003507 asdl_seq* body;
3508
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003509 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003510 int res;
3511 Py_ssize_t len;
3512 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003513 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003514 if (tmp == NULL) goto failed;
3515 if (!PyList_Check(tmp)) {
3516 PyErr_Format(PyExc_TypeError, "Module field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3517 goto failed;
3518 }
3519 len = PyList_GET_SIZE(tmp);
3520 body = asdl_seq_new(len, arena);
3521 if (body == NULL) goto failed;
3522 for (i = 0; i < len; i++) {
3523 stmt_ty value;
3524 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3525 if (res != 0) goto failed;
3526 asdl_seq_SET(body, i, value);
3527 }
3528 Py_XDECREF(tmp);
3529 tmp = NULL;
3530 } else {
3531 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Module");
3532 return 1;
3533 }
3534 *out = Module(body, arena);
3535 if (*out == NULL) goto failed;
3536 return 0;
3537 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003538 isinstance = PyObject_IsInstance(obj, (PyObject*)Interactive_type);
3539 if (isinstance == -1) {
3540 return 1;
3541 }
3542 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003543 asdl_seq* body;
3544
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003545 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003546 int res;
3547 Py_ssize_t len;
3548 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003549 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003550 if (tmp == NULL) goto failed;
3551 if (!PyList_Check(tmp)) {
3552 PyErr_Format(PyExc_TypeError, "Interactive field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3553 goto failed;
3554 }
3555 len = PyList_GET_SIZE(tmp);
3556 body = asdl_seq_new(len, arena);
3557 if (body == NULL) goto failed;
3558 for (i = 0; i < len; i++) {
3559 stmt_ty value;
3560 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3561 if (res != 0) goto failed;
3562 asdl_seq_SET(body, i, value);
3563 }
3564 Py_XDECREF(tmp);
3565 tmp = NULL;
3566 } else {
3567 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Interactive");
3568 return 1;
3569 }
3570 *out = Interactive(body, arena);
3571 if (*out == NULL) goto failed;
3572 return 0;
3573 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003574 isinstance = PyObject_IsInstance(obj, (PyObject*)Expression_type);
3575 if (isinstance == -1) {
3576 return 1;
3577 }
3578 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003579 expr_ty body;
3580
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003581 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003582 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003583 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003584 if (tmp == NULL) goto failed;
3585 res = obj2ast_expr(tmp, &body, arena);
3586 if (res != 0) goto failed;
3587 Py_XDECREF(tmp);
3588 tmp = NULL;
3589 } else {
3590 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Expression");
3591 return 1;
3592 }
3593 *out = Expression(body, arena);
3594 if (*out == NULL) goto failed;
3595 return 0;
3596 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003597 isinstance = PyObject_IsInstance(obj, (PyObject*)Suite_type);
3598 if (isinstance == -1) {
3599 return 1;
3600 }
3601 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003602 asdl_seq* body;
3603
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003604 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003605 int res;
3606 Py_ssize_t len;
3607 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003608 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003609 if (tmp == NULL) goto failed;
3610 if (!PyList_Check(tmp)) {
3611 PyErr_Format(PyExc_TypeError, "Suite field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3612 goto failed;
3613 }
3614 len = PyList_GET_SIZE(tmp);
3615 body = asdl_seq_new(len, arena);
3616 if (body == NULL) goto failed;
3617 for (i = 0; i < len; i++) {
3618 stmt_ty value;
3619 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3620 if (res != 0) goto failed;
3621 asdl_seq_SET(body, i, value);
3622 }
3623 Py_XDECREF(tmp);
3624 tmp = NULL;
3625 } else {
3626 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Suite");
3627 return 1;
3628 }
3629 *out = Suite(body, arena);
3630 if (*out == NULL) goto failed;
3631 return 0;
3632 }
3633
Benjamin Peterson5b066812010-11-20 01:38:49 +00003634 PyErr_Format(PyExc_TypeError, "expected some sort of mod, but got %R", obj);
3635 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00003636 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003637 return 1;
3638}
3639
3640int
3641obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena)
3642{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003643 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003644
Benjamin Petersond8f65972010-11-20 04:31:07 +00003645 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003646 int lineno;
3647 int col_offset;
3648
3649 if (obj == Py_None) {
3650 *out = NULL;
3651 return 0;
3652 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003653 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003654 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003655 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003656 if (tmp == NULL) goto failed;
3657 res = obj2ast_int(tmp, &lineno, arena);
3658 if (res != 0) goto failed;
3659 Py_XDECREF(tmp);
3660 tmp = NULL;
3661 } else {
3662 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from stmt");
3663 return 1;
3664 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003665 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003666 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003667 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003668 if (tmp == NULL) goto failed;
3669 res = obj2ast_int(tmp, &col_offset, arena);
3670 if (res != 0) goto failed;
3671 Py_XDECREF(tmp);
3672 tmp = NULL;
3673 } else {
3674 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from stmt");
3675 return 1;
3676 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003677 isinstance = PyObject_IsInstance(obj, (PyObject*)FunctionDef_type);
3678 if (isinstance == -1) {
3679 return 1;
3680 }
3681 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003682 identifier name;
3683 arguments_ty args;
3684 asdl_seq* body;
3685 asdl_seq* decorator_list;
3686 expr_ty returns;
3687
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003688 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003689 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003690 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003691 if (tmp == NULL) goto failed;
3692 res = obj2ast_identifier(tmp, &name, arena);
3693 if (res != 0) goto failed;
3694 Py_XDECREF(tmp);
3695 tmp = NULL;
3696 } else {
3697 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from FunctionDef");
3698 return 1;
3699 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003700 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003701 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003702 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003703 if (tmp == NULL) goto failed;
3704 res = obj2ast_arguments(tmp, &args, arena);
3705 if (res != 0) goto failed;
3706 Py_XDECREF(tmp);
3707 tmp = NULL;
3708 } else {
3709 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from FunctionDef");
3710 return 1;
3711 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003712 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003713 int res;
3714 Py_ssize_t len;
3715 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003716 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003717 if (tmp == NULL) goto failed;
3718 if (!PyList_Check(tmp)) {
3719 PyErr_Format(PyExc_TypeError, "FunctionDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3720 goto failed;
3721 }
3722 len = PyList_GET_SIZE(tmp);
3723 body = asdl_seq_new(len, arena);
3724 if (body == NULL) goto failed;
3725 for (i = 0; i < len; i++) {
3726 stmt_ty value;
3727 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3728 if (res != 0) goto failed;
3729 asdl_seq_SET(body, i, value);
3730 }
3731 Py_XDECREF(tmp);
3732 tmp = NULL;
3733 } else {
3734 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from FunctionDef");
3735 return 1;
3736 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003737 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003738 int res;
3739 Py_ssize_t len;
3740 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003741 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003742 if (tmp == NULL) goto failed;
3743 if (!PyList_Check(tmp)) {
3744 PyErr_Format(PyExc_TypeError, "FunctionDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3745 goto failed;
3746 }
3747 len = PyList_GET_SIZE(tmp);
3748 decorator_list = asdl_seq_new(len, arena);
3749 if (decorator_list == NULL) goto failed;
3750 for (i = 0; i < len; i++) {
3751 expr_ty value;
3752 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3753 if (res != 0) goto failed;
3754 asdl_seq_SET(decorator_list, i, value);
3755 }
3756 Py_XDECREF(tmp);
3757 tmp = NULL;
3758 } else {
3759 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from FunctionDef");
3760 return 1;
3761 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003762 if (_PyObject_HasAttrId(obj, &PyId_returns)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003763 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003764 tmp = _PyObject_GetAttrId(obj, &PyId_returns);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003765 if (tmp == NULL) goto failed;
3766 res = obj2ast_expr(tmp, &returns, arena);
3767 if (res != 0) goto failed;
3768 Py_XDECREF(tmp);
3769 tmp = NULL;
3770 } else {
3771 returns = NULL;
3772 }
3773 *out = FunctionDef(name, args, body, decorator_list, returns,
3774 lineno, col_offset, arena);
3775 if (*out == NULL) goto failed;
3776 return 0;
3777 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003778 isinstance = PyObject_IsInstance(obj, (PyObject*)ClassDef_type);
3779 if (isinstance == -1) {
3780 return 1;
3781 }
3782 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003783 identifier name;
3784 asdl_seq* bases;
3785 asdl_seq* keywords;
3786 expr_ty starargs;
3787 expr_ty kwargs;
3788 asdl_seq* body;
3789 asdl_seq* decorator_list;
3790
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003791 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003792 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003793 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003794 if (tmp == NULL) goto failed;
3795 res = obj2ast_identifier(tmp, &name, arena);
3796 if (res != 0) goto failed;
3797 Py_XDECREF(tmp);
3798 tmp = NULL;
3799 } else {
3800 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from ClassDef");
3801 return 1;
3802 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003803 if (_PyObject_HasAttrId(obj, &PyId_bases)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003804 int res;
3805 Py_ssize_t len;
3806 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003807 tmp = _PyObject_GetAttrId(obj, &PyId_bases);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003808 if (tmp == NULL) goto failed;
3809 if (!PyList_Check(tmp)) {
3810 PyErr_Format(PyExc_TypeError, "ClassDef field \"bases\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3811 goto failed;
3812 }
3813 len = PyList_GET_SIZE(tmp);
3814 bases = asdl_seq_new(len, arena);
3815 if (bases == NULL) goto failed;
3816 for (i = 0; i < len; i++) {
3817 expr_ty value;
3818 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3819 if (res != 0) goto failed;
3820 asdl_seq_SET(bases, i, value);
3821 }
3822 Py_XDECREF(tmp);
3823 tmp = NULL;
3824 } else {
3825 PyErr_SetString(PyExc_TypeError, "required field \"bases\" missing from ClassDef");
3826 return 1;
3827 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003828 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003829 int res;
3830 Py_ssize_t len;
3831 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003832 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003833 if (tmp == NULL) goto failed;
3834 if (!PyList_Check(tmp)) {
3835 PyErr_Format(PyExc_TypeError, "ClassDef field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3836 goto failed;
3837 }
3838 len = PyList_GET_SIZE(tmp);
3839 keywords = asdl_seq_new(len, arena);
3840 if (keywords == NULL) goto failed;
3841 for (i = 0; i < len; i++) {
3842 keyword_ty value;
3843 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
3844 if (res != 0) goto failed;
3845 asdl_seq_SET(keywords, i, value);
3846 }
3847 Py_XDECREF(tmp);
3848 tmp = NULL;
3849 } else {
3850 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from ClassDef");
3851 return 1;
3852 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003853 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003854 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003855 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003856 if (tmp == NULL) goto failed;
3857 res = obj2ast_expr(tmp, &starargs, arena);
3858 if (res != 0) goto failed;
3859 Py_XDECREF(tmp);
3860 tmp = NULL;
3861 } else {
3862 starargs = NULL;
3863 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003864 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003865 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003866 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003867 if (tmp == NULL) goto failed;
3868 res = obj2ast_expr(tmp, &kwargs, arena);
3869 if (res != 0) goto failed;
3870 Py_XDECREF(tmp);
3871 tmp = NULL;
3872 } else {
3873 kwargs = NULL;
3874 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003875 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003876 int res;
3877 Py_ssize_t len;
3878 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003879 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003880 if (tmp == NULL) goto failed;
3881 if (!PyList_Check(tmp)) {
3882 PyErr_Format(PyExc_TypeError, "ClassDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3883 goto failed;
3884 }
3885 len = PyList_GET_SIZE(tmp);
3886 body = asdl_seq_new(len, arena);
3887 if (body == NULL) goto failed;
3888 for (i = 0; i < len; i++) {
3889 stmt_ty value;
3890 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3891 if (res != 0) goto failed;
3892 asdl_seq_SET(body, i, value);
3893 }
3894 Py_XDECREF(tmp);
3895 tmp = NULL;
3896 } else {
3897 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ClassDef");
3898 return 1;
3899 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003900 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003901 int res;
3902 Py_ssize_t len;
3903 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003904 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003905 if (tmp == NULL) goto failed;
3906 if (!PyList_Check(tmp)) {
3907 PyErr_Format(PyExc_TypeError, "ClassDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3908 goto failed;
3909 }
3910 len = PyList_GET_SIZE(tmp);
3911 decorator_list = asdl_seq_new(len, arena);
3912 if (decorator_list == NULL) goto failed;
3913 for (i = 0; i < len; i++) {
3914 expr_ty value;
3915 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3916 if (res != 0) goto failed;
3917 asdl_seq_SET(decorator_list, i, value);
3918 }
3919 Py_XDECREF(tmp);
3920 tmp = NULL;
3921 } else {
3922 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from ClassDef");
3923 return 1;
3924 }
3925 *out = ClassDef(name, bases, keywords, starargs, kwargs, body,
3926 decorator_list, lineno, col_offset, arena);
3927 if (*out == NULL) goto failed;
3928 return 0;
3929 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003930 isinstance = PyObject_IsInstance(obj, (PyObject*)Return_type);
3931 if (isinstance == -1) {
3932 return 1;
3933 }
3934 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003935 expr_ty value;
3936
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003937 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003938 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003939 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003940 if (tmp == NULL) goto failed;
3941 res = obj2ast_expr(tmp, &value, arena);
3942 if (res != 0) goto failed;
3943 Py_XDECREF(tmp);
3944 tmp = NULL;
3945 } else {
3946 value = NULL;
3947 }
3948 *out = Return(value, lineno, col_offset, arena);
3949 if (*out == NULL) goto failed;
3950 return 0;
3951 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003952 isinstance = PyObject_IsInstance(obj, (PyObject*)Delete_type);
3953 if (isinstance == -1) {
3954 return 1;
3955 }
3956 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003957 asdl_seq* targets;
3958
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003959 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003960 int res;
3961 Py_ssize_t len;
3962 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003963 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003964 if (tmp == NULL) goto failed;
3965 if (!PyList_Check(tmp)) {
3966 PyErr_Format(PyExc_TypeError, "Delete field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3967 goto failed;
3968 }
3969 len = PyList_GET_SIZE(tmp);
3970 targets = asdl_seq_new(len, arena);
3971 if (targets == NULL) goto failed;
3972 for (i = 0; i < len; i++) {
3973 expr_ty value;
3974 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3975 if (res != 0) goto failed;
3976 asdl_seq_SET(targets, i, value);
3977 }
3978 Py_XDECREF(tmp);
3979 tmp = NULL;
3980 } else {
3981 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Delete");
3982 return 1;
3983 }
3984 *out = Delete(targets, lineno, col_offset, arena);
3985 if (*out == NULL) goto failed;
3986 return 0;
3987 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003988 isinstance = PyObject_IsInstance(obj, (PyObject*)Assign_type);
3989 if (isinstance == -1) {
3990 return 1;
3991 }
3992 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003993 asdl_seq* targets;
3994 expr_ty value;
3995
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003996 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003997 int res;
3998 Py_ssize_t len;
3999 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004000 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004001 if (tmp == NULL) goto failed;
4002 if (!PyList_Check(tmp)) {
4003 PyErr_Format(PyExc_TypeError, "Assign field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4004 goto failed;
4005 }
4006 len = PyList_GET_SIZE(tmp);
4007 targets = asdl_seq_new(len, arena);
4008 if (targets == NULL) goto failed;
4009 for (i = 0; i < len; i++) {
4010 expr_ty value;
4011 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4012 if (res != 0) goto failed;
4013 asdl_seq_SET(targets, i, value);
4014 }
4015 Py_XDECREF(tmp);
4016 tmp = NULL;
4017 } else {
4018 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Assign");
4019 return 1;
4020 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004021 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004022 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004023 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004024 if (tmp == NULL) goto failed;
4025 res = obj2ast_expr(tmp, &value, arena);
4026 if (res != 0) goto failed;
4027 Py_XDECREF(tmp);
4028 tmp = NULL;
4029 } else {
4030 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Assign");
4031 return 1;
4032 }
4033 *out = Assign(targets, value, lineno, col_offset, arena);
4034 if (*out == NULL) goto failed;
4035 return 0;
4036 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004037 isinstance = PyObject_IsInstance(obj, (PyObject*)AugAssign_type);
4038 if (isinstance == -1) {
4039 return 1;
4040 }
4041 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004042 expr_ty target;
4043 operator_ty op;
4044 expr_ty value;
4045
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004046 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004047 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004048 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004049 if (tmp == NULL) goto failed;
4050 res = obj2ast_expr(tmp, &target, arena);
4051 if (res != 0) goto failed;
4052 Py_XDECREF(tmp);
4053 tmp = NULL;
4054 } else {
4055 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from AugAssign");
4056 return 1;
4057 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004058 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004059 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004060 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004061 if (tmp == NULL) goto failed;
4062 res = obj2ast_operator(tmp, &op, arena);
4063 if (res != 0) goto failed;
4064 Py_XDECREF(tmp);
4065 tmp = NULL;
4066 } else {
4067 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from AugAssign");
4068 return 1;
4069 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004070 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004071 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004072 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004073 if (tmp == NULL) goto failed;
4074 res = obj2ast_expr(tmp, &value, arena);
4075 if (res != 0) goto failed;
4076 Py_XDECREF(tmp);
4077 tmp = NULL;
4078 } else {
4079 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from AugAssign");
4080 return 1;
4081 }
4082 *out = AugAssign(target, op, value, lineno, col_offset, arena);
4083 if (*out == NULL) goto failed;
4084 return 0;
4085 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004086 isinstance = PyObject_IsInstance(obj, (PyObject*)For_type);
4087 if (isinstance == -1) {
4088 return 1;
4089 }
4090 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004091 expr_ty target;
4092 expr_ty iter;
4093 asdl_seq* body;
4094 asdl_seq* orelse;
4095
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004096 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004097 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004098 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004099 if (tmp == NULL) goto failed;
4100 res = obj2ast_expr(tmp, &target, arena);
4101 if (res != 0) goto failed;
4102 Py_XDECREF(tmp);
4103 tmp = NULL;
4104 } else {
4105 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from For");
4106 return 1;
4107 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004108 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004109 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004110 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004111 if (tmp == NULL) goto failed;
4112 res = obj2ast_expr(tmp, &iter, arena);
4113 if (res != 0) goto failed;
4114 Py_XDECREF(tmp);
4115 tmp = NULL;
4116 } else {
4117 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from For");
4118 return 1;
4119 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004120 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004121 int res;
4122 Py_ssize_t len;
4123 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004124 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004125 if (tmp == NULL) goto failed;
4126 if (!PyList_Check(tmp)) {
4127 PyErr_Format(PyExc_TypeError, "For field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4128 goto failed;
4129 }
4130 len = PyList_GET_SIZE(tmp);
4131 body = asdl_seq_new(len, arena);
4132 if (body == NULL) goto failed;
4133 for (i = 0; i < len; i++) {
4134 stmt_ty value;
4135 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4136 if (res != 0) goto failed;
4137 asdl_seq_SET(body, i, value);
4138 }
4139 Py_XDECREF(tmp);
4140 tmp = NULL;
4141 } else {
4142 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from For");
4143 return 1;
4144 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004145 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004146 int res;
4147 Py_ssize_t len;
4148 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004149 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004150 if (tmp == NULL) goto failed;
4151 if (!PyList_Check(tmp)) {
4152 PyErr_Format(PyExc_TypeError, "For field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4153 goto failed;
4154 }
4155 len = PyList_GET_SIZE(tmp);
4156 orelse = asdl_seq_new(len, arena);
4157 if (orelse == NULL) goto failed;
4158 for (i = 0; i < len; i++) {
4159 stmt_ty value;
4160 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4161 if (res != 0) goto failed;
4162 asdl_seq_SET(orelse, i, value);
4163 }
4164 Py_XDECREF(tmp);
4165 tmp = NULL;
4166 } else {
4167 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from For");
4168 return 1;
4169 }
4170 *out = For(target, iter, body, orelse, lineno, col_offset,
4171 arena);
4172 if (*out == NULL) goto failed;
4173 return 0;
4174 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004175 isinstance = PyObject_IsInstance(obj, (PyObject*)While_type);
4176 if (isinstance == -1) {
4177 return 1;
4178 }
4179 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004180 expr_ty test;
4181 asdl_seq* body;
4182 asdl_seq* orelse;
4183
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004184 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004185 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004186 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004187 if (tmp == NULL) goto failed;
4188 res = obj2ast_expr(tmp, &test, arena);
4189 if (res != 0) goto failed;
4190 Py_XDECREF(tmp);
4191 tmp = NULL;
4192 } else {
4193 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from While");
4194 return 1;
4195 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004196 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004197 int res;
4198 Py_ssize_t len;
4199 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004200 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004201 if (tmp == NULL) goto failed;
4202 if (!PyList_Check(tmp)) {
4203 PyErr_Format(PyExc_TypeError, "While field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4204 goto failed;
4205 }
4206 len = PyList_GET_SIZE(tmp);
4207 body = asdl_seq_new(len, arena);
4208 if (body == NULL) goto failed;
4209 for (i = 0; i < len; i++) {
4210 stmt_ty value;
4211 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4212 if (res != 0) goto failed;
4213 asdl_seq_SET(body, i, value);
4214 }
4215 Py_XDECREF(tmp);
4216 tmp = NULL;
4217 } else {
4218 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from While");
4219 return 1;
4220 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004221 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004222 int res;
4223 Py_ssize_t len;
4224 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004225 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004226 if (tmp == NULL) goto failed;
4227 if (!PyList_Check(tmp)) {
4228 PyErr_Format(PyExc_TypeError, "While field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4229 goto failed;
4230 }
4231 len = PyList_GET_SIZE(tmp);
4232 orelse = asdl_seq_new(len, arena);
4233 if (orelse == NULL) goto failed;
4234 for (i = 0; i < len; i++) {
4235 stmt_ty value;
4236 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4237 if (res != 0) goto failed;
4238 asdl_seq_SET(orelse, i, value);
4239 }
4240 Py_XDECREF(tmp);
4241 tmp = NULL;
4242 } else {
4243 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from While");
4244 return 1;
4245 }
4246 *out = While(test, body, orelse, lineno, col_offset, arena);
4247 if (*out == NULL) goto failed;
4248 return 0;
4249 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004250 isinstance = PyObject_IsInstance(obj, (PyObject*)If_type);
4251 if (isinstance == -1) {
4252 return 1;
4253 }
4254 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004255 expr_ty test;
4256 asdl_seq* body;
4257 asdl_seq* orelse;
4258
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004259 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004260 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004261 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004262 if (tmp == NULL) goto failed;
4263 res = obj2ast_expr(tmp, &test, arena);
4264 if (res != 0) goto failed;
4265 Py_XDECREF(tmp);
4266 tmp = NULL;
4267 } else {
4268 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from If");
4269 return 1;
4270 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004271 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004272 int res;
4273 Py_ssize_t len;
4274 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004275 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004276 if (tmp == NULL) goto failed;
4277 if (!PyList_Check(tmp)) {
4278 PyErr_Format(PyExc_TypeError, "If field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4279 goto failed;
4280 }
4281 len = PyList_GET_SIZE(tmp);
4282 body = asdl_seq_new(len, arena);
4283 if (body == NULL) goto failed;
4284 for (i = 0; i < len; i++) {
4285 stmt_ty value;
4286 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4287 if (res != 0) goto failed;
4288 asdl_seq_SET(body, i, value);
4289 }
4290 Py_XDECREF(tmp);
4291 tmp = NULL;
4292 } else {
4293 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from If");
4294 return 1;
4295 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004296 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004297 int res;
4298 Py_ssize_t len;
4299 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004300 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004301 if (tmp == NULL) goto failed;
4302 if (!PyList_Check(tmp)) {
4303 PyErr_Format(PyExc_TypeError, "If field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4304 goto failed;
4305 }
4306 len = PyList_GET_SIZE(tmp);
4307 orelse = asdl_seq_new(len, arena);
4308 if (orelse == NULL) goto failed;
4309 for (i = 0; i < len; i++) {
4310 stmt_ty value;
4311 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4312 if (res != 0) goto failed;
4313 asdl_seq_SET(orelse, i, value);
4314 }
4315 Py_XDECREF(tmp);
4316 tmp = NULL;
4317 } else {
4318 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from If");
4319 return 1;
4320 }
4321 *out = If(test, body, orelse, lineno, col_offset, arena);
4322 if (*out == NULL) goto failed;
4323 return 0;
4324 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004325 isinstance = PyObject_IsInstance(obj, (PyObject*)With_type);
4326 if (isinstance == -1) {
4327 return 1;
4328 }
4329 if (isinstance) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004330 asdl_seq* items;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004331 asdl_seq* body;
4332
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004333 if (_PyObject_HasAttrId(obj, &PyId_items)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004334 int res;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004335 Py_ssize_t len;
4336 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004337 tmp = _PyObject_GetAttrId(obj, &PyId_items);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004338 if (tmp == NULL) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004339 if (!PyList_Check(tmp)) {
4340 PyErr_Format(PyExc_TypeError, "With field \"items\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4341 goto failed;
4342 }
4343 len = PyList_GET_SIZE(tmp);
4344 items = asdl_seq_new(len, arena);
4345 if (items == NULL) goto failed;
4346 for (i = 0; i < len; i++) {
4347 withitem_ty value;
4348 res = obj2ast_withitem(PyList_GET_ITEM(tmp, i), &value, arena);
4349 if (res != 0) goto failed;
4350 asdl_seq_SET(items, i, value);
4351 }
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004352 Py_XDECREF(tmp);
4353 tmp = NULL;
4354 } else {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004355 PyErr_SetString(PyExc_TypeError, "required field \"items\" missing from With");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004356 return 1;
4357 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004358 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004359 int res;
4360 Py_ssize_t len;
4361 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004362 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004363 if (tmp == NULL) goto failed;
4364 if (!PyList_Check(tmp)) {
4365 PyErr_Format(PyExc_TypeError, "With field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4366 goto failed;
4367 }
4368 len = PyList_GET_SIZE(tmp);
4369 body = asdl_seq_new(len, arena);
4370 if (body == NULL) goto failed;
4371 for (i = 0; i < len; i++) {
4372 stmt_ty value;
4373 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4374 if (res != 0) goto failed;
4375 asdl_seq_SET(body, i, value);
4376 }
4377 Py_XDECREF(tmp);
4378 tmp = NULL;
4379 } else {
4380 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from With");
4381 return 1;
4382 }
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004383 *out = With(items, body, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004384 if (*out == NULL) goto failed;
4385 return 0;
4386 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004387 isinstance = PyObject_IsInstance(obj, (PyObject*)Raise_type);
4388 if (isinstance == -1) {
4389 return 1;
4390 }
4391 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004392 expr_ty exc;
4393 expr_ty cause;
4394
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004395 if (_PyObject_HasAttrId(obj, &PyId_exc)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004396 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004397 tmp = _PyObject_GetAttrId(obj, &PyId_exc);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004398 if (tmp == NULL) goto failed;
4399 res = obj2ast_expr(tmp, &exc, arena);
4400 if (res != 0) goto failed;
4401 Py_XDECREF(tmp);
4402 tmp = NULL;
4403 } else {
4404 exc = NULL;
4405 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004406 if (_PyObject_HasAttrId(obj, &PyId_cause)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004407 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004408 tmp = _PyObject_GetAttrId(obj, &PyId_cause);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004409 if (tmp == NULL) goto failed;
4410 res = obj2ast_expr(tmp, &cause, arena);
4411 if (res != 0) goto failed;
4412 Py_XDECREF(tmp);
4413 tmp = NULL;
4414 } else {
4415 cause = NULL;
4416 }
4417 *out = Raise(exc, cause, lineno, col_offset, arena);
4418 if (*out == NULL) goto failed;
4419 return 0;
4420 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004421 isinstance = PyObject_IsInstance(obj, (PyObject*)Try_type);
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004422 if (isinstance == -1) {
4423 return 1;
4424 }
4425 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004426 asdl_seq* body;
4427 asdl_seq* handlers;
4428 asdl_seq* orelse;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004429 asdl_seq* finalbody;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004430
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004431 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004432 int res;
4433 Py_ssize_t len;
4434 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004435 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004436 if (tmp == NULL) goto failed;
4437 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004438 PyErr_Format(PyExc_TypeError, "Try field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004439 goto failed;
4440 }
4441 len = PyList_GET_SIZE(tmp);
4442 body = asdl_seq_new(len, arena);
4443 if (body == NULL) goto failed;
4444 for (i = 0; i < len; i++) {
4445 stmt_ty value;
4446 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4447 if (res != 0) goto failed;
4448 asdl_seq_SET(body, i, value);
4449 }
4450 Py_XDECREF(tmp);
4451 tmp = NULL;
4452 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004453 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004454 return 1;
4455 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004456 if (_PyObject_HasAttrId(obj, &PyId_handlers)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004457 int res;
4458 Py_ssize_t len;
4459 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004460 tmp = _PyObject_GetAttrId(obj, &PyId_handlers);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004461 if (tmp == NULL) goto failed;
4462 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004463 PyErr_Format(PyExc_TypeError, "Try field \"handlers\" must be a list, not a %.200s", tmp->ob_type->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004464 goto failed;
4465 }
4466 len = PyList_GET_SIZE(tmp);
4467 handlers = asdl_seq_new(len, arena);
4468 if (handlers == NULL) goto failed;
4469 for (i = 0; i < len; i++) {
4470 excepthandler_ty value;
4471 res = obj2ast_excepthandler(PyList_GET_ITEM(tmp, i), &value, arena);
4472 if (res != 0) goto failed;
4473 asdl_seq_SET(handlers, i, value);
4474 }
4475 Py_XDECREF(tmp);
4476 tmp = NULL;
4477 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004478 PyErr_SetString(PyExc_TypeError, "required field \"handlers\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004479 return 1;
4480 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004481 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004482 int res;
4483 Py_ssize_t len;
4484 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004485 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004486 if (tmp == NULL) goto failed;
4487 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004488 PyErr_Format(PyExc_TypeError, "Try field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004489 goto failed;
4490 }
4491 len = PyList_GET_SIZE(tmp);
4492 orelse = asdl_seq_new(len, arena);
4493 if (orelse == NULL) goto failed;
4494 for (i = 0; i < len; i++) {
4495 stmt_ty value;
4496 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4497 if (res != 0) goto failed;
4498 asdl_seq_SET(orelse, i, value);
4499 }
4500 Py_XDECREF(tmp);
4501 tmp = NULL;
4502 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004503 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004504 return 1;
4505 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004506 if (_PyObject_HasAttrId(obj, &PyId_finalbody)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004507 int res;
4508 Py_ssize_t len;
4509 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004510 tmp = _PyObject_GetAttrId(obj, &PyId_finalbody);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004511 if (tmp == NULL) goto failed;
4512 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004513 PyErr_Format(PyExc_TypeError, "Try field \"finalbody\" must be a list, not a %.200s", tmp->ob_type->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004514 goto failed;
4515 }
4516 len = PyList_GET_SIZE(tmp);
4517 finalbody = asdl_seq_new(len, arena);
4518 if (finalbody == NULL) goto failed;
4519 for (i = 0; i < len; i++) {
4520 stmt_ty value;
4521 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4522 if (res != 0) goto failed;
4523 asdl_seq_SET(finalbody, i, value);
4524 }
4525 Py_XDECREF(tmp);
4526 tmp = NULL;
4527 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004528 PyErr_SetString(PyExc_TypeError, "required field \"finalbody\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004529 return 1;
4530 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004531 *out = Try(body, handlers, orelse, finalbody, lineno,
4532 col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004533 if (*out == NULL) goto failed;
4534 return 0;
4535 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004536 isinstance = PyObject_IsInstance(obj, (PyObject*)Assert_type);
4537 if (isinstance == -1) {
4538 return 1;
4539 }
4540 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004541 expr_ty test;
4542 expr_ty msg;
4543
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004544 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004545 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004546 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004547 if (tmp == NULL) goto failed;
4548 res = obj2ast_expr(tmp, &test, arena);
4549 if (res != 0) goto failed;
4550 Py_XDECREF(tmp);
4551 tmp = NULL;
4552 } else {
4553 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from Assert");
4554 return 1;
4555 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004556 if (_PyObject_HasAttrId(obj, &PyId_msg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004557 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004558 tmp = _PyObject_GetAttrId(obj, &PyId_msg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004559 if (tmp == NULL) goto failed;
4560 res = obj2ast_expr(tmp, &msg, arena);
4561 if (res != 0) goto failed;
4562 Py_XDECREF(tmp);
4563 tmp = NULL;
4564 } else {
4565 msg = NULL;
4566 }
4567 *out = Assert(test, msg, lineno, col_offset, arena);
4568 if (*out == NULL) goto failed;
4569 return 0;
4570 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004571 isinstance = PyObject_IsInstance(obj, (PyObject*)Import_type);
4572 if (isinstance == -1) {
4573 return 1;
4574 }
4575 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004576 asdl_seq* names;
4577
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004578 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004579 int res;
4580 Py_ssize_t len;
4581 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004582 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004583 if (tmp == NULL) goto failed;
4584 if (!PyList_Check(tmp)) {
4585 PyErr_Format(PyExc_TypeError, "Import field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4586 goto failed;
4587 }
4588 len = PyList_GET_SIZE(tmp);
4589 names = asdl_seq_new(len, arena);
4590 if (names == NULL) goto failed;
4591 for (i = 0; i < len; i++) {
4592 alias_ty value;
4593 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4594 if (res != 0) goto failed;
4595 asdl_seq_SET(names, i, value);
4596 }
4597 Py_XDECREF(tmp);
4598 tmp = NULL;
4599 } else {
4600 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Import");
4601 return 1;
4602 }
4603 *out = Import(names, lineno, col_offset, arena);
4604 if (*out == NULL) goto failed;
4605 return 0;
4606 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004607 isinstance = PyObject_IsInstance(obj, (PyObject*)ImportFrom_type);
4608 if (isinstance == -1) {
4609 return 1;
4610 }
4611 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004612 identifier module;
4613 asdl_seq* names;
4614 int level;
4615
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004616 if (_PyObject_HasAttrId(obj, &PyId_module)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004617 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004618 tmp = _PyObject_GetAttrId(obj, &PyId_module);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004619 if (tmp == NULL) goto failed;
4620 res = obj2ast_identifier(tmp, &module, arena);
4621 if (res != 0) goto failed;
4622 Py_XDECREF(tmp);
4623 tmp = NULL;
4624 } else {
Benjamin Peterson78565b22009-06-28 19:19:51 +00004625 module = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004626 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004627 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004628 int res;
4629 Py_ssize_t len;
4630 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004631 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004632 if (tmp == NULL) goto failed;
4633 if (!PyList_Check(tmp)) {
4634 PyErr_Format(PyExc_TypeError, "ImportFrom field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4635 goto failed;
4636 }
4637 len = PyList_GET_SIZE(tmp);
4638 names = asdl_seq_new(len, arena);
4639 if (names == NULL) goto failed;
4640 for (i = 0; i < len; i++) {
4641 alias_ty value;
4642 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4643 if (res != 0) goto failed;
4644 asdl_seq_SET(names, i, value);
4645 }
4646 Py_XDECREF(tmp);
4647 tmp = NULL;
4648 } else {
4649 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from ImportFrom");
4650 return 1;
4651 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004652 if (_PyObject_HasAttrId(obj, &PyId_level)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004653 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004654 tmp = _PyObject_GetAttrId(obj, &PyId_level);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004655 if (tmp == NULL) goto failed;
4656 res = obj2ast_int(tmp, &level, arena);
4657 if (res != 0) goto failed;
4658 Py_XDECREF(tmp);
4659 tmp = NULL;
4660 } else {
4661 level = 0;
4662 }
4663 *out = ImportFrom(module, names, level, lineno, col_offset,
4664 arena);
4665 if (*out == NULL) goto failed;
4666 return 0;
4667 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004668 isinstance = PyObject_IsInstance(obj, (PyObject*)Global_type);
4669 if (isinstance == -1) {
4670 return 1;
4671 }
4672 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004673 asdl_seq* names;
4674
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004675 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004676 int res;
4677 Py_ssize_t len;
4678 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004679 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004680 if (tmp == NULL) goto failed;
4681 if (!PyList_Check(tmp)) {
4682 PyErr_Format(PyExc_TypeError, "Global field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4683 goto failed;
4684 }
4685 len = PyList_GET_SIZE(tmp);
4686 names = asdl_seq_new(len, arena);
4687 if (names == NULL) goto failed;
4688 for (i = 0; i < len; i++) {
4689 identifier value;
4690 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4691 if (res != 0) goto failed;
4692 asdl_seq_SET(names, i, value);
4693 }
4694 Py_XDECREF(tmp);
4695 tmp = NULL;
4696 } else {
4697 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Global");
4698 return 1;
4699 }
4700 *out = Global(names, lineno, col_offset, arena);
4701 if (*out == NULL) goto failed;
4702 return 0;
4703 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004704 isinstance = PyObject_IsInstance(obj, (PyObject*)Nonlocal_type);
4705 if (isinstance == -1) {
4706 return 1;
4707 }
4708 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004709 asdl_seq* names;
4710
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004711 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004712 int res;
4713 Py_ssize_t len;
4714 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004715 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004716 if (tmp == NULL) goto failed;
4717 if (!PyList_Check(tmp)) {
4718 PyErr_Format(PyExc_TypeError, "Nonlocal field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4719 goto failed;
4720 }
4721 len = PyList_GET_SIZE(tmp);
4722 names = asdl_seq_new(len, arena);
4723 if (names == NULL) goto failed;
4724 for (i = 0; i < len; i++) {
4725 identifier value;
4726 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4727 if (res != 0) goto failed;
4728 asdl_seq_SET(names, i, value);
4729 }
4730 Py_XDECREF(tmp);
4731 tmp = NULL;
4732 } else {
4733 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Nonlocal");
4734 return 1;
4735 }
4736 *out = Nonlocal(names, lineno, col_offset, arena);
4737 if (*out == NULL) goto failed;
4738 return 0;
4739 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004740 isinstance = PyObject_IsInstance(obj, (PyObject*)Expr_type);
4741 if (isinstance == -1) {
4742 return 1;
4743 }
4744 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004745 expr_ty value;
4746
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004747 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004748 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004749 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004750 if (tmp == NULL) goto failed;
4751 res = obj2ast_expr(tmp, &value, arena);
4752 if (res != 0) goto failed;
4753 Py_XDECREF(tmp);
4754 tmp = NULL;
4755 } else {
4756 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Expr");
4757 return 1;
4758 }
4759 *out = Expr(value, lineno, col_offset, arena);
4760 if (*out == NULL) goto failed;
4761 return 0;
4762 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004763 isinstance = PyObject_IsInstance(obj, (PyObject*)Pass_type);
4764 if (isinstance == -1) {
4765 return 1;
4766 }
4767 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004768
4769 *out = Pass(lineno, col_offset, arena);
4770 if (*out == NULL) goto failed;
4771 return 0;
4772 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004773 isinstance = PyObject_IsInstance(obj, (PyObject*)Break_type);
4774 if (isinstance == -1) {
4775 return 1;
4776 }
4777 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004778
4779 *out = Break(lineno, col_offset, arena);
4780 if (*out == NULL) goto failed;
4781 return 0;
4782 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004783 isinstance = PyObject_IsInstance(obj, (PyObject*)Continue_type);
4784 if (isinstance == -1) {
4785 return 1;
4786 }
4787 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004788
4789 *out = Continue(lineno, col_offset, arena);
4790 if (*out == NULL) goto failed;
4791 return 0;
4792 }
4793
Benjamin Peterson5b066812010-11-20 01:38:49 +00004794 PyErr_Format(PyExc_TypeError, "expected some sort of stmt, but got %R", obj);
4795 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00004796 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004797 return 1;
4798}
4799
4800int
4801obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena)
4802{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004803 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004804
Benjamin Petersond8f65972010-11-20 04:31:07 +00004805 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004806 int lineno;
4807 int col_offset;
4808
4809 if (obj == Py_None) {
4810 *out = NULL;
4811 return 0;
4812 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004813 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004814 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004815 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004816 if (tmp == NULL) goto failed;
4817 res = obj2ast_int(tmp, &lineno, arena);
4818 if (res != 0) goto failed;
4819 Py_XDECREF(tmp);
4820 tmp = NULL;
4821 } else {
4822 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from expr");
4823 return 1;
4824 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004825 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004826 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004827 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004828 if (tmp == NULL) goto failed;
4829 res = obj2ast_int(tmp, &col_offset, arena);
4830 if (res != 0) goto failed;
4831 Py_XDECREF(tmp);
4832 tmp = NULL;
4833 } else {
4834 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from expr");
4835 return 1;
4836 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004837 isinstance = PyObject_IsInstance(obj, (PyObject*)BoolOp_type);
4838 if (isinstance == -1) {
4839 return 1;
4840 }
4841 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004842 boolop_ty op;
4843 asdl_seq* values;
4844
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004845 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004846 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004847 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004848 if (tmp == NULL) goto failed;
4849 res = obj2ast_boolop(tmp, &op, arena);
4850 if (res != 0) goto failed;
4851 Py_XDECREF(tmp);
4852 tmp = NULL;
4853 } else {
4854 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BoolOp");
4855 return 1;
4856 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004857 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004858 int res;
4859 Py_ssize_t len;
4860 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004861 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004862 if (tmp == NULL) goto failed;
4863 if (!PyList_Check(tmp)) {
4864 PyErr_Format(PyExc_TypeError, "BoolOp field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4865 goto failed;
4866 }
4867 len = PyList_GET_SIZE(tmp);
4868 values = asdl_seq_new(len, arena);
4869 if (values == NULL) goto failed;
4870 for (i = 0; i < len; i++) {
4871 expr_ty value;
4872 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4873 if (res != 0) goto failed;
4874 asdl_seq_SET(values, i, value);
4875 }
4876 Py_XDECREF(tmp);
4877 tmp = NULL;
4878 } else {
4879 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from BoolOp");
4880 return 1;
4881 }
4882 *out = BoolOp(op, values, lineno, col_offset, arena);
4883 if (*out == NULL) goto failed;
4884 return 0;
4885 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004886 isinstance = PyObject_IsInstance(obj, (PyObject*)BinOp_type);
4887 if (isinstance == -1) {
4888 return 1;
4889 }
4890 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004891 expr_ty left;
4892 operator_ty op;
4893 expr_ty right;
4894
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004895 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004896 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004897 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004898 if (tmp == NULL) goto failed;
4899 res = obj2ast_expr(tmp, &left, arena);
4900 if (res != 0) goto failed;
4901 Py_XDECREF(tmp);
4902 tmp = NULL;
4903 } else {
4904 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from BinOp");
4905 return 1;
4906 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004907 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004908 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004909 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004910 if (tmp == NULL) goto failed;
4911 res = obj2ast_operator(tmp, &op, arena);
4912 if (res != 0) goto failed;
4913 Py_XDECREF(tmp);
4914 tmp = NULL;
4915 } else {
4916 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BinOp");
4917 return 1;
4918 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004919 if (_PyObject_HasAttrId(obj, &PyId_right)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004920 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004921 tmp = _PyObject_GetAttrId(obj, &PyId_right);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004922 if (tmp == NULL) goto failed;
4923 res = obj2ast_expr(tmp, &right, arena);
4924 if (res != 0) goto failed;
4925 Py_XDECREF(tmp);
4926 tmp = NULL;
4927 } else {
4928 PyErr_SetString(PyExc_TypeError, "required field \"right\" missing from BinOp");
4929 return 1;
4930 }
4931 *out = BinOp(left, op, right, lineno, col_offset, arena);
4932 if (*out == NULL) goto failed;
4933 return 0;
4934 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004935 isinstance = PyObject_IsInstance(obj, (PyObject*)UnaryOp_type);
4936 if (isinstance == -1) {
4937 return 1;
4938 }
4939 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004940 unaryop_ty op;
4941 expr_ty operand;
4942
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004943 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004944 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004945 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004946 if (tmp == NULL) goto failed;
4947 res = obj2ast_unaryop(tmp, &op, arena);
4948 if (res != 0) goto failed;
4949 Py_XDECREF(tmp);
4950 tmp = NULL;
4951 } else {
4952 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from UnaryOp");
4953 return 1;
4954 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004955 if (_PyObject_HasAttrId(obj, &PyId_operand)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004956 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004957 tmp = _PyObject_GetAttrId(obj, &PyId_operand);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004958 if (tmp == NULL) goto failed;
4959 res = obj2ast_expr(tmp, &operand, arena);
4960 if (res != 0) goto failed;
4961 Py_XDECREF(tmp);
4962 tmp = NULL;
4963 } else {
4964 PyErr_SetString(PyExc_TypeError, "required field \"operand\" missing from UnaryOp");
4965 return 1;
4966 }
4967 *out = UnaryOp(op, operand, lineno, col_offset, arena);
4968 if (*out == NULL) goto failed;
4969 return 0;
4970 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004971 isinstance = PyObject_IsInstance(obj, (PyObject*)Lambda_type);
4972 if (isinstance == -1) {
4973 return 1;
4974 }
4975 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004976 arguments_ty args;
4977 expr_ty body;
4978
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004979 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004980 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004981 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004982 if (tmp == NULL) goto failed;
4983 res = obj2ast_arguments(tmp, &args, arena);
4984 if (res != 0) goto failed;
4985 Py_XDECREF(tmp);
4986 tmp = NULL;
4987 } else {
4988 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Lambda");
4989 return 1;
4990 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004991 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004992 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004993 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004994 if (tmp == NULL) goto failed;
4995 res = obj2ast_expr(tmp, &body, arena);
4996 if (res != 0) goto failed;
4997 Py_XDECREF(tmp);
4998 tmp = NULL;
4999 } else {
5000 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Lambda");
5001 return 1;
5002 }
5003 *out = Lambda(args, body, lineno, col_offset, arena);
5004 if (*out == NULL) goto failed;
5005 return 0;
5006 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005007 isinstance = PyObject_IsInstance(obj, (PyObject*)IfExp_type);
5008 if (isinstance == -1) {
5009 return 1;
5010 }
5011 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005012 expr_ty test;
5013 expr_ty body;
5014 expr_ty orelse;
5015
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005016 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005017 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005018 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005019 if (tmp == NULL) goto failed;
5020 res = obj2ast_expr(tmp, &test, arena);
5021 if (res != 0) goto failed;
5022 Py_XDECREF(tmp);
5023 tmp = NULL;
5024 } else {
5025 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from IfExp");
5026 return 1;
5027 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005028 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005029 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005030 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005031 if (tmp == NULL) goto failed;
5032 res = obj2ast_expr(tmp, &body, arena);
5033 if (res != 0) goto failed;
5034 Py_XDECREF(tmp);
5035 tmp = NULL;
5036 } else {
5037 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from IfExp");
5038 return 1;
5039 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005040 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005041 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005042 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005043 if (tmp == NULL) goto failed;
5044 res = obj2ast_expr(tmp, &orelse, arena);
5045 if (res != 0) goto failed;
5046 Py_XDECREF(tmp);
5047 tmp = NULL;
5048 } else {
5049 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from IfExp");
5050 return 1;
5051 }
5052 *out = IfExp(test, body, orelse, lineno, col_offset, arena);
5053 if (*out == NULL) goto failed;
5054 return 0;
5055 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005056 isinstance = PyObject_IsInstance(obj, (PyObject*)Dict_type);
5057 if (isinstance == -1) {
5058 return 1;
5059 }
5060 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005061 asdl_seq* keys;
5062 asdl_seq* values;
5063
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005064 if (_PyObject_HasAttrId(obj, &PyId_keys)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005065 int res;
5066 Py_ssize_t len;
5067 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005068 tmp = _PyObject_GetAttrId(obj, &PyId_keys);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005069 if (tmp == NULL) goto failed;
5070 if (!PyList_Check(tmp)) {
5071 PyErr_Format(PyExc_TypeError, "Dict field \"keys\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5072 goto failed;
5073 }
5074 len = PyList_GET_SIZE(tmp);
5075 keys = asdl_seq_new(len, arena);
5076 if (keys == NULL) goto failed;
5077 for (i = 0; i < len; i++) {
5078 expr_ty value;
5079 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5080 if (res != 0) goto failed;
5081 asdl_seq_SET(keys, i, value);
5082 }
5083 Py_XDECREF(tmp);
5084 tmp = NULL;
5085 } else {
5086 PyErr_SetString(PyExc_TypeError, "required field \"keys\" missing from Dict");
5087 return 1;
5088 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005089 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005090 int res;
5091 Py_ssize_t len;
5092 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005093 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005094 if (tmp == NULL) goto failed;
5095 if (!PyList_Check(tmp)) {
5096 PyErr_Format(PyExc_TypeError, "Dict field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5097 goto failed;
5098 }
5099 len = PyList_GET_SIZE(tmp);
5100 values = asdl_seq_new(len, arena);
5101 if (values == NULL) goto failed;
5102 for (i = 0; i < len; i++) {
5103 expr_ty value;
5104 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5105 if (res != 0) goto failed;
5106 asdl_seq_SET(values, i, value);
5107 }
5108 Py_XDECREF(tmp);
5109 tmp = NULL;
5110 } else {
5111 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from Dict");
5112 return 1;
5113 }
5114 *out = Dict(keys, values, lineno, col_offset, arena);
5115 if (*out == NULL) goto failed;
5116 return 0;
5117 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005118 isinstance = PyObject_IsInstance(obj, (PyObject*)Set_type);
5119 if (isinstance == -1) {
5120 return 1;
5121 }
5122 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005123 asdl_seq* elts;
5124
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005125 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005126 int res;
5127 Py_ssize_t len;
5128 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005129 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005130 if (tmp == NULL) goto failed;
5131 if (!PyList_Check(tmp)) {
5132 PyErr_Format(PyExc_TypeError, "Set field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5133 goto failed;
5134 }
5135 len = PyList_GET_SIZE(tmp);
5136 elts = asdl_seq_new(len, arena);
5137 if (elts == NULL) goto failed;
5138 for (i = 0; i < len; i++) {
5139 expr_ty value;
5140 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5141 if (res != 0) goto failed;
5142 asdl_seq_SET(elts, i, value);
5143 }
5144 Py_XDECREF(tmp);
5145 tmp = NULL;
5146 } else {
5147 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Set");
5148 return 1;
5149 }
5150 *out = Set(elts, lineno, col_offset, arena);
5151 if (*out == NULL) goto failed;
5152 return 0;
5153 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005154 isinstance = PyObject_IsInstance(obj, (PyObject*)ListComp_type);
5155 if (isinstance == -1) {
5156 return 1;
5157 }
5158 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005159 expr_ty elt;
5160 asdl_seq* generators;
5161
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005162 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005163 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005164 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005165 if (tmp == NULL) goto failed;
5166 res = obj2ast_expr(tmp, &elt, arena);
5167 if (res != 0) goto failed;
5168 Py_XDECREF(tmp);
5169 tmp = NULL;
5170 } else {
5171 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from ListComp");
5172 return 1;
5173 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005174 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005175 int res;
5176 Py_ssize_t len;
5177 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005178 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005179 if (tmp == NULL) goto failed;
5180 if (!PyList_Check(tmp)) {
5181 PyErr_Format(PyExc_TypeError, "ListComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5182 goto failed;
5183 }
5184 len = PyList_GET_SIZE(tmp);
5185 generators = asdl_seq_new(len, arena);
5186 if (generators == NULL) goto failed;
5187 for (i = 0; i < len; i++) {
5188 comprehension_ty value;
5189 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5190 if (res != 0) goto failed;
5191 asdl_seq_SET(generators, i, value);
5192 }
5193 Py_XDECREF(tmp);
5194 tmp = NULL;
5195 } else {
5196 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from ListComp");
5197 return 1;
5198 }
5199 *out = ListComp(elt, generators, lineno, col_offset, arena);
5200 if (*out == NULL) goto failed;
5201 return 0;
5202 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005203 isinstance = PyObject_IsInstance(obj, (PyObject*)SetComp_type);
5204 if (isinstance == -1) {
5205 return 1;
5206 }
5207 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005208 expr_ty elt;
5209 asdl_seq* generators;
5210
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005211 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005212 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005213 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005214 if (tmp == NULL) goto failed;
5215 res = obj2ast_expr(tmp, &elt, arena);
5216 if (res != 0) goto failed;
5217 Py_XDECREF(tmp);
5218 tmp = NULL;
5219 } else {
5220 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from SetComp");
5221 return 1;
5222 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005223 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005224 int res;
5225 Py_ssize_t len;
5226 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005227 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005228 if (tmp == NULL) goto failed;
5229 if (!PyList_Check(tmp)) {
5230 PyErr_Format(PyExc_TypeError, "SetComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5231 goto failed;
5232 }
5233 len = PyList_GET_SIZE(tmp);
5234 generators = asdl_seq_new(len, arena);
5235 if (generators == NULL) goto failed;
5236 for (i = 0; i < len; i++) {
5237 comprehension_ty value;
5238 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5239 if (res != 0) goto failed;
5240 asdl_seq_SET(generators, i, value);
5241 }
5242 Py_XDECREF(tmp);
5243 tmp = NULL;
5244 } else {
5245 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from SetComp");
5246 return 1;
5247 }
5248 *out = SetComp(elt, generators, lineno, col_offset, arena);
5249 if (*out == NULL) goto failed;
5250 return 0;
5251 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005252 isinstance = PyObject_IsInstance(obj, (PyObject*)DictComp_type);
5253 if (isinstance == -1) {
5254 return 1;
5255 }
5256 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005257 expr_ty key;
5258 expr_ty value;
5259 asdl_seq* generators;
5260
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005261 if (_PyObject_HasAttrId(obj, &PyId_key)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005262 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005263 tmp = _PyObject_GetAttrId(obj, &PyId_key);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005264 if (tmp == NULL) goto failed;
5265 res = obj2ast_expr(tmp, &key, arena);
5266 if (res != 0) goto failed;
5267 Py_XDECREF(tmp);
5268 tmp = NULL;
5269 } else {
5270 PyErr_SetString(PyExc_TypeError, "required field \"key\" missing from DictComp");
5271 return 1;
5272 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005273 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005274 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005275 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005276 if (tmp == NULL) goto failed;
5277 res = obj2ast_expr(tmp, &value, arena);
5278 if (res != 0) goto failed;
5279 Py_XDECREF(tmp);
5280 tmp = NULL;
5281 } else {
5282 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from DictComp");
5283 return 1;
5284 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005285 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005286 int res;
5287 Py_ssize_t len;
5288 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005289 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005290 if (tmp == NULL) goto failed;
5291 if (!PyList_Check(tmp)) {
5292 PyErr_Format(PyExc_TypeError, "DictComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5293 goto failed;
5294 }
5295 len = PyList_GET_SIZE(tmp);
5296 generators = asdl_seq_new(len, arena);
5297 if (generators == NULL) goto failed;
5298 for (i = 0; i < len; i++) {
5299 comprehension_ty value;
5300 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5301 if (res != 0) goto failed;
5302 asdl_seq_SET(generators, i, value);
5303 }
5304 Py_XDECREF(tmp);
5305 tmp = NULL;
5306 } else {
5307 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from DictComp");
5308 return 1;
5309 }
5310 *out = DictComp(key, value, generators, lineno, col_offset,
5311 arena);
5312 if (*out == NULL) goto failed;
5313 return 0;
5314 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005315 isinstance = PyObject_IsInstance(obj, (PyObject*)GeneratorExp_type);
5316 if (isinstance == -1) {
5317 return 1;
5318 }
5319 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005320 expr_ty elt;
5321 asdl_seq* generators;
5322
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005323 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005324 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005325 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005326 if (tmp == NULL) goto failed;
5327 res = obj2ast_expr(tmp, &elt, arena);
5328 if (res != 0) goto failed;
5329 Py_XDECREF(tmp);
5330 tmp = NULL;
5331 } else {
5332 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from GeneratorExp");
5333 return 1;
5334 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005335 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005336 int res;
5337 Py_ssize_t len;
5338 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005339 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005340 if (tmp == NULL) goto failed;
5341 if (!PyList_Check(tmp)) {
5342 PyErr_Format(PyExc_TypeError, "GeneratorExp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5343 goto failed;
5344 }
5345 len = PyList_GET_SIZE(tmp);
5346 generators = asdl_seq_new(len, arena);
5347 if (generators == NULL) goto failed;
5348 for (i = 0; i < len; i++) {
5349 comprehension_ty value;
5350 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5351 if (res != 0) goto failed;
5352 asdl_seq_SET(generators, i, value);
5353 }
5354 Py_XDECREF(tmp);
5355 tmp = NULL;
5356 } else {
5357 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from GeneratorExp");
5358 return 1;
5359 }
5360 *out = GeneratorExp(elt, generators, lineno, col_offset, arena);
5361 if (*out == NULL) goto failed;
5362 return 0;
5363 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005364 isinstance = PyObject_IsInstance(obj, (PyObject*)Yield_type);
5365 if (isinstance == -1) {
5366 return 1;
5367 }
5368 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005369 expr_ty value;
5370
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005371 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005372 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005373 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005374 if (tmp == NULL) goto failed;
5375 res = obj2ast_expr(tmp, &value, arena);
5376 if (res != 0) goto failed;
5377 Py_XDECREF(tmp);
5378 tmp = NULL;
5379 } else {
5380 value = NULL;
5381 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05005382 *out = Yield(value, lineno, col_offset, arena);
5383 if (*out == NULL) goto failed;
5384 return 0;
5385 }
5386 isinstance = PyObject_IsInstance(obj, (PyObject*)YieldFrom_type);
5387 if (isinstance == -1) {
5388 return 1;
5389 }
5390 if (isinstance) {
5391 expr_ty value;
5392
5393 if (_PyObject_HasAttrId(obj, &PyId_value)) {
5394 int res;
5395 tmp = _PyObject_GetAttrId(obj, &PyId_value);
5396 if (tmp == NULL) goto failed;
5397 res = obj2ast_expr(tmp, &value, arena);
5398 if (res != 0) goto failed;
5399 Py_XDECREF(tmp);
5400 tmp = NULL;
5401 } else {
5402 value = NULL;
5403 }
5404 *out = YieldFrom(value, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005405 if (*out == NULL) goto failed;
5406 return 0;
5407 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005408 isinstance = PyObject_IsInstance(obj, (PyObject*)Compare_type);
5409 if (isinstance == -1) {
5410 return 1;
5411 }
5412 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005413 expr_ty left;
5414 asdl_int_seq* ops;
5415 asdl_seq* comparators;
5416
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005417 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005418 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005419 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005420 if (tmp == NULL) goto failed;
5421 res = obj2ast_expr(tmp, &left, arena);
5422 if (res != 0) goto failed;
5423 Py_XDECREF(tmp);
5424 tmp = NULL;
5425 } else {
5426 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from Compare");
5427 return 1;
5428 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005429 if (_PyObject_HasAttrId(obj, &PyId_ops)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005430 int res;
5431 Py_ssize_t len;
5432 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005433 tmp = _PyObject_GetAttrId(obj, &PyId_ops);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005434 if (tmp == NULL) goto failed;
5435 if (!PyList_Check(tmp)) {
5436 PyErr_Format(PyExc_TypeError, "Compare field \"ops\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5437 goto failed;
5438 }
5439 len = PyList_GET_SIZE(tmp);
5440 ops = asdl_int_seq_new(len, arena);
5441 if (ops == NULL) goto failed;
5442 for (i = 0; i < len; i++) {
5443 cmpop_ty value;
5444 res = obj2ast_cmpop(PyList_GET_ITEM(tmp, i), &value, arena);
5445 if (res != 0) goto failed;
5446 asdl_seq_SET(ops, i, value);
5447 }
5448 Py_XDECREF(tmp);
5449 tmp = NULL;
5450 } else {
5451 PyErr_SetString(PyExc_TypeError, "required field \"ops\" missing from Compare");
5452 return 1;
5453 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005454 if (_PyObject_HasAttrId(obj, &PyId_comparators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005455 int res;
5456 Py_ssize_t len;
5457 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005458 tmp = _PyObject_GetAttrId(obj, &PyId_comparators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005459 if (tmp == NULL) goto failed;
5460 if (!PyList_Check(tmp)) {
5461 PyErr_Format(PyExc_TypeError, "Compare field \"comparators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5462 goto failed;
5463 }
5464 len = PyList_GET_SIZE(tmp);
5465 comparators = asdl_seq_new(len, arena);
5466 if (comparators == NULL) goto failed;
5467 for (i = 0; i < len; i++) {
5468 expr_ty value;
5469 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5470 if (res != 0) goto failed;
5471 asdl_seq_SET(comparators, i, value);
5472 }
5473 Py_XDECREF(tmp);
5474 tmp = NULL;
5475 } else {
5476 PyErr_SetString(PyExc_TypeError, "required field \"comparators\" missing from Compare");
5477 return 1;
5478 }
5479 *out = Compare(left, ops, comparators, lineno, col_offset,
5480 arena);
5481 if (*out == NULL) goto failed;
5482 return 0;
5483 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005484 isinstance = PyObject_IsInstance(obj, (PyObject*)Call_type);
5485 if (isinstance == -1) {
5486 return 1;
5487 }
5488 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005489 expr_ty func;
5490 asdl_seq* args;
5491 asdl_seq* keywords;
5492 expr_ty starargs;
5493 expr_ty kwargs;
5494
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005495 if (_PyObject_HasAttrId(obj, &PyId_func)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005496 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005497 tmp = _PyObject_GetAttrId(obj, &PyId_func);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005498 if (tmp == NULL) goto failed;
5499 res = obj2ast_expr(tmp, &func, arena);
5500 if (res != 0) goto failed;
5501 Py_XDECREF(tmp);
5502 tmp = NULL;
5503 } else {
5504 PyErr_SetString(PyExc_TypeError, "required field \"func\" missing from Call");
5505 return 1;
5506 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005507 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005508 int res;
5509 Py_ssize_t len;
5510 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005511 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005512 if (tmp == NULL) goto failed;
5513 if (!PyList_Check(tmp)) {
5514 PyErr_Format(PyExc_TypeError, "Call field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5515 goto failed;
5516 }
5517 len = PyList_GET_SIZE(tmp);
5518 args = asdl_seq_new(len, arena);
5519 if (args == NULL) goto failed;
5520 for (i = 0; i < len; i++) {
5521 expr_ty value;
5522 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5523 if (res != 0) goto failed;
5524 asdl_seq_SET(args, i, value);
5525 }
5526 Py_XDECREF(tmp);
5527 tmp = NULL;
5528 } else {
5529 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Call");
5530 return 1;
5531 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005532 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005533 int res;
5534 Py_ssize_t len;
5535 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005536 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005537 if (tmp == NULL) goto failed;
5538 if (!PyList_Check(tmp)) {
5539 PyErr_Format(PyExc_TypeError, "Call field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5540 goto failed;
5541 }
5542 len = PyList_GET_SIZE(tmp);
5543 keywords = asdl_seq_new(len, arena);
5544 if (keywords == NULL) goto failed;
5545 for (i = 0; i < len; i++) {
5546 keyword_ty value;
5547 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
5548 if (res != 0) goto failed;
5549 asdl_seq_SET(keywords, i, value);
5550 }
5551 Py_XDECREF(tmp);
5552 tmp = NULL;
5553 } else {
5554 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from Call");
5555 return 1;
5556 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005557 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005558 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005559 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005560 if (tmp == NULL) goto failed;
5561 res = obj2ast_expr(tmp, &starargs, arena);
5562 if (res != 0) goto failed;
5563 Py_XDECREF(tmp);
5564 tmp = NULL;
5565 } else {
5566 starargs = NULL;
5567 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005568 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005569 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005570 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005571 if (tmp == NULL) goto failed;
5572 res = obj2ast_expr(tmp, &kwargs, arena);
5573 if (res != 0) goto failed;
5574 Py_XDECREF(tmp);
5575 tmp = NULL;
5576 } else {
5577 kwargs = NULL;
5578 }
5579 *out = Call(func, args, keywords, starargs, kwargs, lineno,
5580 col_offset, arena);
5581 if (*out == NULL) goto failed;
5582 return 0;
5583 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005584 isinstance = PyObject_IsInstance(obj, (PyObject*)Num_type);
5585 if (isinstance == -1) {
5586 return 1;
5587 }
5588 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005589 object n;
5590
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005591 if (_PyObject_HasAttrId(obj, &PyId_n)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005592 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005593 tmp = _PyObject_GetAttrId(obj, &PyId_n);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005594 if (tmp == NULL) goto failed;
5595 res = obj2ast_object(tmp, &n, arena);
5596 if (res != 0) goto failed;
5597 Py_XDECREF(tmp);
5598 tmp = NULL;
5599 } else {
5600 PyErr_SetString(PyExc_TypeError, "required field \"n\" missing from Num");
5601 return 1;
5602 }
5603 *out = Num(n, lineno, col_offset, arena);
5604 if (*out == NULL) goto failed;
5605 return 0;
5606 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005607 isinstance = PyObject_IsInstance(obj, (PyObject*)Str_type);
5608 if (isinstance == -1) {
5609 return 1;
5610 }
5611 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005612 string s;
5613
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005614 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005615 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005616 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005617 if (tmp == NULL) goto failed;
5618 res = obj2ast_string(tmp, &s, arena);
5619 if (res != 0) goto failed;
5620 Py_XDECREF(tmp);
5621 tmp = NULL;
5622 } else {
5623 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Str");
5624 return 1;
5625 }
5626 *out = Str(s, lineno, col_offset, arena);
5627 if (*out == NULL) goto failed;
5628 return 0;
5629 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005630 isinstance = PyObject_IsInstance(obj, (PyObject*)Bytes_type);
5631 if (isinstance == -1) {
5632 return 1;
5633 }
5634 if (isinstance) {
Benjamin Petersone2498412011-08-09 16:08:39 -05005635 bytes s;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005636
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005637 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005638 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005639 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005640 if (tmp == NULL) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05005641 res = obj2ast_bytes(tmp, &s, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005642 if (res != 0) goto failed;
5643 Py_XDECREF(tmp);
5644 tmp = NULL;
5645 } else {
5646 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Bytes");
5647 return 1;
5648 }
5649 *out = Bytes(s, lineno, col_offset, arena);
5650 if (*out == NULL) goto failed;
5651 return 0;
5652 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005653 isinstance = PyObject_IsInstance(obj, (PyObject*)Ellipsis_type);
5654 if (isinstance == -1) {
5655 return 1;
5656 }
5657 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005658
5659 *out = Ellipsis(lineno, col_offset, arena);
5660 if (*out == NULL) goto failed;
5661 return 0;
5662 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005663 isinstance = PyObject_IsInstance(obj, (PyObject*)Attribute_type);
5664 if (isinstance == -1) {
5665 return 1;
5666 }
5667 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005668 expr_ty value;
5669 identifier attr;
5670 expr_context_ty ctx;
5671
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005672 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005673 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005674 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005675 if (tmp == NULL) goto failed;
5676 res = obj2ast_expr(tmp, &value, arena);
5677 if (res != 0) goto failed;
5678 Py_XDECREF(tmp);
5679 tmp = NULL;
5680 } else {
5681 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Attribute");
5682 return 1;
5683 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005684 if (_PyObject_HasAttrId(obj, &PyId_attr)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005685 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005686 tmp = _PyObject_GetAttrId(obj, &PyId_attr);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005687 if (tmp == NULL) goto failed;
5688 res = obj2ast_identifier(tmp, &attr, arena);
5689 if (res != 0) goto failed;
5690 Py_XDECREF(tmp);
5691 tmp = NULL;
5692 } else {
5693 PyErr_SetString(PyExc_TypeError, "required field \"attr\" missing from Attribute");
5694 return 1;
5695 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005696 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005697 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005698 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005699 if (tmp == NULL) goto failed;
5700 res = obj2ast_expr_context(tmp, &ctx, arena);
5701 if (res != 0) goto failed;
5702 Py_XDECREF(tmp);
5703 tmp = NULL;
5704 } else {
5705 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Attribute");
5706 return 1;
5707 }
5708 *out = Attribute(value, attr, ctx, lineno, col_offset, arena);
5709 if (*out == NULL) goto failed;
5710 return 0;
5711 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005712 isinstance = PyObject_IsInstance(obj, (PyObject*)Subscript_type);
5713 if (isinstance == -1) {
5714 return 1;
5715 }
5716 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005717 expr_ty value;
5718 slice_ty slice;
5719 expr_context_ty ctx;
5720
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005721 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005722 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005723 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005724 if (tmp == NULL) goto failed;
5725 res = obj2ast_expr(tmp, &value, arena);
5726 if (res != 0) goto failed;
5727 Py_XDECREF(tmp);
5728 tmp = NULL;
5729 } else {
5730 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Subscript");
5731 return 1;
5732 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005733 if (_PyObject_HasAttrId(obj, &PyId_slice)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005734 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005735 tmp = _PyObject_GetAttrId(obj, &PyId_slice);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005736 if (tmp == NULL) goto failed;
5737 res = obj2ast_slice(tmp, &slice, arena);
5738 if (res != 0) goto failed;
5739 Py_XDECREF(tmp);
5740 tmp = NULL;
5741 } else {
5742 PyErr_SetString(PyExc_TypeError, "required field \"slice\" missing from Subscript");
5743 return 1;
5744 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005745 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005746 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005747 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005748 if (tmp == NULL) goto failed;
5749 res = obj2ast_expr_context(tmp, &ctx, arena);
5750 if (res != 0) goto failed;
5751 Py_XDECREF(tmp);
5752 tmp = NULL;
5753 } else {
5754 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Subscript");
5755 return 1;
5756 }
5757 *out = Subscript(value, slice, ctx, lineno, col_offset, arena);
5758 if (*out == NULL) goto failed;
5759 return 0;
5760 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005761 isinstance = PyObject_IsInstance(obj, (PyObject*)Starred_type);
5762 if (isinstance == -1) {
5763 return 1;
5764 }
5765 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005766 expr_ty value;
5767 expr_context_ty ctx;
5768
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005769 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005770 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005771 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005772 if (tmp == NULL) goto failed;
5773 res = obj2ast_expr(tmp, &value, arena);
5774 if (res != 0) goto failed;
5775 Py_XDECREF(tmp);
5776 tmp = NULL;
5777 } else {
5778 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Starred");
5779 return 1;
5780 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005781 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005782 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005783 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005784 if (tmp == NULL) goto failed;
5785 res = obj2ast_expr_context(tmp, &ctx, arena);
5786 if (res != 0) goto failed;
5787 Py_XDECREF(tmp);
5788 tmp = NULL;
5789 } else {
5790 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Starred");
5791 return 1;
5792 }
5793 *out = Starred(value, ctx, lineno, col_offset, arena);
5794 if (*out == NULL) goto failed;
5795 return 0;
5796 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005797 isinstance = PyObject_IsInstance(obj, (PyObject*)Name_type);
5798 if (isinstance == -1) {
5799 return 1;
5800 }
5801 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005802 identifier id;
5803 expr_context_ty ctx;
5804
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005805 if (_PyObject_HasAttrId(obj, &PyId_id)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005806 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005807 tmp = _PyObject_GetAttrId(obj, &PyId_id);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005808 if (tmp == NULL) goto failed;
5809 res = obj2ast_identifier(tmp, &id, arena);
5810 if (res != 0) goto failed;
5811 Py_XDECREF(tmp);
5812 tmp = NULL;
5813 } else {
5814 PyErr_SetString(PyExc_TypeError, "required field \"id\" missing from Name");
5815 return 1;
5816 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005817 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005818 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005819 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005820 if (tmp == NULL) goto failed;
5821 res = obj2ast_expr_context(tmp, &ctx, arena);
5822 if (res != 0) goto failed;
5823 Py_XDECREF(tmp);
5824 tmp = NULL;
5825 } else {
5826 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Name");
5827 return 1;
5828 }
5829 *out = Name(id, ctx, lineno, col_offset, arena);
5830 if (*out == NULL) goto failed;
5831 return 0;
5832 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005833 isinstance = PyObject_IsInstance(obj, (PyObject*)List_type);
5834 if (isinstance == -1) {
5835 return 1;
5836 }
5837 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005838 asdl_seq* elts;
5839 expr_context_ty ctx;
5840
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005841 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005842 int res;
5843 Py_ssize_t len;
5844 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005845 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005846 if (tmp == NULL) goto failed;
5847 if (!PyList_Check(tmp)) {
5848 PyErr_Format(PyExc_TypeError, "List field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5849 goto failed;
5850 }
5851 len = PyList_GET_SIZE(tmp);
5852 elts = asdl_seq_new(len, arena);
5853 if (elts == NULL) goto failed;
5854 for (i = 0; i < len; i++) {
5855 expr_ty value;
5856 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5857 if (res != 0) goto failed;
5858 asdl_seq_SET(elts, i, value);
5859 }
5860 Py_XDECREF(tmp);
5861 tmp = NULL;
5862 } else {
5863 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from List");
5864 return 1;
5865 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005866 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005867 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005868 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005869 if (tmp == NULL) goto failed;
5870 res = obj2ast_expr_context(tmp, &ctx, arena);
5871 if (res != 0) goto failed;
5872 Py_XDECREF(tmp);
5873 tmp = NULL;
5874 } else {
5875 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from List");
5876 return 1;
5877 }
5878 *out = List(elts, ctx, lineno, col_offset, arena);
5879 if (*out == NULL) goto failed;
5880 return 0;
5881 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005882 isinstance = PyObject_IsInstance(obj, (PyObject*)Tuple_type);
5883 if (isinstance == -1) {
5884 return 1;
5885 }
5886 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005887 asdl_seq* elts;
5888 expr_context_ty ctx;
5889
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005890 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005891 int res;
5892 Py_ssize_t len;
5893 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005894 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005895 if (tmp == NULL) goto failed;
5896 if (!PyList_Check(tmp)) {
5897 PyErr_Format(PyExc_TypeError, "Tuple field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5898 goto failed;
5899 }
5900 len = PyList_GET_SIZE(tmp);
5901 elts = asdl_seq_new(len, arena);
5902 if (elts == NULL) goto failed;
5903 for (i = 0; i < len; i++) {
5904 expr_ty value;
5905 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5906 if (res != 0) goto failed;
5907 asdl_seq_SET(elts, i, value);
5908 }
5909 Py_XDECREF(tmp);
5910 tmp = NULL;
5911 } else {
5912 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Tuple");
5913 return 1;
5914 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005915 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005916 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005917 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005918 if (tmp == NULL) goto failed;
5919 res = obj2ast_expr_context(tmp, &ctx, arena);
5920 if (res != 0) goto failed;
5921 Py_XDECREF(tmp);
5922 tmp = NULL;
5923 } else {
5924 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Tuple");
5925 return 1;
5926 }
5927 *out = Tuple(elts, ctx, lineno, col_offset, arena);
5928 if (*out == NULL) goto failed;
5929 return 0;
5930 }
5931
Benjamin Peterson5b066812010-11-20 01:38:49 +00005932 PyErr_Format(PyExc_TypeError, "expected some sort of expr, but got %R", obj);
5933 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00005934 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005935 return 1;
5936}
5937
5938int
5939obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena* arena)
5940{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005941 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005942
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005943 isinstance = PyObject_IsInstance(obj, (PyObject *)Load_type);
5944 if (isinstance == -1) {
5945 return 1;
5946 }
5947 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005948 *out = Load;
5949 return 0;
5950 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005951 isinstance = PyObject_IsInstance(obj, (PyObject *)Store_type);
5952 if (isinstance == -1) {
5953 return 1;
5954 }
5955 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005956 *out = Store;
5957 return 0;
5958 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005959 isinstance = PyObject_IsInstance(obj, (PyObject *)Del_type);
5960 if (isinstance == -1) {
5961 return 1;
5962 }
5963 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005964 *out = Del;
5965 return 0;
5966 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005967 isinstance = PyObject_IsInstance(obj, (PyObject *)AugLoad_type);
5968 if (isinstance == -1) {
5969 return 1;
5970 }
5971 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005972 *out = AugLoad;
5973 return 0;
5974 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005975 isinstance = PyObject_IsInstance(obj, (PyObject *)AugStore_type);
5976 if (isinstance == -1) {
5977 return 1;
5978 }
5979 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005980 *out = AugStore;
5981 return 0;
5982 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005983 isinstance = PyObject_IsInstance(obj, (PyObject *)Param_type);
5984 if (isinstance == -1) {
5985 return 1;
5986 }
5987 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005988 *out = Param;
5989 return 0;
5990 }
5991
Benjamin Peterson5b066812010-11-20 01:38:49 +00005992 PyErr_Format(PyExc_TypeError, "expected some sort of expr_context, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005993 return 1;
5994}
5995
5996int
5997obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena)
5998{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005999 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006000
Benjamin Petersond8f65972010-11-20 04:31:07 +00006001 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006002
6003 if (obj == Py_None) {
6004 *out = NULL;
6005 return 0;
6006 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006007 isinstance = PyObject_IsInstance(obj, (PyObject*)Slice_type);
6008 if (isinstance == -1) {
6009 return 1;
6010 }
6011 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006012 expr_ty lower;
6013 expr_ty upper;
6014 expr_ty step;
6015
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006016 if (_PyObject_HasAttrId(obj, &PyId_lower)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006017 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006018 tmp = _PyObject_GetAttrId(obj, &PyId_lower);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006019 if (tmp == NULL) goto failed;
6020 res = obj2ast_expr(tmp, &lower, arena);
6021 if (res != 0) goto failed;
6022 Py_XDECREF(tmp);
6023 tmp = NULL;
6024 } else {
6025 lower = NULL;
6026 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006027 if (_PyObject_HasAttrId(obj, &PyId_upper)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006028 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006029 tmp = _PyObject_GetAttrId(obj, &PyId_upper);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006030 if (tmp == NULL) goto failed;
6031 res = obj2ast_expr(tmp, &upper, arena);
6032 if (res != 0) goto failed;
6033 Py_XDECREF(tmp);
6034 tmp = NULL;
6035 } else {
6036 upper = NULL;
6037 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006038 if (_PyObject_HasAttrId(obj, &PyId_step)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006039 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006040 tmp = _PyObject_GetAttrId(obj, &PyId_step);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006041 if (tmp == NULL) goto failed;
6042 res = obj2ast_expr(tmp, &step, arena);
6043 if (res != 0) goto failed;
6044 Py_XDECREF(tmp);
6045 tmp = NULL;
6046 } else {
6047 step = NULL;
6048 }
6049 *out = Slice(lower, upper, step, arena);
6050 if (*out == NULL) goto failed;
6051 return 0;
6052 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006053 isinstance = PyObject_IsInstance(obj, (PyObject*)ExtSlice_type);
6054 if (isinstance == -1) {
6055 return 1;
6056 }
6057 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006058 asdl_seq* dims;
6059
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006060 if (_PyObject_HasAttrId(obj, &PyId_dims)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006061 int res;
6062 Py_ssize_t len;
6063 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006064 tmp = _PyObject_GetAttrId(obj, &PyId_dims);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006065 if (tmp == NULL) goto failed;
6066 if (!PyList_Check(tmp)) {
6067 PyErr_Format(PyExc_TypeError, "ExtSlice field \"dims\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6068 goto failed;
6069 }
6070 len = PyList_GET_SIZE(tmp);
6071 dims = asdl_seq_new(len, arena);
6072 if (dims == NULL) goto failed;
6073 for (i = 0; i < len; i++) {
6074 slice_ty value;
6075 res = obj2ast_slice(PyList_GET_ITEM(tmp, i), &value, arena);
6076 if (res != 0) goto failed;
6077 asdl_seq_SET(dims, i, value);
6078 }
6079 Py_XDECREF(tmp);
6080 tmp = NULL;
6081 } else {
6082 PyErr_SetString(PyExc_TypeError, "required field \"dims\" missing from ExtSlice");
6083 return 1;
6084 }
6085 *out = ExtSlice(dims, arena);
6086 if (*out == NULL) goto failed;
6087 return 0;
6088 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006089 isinstance = PyObject_IsInstance(obj, (PyObject*)Index_type);
6090 if (isinstance == -1) {
6091 return 1;
6092 }
6093 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006094 expr_ty value;
6095
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006096 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006097 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006098 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006099 if (tmp == NULL) goto failed;
6100 res = obj2ast_expr(tmp, &value, arena);
6101 if (res != 0) goto failed;
6102 Py_XDECREF(tmp);
6103 tmp = NULL;
6104 } else {
6105 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Index");
6106 return 1;
6107 }
6108 *out = Index(value, arena);
6109 if (*out == NULL) goto failed;
6110 return 0;
6111 }
6112
Benjamin Peterson5b066812010-11-20 01:38:49 +00006113 PyErr_Format(PyExc_TypeError, "expected some sort of slice, but got %R", obj);
6114 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006115 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006116 return 1;
6117}
6118
6119int
6120obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena)
6121{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006122 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006123
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006124 isinstance = PyObject_IsInstance(obj, (PyObject *)And_type);
6125 if (isinstance == -1) {
6126 return 1;
6127 }
6128 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006129 *out = And;
6130 return 0;
6131 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006132 isinstance = PyObject_IsInstance(obj, (PyObject *)Or_type);
6133 if (isinstance == -1) {
6134 return 1;
6135 }
6136 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006137 *out = Or;
6138 return 0;
6139 }
6140
Benjamin Peterson5b066812010-11-20 01:38:49 +00006141 PyErr_Format(PyExc_TypeError, "expected some sort of boolop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006142 return 1;
6143}
6144
6145int
6146obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena)
6147{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006148 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006149
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006150 isinstance = PyObject_IsInstance(obj, (PyObject *)Add_type);
6151 if (isinstance == -1) {
6152 return 1;
6153 }
6154 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006155 *out = Add;
6156 return 0;
6157 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006158 isinstance = PyObject_IsInstance(obj, (PyObject *)Sub_type);
6159 if (isinstance == -1) {
6160 return 1;
6161 }
6162 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006163 *out = Sub;
6164 return 0;
6165 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006166 isinstance = PyObject_IsInstance(obj, (PyObject *)Mult_type);
6167 if (isinstance == -1) {
6168 return 1;
6169 }
6170 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006171 *out = Mult;
6172 return 0;
6173 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006174 isinstance = PyObject_IsInstance(obj, (PyObject *)Div_type);
6175 if (isinstance == -1) {
6176 return 1;
6177 }
6178 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006179 *out = Div;
6180 return 0;
6181 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006182 isinstance = PyObject_IsInstance(obj, (PyObject *)Mod_type);
6183 if (isinstance == -1) {
6184 return 1;
6185 }
6186 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006187 *out = Mod;
6188 return 0;
6189 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006190 isinstance = PyObject_IsInstance(obj, (PyObject *)Pow_type);
6191 if (isinstance == -1) {
6192 return 1;
6193 }
6194 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006195 *out = Pow;
6196 return 0;
6197 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006198 isinstance = PyObject_IsInstance(obj, (PyObject *)LShift_type);
6199 if (isinstance == -1) {
6200 return 1;
6201 }
6202 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006203 *out = LShift;
6204 return 0;
6205 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006206 isinstance = PyObject_IsInstance(obj, (PyObject *)RShift_type);
6207 if (isinstance == -1) {
6208 return 1;
6209 }
6210 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006211 *out = RShift;
6212 return 0;
6213 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006214 isinstance = PyObject_IsInstance(obj, (PyObject *)BitOr_type);
6215 if (isinstance == -1) {
6216 return 1;
6217 }
6218 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006219 *out = BitOr;
6220 return 0;
6221 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006222 isinstance = PyObject_IsInstance(obj, (PyObject *)BitXor_type);
6223 if (isinstance == -1) {
6224 return 1;
6225 }
6226 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006227 *out = BitXor;
6228 return 0;
6229 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006230 isinstance = PyObject_IsInstance(obj, (PyObject *)BitAnd_type);
6231 if (isinstance == -1) {
6232 return 1;
6233 }
6234 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006235 *out = BitAnd;
6236 return 0;
6237 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006238 isinstance = PyObject_IsInstance(obj, (PyObject *)FloorDiv_type);
6239 if (isinstance == -1) {
6240 return 1;
6241 }
6242 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006243 *out = FloorDiv;
6244 return 0;
6245 }
6246
Benjamin Peterson5b066812010-11-20 01:38:49 +00006247 PyErr_Format(PyExc_TypeError, "expected some sort of operator, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006248 return 1;
6249}
6250
6251int
6252obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena)
6253{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006254 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006255
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006256 isinstance = PyObject_IsInstance(obj, (PyObject *)Invert_type);
6257 if (isinstance == -1) {
6258 return 1;
6259 }
6260 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006261 *out = Invert;
6262 return 0;
6263 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006264 isinstance = PyObject_IsInstance(obj, (PyObject *)Not_type);
6265 if (isinstance == -1) {
6266 return 1;
6267 }
6268 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006269 *out = Not;
6270 return 0;
6271 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006272 isinstance = PyObject_IsInstance(obj, (PyObject *)UAdd_type);
6273 if (isinstance == -1) {
6274 return 1;
6275 }
6276 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006277 *out = UAdd;
6278 return 0;
6279 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006280 isinstance = PyObject_IsInstance(obj, (PyObject *)USub_type);
6281 if (isinstance == -1) {
6282 return 1;
6283 }
6284 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006285 *out = USub;
6286 return 0;
6287 }
6288
Benjamin Peterson5b066812010-11-20 01:38:49 +00006289 PyErr_Format(PyExc_TypeError, "expected some sort of unaryop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006290 return 1;
6291}
6292
6293int
6294obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena)
6295{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006296 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006297
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006298 isinstance = PyObject_IsInstance(obj, (PyObject *)Eq_type);
6299 if (isinstance == -1) {
6300 return 1;
6301 }
6302 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006303 *out = Eq;
6304 return 0;
6305 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006306 isinstance = PyObject_IsInstance(obj, (PyObject *)NotEq_type);
6307 if (isinstance == -1) {
6308 return 1;
6309 }
6310 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006311 *out = NotEq;
6312 return 0;
6313 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006314 isinstance = PyObject_IsInstance(obj, (PyObject *)Lt_type);
6315 if (isinstance == -1) {
6316 return 1;
6317 }
6318 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006319 *out = Lt;
6320 return 0;
6321 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006322 isinstance = PyObject_IsInstance(obj, (PyObject *)LtE_type);
6323 if (isinstance == -1) {
6324 return 1;
6325 }
6326 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006327 *out = LtE;
6328 return 0;
6329 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006330 isinstance = PyObject_IsInstance(obj, (PyObject *)Gt_type);
6331 if (isinstance == -1) {
6332 return 1;
6333 }
6334 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006335 *out = Gt;
6336 return 0;
6337 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006338 isinstance = PyObject_IsInstance(obj, (PyObject *)GtE_type);
6339 if (isinstance == -1) {
6340 return 1;
6341 }
6342 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006343 *out = GtE;
6344 return 0;
6345 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006346 isinstance = PyObject_IsInstance(obj, (PyObject *)Is_type);
6347 if (isinstance == -1) {
6348 return 1;
6349 }
6350 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006351 *out = Is;
6352 return 0;
6353 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006354 isinstance = PyObject_IsInstance(obj, (PyObject *)IsNot_type);
6355 if (isinstance == -1) {
6356 return 1;
6357 }
6358 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006359 *out = IsNot;
6360 return 0;
6361 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006362 isinstance = PyObject_IsInstance(obj, (PyObject *)In_type);
6363 if (isinstance == -1) {
6364 return 1;
6365 }
6366 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006367 *out = In;
6368 return 0;
6369 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006370 isinstance = PyObject_IsInstance(obj, (PyObject *)NotIn_type);
6371 if (isinstance == -1) {
6372 return 1;
6373 }
6374 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006375 *out = NotIn;
6376 return 0;
6377 }
6378
Benjamin Peterson5b066812010-11-20 01:38:49 +00006379 PyErr_Format(PyExc_TypeError, "expected some sort of cmpop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006380 return 1;
6381}
6382
6383int
6384obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena* arena)
6385{
6386 PyObject* tmp = NULL;
6387 expr_ty target;
6388 expr_ty iter;
6389 asdl_seq* ifs;
6390
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006391 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006392 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006393 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006394 if (tmp == NULL) goto failed;
6395 res = obj2ast_expr(tmp, &target, arena);
6396 if (res != 0) goto failed;
6397 Py_XDECREF(tmp);
6398 tmp = NULL;
6399 } else {
6400 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from comprehension");
6401 return 1;
6402 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006403 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006404 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006405 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006406 if (tmp == NULL) goto failed;
6407 res = obj2ast_expr(tmp, &iter, arena);
6408 if (res != 0) goto failed;
6409 Py_XDECREF(tmp);
6410 tmp = NULL;
6411 } else {
6412 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from comprehension");
6413 return 1;
6414 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006415 if (_PyObject_HasAttrId(obj, &PyId_ifs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006416 int res;
6417 Py_ssize_t len;
6418 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006419 tmp = _PyObject_GetAttrId(obj, &PyId_ifs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006420 if (tmp == NULL) goto failed;
6421 if (!PyList_Check(tmp)) {
6422 PyErr_Format(PyExc_TypeError, "comprehension field \"ifs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6423 goto failed;
6424 }
6425 len = PyList_GET_SIZE(tmp);
6426 ifs = asdl_seq_new(len, arena);
6427 if (ifs == NULL) goto failed;
6428 for (i = 0; i < len; i++) {
6429 expr_ty value;
6430 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6431 if (res != 0) goto failed;
6432 asdl_seq_SET(ifs, i, value);
6433 }
6434 Py_XDECREF(tmp);
6435 tmp = NULL;
6436 } else {
6437 PyErr_SetString(PyExc_TypeError, "required field \"ifs\" missing from comprehension");
6438 return 1;
6439 }
6440 *out = comprehension(target, iter, ifs, arena);
6441 return 0;
6442failed:
6443 Py_XDECREF(tmp);
6444 return 1;
6445}
6446
6447int
6448obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena* arena)
6449{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006450 int isinstance;
Neal Norwitzad74aa82008-03-31 05:14:30 +00006451
Benjamin Petersond8f65972010-11-20 04:31:07 +00006452 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006453 int lineno;
6454 int col_offset;
6455
Neal Norwitzad74aa82008-03-31 05:14:30 +00006456 if (obj == Py_None) {
6457 *out = NULL;
6458 return 0;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006459 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006460 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006461 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006462 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006463 if (tmp == NULL) goto failed;
6464 res = obj2ast_int(tmp, &lineno, arena);
6465 if (res != 0) goto failed;
6466 Py_XDECREF(tmp);
6467 tmp = NULL;
6468 } else {
6469 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from excepthandler");
6470 return 1;
6471 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006472 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006473 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006474 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006475 if (tmp == NULL) goto failed;
6476 res = obj2ast_int(tmp, &col_offset, arena);
6477 if (res != 0) goto failed;
6478 Py_XDECREF(tmp);
6479 tmp = NULL;
6480 } else {
6481 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from excepthandler");
6482 return 1;
6483 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006484 isinstance = PyObject_IsInstance(obj, (PyObject*)ExceptHandler_type);
6485 if (isinstance == -1) {
6486 return 1;
6487 }
6488 if (isinstance) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006489 expr_ty type;
6490 identifier name;
6491 asdl_seq* body;
6492
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006493 if (_PyObject_HasAttrId(obj, &PyId_type)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006494 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006495 tmp = _PyObject_GetAttrId(obj, &PyId_type);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006496 if (tmp == NULL) goto failed;
6497 res = obj2ast_expr(tmp, &type, arena);
6498 if (res != 0) goto failed;
6499 Py_XDECREF(tmp);
6500 tmp = NULL;
6501 } else {
6502 type = NULL;
6503 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006504 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006505 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006506 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006507 if (tmp == NULL) goto failed;
6508 res = obj2ast_identifier(tmp, &name, arena);
6509 if (res != 0) goto failed;
6510 Py_XDECREF(tmp);
6511 tmp = NULL;
6512 } else {
6513 name = NULL;
6514 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006515 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006516 int res;
6517 Py_ssize_t len;
6518 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006519 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006520 if (tmp == NULL) goto failed;
6521 if (!PyList_Check(tmp)) {
6522 PyErr_Format(PyExc_TypeError, "ExceptHandler field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6523 goto failed;
6524 }
6525 len = PyList_GET_SIZE(tmp);
6526 body = asdl_seq_new(len, arena);
6527 if (body == NULL) goto failed;
6528 for (i = 0; i < len; i++) {
6529 stmt_ty value;
6530 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
6531 if (res != 0) goto failed;
6532 asdl_seq_SET(body, i, value);
6533 }
6534 Py_XDECREF(tmp);
6535 tmp = NULL;
6536 } else {
6537 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ExceptHandler");
6538 return 1;
6539 }
6540 *out = ExceptHandler(type, name, body, lineno, col_offset,
6541 arena);
6542 if (*out == NULL) goto failed;
6543 return 0;
6544 }
6545
Benjamin Peterson5b066812010-11-20 01:38:49 +00006546 PyErr_Format(PyExc_TypeError, "expected some sort of excepthandler, but got %R", obj);
6547 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006548 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006549 return 1;
6550}
6551
6552int
6553obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena)
6554{
6555 PyObject* tmp = NULL;
6556 asdl_seq* args;
6557 identifier vararg;
6558 expr_ty varargannotation;
6559 asdl_seq* kwonlyargs;
6560 identifier kwarg;
6561 expr_ty kwargannotation;
6562 asdl_seq* defaults;
6563 asdl_seq* kw_defaults;
6564
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006565 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006566 int res;
6567 Py_ssize_t len;
6568 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006569 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006570 if (tmp == NULL) goto failed;
6571 if (!PyList_Check(tmp)) {
6572 PyErr_Format(PyExc_TypeError, "arguments field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6573 goto failed;
6574 }
6575 len = PyList_GET_SIZE(tmp);
6576 args = asdl_seq_new(len, arena);
6577 if (args == NULL) goto failed;
6578 for (i = 0; i < len; i++) {
6579 arg_ty value;
6580 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6581 if (res != 0) goto failed;
6582 asdl_seq_SET(args, i, value);
6583 }
6584 Py_XDECREF(tmp);
6585 tmp = NULL;
6586 } else {
6587 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from arguments");
6588 return 1;
6589 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006590 if (_PyObject_HasAttrId(obj, &PyId_vararg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006591 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006592 tmp = _PyObject_GetAttrId(obj, &PyId_vararg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006593 if (tmp == NULL) goto failed;
6594 res = obj2ast_identifier(tmp, &vararg, arena);
6595 if (res != 0) goto failed;
6596 Py_XDECREF(tmp);
6597 tmp = NULL;
6598 } else {
6599 vararg = NULL;
6600 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006601 if (_PyObject_HasAttrId(obj, &PyId_varargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006602 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006603 tmp = _PyObject_GetAttrId(obj, &PyId_varargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006604 if (tmp == NULL) goto failed;
6605 res = obj2ast_expr(tmp, &varargannotation, arena);
6606 if (res != 0) goto failed;
6607 Py_XDECREF(tmp);
6608 tmp = NULL;
6609 } else {
6610 varargannotation = NULL;
6611 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006612 if (_PyObject_HasAttrId(obj, &PyId_kwonlyargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006613 int res;
6614 Py_ssize_t len;
6615 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006616 tmp = _PyObject_GetAttrId(obj, &PyId_kwonlyargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006617 if (tmp == NULL) goto failed;
6618 if (!PyList_Check(tmp)) {
6619 PyErr_Format(PyExc_TypeError, "arguments field \"kwonlyargs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6620 goto failed;
6621 }
6622 len = PyList_GET_SIZE(tmp);
6623 kwonlyargs = asdl_seq_new(len, arena);
6624 if (kwonlyargs == NULL) goto failed;
6625 for (i = 0; i < len; i++) {
6626 arg_ty value;
6627 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6628 if (res != 0) goto failed;
6629 asdl_seq_SET(kwonlyargs, i, value);
6630 }
6631 Py_XDECREF(tmp);
6632 tmp = NULL;
6633 } else {
6634 PyErr_SetString(PyExc_TypeError, "required field \"kwonlyargs\" missing from arguments");
6635 return 1;
6636 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006637 if (_PyObject_HasAttrId(obj, &PyId_kwarg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006638 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006639 tmp = _PyObject_GetAttrId(obj, &PyId_kwarg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006640 if (tmp == NULL) goto failed;
6641 res = obj2ast_identifier(tmp, &kwarg, arena);
6642 if (res != 0) goto failed;
6643 Py_XDECREF(tmp);
6644 tmp = NULL;
6645 } else {
6646 kwarg = NULL;
6647 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006648 if (_PyObject_HasAttrId(obj, &PyId_kwargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006649 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006650 tmp = _PyObject_GetAttrId(obj, &PyId_kwargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006651 if (tmp == NULL) goto failed;
6652 res = obj2ast_expr(tmp, &kwargannotation, arena);
6653 if (res != 0) goto failed;
6654 Py_XDECREF(tmp);
6655 tmp = NULL;
6656 } else {
6657 kwargannotation = NULL;
6658 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006659 if (_PyObject_HasAttrId(obj, &PyId_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006660 int res;
6661 Py_ssize_t len;
6662 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006663 tmp = _PyObject_GetAttrId(obj, &PyId_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006664 if (tmp == NULL) goto failed;
6665 if (!PyList_Check(tmp)) {
6666 PyErr_Format(PyExc_TypeError, "arguments field \"defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6667 goto failed;
6668 }
6669 len = PyList_GET_SIZE(tmp);
6670 defaults = asdl_seq_new(len, arena);
6671 if (defaults == NULL) goto failed;
6672 for (i = 0; i < len; i++) {
6673 expr_ty value;
6674 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6675 if (res != 0) goto failed;
6676 asdl_seq_SET(defaults, i, value);
6677 }
6678 Py_XDECREF(tmp);
6679 tmp = NULL;
6680 } else {
6681 PyErr_SetString(PyExc_TypeError, "required field \"defaults\" missing from arguments");
6682 return 1;
6683 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006684 if (_PyObject_HasAttrId(obj, &PyId_kw_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006685 int res;
6686 Py_ssize_t len;
6687 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006688 tmp = _PyObject_GetAttrId(obj, &PyId_kw_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006689 if (tmp == NULL) goto failed;
6690 if (!PyList_Check(tmp)) {
6691 PyErr_Format(PyExc_TypeError, "arguments field \"kw_defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6692 goto failed;
6693 }
6694 len = PyList_GET_SIZE(tmp);
6695 kw_defaults = asdl_seq_new(len, arena);
6696 if (kw_defaults == NULL) goto failed;
6697 for (i = 0; i < len; i++) {
6698 expr_ty value;
6699 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6700 if (res != 0) goto failed;
6701 asdl_seq_SET(kw_defaults, i, value);
6702 }
6703 Py_XDECREF(tmp);
6704 tmp = NULL;
6705 } else {
6706 PyErr_SetString(PyExc_TypeError, "required field \"kw_defaults\" missing from arguments");
6707 return 1;
6708 }
6709 *out = arguments(args, vararg, varargannotation, kwonlyargs, kwarg,
6710 kwargannotation, defaults, kw_defaults, arena);
6711 return 0;
6712failed:
6713 Py_XDECREF(tmp);
6714 return 1;
6715}
6716
6717int
6718obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena)
6719{
6720 PyObject* tmp = NULL;
6721 identifier arg;
6722 expr_ty annotation;
6723
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006724 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006725 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006726 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006727 if (tmp == NULL) goto failed;
6728 res = obj2ast_identifier(tmp, &arg, arena);
6729 if (res != 0) goto failed;
6730 Py_XDECREF(tmp);
6731 tmp = NULL;
6732 } else {
6733 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from arg");
6734 return 1;
6735 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006736 if (_PyObject_HasAttrId(obj, &PyId_annotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006737 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006738 tmp = _PyObject_GetAttrId(obj, &PyId_annotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006739 if (tmp == NULL) goto failed;
6740 res = obj2ast_expr(tmp, &annotation, arena);
6741 if (res != 0) goto failed;
6742 Py_XDECREF(tmp);
6743 tmp = NULL;
6744 } else {
6745 annotation = NULL;
6746 }
6747 *out = arg(arg, annotation, arena);
6748 return 0;
6749failed:
6750 Py_XDECREF(tmp);
6751 return 1;
6752}
6753
6754int
6755obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena)
6756{
6757 PyObject* tmp = NULL;
6758 identifier arg;
6759 expr_ty value;
6760
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006761 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006762 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006763 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006764 if (tmp == NULL) goto failed;
6765 res = obj2ast_identifier(tmp, &arg, arena);
6766 if (res != 0) goto failed;
6767 Py_XDECREF(tmp);
6768 tmp = NULL;
6769 } else {
6770 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from keyword");
6771 return 1;
6772 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006773 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006774 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006775 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006776 if (tmp == NULL) goto failed;
6777 res = obj2ast_expr(tmp, &value, arena);
6778 if (res != 0) goto failed;
6779 Py_XDECREF(tmp);
6780 tmp = NULL;
6781 } else {
6782 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from keyword");
6783 return 1;
6784 }
6785 *out = keyword(arg, value, arena);
6786 return 0;
6787failed:
6788 Py_XDECREF(tmp);
6789 return 1;
6790}
6791
6792int
6793obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena)
6794{
6795 PyObject* tmp = NULL;
6796 identifier name;
6797 identifier asname;
6798
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006799 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006800 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006801 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006802 if (tmp == NULL) goto failed;
6803 res = obj2ast_identifier(tmp, &name, arena);
6804 if (res != 0) goto failed;
6805 Py_XDECREF(tmp);
6806 tmp = NULL;
6807 } else {
6808 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from alias");
6809 return 1;
6810 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006811 if (_PyObject_HasAttrId(obj, &PyId_asname)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006812 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006813 tmp = _PyObject_GetAttrId(obj, &PyId_asname);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006814 if (tmp == NULL) goto failed;
6815 res = obj2ast_identifier(tmp, &asname, arena);
6816 if (res != 0) goto failed;
6817 Py_XDECREF(tmp);
6818 tmp = NULL;
6819 } else {
6820 asname = NULL;
6821 }
6822 *out = alias(name, asname, arena);
6823 return 0;
6824failed:
6825 Py_XDECREF(tmp);
6826 return 1;
6827}
6828
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006829int
6830obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena)
6831{
6832 PyObject* tmp = NULL;
6833 expr_ty context_expr;
6834 expr_ty optional_vars;
6835
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006836 if (_PyObject_HasAttrId(obj, &PyId_context_expr)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006837 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006838 tmp = _PyObject_GetAttrId(obj, &PyId_context_expr);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006839 if (tmp == NULL) goto failed;
6840 res = obj2ast_expr(tmp, &context_expr, arena);
6841 if (res != 0) goto failed;
6842 Py_XDECREF(tmp);
6843 tmp = NULL;
6844 } else {
6845 PyErr_SetString(PyExc_TypeError, "required field \"context_expr\" missing from withitem");
6846 return 1;
6847 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006848 if (_PyObject_HasAttrId(obj, &PyId_optional_vars)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006849 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006850 tmp = _PyObject_GetAttrId(obj, &PyId_optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006851 if (tmp == NULL) goto failed;
6852 res = obj2ast_expr(tmp, &optional_vars, arena);
6853 if (res != 0) goto failed;
6854 Py_XDECREF(tmp);
6855 tmp = NULL;
6856 } else {
6857 optional_vars = NULL;
6858 }
6859 *out = withitem(context_expr, optional_vars, arena);
6860 return 0;
6861failed:
6862 Py_XDECREF(tmp);
6863 return 1;
6864}
6865
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006866
Martin v. Löwis1a214512008-06-11 05:26:20 +00006867static struct PyModuleDef _astmodule = {
6868 PyModuleDef_HEAD_INIT, "_ast"
6869};
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006870PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00006871PyInit__ast(void)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006872{
6873 PyObject *m, *d;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006874 if (!init_types()) return NULL;
6875 m = PyModule_Create(&_astmodule);
6876 if (!m) return NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006877 d = PyModule_GetDict(m);
Martin v. Löwis1a214512008-06-11 05:26:20 +00006878 if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return
6879 NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006880 if (PyModule_AddIntConstant(m, "PyCF_ONLY_AST", PyCF_ONLY_AST) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006881 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006882 if (PyDict_SetItemString(d, "mod", (PyObject*)mod_type) < 0) return
6883 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006884 if (PyDict_SetItemString(d, "Module", (PyObject*)Module_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006885 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006886 if (PyDict_SetItemString(d, "Interactive", (PyObject*)Interactive_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006887 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006888 if (PyDict_SetItemString(d, "Expression", (PyObject*)Expression_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006889 0) return NULL;
6890 if (PyDict_SetItemString(d, "Suite", (PyObject*)Suite_type) < 0) return
6891 NULL;
6892 if (PyDict_SetItemString(d, "stmt", (PyObject*)stmt_type) < 0) return
6893 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006894 if (PyDict_SetItemString(d, "FunctionDef", (PyObject*)FunctionDef_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006895 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006896 if (PyDict_SetItemString(d, "ClassDef", (PyObject*)ClassDef_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006897 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006898 if (PyDict_SetItemString(d, "Return", (PyObject*)Return_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006899 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006900 if (PyDict_SetItemString(d, "Delete", (PyObject*)Delete_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006901 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006902 if (PyDict_SetItemString(d, "Assign", (PyObject*)Assign_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006903 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006904 if (PyDict_SetItemString(d, "AugAssign", (PyObject*)AugAssign_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006905 0) return NULL;
6906 if (PyDict_SetItemString(d, "For", (PyObject*)For_type) < 0) return
6907 NULL;
6908 if (PyDict_SetItemString(d, "While", (PyObject*)While_type) < 0) return
6909 NULL;
6910 if (PyDict_SetItemString(d, "If", (PyObject*)If_type) < 0) return NULL;
6911 if (PyDict_SetItemString(d, "With", (PyObject*)With_type) < 0) return
6912 NULL;
6913 if (PyDict_SetItemString(d, "Raise", (PyObject*)Raise_type) < 0) return
6914 NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05006915 if (PyDict_SetItemString(d, "Try", (PyObject*)Try_type) < 0) return
6916 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006917 if (PyDict_SetItemString(d, "Assert", (PyObject*)Assert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006918 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006919 if (PyDict_SetItemString(d, "Import", (PyObject*)Import_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006920 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006921 if (PyDict_SetItemString(d, "ImportFrom", (PyObject*)ImportFrom_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006922 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006923 if (PyDict_SetItemString(d, "Global", (PyObject*)Global_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006924 return NULL;
Jeremy Hylton81e95022007-02-27 06:50:52 +00006925 if (PyDict_SetItemString(d, "Nonlocal", (PyObject*)Nonlocal_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006926 return NULL;
6927 if (PyDict_SetItemString(d, "Expr", (PyObject*)Expr_type) < 0) return
6928 NULL;
6929 if (PyDict_SetItemString(d, "Pass", (PyObject*)Pass_type) < 0) return
6930 NULL;
6931 if (PyDict_SetItemString(d, "Break", (PyObject*)Break_type) < 0) return
6932 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006933 if (PyDict_SetItemString(d, "Continue", (PyObject*)Continue_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006934 return NULL;
6935 if (PyDict_SetItemString(d, "expr", (PyObject*)expr_type) < 0) return
6936 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006937 if (PyDict_SetItemString(d, "BoolOp", (PyObject*)BoolOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006938 return NULL;
6939 if (PyDict_SetItemString(d, "BinOp", (PyObject*)BinOp_type) < 0) return
6940 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006941 if (PyDict_SetItemString(d, "UnaryOp", (PyObject*)UnaryOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006942 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006943 if (PyDict_SetItemString(d, "Lambda", (PyObject*)Lambda_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006944 return NULL;
6945 if (PyDict_SetItemString(d, "IfExp", (PyObject*)IfExp_type) < 0) return
6946 NULL;
6947 if (PyDict_SetItemString(d, "Dict", (PyObject*)Dict_type) < 0) return
6948 NULL;
6949 if (PyDict_SetItemString(d, "Set", (PyObject*)Set_type) < 0) return
6950 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006951 if (PyDict_SetItemString(d, "ListComp", (PyObject*)ListComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006952 return NULL;
Nick Coghlan650f0d02007-04-15 12:05:43 +00006953 if (PyDict_SetItemString(d, "SetComp", (PyObject*)SetComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006954 return NULL;
Guido van Rossum992d4a32007-07-11 13:09:30 +00006955 if (PyDict_SetItemString(d, "DictComp", (PyObject*)DictComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006956 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006957 if (PyDict_SetItemString(d, "GeneratorExp",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006958 (PyObject*)GeneratorExp_type) < 0) return NULL;
6959 if (PyDict_SetItemString(d, "Yield", (PyObject*)Yield_type) < 0) return
6960 NULL;
Benjamin Peterson527c6222012-01-14 08:58:23 -05006961 if (PyDict_SetItemString(d, "YieldFrom", (PyObject*)YieldFrom_type) <
6962 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006963 if (PyDict_SetItemString(d, "Compare", (PyObject*)Compare_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006964 return NULL;
6965 if (PyDict_SetItemString(d, "Call", (PyObject*)Call_type) < 0) return
6966 NULL;
6967 if (PyDict_SetItemString(d, "Num", (PyObject*)Num_type) < 0) return
6968 NULL;
6969 if (PyDict_SetItemString(d, "Str", (PyObject*)Str_type) < 0) return
6970 NULL;
6971 if (PyDict_SetItemString(d, "Bytes", (PyObject*)Bytes_type) < 0) return
6972 NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00006973 if (PyDict_SetItemString(d, "Ellipsis", (PyObject*)Ellipsis_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006974 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006975 if (PyDict_SetItemString(d, "Attribute", (PyObject*)Attribute_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006976 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006977 if (PyDict_SetItemString(d, "Subscript", (PyObject*)Subscript_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006978 0) return NULL;
Guido van Rossum0368b722007-05-11 16:50:42 +00006979 if (PyDict_SetItemString(d, "Starred", (PyObject*)Starred_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006980 return NULL;
6981 if (PyDict_SetItemString(d, "Name", (PyObject*)Name_type) < 0) return
6982 NULL;
6983 if (PyDict_SetItemString(d, "List", (PyObject*)List_type) < 0) return
6984 NULL;
6985 if (PyDict_SetItemString(d, "Tuple", (PyObject*)Tuple_type) < 0) return
6986 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006987 if (PyDict_SetItemString(d, "expr_context",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006988 (PyObject*)expr_context_type) < 0) return NULL;
6989 if (PyDict_SetItemString(d, "Load", (PyObject*)Load_type) < 0) return
6990 NULL;
6991 if (PyDict_SetItemString(d, "Store", (PyObject*)Store_type) < 0) return
6992 NULL;
6993 if (PyDict_SetItemString(d, "Del", (PyObject*)Del_type) < 0) return
6994 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006995 if (PyDict_SetItemString(d, "AugLoad", (PyObject*)AugLoad_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006996 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006997 if (PyDict_SetItemString(d, "AugStore", (PyObject*)AugStore_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006998 return NULL;
6999 if (PyDict_SetItemString(d, "Param", (PyObject*)Param_type) < 0) return
7000 NULL;
7001 if (PyDict_SetItemString(d, "slice", (PyObject*)slice_type) < 0) return
7002 NULL;
7003 if (PyDict_SetItemString(d, "Slice", (PyObject*)Slice_type) < 0) return
7004 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007005 if (PyDict_SetItemString(d, "ExtSlice", (PyObject*)ExtSlice_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007006 return NULL;
7007 if (PyDict_SetItemString(d, "Index", (PyObject*)Index_type) < 0) return
7008 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007009 if (PyDict_SetItemString(d, "boolop", (PyObject*)boolop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007010 return NULL;
7011 if (PyDict_SetItemString(d, "And", (PyObject*)And_type) < 0) return
7012 NULL;
7013 if (PyDict_SetItemString(d, "Or", (PyObject*)Or_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007014 if (PyDict_SetItemString(d, "operator", (PyObject*)operator_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007015 return NULL;
7016 if (PyDict_SetItemString(d, "Add", (PyObject*)Add_type) < 0) return
7017 NULL;
7018 if (PyDict_SetItemString(d, "Sub", (PyObject*)Sub_type) < 0) return
7019 NULL;
7020 if (PyDict_SetItemString(d, "Mult", (PyObject*)Mult_type) < 0) return
7021 NULL;
7022 if (PyDict_SetItemString(d, "Div", (PyObject*)Div_type) < 0) return
7023 NULL;
7024 if (PyDict_SetItemString(d, "Mod", (PyObject*)Mod_type) < 0) return
7025 NULL;
7026 if (PyDict_SetItemString(d, "Pow", (PyObject*)Pow_type) < 0) return
7027 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007028 if (PyDict_SetItemString(d, "LShift", (PyObject*)LShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007029 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007030 if (PyDict_SetItemString(d, "RShift", (PyObject*)RShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007031 return NULL;
7032 if (PyDict_SetItemString(d, "BitOr", (PyObject*)BitOr_type) < 0) return
7033 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007034 if (PyDict_SetItemString(d, "BitXor", (PyObject*)BitXor_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007035 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007036 if (PyDict_SetItemString(d, "BitAnd", (PyObject*)BitAnd_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007037 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007038 if (PyDict_SetItemString(d, "FloorDiv", (PyObject*)FloorDiv_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007039 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007040 if (PyDict_SetItemString(d, "unaryop", (PyObject*)unaryop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007041 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007042 if (PyDict_SetItemString(d, "Invert", (PyObject*)Invert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007043 return NULL;
7044 if (PyDict_SetItemString(d, "Not", (PyObject*)Not_type) < 0) return
7045 NULL;
7046 if (PyDict_SetItemString(d, "UAdd", (PyObject*)UAdd_type) < 0) return
7047 NULL;
7048 if (PyDict_SetItemString(d, "USub", (PyObject*)USub_type) < 0) return
7049 NULL;
7050 if (PyDict_SetItemString(d, "cmpop", (PyObject*)cmpop_type) < 0) return
7051 NULL;
7052 if (PyDict_SetItemString(d, "Eq", (PyObject*)Eq_type) < 0) return NULL;
7053 if (PyDict_SetItemString(d, "NotEq", (PyObject*)NotEq_type) < 0) return
7054 NULL;
7055 if (PyDict_SetItemString(d, "Lt", (PyObject*)Lt_type) < 0) return NULL;
7056 if (PyDict_SetItemString(d, "LtE", (PyObject*)LtE_type) < 0) return
7057 NULL;
7058 if (PyDict_SetItemString(d, "Gt", (PyObject*)Gt_type) < 0) return NULL;
7059 if (PyDict_SetItemString(d, "GtE", (PyObject*)GtE_type) < 0) return
7060 NULL;
7061 if (PyDict_SetItemString(d, "Is", (PyObject*)Is_type) < 0) return NULL;
7062 if (PyDict_SetItemString(d, "IsNot", (PyObject*)IsNot_type) < 0) return
7063 NULL;
7064 if (PyDict_SetItemString(d, "In", (PyObject*)In_type) < 0) return NULL;
7065 if (PyDict_SetItemString(d, "NotIn", (PyObject*)NotIn_type) < 0) return
7066 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007067 if (PyDict_SetItemString(d, "comprehension",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007068 (PyObject*)comprehension_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007069 if (PyDict_SetItemString(d, "excepthandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007070 (PyObject*)excepthandler_type) < 0) return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00007071 if (PyDict_SetItemString(d, "ExceptHandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007072 (PyObject*)ExceptHandler_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007073 if (PyDict_SetItemString(d, "arguments", (PyObject*)arguments_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007074 0) return NULL;
7075 if (PyDict_SetItemString(d, "arg", (PyObject*)arg_type) < 0) return
7076 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007077 if (PyDict_SetItemString(d, "keyword", (PyObject*)keyword_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007078 return NULL;
7079 if (PyDict_SetItemString(d, "alias", (PyObject*)alias_type) < 0) return
7080 NULL;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05007081 if (PyDict_SetItemString(d, "withitem", (PyObject*)withitem_type) < 0)
7082 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00007083 return m;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00007084}
7085
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007086
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007087PyObject* PyAST_mod2obj(mod_ty t)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007088{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007089 init_types();
7090 return ast2obj_mod(t);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007091}
Martin v. Löwis5b222132007-06-10 09:51:05 +00007092
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007093/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
7094mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007095{
7096 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007097 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
7098 (PyObject*)Interactive_type};
7099 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007100 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007101 assert(0 <= mode && mode <= 2);
7102
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007103 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007104
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007105 isinstance = PyObject_IsInstance(ast, req_type[mode]);
7106 if (isinstance == -1)
7107 return NULL;
7108 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007109 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
7110 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007111 return NULL;
7112 }
7113 if (obj2ast_mod(ast, &res, arena) != 0)
7114 return NULL;
7115 else
7116 return res;
7117}
7118
7119int PyAST_Check(PyObject* obj)
7120{
7121 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00007122 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007123}
7124
Martin v. Löwis5b222132007-06-10 09:51:05 +00007125