blob: 3cfb8a87aa68f267bcfe0d2b551b81a5555c64a3 [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[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000235 "value",
236};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000237static PyTypeObject *Compare_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200238_Py_IDENTIFIER(ops);
239_Py_IDENTIFIER(comparators);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000240static char *Compare_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000241 "left",
242 "ops",
243 "comparators",
244};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000245static PyTypeObject *Call_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200246_Py_IDENTIFIER(func);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000247static char *Call_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000248 "func",
249 "args",
250 "keywords",
251 "starargs",
252 "kwargs",
253};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000254static PyTypeObject *Num_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200255_Py_IDENTIFIER(n);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000256static char *Num_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000257 "n",
258};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000259static PyTypeObject *Str_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200260_Py_IDENTIFIER(s);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000261static char *Str_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000262 "s",
263};
Thomas Wouters00e41de2007-02-23 19:56:57 +0000264static PyTypeObject *Bytes_type;
265static char *Bytes_fields[]={
266 "s",
267};
Georg Brandl52318d62006-09-06 07:06:08 +0000268static PyTypeObject *Ellipsis_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +0000269static PyTypeObject *Attribute_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200270_Py_IDENTIFIER(attr);
271_Py_IDENTIFIER(ctx);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000272static char *Attribute_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000273 "value",
274 "attr",
275 "ctx",
276};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000277static PyTypeObject *Subscript_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200278_Py_IDENTIFIER(slice);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000279static char *Subscript_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000280 "value",
281 "slice",
282 "ctx",
283};
Guido van Rossum0368b722007-05-11 16:50:42 +0000284static PyTypeObject *Starred_type;
285static char *Starred_fields[]={
286 "value",
287 "ctx",
288};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000289static PyTypeObject *Name_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200290_Py_IDENTIFIER(id);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000291static char *Name_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000292 "id",
293 "ctx",
294};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000295static PyTypeObject *List_type;
296static char *List_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000297 "elts",
298 "ctx",
299};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000300static PyTypeObject *Tuple_type;
301static char *Tuple_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000302 "elts",
303 "ctx",
304};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000305static PyTypeObject *expr_context_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000306static PyObject *Load_singleton, *Store_singleton, *Del_singleton,
307*AugLoad_singleton, *AugStore_singleton, *Param_singleton;
308static PyObject* ast2obj_expr_context(expr_context_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000309static PyTypeObject *Load_type;
310static PyTypeObject *Store_type;
311static PyTypeObject *Del_type;
312static PyTypeObject *AugLoad_type;
313static PyTypeObject *AugStore_type;
314static PyTypeObject *Param_type;
315static PyTypeObject *slice_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000316static PyObject* ast2obj_slice(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000317static PyTypeObject *Slice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200318_Py_IDENTIFIER(lower);
319_Py_IDENTIFIER(upper);
320_Py_IDENTIFIER(step);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000321static char *Slice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000322 "lower",
323 "upper",
324 "step",
325};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000326static PyTypeObject *ExtSlice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200327_Py_IDENTIFIER(dims);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000328static char *ExtSlice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000329 "dims",
330};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000331static PyTypeObject *Index_type;
332static char *Index_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000333 "value",
334};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000335static PyTypeObject *boolop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000336static PyObject *And_singleton, *Or_singleton;
337static PyObject* ast2obj_boolop(boolop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000338static PyTypeObject *And_type;
339static PyTypeObject *Or_type;
340static PyTypeObject *operator_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000341static PyObject *Add_singleton, *Sub_singleton, *Mult_singleton,
342*Div_singleton, *Mod_singleton, *Pow_singleton, *LShift_singleton,
343*RShift_singleton, *BitOr_singleton, *BitXor_singleton, *BitAnd_singleton,
344*FloorDiv_singleton;
345static PyObject* ast2obj_operator(operator_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000346static PyTypeObject *Add_type;
347static PyTypeObject *Sub_type;
348static PyTypeObject *Mult_type;
349static PyTypeObject *Div_type;
350static PyTypeObject *Mod_type;
351static PyTypeObject *Pow_type;
352static PyTypeObject *LShift_type;
353static PyTypeObject *RShift_type;
354static PyTypeObject *BitOr_type;
355static PyTypeObject *BitXor_type;
356static PyTypeObject *BitAnd_type;
357static PyTypeObject *FloorDiv_type;
358static PyTypeObject *unaryop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000359static PyObject *Invert_singleton, *Not_singleton, *UAdd_singleton,
360*USub_singleton;
361static PyObject* ast2obj_unaryop(unaryop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000362static PyTypeObject *Invert_type;
363static PyTypeObject *Not_type;
364static PyTypeObject *UAdd_type;
365static PyTypeObject *USub_type;
366static PyTypeObject *cmpop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000367static PyObject *Eq_singleton, *NotEq_singleton, *Lt_singleton, *LtE_singleton,
368*Gt_singleton, *GtE_singleton, *Is_singleton, *IsNot_singleton, *In_singleton,
369*NotIn_singleton;
370static PyObject* ast2obj_cmpop(cmpop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000371static PyTypeObject *Eq_type;
372static PyTypeObject *NotEq_type;
373static PyTypeObject *Lt_type;
374static PyTypeObject *LtE_type;
375static PyTypeObject *Gt_type;
376static PyTypeObject *GtE_type;
377static PyTypeObject *Is_type;
378static PyTypeObject *IsNot_type;
379static PyTypeObject *In_type;
380static PyTypeObject *NotIn_type;
381static PyTypeObject *comprehension_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000382static PyObject* ast2obj_comprehension(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200383_Py_IDENTIFIER(ifs);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000384static char *comprehension_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000385 "target",
386 "iter",
387 "ifs",
388};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000389static PyTypeObject *excepthandler_type;
Neal Norwitzad74aa82008-03-31 05:14:30 +0000390static char *excepthandler_attributes[] = {
391 "lineno",
392 "col_offset",
393};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000394static PyObject* ast2obj_excepthandler(void*);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000395static PyTypeObject *ExceptHandler_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200396_Py_IDENTIFIER(type);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000397static char *ExceptHandler_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000398 "type",
399 "name",
400 "body",
401};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000402static PyTypeObject *arguments_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000403static PyObject* ast2obj_arguments(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200404_Py_IDENTIFIER(vararg);
405_Py_IDENTIFIER(varargannotation);
406_Py_IDENTIFIER(kwonlyargs);
407_Py_IDENTIFIER(kwarg);
408_Py_IDENTIFIER(kwargannotation);
409_Py_IDENTIFIER(defaults);
410_Py_IDENTIFIER(kw_defaults);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000411static char *arguments_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000412 "args",
413 "vararg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000414 "varargannotation",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000415 "kwonlyargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000416 "kwarg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000417 "kwargannotation",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000418 "defaults",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000419 "kw_defaults",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000420};
Neal Norwitzc1505362006-12-28 06:47:50 +0000421static PyTypeObject *arg_type;
422static PyObject* ast2obj_arg(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200423_Py_IDENTIFIER(arg);
424_Py_IDENTIFIER(annotation);
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000425static char *arg_fields[]={
Neal Norwitzc1505362006-12-28 06:47:50 +0000426 "arg",
427 "annotation",
428};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000429static PyTypeObject *keyword_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000430static PyObject* ast2obj_keyword(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000431static char *keyword_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000432 "arg",
433 "value",
434};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000435static PyTypeObject *alias_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000436static PyObject* ast2obj_alias(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200437_Py_IDENTIFIER(asname);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000438static char *alias_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000439 "name",
440 "asname",
441};
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500442static PyTypeObject *withitem_type;
443static PyObject* ast2obj_withitem(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200444_Py_IDENTIFIER(context_expr);
445_Py_IDENTIFIER(optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500446static char *withitem_fields[]={
447 "context_expr",
448 "optional_vars",
449};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000450
451
Neal Norwitz207c9f32008-03-31 04:42:11 +0000452static int
453ast_type_init(PyObject *self, PyObject *args, PyObject *kw)
454{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200455 _Py_IDENTIFIER(_fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000456 Py_ssize_t i, numfields = 0;
457 int res = -1;
458 PyObject *key, *value, *fields;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200459 fields = _PyObject_GetAttrId((PyObject*)Py_TYPE(self), &PyId__fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000460 if (!fields)
461 PyErr_Clear();
462 if (fields) {
463 numfields = PySequence_Size(fields);
464 if (numfields == -1)
465 goto cleanup;
466 }
467 res = 0; /* if no error occurs, this stays 0 to the end */
468 if (PyTuple_GET_SIZE(args) > 0) {
469 if (numfields != PyTuple_GET_SIZE(args)) {
470 PyErr_Format(PyExc_TypeError, "%.400s constructor takes %s"
Amaury Forgeot d'Arc245c70b2008-09-10 22:24:24 +0000471 "%zd positional argument%s",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000472 Py_TYPE(self)->tp_name,
473 numfields == 0 ? "" : "either 0 or ",
474 numfields, numfields == 1 ? "" : "s");
475 res = -1;
476 goto cleanup;
477 }
478 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
479 /* cannot be reached when fields is NULL */
480 PyObject *name = PySequence_GetItem(fields, i);
481 if (!name) {
482 res = -1;
483 goto cleanup;
484 }
485 res = PyObject_SetAttr(self, name, PyTuple_GET_ITEM(args, i));
486 Py_DECREF(name);
487 if (res < 0)
488 goto cleanup;
489 }
490 }
491 if (kw) {
492 i = 0; /* needed by PyDict_Next */
493 while (PyDict_Next(kw, &i, &key, &value)) {
494 res = PyObject_SetAttr(self, key, value);
495 if (res < 0)
496 goto cleanup;
497 }
498 }
499 cleanup:
500 Py_XDECREF(fields);
501 return res;
502}
503
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000504/* Pickling support */
505static PyObject *
506ast_type_reduce(PyObject *self, PyObject *unused)
507{
508 PyObject *res;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200509 _Py_IDENTIFIER(__dict__);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200510 PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__);
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000511 if (dict == NULL) {
512 if (PyErr_ExceptionMatches(PyExc_AttributeError))
513 PyErr_Clear();
514 else
515 return NULL;
516 }
517 if (dict) {
518 res = Py_BuildValue("O()O", Py_TYPE(self), dict);
519 Py_DECREF(dict);
520 return res;
521 }
522 return Py_BuildValue("O()", Py_TYPE(self));
523}
524
525static PyMethodDef ast_type_methods[] = {
526 {"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
527 {NULL}
528};
529
Neal Norwitz207c9f32008-03-31 04:42:11 +0000530static PyTypeObject AST_type = {
531 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000532 "_ast.AST",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000533 sizeof(PyObject),
534 0,
535 0, /* tp_dealloc */
536 0, /* tp_print */
537 0, /* tp_getattr */
538 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000539 0, /* tp_reserved */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000540 0, /* tp_repr */
541 0, /* tp_as_number */
542 0, /* tp_as_sequence */
543 0, /* tp_as_mapping */
544 0, /* tp_hash */
545 0, /* tp_call */
546 0, /* tp_str */
547 PyObject_GenericGetAttr, /* tp_getattro */
548 PyObject_GenericSetAttr, /* tp_setattro */
549 0, /* tp_as_buffer */
550 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
551 0, /* tp_doc */
552 0, /* tp_traverse */
553 0, /* tp_clear */
554 0, /* tp_richcompare */
555 0, /* tp_weaklistoffset */
556 0, /* tp_iter */
557 0, /* tp_iternext */
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000558 ast_type_methods, /* tp_methods */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000559 0, /* tp_members */
560 0, /* tp_getset */
561 0, /* tp_base */
562 0, /* tp_dict */
563 0, /* tp_descr_get */
564 0, /* tp_descr_set */
565 0, /* tp_dictoffset */
566 (initproc)ast_type_init, /* tp_init */
567 PyType_GenericAlloc, /* tp_alloc */
568 PyType_GenericNew, /* tp_new */
569 PyObject_Del, /* tp_free */
570};
571
572
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000573static PyTypeObject* make_type(char *type, PyTypeObject* base, char**fields, int num_fields)
574{
575 PyObject *fnames, *result;
576 int i;
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000577 fnames = PyTuple_New(num_fields);
578 if (!fnames) return NULL;
579 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000580 PyObject *field = PyUnicode_FromString(fields[i]);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000581 if (!field) {
582 Py_DECREF(fnames);
583 return NULL;
584 }
585 PyTuple_SET_ITEM(fnames, i, field);
586 }
Victor Stinner7eeb5b52010-06-07 19:57:46 +0000587 result = PyObject_CallFunction((PyObject*)&PyType_Type, "s(O){sOss}",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000588 type, base, "_fields", fnames, "__module__", "_ast");
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000589 Py_DECREF(fnames);
590 return (PyTypeObject*)result;
591}
592
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000593static int add_attributes(PyTypeObject* type, char**attrs, int num_fields)
594{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000595 int i, result;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200596 _Py_IDENTIFIER(_attributes);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000597 PyObject *s, *l = PyTuple_New(num_fields);
Benjamin Peterson3e5cd1d2010-06-27 21:45:24 +0000598 if (!l)
599 return 0;
600 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000601 s = PyUnicode_FromString(attrs[i]);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000602 if (!s) {
603 Py_DECREF(l);
604 return 0;
605 }
Neal Norwitz207c9f32008-03-31 04:42:11 +0000606 PyTuple_SET_ITEM(l, i, s);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000607 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200608 result = _PyObject_SetAttrId((PyObject*)type, &PyId__attributes, l) >= 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000609 Py_DECREF(l);
610 return result;
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000611}
612
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000613/* Conversion AST -> Python */
614
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000615static PyObject* ast2obj_list(asdl_seq *seq, PyObject* (*func)(void*))
616{
617 int i, n = asdl_seq_LEN(seq);
618 PyObject *result = PyList_New(n);
619 PyObject *value;
620 if (!result)
621 return NULL;
622 for (i = 0; i < n; i++) {
623 value = func(asdl_seq_GET(seq, i));
624 if (!value) {
625 Py_DECREF(result);
626 return NULL;
627 }
628 PyList_SET_ITEM(result, i, value);
629 }
630 return result;
631}
632
633static PyObject* ast2obj_object(void *o)
634{
635 if (!o)
636 o = Py_None;
637 Py_INCREF((PyObject*)o);
638 return (PyObject*)o;
639}
640#define ast2obj_identifier ast2obj_object
641#define ast2obj_string ast2obj_object
Benjamin Petersone2498412011-08-09 16:08:39 -0500642#define ast2obj_bytes ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000643
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000644static PyObject* ast2obj_int(long b)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000645{
Christian Heimes217cfd12007-12-02 14:31:20 +0000646 return PyLong_FromLong(b);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000647}
648
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000649/* Conversion Python -> AST */
650
651static int obj2ast_object(PyObject* obj, PyObject** out, PyArena* arena)
652{
653 if (obj == Py_None)
654 obj = NULL;
655 if (obj)
656 PyArena_AddPyObject(arena, obj);
657 Py_XINCREF(obj);
658 *out = obj;
659 return 0;
660}
661
Benjamin Peterson180e6352011-07-22 11:09:07 -0500662static int obj2ast_identifier(PyObject* obj, PyObject** out, PyArena* arena)
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500663{
Benjamin Peterson180e6352011-07-22 11:09:07 -0500664 if (!PyUnicode_CheckExact(obj) && obj != Py_None) {
665 PyErr_SetString(PyExc_TypeError, "AST identifier must be of type str");
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500666 return 1;
667 }
668 return obj2ast_object(obj, out, arena);
669}
670
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500671static int obj2ast_string(PyObject* obj, PyObject** out, PyArena* arena)
672{
Benjamin Peterson0224d4e2011-08-31 22:13:03 -0400673 if (!PyUnicode_CheckExact(obj) && !PyBytes_CheckExact(obj)) {
Benjamin Peterson180e6352011-07-22 11:09:07 -0500674 PyErr_SetString(PyExc_TypeError, "AST string must be of type str");
675 return 1;
676 }
677 return obj2ast_object(obj, out, arena);
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500678}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000679
Benjamin Petersone2498412011-08-09 16:08:39 -0500680static int obj2ast_bytes(PyObject* obj, PyObject** out, PyArena* arena)
681{
682 if (!PyBytes_CheckExact(obj)) {
683 PyErr_SetString(PyExc_TypeError, "AST bytes must be of type bytes");
684 return 1;
685 }
686 return obj2ast_object(obj, out, arena);
687}
688
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000689static int obj2ast_int(PyObject* obj, int* out, PyArena* arena)
690{
691 int i;
692 if (!PyLong_Check(obj)) {
693 PyObject *s = PyObject_Repr(obj);
694 if (s == NULL) return 1;
695 PyErr_Format(PyExc_ValueError, "invalid integer value: %.400s",
Christian Heimes72b710a2008-05-26 13:28:38 +0000696 PyBytes_AS_STRING(s));
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000697 Py_DECREF(s);
698 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;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000817 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;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000819 Compare_type = make_type("Compare", expr_type, Compare_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000820 if (!Compare_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000821 Call_type = make_type("Call", expr_type, Call_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000822 if (!Call_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000823 Num_type = make_type("Num", expr_type, Num_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000824 if (!Num_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000825 Str_type = make_type("Str", expr_type, Str_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000826 if (!Str_type) return 0;
Thomas Wouters00e41de2007-02-23 19:56:57 +0000827 Bytes_type = make_type("Bytes", expr_type, Bytes_fields, 1);
828 if (!Bytes_type) return 0;
Georg Brandl52318d62006-09-06 07:06:08 +0000829 Ellipsis_type = make_type("Ellipsis", expr_type, NULL, 0);
830 if (!Ellipsis_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000831 Attribute_type = make_type("Attribute", expr_type, Attribute_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000832 if (!Attribute_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000833 Subscript_type = make_type("Subscript", expr_type, Subscript_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000834 if (!Subscript_type) return 0;
Guido van Rossum0368b722007-05-11 16:50:42 +0000835 Starred_type = make_type("Starred", expr_type, Starred_fields, 2);
836 if (!Starred_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000837 Name_type = make_type("Name", expr_type, Name_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000838 if (!Name_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000839 List_type = make_type("List", expr_type, List_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000840 if (!List_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000841 Tuple_type = make_type("Tuple", expr_type, Tuple_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000842 if (!Tuple_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000843 expr_context_type = make_type("expr_context", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000844 if (!expr_context_type) return 0;
845 if (!add_attributes(expr_context_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000846 Load_type = make_type("Load", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000847 if (!Load_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000848 Load_singleton = PyType_GenericNew(Load_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000849 if (!Load_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000850 Store_type = make_type("Store", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000851 if (!Store_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000852 Store_singleton = PyType_GenericNew(Store_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000853 if (!Store_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000854 Del_type = make_type("Del", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000855 if (!Del_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000856 Del_singleton = PyType_GenericNew(Del_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000857 if (!Del_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000858 AugLoad_type = make_type("AugLoad", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000859 if (!AugLoad_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000860 AugLoad_singleton = PyType_GenericNew(AugLoad_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000861 if (!AugLoad_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000862 AugStore_type = make_type("AugStore", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000863 if (!AugStore_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000864 AugStore_singleton = PyType_GenericNew(AugStore_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000865 if (!AugStore_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000866 Param_type = make_type("Param", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000867 if (!Param_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000868 Param_singleton = PyType_GenericNew(Param_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000869 if (!Param_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000870 slice_type = make_type("slice", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000871 if (!slice_type) return 0;
872 if (!add_attributes(slice_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000873 Slice_type = make_type("Slice", slice_type, Slice_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000874 if (!Slice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000875 ExtSlice_type = make_type("ExtSlice", slice_type, ExtSlice_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000876 if (!ExtSlice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000877 Index_type = make_type("Index", slice_type, Index_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000878 if (!Index_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000879 boolop_type = make_type("boolop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000880 if (!boolop_type) return 0;
881 if (!add_attributes(boolop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000882 And_type = make_type("And", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000883 if (!And_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000884 And_singleton = PyType_GenericNew(And_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000885 if (!And_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000886 Or_type = make_type("Or", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000887 if (!Or_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000888 Or_singleton = PyType_GenericNew(Or_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000889 if (!Or_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000890 operator_type = make_type("operator", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000891 if (!operator_type) return 0;
892 if (!add_attributes(operator_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000893 Add_type = make_type("Add", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000894 if (!Add_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000895 Add_singleton = PyType_GenericNew(Add_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000896 if (!Add_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000897 Sub_type = make_type("Sub", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000898 if (!Sub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000899 Sub_singleton = PyType_GenericNew(Sub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000900 if (!Sub_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000901 Mult_type = make_type("Mult", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000902 if (!Mult_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000903 Mult_singleton = PyType_GenericNew(Mult_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000904 if (!Mult_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000905 Div_type = make_type("Div", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000906 if (!Div_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000907 Div_singleton = PyType_GenericNew(Div_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000908 if (!Div_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000909 Mod_type = make_type("Mod", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000910 if (!Mod_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000911 Mod_singleton = PyType_GenericNew(Mod_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000912 if (!Mod_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000913 Pow_type = make_type("Pow", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000914 if (!Pow_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000915 Pow_singleton = PyType_GenericNew(Pow_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000916 if (!Pow_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000917 LShift_type = make_type("LShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000918 if (!LShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000919 LShift_singleton = PyType_GenericNew(LShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000920 if (!LShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000921 RShift_type = make_type("RShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000922 if (!RShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000923 RShift_singleton = PyType_GenericNew(RShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000924 if (!RShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000925 BitOr_type = make_type("BitOr", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000926 if (!BitOr_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000927 BitOr_singleton = PyType_GenericNew(BitOr_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000928 if (!BitOr_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000929 BitXor_type = make_type("BitXor", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000930 if (!BitXor_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000931 BitXor_singleton = PyType_GenericNew(BitXor_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000932 if (!BitXor_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000933 BitAnd_type = make_type("BitAnd", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000934 if (!BitAnd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000935 BitAnd_singleton = PyType_GenericNew(BitAnd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000936 if (!BitAnd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000937 FloorDiv_type = make_type("FloorDiv", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000938 if (!FloorDiv_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000939 FloorDiv_singleton = PyType_GenericNew(FloorDiv_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000940 if (!FloorDiv_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000941 unaryop_type = make_type("unaryop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000942 if (!unaryop_type) return 0;
943 if (!add_attributes(unaryop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000944 Invert_type = make_type("Invert", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000945 if (!Invert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000946 Invert_singleton = PyType_GenericNew(Invert_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000947 if (!Invert_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000948 Not_type = make_type("Not", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000949 if (!Not_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000950 Not_singleton = PyType_GenericNew(Not_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000951 if (!Not_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000952 UAdd_type = make_type("UAdd", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000953 if (!UAdd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000954 UAdd_singleton = PyType_GenericNew(UAdd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000955 if (!UAdd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000956 USub_type = make_type("USub", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000957 if (!USub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000958 USub_singleton = PyType_GenericNew(USub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000959 if (!USub_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000960 cmpop_type = make_type("cmpop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000961 if (!cmpop_type) return 0;
962 if (!add_attributes(cmpop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000963 Eq_type = make_type("Eq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000964 if (!Eq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000965 Eq_singleton = PyType_GenericNew(Eq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000966 if (!Eq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000967 NotEq_type = make_type("NotEq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000968 if (!NotEq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000969 NotEq_singleton = PyType_GenericNew(NotEq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000970 if (!NotEq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000971 Lt_type = make_type("Lt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000972 if (!Lt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000973 Lt_singleton = PyType_GenericNew(Lt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000974 if (!Lt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000975 LtE_type = make_type("LtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000976 if (!LtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000977 LtE_singleton = PyType_GenericNew(LtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000978 if (!LtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000979 Gt_type = make_type("Gt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000980 if (!Gt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000981 Gt_singleton = PyType_GenericNew(Gt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000982 if (!Gt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000983 GtE_type = make_type("GtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000984 if (!GtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000985 GtE_singleton = PyType_GenericNew(GtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000986 if (!GtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000987 Is_type = make_type("Is", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000988 if (!Is_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000989 Is_singleton = PyType_GenericNew(Is_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000990 if (!Is_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000991 IsNot_type = make_type("IsNot", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000992 if (!IsNot_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000993 IsNot_singleton = PyType_GenericNew(IsNot_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000994 if (!IsNot_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000995 In_type = make_type("In", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000996 if (!In_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000997 In_singleton = PyType_GenericNew(In_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000998 if (!In_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000999 NotIn_type = make_type("NotIn", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001000 if (!NotIn_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001001 NotIn_singleton = PyType_GenericNew(NotIn_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001002 if (!NotIn_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001003 comprehension_type = make_type("comprehension", &AST_type,
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001004 comprehension_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001005 if (!comprehension_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001006 excepthandler_type = make_type("excepthandler", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001007 if (!excepthandler_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001008 if (!add_attributes(excepthandler_type, excepthandler_attributes, 2))
1009 return 0;
1010 ExceptHandler_type = make_type("ExceptHandler", excepthandler_type,
1011 ExceptHandler_fields, 3);
1012 if (!ExceptHandler_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001013 arguments_type = make_type("arguments", &AST_type, arguments_fields, 8);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001014 if (!arguments_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001015 arg_type = make_type("arg", &AST_type, arg_fields, 2);
Neal Norwitzc1505362006-12-28 06:47:50 +00001016 if (!arg_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001017 keyword_type = make_type("keyword", &AST_type, keyword_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001018 if (!keyword_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001019 alias_type = make_type("alias", &AST_type, alias_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001020 if (!alias_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001021 withitem_type = make_type("withitem", &AST_type, withitem_fields, 2);
1022 if (!withitem_type) return 0;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001023 initialized = 1;
1024 return 1;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001025}
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001026
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001027static int obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena);
1028static int obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena);
1029static int obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena);
1030static int obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena*
1031 arena);
1032static int obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena);
1033static int obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena);
1034static int obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena);
1035static int obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena);
1036static int obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena);
1037static int obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena*
1038 arena);
1039static int obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena*
1040 arena);
1041static int obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena);
1042static int obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena);
1043static int obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena);
1044static int obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001045static int obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001046
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001047mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001048Module(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001049{
1050 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001051 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001052 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001053 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001054 p->kind = Module_kind;
1055 p->v.Module.body = body;
1056 return p;
1057}
1058
1059mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001060Interactive(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001061{
1062 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001063 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001064 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001065 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001066 p->kind = Interactive_kind;
1067 p->v.Interactive.body = body;
1068 return p;
1069}
1070
1071mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001072Expression(expr_ty body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001073{
1074 mod_ty p;
1075 if (!body) {
1076 PyErr_SetString(PyExc_ValueError,
1077 "field body is required for Expression");
1078 return NULL;
1079 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001080 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001081 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001082 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001083 p->kind = Expression_kind;
1084 p->v.Expression.body = body;
1085 return p;
1086}
1087
1088mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001089Suite(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001090{
1091 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001092 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001093 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001094 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001095 p->kind = Suite_kind;
1096 p->v.Suite.body = body;
1097 return p;
1098}
1099
1100stmt_ty
1101FunctionDef(identifier name, arguments_ty args, asdl_seq * body, asdl_seq *
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001102 decorator_list, expr_ty returns, int lineno, int col_offset,
1103 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001104{
1105 stmt_ty p;
1106 if (!name) {
1107 PyErr_SetString(PyExc_ValueError,
1108 "field name is required for FunctionDef");
1109 return NULL;
1110 }
1111 if (!args) {
1112 PyErr_SetString(PyExc_ValueError,
1113 "field args is required for FunctionDef");
1114 return NULL;
1115 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001116 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001117 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001118 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001119 p->kind = FunctionDef_kind;
1120 p->v.FunctionDef.name = name;
1121 p->v.FunctionDef.args = args;
1122 p->v.FunctionDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001123 p->v.FunctionDef.decorator_list = decorator_list;
Neal Norwitzc1505362006-12-28 06:47:50 +00001124 p->v.FunctionDef.returns = returns;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001125 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001126 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001127 return p;
1128}
1129
1130stmt_ty
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001131ClassDef(identifier name, asdl_seq * bases, asdl_seq * keywords, expr_ty
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001132 starargs, expr_ty kwargs, asdl_seq * body, asdl_seq * decorator_list,
1133 int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001134{
1135 stmt_ty p;
1136 if (!name) {
1137 PyErr_SetString(PyExc_ValueError,
1138 "field name is required for ClassDef");
1139 return NULL;
1140 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001141 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001142 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001143 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001144 p->kind = ClassDef_kind;
1145 p->v.ClassDef.name = name;
1146 p->v.ClassDef.bases = bases;
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001147 p->v.ClassDef.keywords = keywords;
1148 p->v.ClassDef.starargs = starargs;
1149 p->v.ClassDef.kwargs = kwargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001150 p->v.ClassDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001151 p->v.ClassDef.decorator_list = decorator_list;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001152 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001153 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001154 return p;
1155}
1156
1157stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001158Return(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001159{
1160 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001161 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001162 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001163 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001164 p->kind = Return_kind;
1165 p->v.Return.value = value;
1166 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001167 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001168 return p;
1169}
1170
1171stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001172Delete(asdl_seq * targets, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001173{
1174 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001175 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001176 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001177 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001178 p->kind = Delete_kind;
1179 p->v.Delete.targets = targets;
1180 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001181 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001182 return p;
1183}
1184
1185stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001186Assign(asdl_seq * targets, expr_ty value, int lineno, int col_offset, PyArena
1187 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001188{
1189 stmt_ty p;
1190 if (!value) {
1191 PyErr_SetString(PyExc_ValueError,
1192 "field value is required for Assign");
1193 return NULL;
1194 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001195 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001196 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001197 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001198 p->kind = Assign_kind;
1199 p->v.Assign.targets = targets;
1200 p->v.Assign.value = value;
1201 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001202 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001203 return p;
1204}
1205
1206stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001207AugAssign(expr_ty target, operator_ty op, expr_ty value, int lineno, int
1208 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001209{
1210 stmt_ty p;
1211 if (!target) {
1212 PyErr_SetString(PyExc_ValueError,
1213 "field target is required for AugAssign");
1214 return NULL;
1215 }
1216 if (!op) {
1217 PyErr_SetString(PyExc_ValueError,
1218 "field op is required for AugAssign");
1219 return NULL;
1220 }
1221 if (!value) {
1222 PyErr_SetString(PyExc_ValueError,
1223 "field value is required for AugAssign");
1224 return NULL;
1225 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001226 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001227 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001228 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001229 p->kind = AugAssign_kind;
1230 p->v.AugAssign.target = target;
1231 p->v.AugAssign.op = op;
1232 p->v.AugAssign.value = value;
1233 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001234 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001235 return p;
1236}
1237
1238stmt_ty
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001239For(expr_ty target, expr_ty iter, asdl_seq * body, asdl_seq * orelse, int
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001240 lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001241{
1242 stmt_ty p;
1243 if (!target) {
1244 PyErr_SetString(PyExc_ValueError,
1245 "field target is required for For");
1246 return NULL;
1247 }
1248 if (!iter) {
1249 PyErr_SetString(PyExc_ValueError,
1250 "field iter is required for For");
1251 return NULL;
1252 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001253 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001254 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001255 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001256 p->kind = For_kind;
1257 p->v.For.target = target;
1258 p->v.For.iter = iter;
1259 p->v.For.body = body;
1260 p->v.For.orelse = orelse;
1261 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001262 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001263 return p;
1264}
1265
1266stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001267While(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1268 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001269{
1270 stmt_ty p;
1271 if (!test) {
1272 PyErr_SetString(PyExc_ValueError,
1273 "field test is required for While");
1274 return NULL;
1275 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001276 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001277 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001278 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001279 p->kind = While_kind;
1280 p->v.While.test = test;
1281 p->v.While.body = body;
1282 p->v.While.orelse = orelse;
1283 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001284 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001285 return p;
1286}
1287
1288stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001289If(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1290 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001291{
1292 stmt_ty p;
1293 if (!test) {
1294 PyErr_SetString(PyExc_ValueError,
1295 "field test is required for If");
1296 return NULL;
1297 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001298 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001299 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001300 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001301 p->kind = If_kind;
1302 p->v.If.test = test;
1303 p->v.If.body = body;
1304 p->v.If.orelse = orelse;
1305 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001306 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001307 return p;
1308}
1309
1310stmt_ty
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001311With(asdl_seq * items, asdl_seq * body, int lineno, int col_offset, PyArena
1312 *arena)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001313{
1314 stmt_ty p;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001315 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001316 if (!p)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001317 return NULL;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001318 p->kind = With_kind;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001319 p->v.With.items = items;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001320 p->v.With.body = body;
1321 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001322 p->col_offset = col_offset;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001323 return p;
1324}
1325
1326stmt_ty
Collin Winter828f04a2007-08-31 00:04:24 +00001327Raise(expr_ty exc, expr_ty cause, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001328{
1329 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001330 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001331 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001332 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001333 p->kind = Raise_kind;
Collin Winter828f04a2007-08-31 00:04:24 +00001334 p->v.Raise.exc = exc;
1335 p->v.Raise.cause = cause;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001336 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001337 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001338 return p;
1339}
1340
1341stmt_ty
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001342Try(asdl_seq * body, asdl_seq * handlers, asdl_seq * orelse, asdl_seq *
1343 finalbody, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001344{
1345 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001346 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001347 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001348 return NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001349 p->kind = Try_kind;
1350 p->v.Try.body = body;
1351 p->v.Try.handlers = handlers;
1352 p->v.Try.orelse = orelse;
1353 p->v.Try.finalbody = finalbody;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001354 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001355 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001356 return p;
1357}
1358
1359stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001360Assert(expr_ty test, expr_ty msg, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001361{
1362 stmt_ty p;
1363 if (!test) {
1364 PyErr_SetString(PyExc_ValueError,
1365 "field test is required for Assert");
1366 return NULL;
1367 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001368 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001369 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001371 p->kind = Assert_kind;
1372 p->v.Assert.test = test;
1373 p->v.Assert.msg = msg;
1374 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001375 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001376 return p;
1377}
1378
1379stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001380Import(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001381{
1382 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001383 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001384 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001385 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001386 p->kind = Import_kind;
1387 p->v.Import.names = names;
1388 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001389 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001390 return p;
1391}
1392
1393stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001394ImportFrom(identifier module, asdl_seq * names, int level, int lineno, int
1395 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001396{
1397 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001398 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001399 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001400 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001401 p->kind = ImportFrom_kind;
1402 p->v.ImportFrom.module = module;
1403 p->v.ImportFrom.names = names;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001404 p->v.ImportFrom.level = level;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001405 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001406 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001407 return p;
1408}
1409
1410stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001411Global(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001412{
1413 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001414 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001415 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001416 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001417 p->kind = Global_kind;
1418 p->v.Global.names = names;
1419 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001420 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001421 return p;
1422}
1423
1424stmt_ty
Jeremy Hylton81e95022007-02-27 06:50:52 +00001425Nonlocal(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
1426{
1427 stmt_ty p;
1428 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
1429 if (!p)
1430 return NULL;
1431 p->kind = Nonlocal_kind;
1432 p->v.Nonlocal.names = names;
1433 p->lineno = lineno;
1434 p->col_offset = col_offset;
1435 return p;
1436}
1437
1438stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001439Expr(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001440{
1441 stmt_ty p;
1442 if (!value) {
1443 PyErr_SetString(PyExc_ValueError,
1444 "field value is required for Expr");
1445 return NULL;
1446 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001447 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001448 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001449 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001450 p->kind = Expr_kind;
1451 p->v.Expr.value = value;
1452 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001453 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001454 return p;
1455}
1456
1457stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001458Pass(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001459{
1460 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001461 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001462 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001463 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001464 p->kind = Pass_kind;
1465 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001466 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001467 return p;
1468}
1469
1470stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001471Break(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001472{
1473 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001474 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001475 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001476 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001477 p->kind = Break_kind;
1478 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001479 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001480 return p;
1481}
1482
1483stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001484Continue(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001485{
1486 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001487 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001488 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001489 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001490 p->kind = Continue_kind;
1491 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001492 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001493 return p;
1494}
1495
1496expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001497BoolOp(boolop_ty op, asdl_seq * values, int lineno, int col_offset, PyArena
1498 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001499{
1500 expr_ty p;
1501 if (!op) {
1502 PyErr_SetString(PyExc_ValueError,
1503 "field op is required for BoolOp");
1504 return NULL;
1505 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001506 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001507 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001508 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001509 p->kind = BoolOp_kind;
1510 p->v.BoolOp.op = op;
1511 p->v.BoolOp.values = values;
1512 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001513 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001514 return p;
1515}
1516
1517expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001518BinOp(expr_ty left, operator_ty op, expr_ty right, int lineno, int col_offset,
1519 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001520{
1521 expr_ty p;
1522 if (!left) {
1523 PyErr_SetString(PyExc_ValueError,
1524 "field left is required for BinOp");
1525 return NULL;
1526 }
1527 if (!op) {
1528 PyErr_SetString(PyExc_ValueError,
1529 "field op is required for BinOp");
1530 return NULL;
1531 }
1532 if (!right) {
1533 PyErr_SetString(PyExc_ValueError,
1534 "field right is required for BinOp");
1535 return NULL;
1536 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001537 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001538 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001539 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001540 p->kind = BinOp_kind;
1541 p->v.BinOp.left = left;
1542 p->v.BinOp.op = op;
1543 p->v.BinOp.right = right;
1544 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001545 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001546 return p;
1547}
1548
1549expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001550UnaryOp(unaryop_ty op, expr_ty operand, int lineno, int col_offset, PyArena
1551 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001552{
1553 expr_ty p;
1554 if (!op) {
1555 PyErr_SetString(PyExc_ValueError,
1556 "field op is required for UnaryOp");
1557 return NULL;
1558 }
1559 if (!operand) {
1560 PyErr_SetString(PyExc_ValueError,
1561 "field operand is required for UnaryOp");
1562 return NULL;
1563 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001564 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001565 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001566 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001567 p->kind = UnaryOp_kind;
1568 p->v.UnaryOp.op = op;
1569 p->v.UnaryOp.operand = operand;
1570 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001571 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001572 return p;
1573}
1574
1575expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001576Lambda(arguments_ty args, expr_ty body, int lineno, int col_offset, PyArena
1577 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001578{
1579 expr_ty p;
1580 if (!args) {
1581 PyErr_SetString(PyExc_ValueError,
1582 "field args is required for Lambda");
1583 return NULL;
1584 }
1585 if (!body) {
1586 PyErr_SetString(PyExc_ValueError,
1587 "field body is required for Lambda");
1588 return NULL;
1589 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001590 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001591 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001592 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001593 p->kind = Lambda_kind;
1594 p->v.Lambda.args = args;
1595 p->v.Lambda.body = body;
1596 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001597 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001598 return p;
1599}
1600
1601expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001602IfExp(expr_ty test, expr_ty body, expr_ty orelse, int lineno, int col_offset,
1603 PyArena *arena)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001604{
1605 expr_ty p;
1606 if (!test) {
1607 PyErr_SetString(PyExc_ValueError,
1608 "field test is required for IfExp");
1609 return NULL;
1610 }
1611 if (!body) {
1612 PyErr_SetString(PyExc_ValueError,
1613 "field body is required for IfExp");
1614 return NULL;
1615 }
1616 if (!orelse) {
1617 PyErr_SetString(PyExc_ValueError,
1618 "field orelse is required for IfExp");
1619 return NULL;
1620 }
1621 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001622 if (!p)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001623 return NULL;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001624 p->kind = IfExp_kind;
1625 p->v.IfExp.test = test;
1626 p->v.IfExp.body = body;
1627 p->v.IfExp.orelse = orelse;
1628 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001629 p->col_offset = col_offset;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001630 return p;
1631}
1632
1633expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001634Dict(asdl_seq * keys, asdl_seq * values, int lineno, int col_offset, PyArena
1635 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001636{
1637 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001638 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001639 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001640 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001641 p->kind = Dict_kind;
1642 p->v.Dict.keys = keys;
1643 p->v.Dict.values = values;
1644 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001645 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001646 return p;
1647}
1648
1649expr_ty
Guido van Rossum86e58e22006-08-28 15:27:34 +00001650Set(asdl_seq * elts, int lineno, int col_offset, PyArena *arena)
1651{
1652 expr_ty p;
1653 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001654 if (!p)
Guido van Rossum86e58e22006-08-28 15:27:34 +00001655 return NULL;
Guido van Rossum86e58e22006-08-28 15:27:34 +00001656 p->kind = Set_kind;
1657 p->v.Set.elts = elts;
1658 p->lineno = lineno;
1659 p->col_offset = col_offset;
1660 return p;
1661}
1662
1663expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001664ListComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1665 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001666{
1667 expr_ty p;
1668 if (!elt) {
1669 PyErr_SetString(PyExc_ValueError,
1670 "field elt is required for ListComp");
1671 return NULL;
1672 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001673 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001674 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001675 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001676 p->kind = ListComp_kind;
1677 p->v.ListComp.elt = elt;
1678 p->v.ListComp.generators = generators;
1679 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001680 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001681 return p;
1682}
1683
1684expr_ty
Nick Coghlan650f0d02007-04-15 12:05:43 +00001685SetComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset, PyArena
1686 *arena)
1687{
1688 expr_ty p;
1689 if (!elt) {
1690 PyErr_SetString(PyExc_ValueError,
1691 "field elt is required for SetComp");
1692 return NULL;
1693 }
1694 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1695 if (!p)
1696 return NULL;
1697 p->kind = SetComp_kind;
1698 p->v.SetComp.elt = elt;
1699 p->v.SetComp.generators = generators;
1700 p->lineno = lineno;
1701 p->col_offset = col_offset;
1702 return p;
1703}
1704
1705expr_ty
Guido van Rossum992d4a32007-07-11 13:09:30 +00001706DictComp(expr_ty key, expr_ty value, asdl_seq * generators, int lineno, int
1707 col_offset, PyArena *arena)
1708{
1709 expr_ty p;
1710 if (!key) {
1711 PyErr_SetString(PyExc_ValueError,
1712 "field key is required for DictComp");
1713 return NULL;
1714 }
1715 if (!value) {
1716 PyErr_SetString(PyExc_ValueError,
1717 "field value is required for DictComp");
1718 return NULL;
1719 }
1720 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1721 if (!p)
1722 return NULL;
1723 p->kind = DictComp_kind;
1724 p->v.DictComp.key = key;
1725 p->v.DictComp.value = value;
1726 p->v.DictComp.generators = generators;
1727 p->lineno = lineno;
1728 p->col_offset = col_offset;
1729 return p;
1730}
1731
1732expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001733GeneratorExp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1734 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001735{
1736 expr_ty p;
1737 if (!elt) {
1738 PyErr_SetString(PyExc_ValueError,
1739 "field elt is required for GeneratorExp");
1740 return NULL;
1741 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001742 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001743 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001744 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001745 p->kind = GeneratorExp_kind;
1746 p->v.GeneratorExp.elt = elt;
1747 p->v.GeneratorExp.generators = generators;
1748 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001749 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001750 return p;
1751}
1752
1753expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001754Yield(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001755{
1756 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001757 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001758 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001759 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001760 p->kind = Yield_kind;
1761 p->v.Yield.value = value;
1762 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001763 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001764 return p;
1765}
1766
1767expr_ty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001768Compare(expr_ty left, asdl_int_seq * ops, asdl_seq * comparators, int lineno,
1769 int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001770{
1771 expr_ty p;
1772 if (!left) {
1773 PyErr_SetString(PyExc_ValueError,
1774 "field left is required for Compare");
1775 return NULL;
1776 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001777 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001778 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001779 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001780 p->kind = Compare_kind;
1781 p->v.Compare.left = left;
1782 p->v.Compare.ops = ops;
1783 p->v.Compare.comparators = comparators;
1784 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001785 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001786 return p;
1787}
1788
1789expr_ty
1790Call(expr_ty func, asdl_seq * args, asdl_seq * keywords, expr_ty starargs,
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001791 expr_ty kwargs, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001792{
1793 expr_ty p;
1794 if (!func) {
1795 PyErr_SetString(PyExc_ValueError,
1796 "field func is required for Call");
1797 return NULL;
1798 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001799 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001800 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001801 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001802 p->kind = Call_kind;
1803 p->v.Call.func = func;
1804 p->v.Call.args = args;
1805 p->v.Call.keywords = keywords;
1806 p->v.Call.starargs = starargs;
1807 p->v.Call.kwargs = kwargs;
1808 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001809 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001810 return p;
1811}
1812
1813expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001814Num(object n, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001815{
1816 expr_ty p;
1817 if (!n) {
1818 PyErr_SetString(PyExc_ValueError,
1819 "field n is required for Num");
1820 return NULL;
1821 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001822 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001823 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001824 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001825 p->kind = Num_kind;
1826 p->v.Num.n = n;
1827 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001828 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001829 return p;
1830}
1831
1832expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001833Str(string s, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001834{
1835 expr_ty p;
1836 if (!s) {
1837 PyErr_SetString(PyExc_ValueError,
1838 "field s is required for Str");
1839 return NULL;
1840 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001841 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001842 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001843 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001844 p->kind = Str_kind;
1845 p->v.Str.s = s;
1846 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001847 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001848 return p;
1849}
1850
1851expr_ty
Benjamin Petersone2498412011-08-09 16:08:39 -05001852Bytes(bytes s, int lineno, int col_offset, PyArena *arena)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001853{
1854 expr_ty p;
1855 if (!s) {
1856 PyErr_SetString(PyExc_ValueError,
1857 "field s is required for Bytes");
1858 return NULL;
1859 }
1860 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001861 if (!p)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001862 return NULL;
Thomas Wouters00e41de2007-02-23 19:56:57 +00001863 p->kind = Bytes_kind;
1864 p->v.Bytes.s = s;
1865 p->lineno = lineno;
1866 p->col_offset = col_offset;
1867 return p;
1868}
1869
1870expr_ty
Georg Brandl52318d62006-09-06 07:06:08 +00001871Ellipsis(int lineno, int col_offset, PyArena *arena)
1872{
1873 expr_ty p;
1874 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001875 if (!p)
Georg Brandl52318d62006-09-06 07:06:08 +00001876 return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00001877 p->kind = Ellipsis_kind;
1878 p->lineno = lineno;
1879 p->col_offset = col_offset;
1880 return p;
1881}
1882
1883expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001884Attribute(expr_ty value, identifier attr, expr_context_ty ctx, int lineno, int
1885 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001886{
1887 expr_ty p;
1888 if (!value) {
1889 PyErr_SetString(PyExc_ValueError,
1890 "field value is required for Attribute");
1891 return NULL;
1892 }
1893 if (!attr) {
1894 PyErr_SetString(PyExc_ValueError,
1895 "field attr is required for Attribute");
1896 return NULL;
1897 }
1898 if (!ctx) {
1899 PyErr_SetString(PyExc_ValueError,
1900 "field ctx is required for Attribute");
1901 return NULL;
1902 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001903 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001904 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001905 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001906 p->kind = Attribute_kind;
1907 p->v.Attribute.value = value;
1908 p->v.Attribute.attr = attr;
1909 p->v.Attribute.ctx = ctx;
1910 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001911 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001912 return p;
1913}
1914
1915expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001916Subscript(expr_ty value, slice_ty slice, expr_context_ty ctx, int lineno, int
1917 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001918{
1919 expr_ty p;
1920 if (!value) {
1921 PyErr_SetString(PyExc_ValueError,
1922 "field value is required for Subscript");
1923 return NULL;
1924 }
1925 if (!slice) {
1926 PyErr_SetString(PyExc_ValueError,
1927 "field slice is required for Subscript");
1928 return NULL;
1929 }
1930 if (!ctx) {
1931 PyErr_SetString(PyExc_ValueError,
1932 "field ctx is required for Subscript");
1933 return NULL;
1934 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001935 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001936 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001937 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001938 p->kind = Subscript_kind;
1939 p->v.Subscript.value = value;
1940 p->v.Subscript.slice = slice;
1941 p->v.Subscript.ctx = ctx;
1942 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001943 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001944 return p;
1945}
1946
1947expr_ty
Guido van Rossum0368b722007-05-11 16:50:42 +00001948Starred(expr_ty value, expr_context_ty ctx, int lineno, int col_offset, PyArena
1949 *arena)
1950{
1951 expr_ty p;
1952 if (!value) {
1953 PyErr_SetString(PyExc_ValueError,
1954 "field value is required for Starred");
1955 return NULL;
1956 }
1957 if (!ctx) {
1958 PyErr_SetString(PyExc_ValueError,
1959 "field ctx is required for Starred");
1960 return NULL;
1961 }
1962 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1963 if (!p)
1964 return NULL;
1965 p->kind = Starred_kind;
1966 p->v.Starred.value = value;
1967 p->v.Starred.ctx = ctx;
1968 p->lineno = lineno;
1969 p->col_offset = col_offset;
1970 return p;
1971}
1972
1973expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001974Name(identifier id, expr_context_ty ctx, int lineno, int col_offset, PyArena
1975 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001976{
1977 expr_ty p;
1978 if (!id) {
1979 PyErr_SetString(PyExc_ValueError,
1980 "field id is required for Name");
1981 return NULL;
1982 }
1983 if (!ctx) {
1984 PyErr_SetString(PyExc_ValueError,
1985 "field ctx is required for Name");
1986 return NULL;
1987 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001988 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001989 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001990 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001991 p->kind = Name_kind;
1992 p->v.Name.id = id;
1993 p->v.Name.ctx = ctx;
1994 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001995 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001996 return p;
1997}
1998
1999expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002000List(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2001 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002002{
2003 expr_ty p;
2004 if (!ctx) {
2005 PyErr_SetString(PyExc_ValueError,
2006 "field ctx is required for List");
2007 return NULL;
2008 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002009 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002010 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002011 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002012 p->kind = List_kind;
2013 p->v.List.elts = elts;
2014 p->v.List.ctx = ctx;
2015 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002016 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002017 return p;
2018}
2019
2020expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002021Tuple(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2022 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002023{
2024 expr_ty p;
2025 if (!ctx) {
2026 PyErr_SetString(PyExc_ValueError,
2027 "field ctx is required for Tuple");
2028 return NULL;
2029 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002030 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002031 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002032 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002033 p->kind = Tuple_kind;
2034 p->v.Tuple.elts = elts;
2035 p->v.Tuple.ctx = ctx;
2036 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002037 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002038 return p;
2039}
2040
2041slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002042Slice(expr_ty lower, expr_ty upper, expr_ty step, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002043{
2044 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002045 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002046 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002047 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002048 p->kind = Slice_kind;
2049 p->v.Slice.lower = lower;
2050 p->v.Slice.upper = upper;
2051 p->v.Slice.step = step;
2052 return p;
2053}
2054
2055slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002056ExtSlice(asdl_seq * dims, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002057{
2058 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002059 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002060 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002061 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002062 p->kind = ExtSlice_kind;
2063 p->v.ExtSlice.dims = dims;
2064 return p;
2065}
2066
2067slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002068Index(expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002069{
2070 slice_ty p;
2071 if (!value) {
2072 PyErr_SetString(PyExc_ValueError,
2073 "field value is required for Index");
2074 return NULL;
2075 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002076 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002077 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002078 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002079 p->kind = Index_kind;
2080 p->v.Index.value = value;
2081 return p;
2082}
2083
2084comprehension_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002085comprehension(expr_ty target, expr_ty iter, asdl_seq * ifs, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002086{
2087 comprehension_ty p;
2088 if (!target) {
2089 PyErr_SetString(PyExc_ValueError,
2090 "field target is required for comprehension");
2091 return NULL;
2092 }
2093 if (!iter) {
2094 PyErr_SetString(PyExc_ValueError,
2095 "field iter is required for comprehension");
2096 return NULL;
2097 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002098 p = (comprehension_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002099 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002100 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002101 p->target = target;
2102 p->iter = iter;
2103 p->ifs = ifs;
2104 return p;
2105}
2106
2107excepthandler_ty
Neal Norwitzad74aa82008-03-31 05:14:30 +00002108ExceptHandler(expr_ty type, identifier name, asdl_seq * body, int lineno, int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002109 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002110{
2111 excepthandler_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002112 p = (excepthandler_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002113 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002114 return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00002115 p->kind = ExceptHandler_kind;
2116 p->v.ExceptHandler.type = type;
2117 p->v.ExceptHandler.name = name;
2118 p->v.ExceptHandler.body = body;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002119 p->lineno = lineno;
2120 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002121 return p;
2122}
2123
2124arguments_ty
Neal Norwitzc1505362006-12-28 06:47:50 +00002125arguments(asdl_seq * args, identifier vararg, expr_ty varargannotation,
2126 asdl_seq * kwonlyargs, identifier kwarg, expr_ty kwargannotation,
2127 asdl_seq * defaults, asdl_seq * kw_defaults, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002128{
2129 arguments_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002130 p = (arguments_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002131 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002132 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002133 p->args = args;
2134 p->vararg = vararg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002135 p->varargannotation = varargannotation;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002136 p->kwonlyargs = kwonlyargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002137 p->kwarg = kwarg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002138 p->kwargannotation = kwargannotation;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002139 p->defaults = defaults;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002140 p->kw_defaults = kw_defaults;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002141 return p;
2142}
2143
Neal Norwitzc1505362006-12-28 06:47:50 +00002144arg_ty
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002145arg(identifier arg, expr_ty annotation, PyArena *arena)
Neal Norwitzc1505362006-12-28 06:47:50 +00002146{
2147 arg_ty p;
2148 if (!arg) {
2149 PyErr_SetString(PyExc_ValueError,
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002150 "field arg is required for arg");
Neal Norwitzc1505362006-12-28 06:47:50 +00002151 return NULL;
2152 }
2153 p = (arg_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002154 if (!p)
Neal Norwitzc1505362006-12-28 06:47:50 +00002155 return NULL;
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002156 p->arg = arg;
2157 p->annotation = annotation;
Neal Norwitzc1505362006-12-28 06:47:50 +00002158 return p;
2159}
2160
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002161keyword_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002162keyword(identifier arg, expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002163{
2164 keyword_ty p;
2165 if (!arg) {
2166 PyErr_SetString(PyExc_ValueError,
2167 "field arg is required for keyword");
2168 return NULL;
2169 }
2170 if (!value) {
2171 PyErr_SetString(PyExc_ValueError,
2172 "field value is required for keyword");
2173 return NULL;
2174 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002175 p = (keyword_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002176 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002177 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002178 p->arg = arg;
2179 p->value = value;
2180 return p;
2181}
2182
2183alias_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002184alias(identifier name, identifier asname, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002185{
2186 alias_ty p;
2187 if (!name) {
2188 PyErr_SetString(PyExc_ValueError,
2189 "field name is required for alias");
2190 return NULL;
2191 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002192 p = (alias_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002193 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002194 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002195 p->name = name;
2196 p->asname = asname;
2197 return p;
2198}
2199
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002200withitem_ty
2201withitem(expr_ty context_expr, expr_ty optional_vars, PyArena *arena)
2202{
2203 withitem_ty p;
2204 if (!context_expr) {
2205 PyErr_SetString(PyExc_ValueError,
2206 "field context_expr is required for withitem");
2207 return NULL;
2208 }
2209 p = (withitem_ty)PyArena_Malloc(arena, sizeof(*p));
2210 if (!p)
2211 return NULL;
2212 p->context_expr = context_expr;
2213 p->optional_vars = optional_vars;
2214 return p;
2215}
2216
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002217
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002218PyObject*
2219ast2obj_mod(void* _o)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002220{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002221 mod_ty o = (mod_ty)_o;
2222 PyObject *result = NULL, *value = NULL;
2223 if (!o) {
2224 Py_INCREF(Py_None);
2225 return Py_None;
2226 }
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002227
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002228 switch (o->kind) {
2229 case Module_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002230 result = PyType_GenericNew(Module_type, NULL, NULL);
2231 if (!result) goto failed;
2232 value = ast2obj_list(o->v.Module.body, ast2obj_stmt);
2233 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002234 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002235 goto failed;
2236 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002237 break;
2238 case Interactive_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002239 result = PyType_GenericNew(Interactive_type, NULL, NULL);
2240 if (!result) goto failed;
2241 value = ast2obj_list(o->v.Interactive.body, ast2obj_stmt);
2242 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002243 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002244 goto failed;
2245 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002246 break;
2247 case Expression_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002248 result = PyType_GenericNew(Expression_type, NULL, NULL);
2249 if (!result) goto failed;
2250 value = ast2obj_expr(o->v.Expression.body);
2251 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002252 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002253 goto failed;
2254 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002255 break;
2256 case Suite_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002257 result = PyType_GenericNew(Suite_type, NULL, NULL);
2258 if (!result) goto failed;
2259 value = ast2obj_list(o->v.Suite.body, ast2obj_stmt);
2260 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002261 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002262 goto failed;
2263 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002264 break;
2265 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002266 return result;
2267failed:
2268 Py_XDECREF(value);
2269 Py_XDECREF(result);
2270 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002271}
2272
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002273PyObject*
2274ast2obj_stmt(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002275{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002276 stmt_ty o = (stmt_ty)_o;
2277 PyObject *result = NULL, *value = NULL;
2278 if (!o) {
2279 Py_INCREF(Py_None);
2280 return Py_None;
2281 }
2282
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002283 switch (o->kind) {
2284 case FunctionDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002285 result = PyType_GenericNew(FunctionDef_type, NULL, NULL);
2286 if (!result) goto failed;
2287 value = ast2obj_identifier(o->v.FunctionDef.name);
2288 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002289 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002290 goto failed;
2291 Py_DECREF(value);
2292 value = ast2obj_arguments(o->v.FunctionDef.args);
2293 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002294 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002295 goto failed;
2296 Py_DECREF(value);
2297 value = ast2obj_list(o->v.FunctionDef.body, ast2obj_stmt);
2298 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002299 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002300 goto failed;
2301 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002302 value = ast2obj_list(o->v.FunctionDef.decorator_list,
2303 ast2obj_expr);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002304 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002305 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002306 -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002307 goto failed;
2308 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00002309 value = ast2obj_expr(o->v.FunctionDef.returns);
2310 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002311 if (_PyObject_SetAttrId(result, &PyId_returns, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00002312 goto failed;
2313 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002314 break;
2315 case ClassDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002316 result = PyType_GenericNew(ClassDef_type, NULL, NULL);
2317 if (!result) goto failed;
2318 value = ast2obj_identifier(o->v.ClassDef.name);
2319 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002320 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002321 goto failed;
2322 Py_DECREF(value);
2323 value = ast2obj_list(o->v.ClassDef.bases, ast2obj_expr);
2324 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002325 if (_PyObject_SetAttrId(result, &PyId_bases, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002326 goto failed;
2327 Py_DECREF(value);
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002328 value = ast2obj_list(o->v.ClassDef.keywords, ast2obj_keyword);
2329 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002330 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002331 goto failed;
2332 Py_DECREF(value);
2333 value = ast2obj_expr(o->v.ClassDef.starargs);
2334 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002335 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002336 goto failed;
2337 Py_DECREF(value);
2338 value = ast2obj_expr(o->v.ClassDef.kwargs);
2339 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002340 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002341 goto failed;
2342 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002343 value = ast2obj_list(o->v.ClassDef.body, ast2obj_stmt);
2344 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002345 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002346 goto failed;
2347 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002348 value = ast2obj_list(o->v.ClassDef.decorator_list,
2349 ast2obj_expr);
2350 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002351 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002352 -1)
2353 goto failed;
2354 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002355 break;
2356 case Return_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002357 result = PyType_GenericNew(Return_type, NULL, NULL);
2358 if (!result) goto failed;
2359 value = ast2obj_expr(o->v.Return.value);
2360 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002361 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002362 goto failed;
2363 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002364 break;
2365 case Delete_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002366 result = PyType_GenericNew(Delete_type, NULL, NULL);
2367 if (!result) goto failed;
2368 value = ast2obj_list(o->v.Delete.targets, ast2obj_expr);
2369 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002370 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002371 goto failed;
2372 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002373 break;
2374 case Assign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002375 result = PyType_GenericNew(Assign_type, NULL, NULL);
2376 if (!result) goto failed;
2377 value = ast2obj_list(o->v.Assign.targets, ast2obj_expr);
2378 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002379 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002380 goto failed;
2381 Py_DECREF(value);
2382 value = ast2obj_expr(o->v.Assign.value);
2383 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002384 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002385 goto failed;
2386 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002387 break;
2388 case AugAssign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002389 result = PyType_GenericNew(AugAssign_type, NULL, NULL);
2390 if (!result) goto failed;
2391 value = ast2obj_expr(o->v.AugAssign.target);
2392 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002393 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002394 goto failed;
2395 Py_DECREF(value);
2396 value = ast2obj_operator(o->v.AugAssign.op);
2397 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002398 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002399 goto failed;
2400 Py_DECREF(value);
2401 value = ast2obj_expr(o->v.AugAssign.value);
2402 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002403 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002404 goto failed;
2405 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002406 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002407 case For_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002408 result = PyType_GenericNew(For_type, NULL, NULL);
2409 if (!result) goto failed;
2410 value = ast2obj_expr(o->v.For.target);
2411 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002412 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002413 goto failed;
2414 Py_DECREF(value);
2415 value = ast2obj_expr(o->v.For.iter);
2416 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002417 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002418 goto failed;
2419 Py_DECREF(value);
2420 value = ast2obj_list(o->v.For.body, ast2obj_stmt);
2421 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002422 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002423 goto failed;
2424 Py_DECREF(value);
2425 value = ast2obj_list(o->v.For.orelse, ast2obj_stmt);
2426 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002427 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002428 goto failed;
2429 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002430 break;
2431 case While_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002432 result = PyType_GenericNew(While_type, NULL, NULL);
2433 if (!result) goto failed;
2434 value = ast2obj_expr(o->v.While.test);
2435 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002436 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002437 goto failed;
2438 Py_DECREF(value);
2439 value = ast2obj_list(o->v.While.body, ast2obj_stmt);
2440 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002441 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002442 goto failed;
2443 Py_DECREF(value);
2444 value = ast2obj_list(o->v.While.orelse, ast2obj_stmt);
2445 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002446 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002447 goto failed;
2448 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002449 break;
2450 case If_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002451 result = PyType_GenericNew(If_type, NULL, NULL);
2452 if (!result) goto failed;
2453 value = ast2obj_expr(o->v.If.test);
2454 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002455 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002456 goto failed;
2457 Py_DECREF(value);
2458 value = ast2obj_list(o->v.If.body, ast2obj_stmt);
2459 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002460 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002461 goto failed;
2462 Py_DECREF(value);
2463 value = ast2obj_list(o->v.If.orelse, ast2obj_stmt);
2464 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002465 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002466 goto failed;
2467 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002468 break;
Guido van Rossumc2e20742006-02-27 22:32:47 +00002469 case With_kind:
2470 result = PyType_GenericNew(With_type, NULL, NULL);
2471 if (!result) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002472 value = ast2obj_list(o->v.With.items, ast2obj_withitem);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002473 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002474 if (_PyObject_SetAttrId(result, &PyId_items, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002475 goto failed;
2476 Py_DECREF(value);
2477 value = ast2obj_list(o->v.With.body, ast2obj_stmt);
2478 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002479 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002480 goto failed;
2481 Py_DECREF(value);
2482 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002483 case Raise_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002484 result = PyType_GenericNew(Raise_type, NULL, NULL);
2485 if (!result) goto failed;
Collin Winter828f04a2007-08-31 00:04:24 +00002486 value = ast2obj_expr(o->v.Raise.exc);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002487 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002488 if (_PyObject_SetAttrId(result, &PyId_exc, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002489 goto failed;
2490 Py_DECREF(value);
Collin Winter828f04a2007-08-31 00:04:24 +00002491 value = ast2obj_expr(o->v.Raise.cause);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002492 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002493 if (_PyObject_SetAttrId(result, &PyId_cause, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002494 goto failed;
2495 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002496 break;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002497 case Try_kind:
2498 result = PyType_GenericNew(Try_type, NULL, NULL);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002499 if (!result) goto failed;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002500 value = ast2obj_list(o->v.Try.body, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002501 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002502 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002503 goto failed;
2504 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002505 value = ast2obj_list(o->v.Try.handlers, ast2obj_excepthandler);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002506 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002507 if (_PyObject_SetAttrId(result, &PyId_handlers, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002508 goto failed;
2509 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002510 value = ast2obj_list(o->v.Try.orelse, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002511 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002512 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002513 goto failed;
2514 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002515 value = ast2obj_list(o->v.Try.finalbody, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002516 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002517 if (_PyObject_SetAttrId(result, &PyId_finalbody, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002518 goto failed;
2519 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002520 break;
2521 case Assert_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002522 result = PyType_GenericNew(Assert_type, NULL, NULL);
2523 if (!result) goto failed;
2524 value = ast2obj_expr(o->v.Assert.test);
2525 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002526 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002527 goto failed;
2528 Py_DECREF(value);
2529 value = ast2obj_expr(o->v.Assert.msg);
2530 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002531 if (_PyObject_SetAttrId(result, &PyId_msg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002532 goto failed;
2533 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002534 break;
2535 case Import_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002536 result = PyType_GenericNew(Import_type, NULL, NULL);
2537 if (!result) goto failed;
2538 value = ast2obj_list(o->v.Import.names, ast2obj_alias);
2539 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002540 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002541 goto failed;
2542 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002543 break;
2544 case ImportFrom_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002545 result = PyType_GenericNew(ImportFrom_type, NULL, NULL);
2546 if (!result) goto failed;
2547 value = ast2obj_identifier(o->v.ImportFrom.module);
2548 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002549 if (_PyObject_SetAttrId(result, &PyId_module, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002550 goto failed;
2551 Py_DECREF(value);
2552 value = ast2obj_list(o->v.ImportFrom.names, ast2obj_alias);
2553 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002554 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002555 goto failed;
2556 Py_DECREF(value);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002557 value = ast2obj_int(o->v.ImportFrom.level);
2558 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002559 if (_PyObject_SetAttrId(result, &PyId_level, value) == -1)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002560 goto failed;
2561 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002562 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002563 case Global_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002564 result = PyType_GenericNew(Global_type, NULL, NULL);
2565 if (!result) goto failed;
2566 value = ast2obj_list(o->v.Global.names, ast2obj_identifier);
2567 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002568 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002569 goto failed;
2570 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002571 break;
Jeremy Hylton81e95022007-02-27 06:50:52 +00002572 case Nonlocal_kind:
2573 result = PyType_GenericNew(Nonlocal_type, NULL, NULL);
2574 if (!result) goto failed;
2575 value = ast2obj_list(o->v.Nonlocal.names, ast2obj_identifier);
2576 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002577 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Jeremy Hylton81e95022007-02-27 06:50:52 +00002578 goto failed;
2579 Py_DECREF(value);
2580 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002581 case Expr_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002582 result = PyType_GenericNew(Expr_type, NULL, NULL);
2583 if (!result) goto failed;
2584 value = ast2obj_expr(o->v.Expr.value);
2585 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002586 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002587 goto failed;
2588 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002589 break;
2590 case Pass_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002591 result = PyType_GenericNew(Pass_type, NULL, NULL);
2592 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002593 break;
2594 case Break_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002595 result = PyType_GenericNew(Break_type, NULL, NULL);
2596 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002597 break;
2598 case Continue_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002599 result = PyType_GenericNew(Continue_type, NULL, NULL);
2600 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002601 break;
2602 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002603 value = ast2obj_int(o->lineno);
2604 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002605 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002606 goto failed;
2607 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002608 value = ast2obj_int(o->col_offset);
2609 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002610 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002611 goto failed;
2612 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002613 return result;
2614failed:
2615 Py_XDECREF(value);
2616 Py_XDECREF(result);
2617 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002618}
2619
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002620PyObject*
2621ast2obj_expr(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002622{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002623 expr_ty o = (expr_ty)_o;
2624 PyObject *result = NULL, *value = NULL;
2625 if (!o) {
2626 Py_INCREF(Py_None);
2627 return Py_None;
2628 }
2629
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002630 switch (o->kind) {
2631 case BoolOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002632 result = PyType_GenericNew(BoolOp_type, NULL, NULL);
2633 if (!result) goto failed;
2634 value = ast2obj_boolop(o->v.BoolOp.op);
2635 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002636 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002637 goto failed;
2638 Py_DECREF(value);
2639 value = ast2obj_list(o->v.BoolOp.values, ast2obj_expr);
2640 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002641 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002642 goto failed;
2643 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002644 break;
2645 case BinOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002646 result = PyType_GenericNew(BinOp_type, NULL, NULL);
2647 if (!result) goto failed;
2648 value = ast2obj_expr(o->v.BinOp.left);
2649 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002650 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002651 goto failed;
2652 Py_DECREF(value);
2653 value = ast2obj_operator(o->v.BinOp.op);
2654 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002655 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002656 goto failed;
2657 Py_DECREF(value);
2658 value = ast2obj_expr(o->v.BinOp.right);
2659 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002660 if (_PyObject_SetAttrId(result, &PyId_right, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002661 goto failed;
2662 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002663 break;
2664 case UnaryOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002665 result = PyType_GenericNew(UnaryOp_type, NULL, NULL);
2666 if (!result) goto failed;
2667 value = ast2obj_unaryop(o->v.UnaryOp.op);
2668 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002669 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002670 goto failed;
2671 Py_DECREF(value);
2672 value = ast2obj_expr(o->v.UnaryOp.operand);
2673 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002674 if (_PyObject_SetAttrId(result, &PyId_operand, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002675 goto failed;
2676 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002677 break;
2678 case Lambda_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002679 result = PyType_GenericNew(Lambda_type, NULL, NULL);
2680 if (!result) goto failed;
2681 value = ast2obj_arguments(o->v.Lambda.args);
2682 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002683 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002684 goto failed;
2685 Py_DECREF(value);
2686 value = ast2obj_expr(o->v.Lambda.body);
2687 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002688 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002689 goto failed;
2690 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002691 break;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002692 case IfExp_kind:
2693 result = PyType_GenericNew(IfExp_type, NULL, NULL);
2694 if (!result) goto failed;
2695 value = ast2obj_expr(o->v.IfExp.test);
2696 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002697 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002698 goto failed;
2699 Py_DECREF(value);
2700 value = ast2obj_expr(o->v.IfExp.body);
2701 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002702 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002703 goto failed;
2704 Py_DECREF(value);
2705 value = ast2obj_expr(o->v.IfExp.orelse);
2706 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002707 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002708 goto failed;
2709 Py_DECREF(value);
2710 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002711 case Dict_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002712 result = PyType_GenericNew(Dict_type, NULL, NULL);
2713 if (!result) goto failed;
2714 value = ast2obj_list(o->v.Dict.keys, ast2obj_expr);
2715 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002716 if (_PyObject_SetAttrId(result, &PyId_keys, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002717 goto failed;
2718 Py_DECREF(value);
2719 value = ast2obj_list(o->v.Dict.values, ast2obj_expr);
2720 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002721 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002722 goto failed;
2723 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002724 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002725 case Set_kind:
2726 result = PyType_GenericNew(Set_type, NULL, NULL);
2727 if (!result) goto failed;
2728 value = ast2obj_list(o->v.Set.elts, ast2obj_expr);
2729 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002730 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Guido van Rossum86e58e22006-08-28 15:27:34 +00002731 goto failed;
2732 Py_DECREF(value);
2733 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002734 case ListComp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002735 result = PyType_GenericNew(ListComp_type, NULL, NULL);
2736 if (!result) goto failed;
2737 value = ast2obj_expr(o->v.ListComp.elt);
2738 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002739 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002740 goto failed;
2741 Py_DECREF(value);
2742 value = ast2obj_list(o->v.ListComp.generators,
2743 ast2obj_comprehension);
2744 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002745 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002746 goto failed;
2747 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002748 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00002749 case SetComp_kind:
2750 result = PyType_GenericNew(SetComp_type, NULL, NULL);
2751 if (!result) goto failed;
2752 value = ast2obj_expr(o->v.SetComp.elt);
2753 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002754 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002755 goto failed;
2756 Py_DECREF(value);
2757 value = ast2obj_list(o->v.SetComp.generators,
2758 ast2obj_comprehension);
2759 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002760 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002761 goto failed;
2762 Py_DECREF(value);
2763 break;
Guido van Rossum992d4a32007-07-11 13:09:30 +00002764 case DictComp_kind:
2765 result = PyType_GenericNew(DictComp_type, NULL, NULL);
2766 if (!result) goto failed;
2767 value = ast2obj_expr(o->v.DictComp.key);
2768 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002769 if (_PyObject_SetAttrId(result, &PyId_key, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002770 goto failed;
2771 Py_DECREF(value);
2772 value = ast2obj_expr(o->v.DictComp.value);
2773 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002774 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002775 goto failed;
2776 Py_DECREF(value);
2777 value = ast2obj_list(o->v.DictComp.generators,
2778 ast2obj_comprehension);
2779 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002780 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002781 goto failed;
2782 Py_DECREF(value);
2783 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002784 case GeneratorExp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002785 result = PyType_GenericNew(GeneratorExp_type, NULL, NULL);
2786 if (!result) goto failed;
2787 value = ast2obj_expr(o->v.GeneratorExp.elt);
2788 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002789 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002790 goto failed;
2791 Py_DECREF(value);
2792 value = ast2obj_list(o->v.GeneratorExp.generators,
2793 ast2obj_comprehension);
2794 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002795 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002796 goto failed;
2797 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002798 break;
2799 case Yield_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002800 result = PyType_GenericNew(Yield_type, NULL, NULL);
2801 if (!result) goto failed;
2802 value = ast2obj_expr(o->v.Yield.value);
2803 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002804 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002805 goto failed;
2806 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002807 break;
2808 case Compare_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002809 result = PyType_GenericNew(Compare_type, NULL, NULL);
2810 if (!result) goto failed;
2811 value = ast2obj_expr(o->v.Compare.left);
2812 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002813 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002814 goto failed;
2815 Py_DECREF(value);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002816 {
2817 int i, n = asdl_seq_LEN(o->v.Compare.ops);
2818 value = PyList_New(n);
2819 if (!value) goto failed;
2820 for(i = 0; i < n; i++)
2821 PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(o->v.Compare.ops, i)));
2822 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002823 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002824 if (_PyObject_SetAttrId(result, &PyId_ops, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002825 goto failed;
2826 Py_DECREF(value);
2827 value = ast2obj_list(o->v.Compare.comparators, ast2obj_expr);
2828 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002829 if (_PyObject_SetAttrId(result, &PyId_comparators, 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 Call_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002834 result = PyType_GenericNew(Call_type, NULL, NULL);
2835 if (!result) goto failed;
2836 value = ast2obj_expr(o->v.Call.func);
2837 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002838 if (_PyObject_SetAttrId(result, &PyId_func, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002839 goto failed;
2840 Py_DECREF(value);
2841 value = ast2obj_list(o->v.Call.args, ast2obj_expr);
2842 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002843 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002844 goto failed;
2845 Py_DECREF(value);
2846 value = ast2obj_list(o->v.Call.keywords, ast2obj_keyword);
2847 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002848 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002849 goto failed;
2850 Py_DECREF(value);
2851 value = ast2obj_expr(o->v.Call.starargs);
2852 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002853 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002854 goto failed;
2855 Py_DECREF(value);
2856 value = ast2obj_expr(o->v.Call.kwargs);
2857 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002858 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002859 goto failed;
2860 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002861 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002862 case Num_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002863 result = PyType_GenericNew(Num_type, NULL, NULL);
2864 if (!result) goto failed;
2865 value = ast2obj_object(o->v.Num.n);
2866 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002867 if (_PyObject_SetAttrId(result, &PyId_n, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002868 goto failed;
2869 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002870 break;
2871 case Str_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002872 result = PyType_GenericNew(Str_type, NULL, NULL);
2873 if (!result) goto failed;
2874 value = ast2obj_string(o->v.Str.s);
2875 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002876 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002877 goto failed;
2878 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002879 break;
Thomas Wouters00e41de2007-02-23 19:56:57 +00002880 case Bytes_kind:
2881 result = PyType_GenericNew(Bytes_type, NULL, NULL);
2882 if (!result) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05002883 value = ast2obj_bytes(o->v.Bytes.s);
Thomas Wouters00e41de2007-02-23 19:56:57 +00002884 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002885 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Thomas Wouters00e41de2007-02-23 19:56:57 +00002886 goto failed;
2887 Py_DECREF(value);
2888 break;
Georg Brandl52318d62006-09-06 07:06:08 +00002889 case Ellipsis_kind:
2890 result = PyType_GenericNew(Ellipsis_type, NULL, NULL);
2891 if (!result) goto failed;
2892 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002893 case Attribute_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002894 result = PyType_GenericNew(Attribute_type, NULL, NULL);
2895 if (!result) goto failed;
2896 value = ast2obj_expr(o->v.Attribute.value);
2897 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002898 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002899 goto failed;
2900 Py_DECREF(value);
2901 value = ast2obj_identifier(o->v.Attribute.attr);
2902 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002903 if (_PyObject_SetAttrId(result, &PyId_attr, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002904 goto failed;
2905 Py_DECREF(value);
2906 value = ast2obj_expr_context(o->v.Attribute.ctx);
2907 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002908 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002909 goto failed;
2910 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002911 break;
2912 case Subscript_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002913 result = PyType_GenericNew(Subscript_type, NULL, NULL);
2914 if (!result) goto failed;
2915 value = ast2obj_expr(o->v.Subscript.value);
2916 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002917 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002918 goto failed;
2919 Py_DECREF(value);
2920 value = ast2obj_slice(o->v.Subscript.slice);
2921 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002922 if (_PyObject_SetAttrId(result, &PyId_slice, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002923 goto failed;
2924 Py_DECREF(value);
2925 value = ast2obj_expr_context(o->v.Subscript.ctx);
2926 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002927 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002928 goto failed;
2929 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002930 break;
Guido van Rossum0368b722007-05-11 16:50:42 +00002931 case Starred_kind:
2932 result = PyType_GenericNew(Starred_type, NULL, NULL);
2933 if (!result) goto failed;
2934 value = ast2obj_expr(o->v.Starred.value);
2935 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002936 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002937 goto failed;
2938 Py_DECREF(value);
2939 value = ast2obj_expr_context(o->v.Starred.ctx);
2940 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002941 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002942 goto failed;
2943 Py_DECREF(value);
2944 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002945 case Name_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002946 result = PyType_GenericNew(Name_type, NULL, NULL);
2947 if (!result) goto failed;
2948 value = ast2obj_identifier(o->v.Name.id);
2949 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002950 if (_PyObject_SetAttrId(result, &PyId_id, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002951 goto failed;
2952 Py_DECREF(value);
2953 value = ast2obj_expr_context(o->v.Name.ctx);
2954 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002955 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002956 goto failed;
2957 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002958 break;
2959 case List_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002960 result = PyType_GenericNew(List_type, NULL, NULL);
2961 if (!result) goto failed;
2962 value = ast2obj_list(o->v.List.elts, ast2obj_expr);
2963 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002964 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002965 goto failed;
2966 Py_DECREF(value);
2967 value = ast2obj_expr_context(o->v.List.ctx);
2968 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002969 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002970 goto failed;
2971 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002972 break;
2973 case Tuple_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002974 result = PyType_GenericNew(Tuple_type, NULL, NULL);
2975 if (!result) goto failed;
2976 value = ast2obj_list(o->v.Tuple.elts, ast2obj_expr);
2977 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002978 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002979 goto failed;
2980 Py_DECREF(value);
2981 value = ast2obj_expr_context(o->v.Tuple.ctx);
2982 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002983 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002984 goto failed;
2985 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002986 break;
2987 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002988 value = ast2obj_int(o->lineno);
2989 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002990 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002991 goto failed;
2992 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002993 value = ast2obj_int(o->col_offset);
2994 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002995 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002996 goto failed;
2997 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002998 return result;
2999failed:
3000 Py_XDECREF(value);
3001 Py_XDECREF(result);
3002 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003003}
3004
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003005PyObject* ast2obj_expr_context(expr_context_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003006{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003007 switch(o) {
3008 case Load:
3009 Py_INCREF(Load_singleton);
3010 return Load_singleton;
3011 case Store:
3012 Py_INCREF(Store_singleton);
3013 return Store_singleton;
3014 case Del:
3015 Py_INCREF(Del_singleton);
3016 return Del_singleton;
3017 case AugLoad:
3018 Py_INCREF(AugLoad_singleton);
3019 return AugLoad_singleton;
3020 case AugStore:
3021 Py_INCREF(AugStore_singleton);
3022 return AugStore_singleton;
3023 case Param:
3024 Py_INCREF(Param_singleton);
3025 return Param_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003026 default:
3027 /* should never happen, but just in case ... */
3028 PyErr_Format(PyExc_SystemError, "unknown expr_context found");
3029 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003030 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003031}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003032PyObject*
3033ast2obj_slice(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003034{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003035 slice_ty o = (slice_ty)_o;
3036 PyObject *result = NULL, *value = NULL;
3037 if (!o) {
3038 Py_INCREF(Py_None);
3039 return Py_None;
3040 }
3041
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003042 switch (o->kind) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003043 case Slice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003044 result = PyType_GenericNew(Slice_type, NULL, NULL);
3045 if (!result) goto failed;
3046 value = ast2obj_expr(o->v.Slice.lower);
3047 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003048 if (_PyObject_SetAttrId(result, &PyId_lower, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003049 goto failed;
3050 Py_DECREF(value);
3051 value = ast2obj_expr(o->v.Slice.upper);
3052 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003053 if (_PyObject_SetAttrId(result, &PyId_upper, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003054 goto failed;
3055 Py_DECREF(value);
3056 value = ast2obj_expr(o->v.Slice.step);
3057 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003058 if (_PyObject_SetAttrId(result, &PyId_step, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003059 goto failed;
3060 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003061 break;
3062 case ExtSlice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003063 result = PyType_GenericNew(ExtSlice_type, NULL, NULL);
3064 if (!result) goto failed;
3065 value = ast2obj_list(o->v.ExtSlice.dims, ast2obj_slice);
3066 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003067 if (_PyObject_SetAttrId(result, &PyId_dims, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003068 goto failed;
3069 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003070 break;
3071 case Index_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003072 result = PyType_GenericNew(Index_type, NULL, NULL);
3073 if (!result) goto failed;
3074 value = ast2obj_expr(o->v.Index.value);
3075 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003076 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003077 goto failed;
3078 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003079 break;
3080 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003081 return result;
3082failed:
3083 Py_XDECREF(value);
3084 Py_XDECREF(result);
3085 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003086}
3087
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003088PyObject* ast2obj_boolop(boolop_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003089{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003090 switch(o) {
3091 case And:
3092 Py_INCREF(And_singleton);
3093 return And_singleton;
3094 case Or:
3095 Py_INCREF(Or_singleton);
3096 return Or_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003097 default:
3098 /* should never happen, but just in case ... */
3099 PyErr_Format(PyExc_SystemError, "unknown boolop found");
3100 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003101 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003102}
3103PyObject* ast2obj_operator(operator_ty o)
3104{
3105 switch(o) {
3106 case Add:
3107 Py_INCREF(Add_singleton);
3108 return Add_singleton;
3109 case Sub:
3110 Py_INCREF(Sub_singleton);
3111 return Sub_singleton;
3112 case Mult:
3113 Py_INCREF(Mult_singleton);
3114 return Mult_singleton;
3115 case Div:
3116 Py_INCREF(Div_singleton);
3117 return Div_singleton;
3118 case Mod:
3119 Py_INCREF(Mod_singleton);
3120 return Mod_singleton;
3121 case Pow:
3122 Py_INCREF(Pow_singleton);
3123 return Pow_singleton;
3124 case LShift:
3125 Py_INCREF(LShift_singleton);
3126 return LShift_singleton;
3127 case RShift:
3128 Py_INCREF(RShift_singleton);
3129 return RShift_singleton;
3130 case BitOr:
3131 Py_INCREF(BitOr_singleton);
3132 return BitOr_singleton;
3133 case BitXor:
3134 Py_INCREF(BitXor_singleton);
3135 return BitXor_singleton;
3136 case BitAnd:
3137 Py_INCREF(BitAnd_singleton);
3138 return BitAnd_singleton;
3139 case FloorDiv:
3140 Py_INCREF(FloorDiv_singleton);
3141 return FloorDiv_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003142 default:
3143 /* should never happen, but just in case ... */
3144 PyErr_Format(PyExc_SystemError, "unknown operator found");
3145 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003146 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003147}
3148PyObject* ast2obj_unaryop(unaryop_ty o)
3149{
3150 switch(o) {
3151 case Invert:
3152 Py_INCREF(Invert_singleton);
3153 return Invert_singleton;
3154 case Not:
3155 Py_INCREF(Not_singleton);
3156 return Not_singleton;
3157 case UAdd:
3158 Py_INCREF(UAdd_singleton);
3159 return UAdd_singleton;
3160 case USub:
3161 Py_INCREF(USub_singleton);
3162 return USub_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003163 default:
3164 /* should never happen, but just in case ... */
3165 PyErr_Format(PyExc_SystemError, "unknown unaryop found");
3166 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003167 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003168}
3169PyObject* ast2obj_cmpop(cmpop_ty o)
3170{
3171 switch(o) {
3172 case Eq:
3173 Py_INCREF(Eq_singleton);
3174 return Eq_singleton;
3175 case NotEq:
3176 Py_INCREF(NotEq_singleton);
3177 return NotEq_singleton;
3178 case Lt:
3179 Py_INCREF(Lt_singleton);
3180 return Lt_singleton;
3181 case LtE:
3182 Py_INCREF(LtE_singleton);
3183 return LtE_singleton;
3184 case Gt:
3185 Py_INCREF(Gt_singleton);
3186 return Gt_singleton;
3187 case GtE:
3188 Py_INCREF(GtE_singleton);
3189 return GtE_singleton;
3190 case Is:
3191 Py_INCREF(Is_singleton);
3192 return Is_singleton;
3193 case IsNot:
3194 Py_INCREF(IsNot_singleton);
3195 return IsNot_singleton;
3196 case In:
3197 Py_INCREF(In_singleton);
3198 return In_singleton;
3199 case NotIn:
3200 Py_INCREF(NotIn_singleton);
3201 return NotIn_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003202 default:
3203 /* should never happen, but just in case ... */
3204 PyErr_Format(PyExc_SystemError, "unknown cmpop found");
3205 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003206 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003207}
3208PyObject*
3209ast2obj_comprehension(void* _o)
3210{
3211 comprehension_ty o = (comprehension_ty)_o;
3212 PyObject *result = NULL, *value = NULL;
3213 if (!o) {
3214 Py_INCREF(Py_None);
3215 return Py_None;
3216 }
3217
3218 result = PyType_GenericNew(comprehension_type, NULL, NULL);
3219 if (!result) return NULL;
3220 value = ast2obj_expr(o->target);
3221 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003222 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003223 goto failed;
3224 Py_DECREF(value);
3225 value = ast2obj_expr(o->iter);
3226 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003227 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003228 goto failed;
3229 Py_DECREF(value);
3230 value = ast2obj_list(o->ifs, ast2obj_expr);
3231 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003232 if (_PyObject_SetAttrId(result, &PyId_ifs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003233 goto failed;
3234 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003235 return result;
3236failed:
3237 Py_XDECREF(value);
3238 Py_XDECREF(result);
3239 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003240}
3241
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003242PyObject*
3243ast2obj_excepthandler(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003244{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003245 excepthandler_ty o = (excepthandler_ty)_o;
3246 PyObject *result = NULL, *value = NULL;
3247 if (!o) {
3248 Py_INCREF(Py_None);
3249 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003250 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003251
Neal Norwitzad74aa82008-03-31 05:14:30 +00003252 switch (o->kind) {
3253 case ExceptHandler_kind:
3254 result = PyType_GenericNew(ExceptHandler_type, NULL, NULL);
3255 if (!result) goto failed;
3256 value = ast2obj_expr(o->v.ExceptHandler.type);
3257 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003258 if (_PyObject_SetAttrId(result, &PyId_type, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003259 goto failed;
3260 Py_DECREF(value);
3261 value = ast2obj_identifier(o->v.ExceptHandler.name);
3262 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003263 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003264 goto failed;
3265 Py_DECREF(value);
3266 value = ast2obj_list(o->v.ExceptHandler.body, ast2obj_stmt);
3267 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003268 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003269 goto failed;
3270 Py_DECREF(value);
3271 break;
3272 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003273 value = ast2obj_int(o->lineno);
3274 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003275 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003276 goto failed;
3277 Py_DECREF(value);
3278 value = ast2obj_int(o->col_offset);
3279 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003280 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003281 goto failed;
3282 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003283 return result;
3284failed:
3285 Py_XDECREF(value);
3286 Py_XDECREF(result);
3287 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003288}
3289
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003290PyObject*
3291ast2obj_arguments(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003292{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003293 arguments_ty o = (arguments_ty)_o;
3294 PyObject *result = NULL, *value = NULL;
3295 if (!o) {
3296 Py_INCREF(Py_None);
3297 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003298 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003299
3300 result = PyType_GenericNew(arguments_type, NULL, NULL);
3301 if (!result) return NULL;
Neal Norwitzc1505362006-12-28 06:47:50 +00003302 value = ast2obj_list(o->args, ast2obj_arg);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003303 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003304 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003305 goto failed;
3306 Py_DECREF(value);
3307 value = ast2obj_identifier(o->vararg);
3308 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003309 if (_PyObject_SetAttrId(result, &PyId_vararg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003310 goto failed;
3311 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003312 value = ast2obj_expr(o->varargannotation);
3313 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003314 if (_PyObject_SetAttrId(result, &PyId_varargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003315 goto failed;
3316 Py_DECREF(value);
3317 value = ast2obj_list(o->kwonlyargs, ast2obj_arg);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003318 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003319 if (_PyObject_SetAttrId(result, &PyId_kwonlyargs, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003320 goto failed;
3321 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003322 value = ast2obj_identifier(o->kwarg);
3323 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003324 if (_PyObject_SetAttrId(result, &PyId_kwarg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003325 goto failed;
3326 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003327 value = ast2obj_expr(o->kwargannotation);
3328 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003329 if (_PyObject_SetAttrId(result, &PyId_kwargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003330 goto failed;
3331 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003332 value = ast2obj_list(o->defaults, ast2obj_expr);
3333 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003334 if (_PyObject_SetAttrId(result, &PyId_defaults, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003335 goto failed;
3336 Py_DECREF(value);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003337 value = ast2obj_list(o->kw_defaults, ast2obj_expr);
3338 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003339 if (_PyObject_SetAttrId(result, &PyId_kw_defaults, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003340 goto failed;
3341 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003342 return result;
3343failed:
3344 Py_XDECREF(value);
3345 Py_XDECREF(result);
3346 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003347}
3348
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003349PyObject*
Neal Norwitzc1505362006-12-28 06:47:50 +00003350ast2obj_arg(void* _o)
3351{
3352 arg_ty o = (arg_ty)_o;
3353 PyObject *result = NULL, *value = NULL;
3354 if (!o) {
3355 Py_INCREF(Py_None);
3356 return Py_None;
3357 }
3358
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003359 result = PyType_GenericNew(arg_type, NULL, NULL);
3360 if (!result) return NULL;
3361 value = ast2obj_identifier(o->arg);
3362 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003363 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003364 goto failed;
3365 Py_DECREF(value);
3366 value = ast2obj_expr(o->annotation);
3367 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003368 if (_PyObject_SetAttrId(result, &PyId_annotation, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003369 goto failed;
3370 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003371 return result;
3372failed:
3373 Py_XDECREF(value);
3374 Py_XDECREF(result);
3375 return NULL;
3376}
3377
3378PyObject*
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003379ast2obj_keyword(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003380{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003381 keyword_ty o = (keyword_ty)_o;
3382 PyObject *result = NULL, *value = NULL;
3383 if (!o) {
3384 Py_INCREF(Py_None);
3385 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003386 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003387
3388 result = PyType_GenericNew(keyword_type, NULL, NULL);
3389 if (!result) return NULL;
3390 value = ast2obj_identifier(o->arg);
3391 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003392 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003393 goto failed;
3394 Py_DECREF(value);
3395 value = ast2obj_expr(o->value);
3396 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003397 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003398 goto failed;
3399 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003400 return result;
3401failed:
3402 Py_XDECREF(value);
3403 Py_XDECREF(result);
3404 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003405}
3406
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003407PyObject*
3408ast2obj_alias(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003409{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003410 alias_ty o = (alias_ty)_o;
3411 PyObject *result = NULL, *value = NULL;
3412 if (!o) {
3413 Py_INCREF(Py_None);
3414 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003415 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003416
3417 result = PyType_GenericNew(alias_type, NULL, NULL);
3418 if (!result) return NULL;
3419 value = ast2obj_identifier(o->name);
3420 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003421 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003422 goto failed;
3423 Py_DECREF(value);
3424 value = ast2obj_identifier(o->asname);
3425 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003426 if (_PyObject_SetAttrId(result, &PyId_asname, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003427 goto failed;
3428 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003429 return result;
3430failed:
3431 Py_XDECREF(value);
3432 Py_XDECREF(result);
3433 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003434}
3435
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003436PyObject*
3437ast2obj_withitem(void* _o)
3438{
3439 withitem_ty o = (withitem_ty)_o;
3440 PyObject *result = NULL, *value = NULL;
3441 if (!o) {
3442 Py_INCREF(Py_None);
3443 return Py_None;
3444 }
3445
3446 result = PyType_GenericNew(withitem_type, NULL, NULL);
3447 if (!result) return NULL;
3448 value = ast2obj_expr(o->context_expr);
3449 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003450 if (_PyObject_SetAttrId(result, &PyId_context_expr, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003451 goto failed;
3452 Py_DECREF(value);
3453 value = ast2obj_expr(o->optional_vars);
3454 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003455 if (_PyObject_SetAttrId(result, &PyId_optional_vars, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003456 goto failed;
3457 Py_DECREF(value);
3458 return result;
3459failed:
3460 Py_XDECREF(value);
3461 Py_XDECREF(result);
3462 return NULL;
3463}
3464
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003465
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003466int
3467obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena)
3468{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003469 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003470
Benjamin Petersond8f65972010-11-20 04:31:07 +00003471 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003472
3473 if (obj == Py_None) {
3474 *out = NULL;
3475 return 0;
3476 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003477 isinstance = PyObject_IsInstance(obj, (PyObject*)Module_type);
3478 if (isinstance == -1) {
3479 return 1;
3480 }
3481 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003482 asdl_seq* body;
3483
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003484 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003485 int res;
3486 Py_ssize_t len;
3487 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003488 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003489 if (tmp == NULL) goto failed;
3490 if (!PyList_Check(tmp)) {
3491 PyErr_Format(PyExc_TypeError, "Module field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3492 goto failed;
3493 }
3494 len = PyList_GET_SIZE(tmp);
3495 body = asdl_seq_new(len, arena);
3496 if (body == NULL) goto failed;
3497 for (i = 0; i < len; i++) {
3498 stmt_ty value;
3499 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3500 if (res != 0) goto failed;
3501 asdl_seq_SET(body, i, value);
3502 }
3503 Py_XDECREF(tmp);
3504 tmp = NULL;
3505 } else {
3506 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Module");
3507 return 1;
3508 }
3509 *out = Module(body, arena);
3510 if (*out == NULL) goto failed;
3511 return 0;
3512 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003513 isinstance = PyObject_IsInstance(obj, (PyObject*)Interactive_type);
3514 if (isinstance == -1) {
3515 return 1;
3516 }
3517 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003518 asdl_seq* body;
3519
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003520 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003521 int res;
3522 Py_ssize_t len;
3523 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003524 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003525 if (tmp == NULL) goto failed;
3526 if (!PyList_Check(tmp)) {
3527 PyErr_Format(PyExc_TypeError, "Interactive field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3528 goto failed;
3529 }
3530 len = PyList_GET_SIZE(tmp);
3531 body = asdl_seq_new(len, arena);
3532 if (body == NULL) goto failed;
3533 for (i = 0; i < len; i++) {
3534 stmt_ty value;
3535 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3536 if (res != 0) goto failed;
3537 asdl_seq_SET(body, i, value);
3538 }
3539 Py_XDECREF(tmp);
3540 tmp = NULL;
3541 } else {
3542 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Interactive");
3543 return 1;
3544 }
3545 *out = Interactive(body, arena);
3546 if (*out == NULL) goto failed;
3547 return 0;
3548 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003549 isinstance = PyObject_IsInstance(obj, (PyObject*)Expression_type);
3550 if (isinstance == -1) {
3551 return 1;
3552 }
3553 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003554 expr_ty body;
3555
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003556 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003557 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003558 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003559 if (tmp == NULL) goto failed;
3560 res = obj2ast_expr(tmp, &body, arena);
3561 if (res != 0) goto failed;
3562 Py_XDECREF(tmp);
3563 tmp = NULL;
3564 } else {
3565 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Expression");
3566 return 1;
3567 }
3568 *out = Expression(body, arena);
3569 if (*out == NULL) goto failed;
3570 return 0;
3571 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003572 isinstance = PyObject_IsInstance(obj, (PyObject*)Suite_type);
3573 if (isinstance == -1) {
3574 return 1;
3575 }
3576 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003577 asdl_seq* body;
3578
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003579 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003580 int res;
3581 Py_ssize_t len;
3582 Py_ssize_t i;
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 if (!PyList_Check(tmp)) {
3586 PyErr_Format(PyExc_TypeError, "Suite field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3587 goto failed;
3588 }
3589 len = PyList_GET_SIZE(tmp);
3590 body = asdl_seq_new(len, arena);
3591 if (body == NULL) goto failed;
3592 for (i = 0; i < len; i++) {
3593 stmt_ty value;
3594 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3595 if (res != 0) goto failed;
3596 asdl_seq_SET(body, i, value);
3597 }
3598 Py_XDECREF(tmp);
3599 tmp = NULL;
3600 } else {
3601 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Suite");
3602 return 1;
3603 }
3604 *out = Suite(body, arena);
3605 if (*out == NULL) goto failed;
3606 return 0;
3607 }
3608
Benjamin Peterson5b066812010-11-20 01:38:49 +00003609 PyErr_Format(PyExc_TypeError, "expected some sort of mod, but got %R", obj);
3610 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00003611 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003612 return 1;
3613}
3614
3615int
3616obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena)
3617{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003618 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003619
Benjamin Petersond8f65972010-11-20 04:31:07 +00003620 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003621 int lineno;
3622 int col_offset;
3623
3624 if (obj == Py_None) {
3625 *out = NULL;
3626 return 0;
3627 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003628 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003629 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003630 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003631 if (tmp == NULL) goto failed;
3632 res = obj2ast_int(tmp, &lineno, arena);
3633 if (res != 0) goto failed;
3634 Py_XDECREF(tmp);
3635 tmp = NULL;
3636 } else {
3637 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from stmt");
3638 return 1;
3639 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003640 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003641 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003642 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003643 if (tmp == NULL) goto failed;
3644 res = obj2ast_int(tmp, &col_offset, arena);
3645 if (res != 0) goto failed;
3646 Py_XDECREF(tmp);
3647 tmp = NULL;
3648 } else {
3649 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from stmt");
3650 return 1;
3651 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003652 isinstance = PyObject_IsInstance(obj, (PyObject*)FunctionDef_type);
3653 if (isinstance == -1) {
3654 return 1;
3655 }
3656 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003657 identifier name;
3658 arguments_ty args;
3659 asdl_seq* body;
3660 asdl_seq* decorator_list;
3661 expr_ty returns;
3662
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003663 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003664 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003665 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003666 if (tmp == NULL) goto failed;
3667 res = obj2ast_identifier(tmp, &name, arena);
3668 if (res != 0) goto failed;
3669 Py_XDECREF(tmp);
3670 tmp = NULL;
3671 } else {
3672 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from FunctionDef");
3673 return 1;
3674 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003675 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003676 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003677 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003678 if (tmp == NULL) goto failed;
3679 res = obj2ast_arguments(tmp, &args, arena);
3680 if (res != 0) goto failed;
3681 Py_XDECREF(tmp);
3682 tmp = NULL;
3683 } else {
3684 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from FunctionDef");
3685 return 1;
3686 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003687 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003688 int res;
3689 Py_ssize_t len;
3690 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003691 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003692 if (tmp == NULL) goto failed;
3693 if (!PyList_Check(tmp)) {
3694 PyErr_Format(PyExc_TypeError, "FunctionDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3695 goto failed;
3696 }
3697 len = PyList_GET_SIZE(tmp);
3698 body = asdl_seq_new(len, arena);
3699 if (body == NULL) goto failed;
3700 for (i = 0; i < len; i++) {
3701 stmt_ty value;
3702 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3703 if (res != 0) goto failed;
3704 asdl_seq_SET(body, i, value);
3705 }
3706 Py_XDECREF(tmp);
3707 tmp = NULL;
3708 } else {
3709 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from FunctionDef");
3710 return 1;
3711 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003712 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
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_decorator_list);
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 \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3720 goto failed;
3721 }
3722 len = PyList_GET_SIZE(tmp);
3723 decorator_list = asdl_seq_new(len, arena);
3724 if (decorator_list == NULL) goto failed;
3725 for (i = 0; i < len; i++) {
3726 expr_ty value;
3727 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3728 if (res != 0) goto failed;
3729 asdl_seq_SET(decorator_list, i, value);
3730 }
3731 Py_XDECREF(tmp);
3732 tmp = NULL;
3733 } else {
3734 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from FunctionDef");
3735 return 1;
3736 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003737 if (_PyObject_HasAttrId(obj, &PyId_returns)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003738 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003739 tmp = _PyObject_GetAttrId(obj, &PyId_returns);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003740 if (tmp == NULL) goto failed;
3741 res = obj2ast_expr(tmp, &returns, arena);
3742 if (res != 0) goto failed;
3743 Py_XDECREF(tmp);
3744 tmp = NULL;
3745 } else {
3746 returns = NULL;
3747 }
3748 *out = FunctionDef(name, args, body, decorator_list, returns,
3749 lineno, col_offset, arena);
3750 if (*out == NULL) goto failed;
3751 return 0;
3752 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003753 isinstance = PyObject_IsInstance(obj, (PyObject*)ClassDef_type);
3754 if (isinstance == -1) {
3755 return 1;
3756 }
3757 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003758 identifier name;
3759 asdl_seq* bases;
3760 asdl_seq* keywords;
3761 expr_ty starargs;
3762 expr_ty kwargs;
3763 asdl_seq* body;
3764 asdl_seq* decorator_list;
3765
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003766 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003767 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003768 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003769 if (tmp == NULL) goto failed;
3770 res = obj2ast_identifier(tmp, &name, arena);
3771 if (res != 0) goto failed;
3772 Py_XDECREF(tmp);
3773 tmp = NULL;
3774 } else {
3775 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from ClassDef");
3776 return 1;
3777 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003778 if (_PyObject_HasAttrId(obj, &PyId_bases)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003779 int res;
3780 Py_ssize_t len;
3781 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003782 tmp = _PyObject_GetAttrId(obj, &PyId_bases);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003783 if (tmp == NULL) goto failed;
3784 if (!PyList_Check(tmp)) {
3785 PyErr_Format(PyExc_TypeError, "ClassDef field \"bases\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3786 goto failed;
3787 }
3788 len = PyList_GET_SIZE(tmp);
3789 bases = asdl_seq_new(len, arena);
3790 if (bases == NULL) goto failed;
3791 for (i = 0; i < len; i++) {
3792 expr_ty value;
3793 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3794 if (res != 0) goto failed;
3795 asdl_seq_SET(bases, i, value);
3796 }
3797 Py_XDECREF(tmp);
3798 tmp = NULL;
3799 } else {
3800 PyErr_SetString(PyExc_TypeError, "required field \"bases\" missing from ClassDef");
3801 return 1;
3802 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003803 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
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_keywords);
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 \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3811 goto failed;
3812 }
3813 len = PyList_GET_SIZE(tmp);
3814 keywords = asdl_seq_new(len, arena);
3815 if (keywords == NULL) goto failed;
3816 for (i = 0; i < len; i++) {
3817 keyword_ty value;
3818 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
3819 if (res != 0) goto failed;
3820 asdl_seq_SET(keywords, i, value);
3821 }
3822 Py_XDECREF(tmp);
3823 tmp = NULL;
3824 } else {
3825 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from ClassDef");
3826 return 1;
3827 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003828 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003829 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003830 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003831 if (tmp == NULL) goto failed;
3832 res = obj2ast_expr(tmp, &starargs, arena);
3833 if (res != 0) goto failed;
3834 Py_XDECREF(tmp);
3835 tmp = NULL;
3836 } else {
3837 starargs = NULL;
3838 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003839 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003840 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003841 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003842 if (tmp == NULL) goto failed;
3843 res = obj2ast_expr(tmp, &kwargs, arena);
3844 if (res != 0) goto failed;
3845 Py_XDECREF(tmp);
3846 tmp = NULL;
3847 } else {
3848 kwargs = NULL;
3849 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003850 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003851 int res;
3852 Py_ssize_t len;
3853 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003854 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003855 if (tmp == NULL) goto failed;
3856 if (!PyList_Check(tmp)) {
3857 PyErr_Format(PyExc_TypeError, "ClassDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3858 goto failed;
3859 }
3860 len = PyList_GET_SIZE(tmp);
3861 body = asdl_seq_new(len, arena);
3862 if (body == NULL) goto failed;
3863 for (i = 0; i < len; i++) {
3864 stmt_ty value;
3865 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3866 if (res != 0) goto failed;
3867 asdl_seq_SET(body, i, value);
3868 }
3869 Py_XDECREF(tmp);
3870 tmp = NULL;
3871 } else {
3872 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ClassDef");
3873 return 1;
3874 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003875 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
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_decorator_list);
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 \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3883 goto failed;
3884 }
3885 len = PyList_GET_SIZE(tmp);
3886 decorator_list = asdl_seq_new(len, arena);
3887 if (decorator_list == NULL) goto failed;
3888 for (i = 0; i < len; i++) {
3889 expr_ty value;
3890 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3891 if (res != 0) goto failed;
3892 asdl_seq_SET(decorator_list, i, value);
3893 }
3894 Py_XDECREF(tmp);
3895 tmp = NULL;
3896 } else {
3897 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from ClassDef");
3898 return 1;
3899 }
3900 *out = ClassDef(name, bases, keywords, starargs, kwargs, body,
3901 decorator_list, lineno, col_offset, arena);
3902 if (*out == NULL) goto failed;
3903 return 0;
3904 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003905 isinstance = PyObject_IsInstance(obj, (PyObject*)Return_type);
3906 if (isinstance == -1) {
3907 return 1;
3908 }
3909 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003910 expr_ty value;
3911
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003912 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003913 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003914 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003915 if (tmp == NULL) goto failed;
3916 res = obj2ast_expr(tmp, &value, arena);
3917 if (res != 0) goto failed;
3918 Py_XDECREF(tmp);
3919 tmp = NULL;
3920 } else {
3921 value = NULL;
3922 }
3923 *out = Return(value, lineno, col_offset, arena);
3924 if (*out == NULL) goto failed;
3925 return 0;
3926 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003927 isinstance = PyObject_IsInstance(obj, (PyObject*)Delete_type);
3928 if (isinstance == -1) {
3929 return 1;
3930 }
3931 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003932 asdl_seq* targets;
3933
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003934 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003935 int res;
3936 Py_ssize_t len;
3937 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003938 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003939 if (tmp == NULL) goto failed;
3940 if (!PyList_Check(tmp)) {
3941 PyErr_Format(PyExc_TypeError, "Delete field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3942 goto failed;
3943 }
3944 len = PyList_GET_SIZE(tmp);
3945 targets = asdl_seq_new(len, arena);
3946 if (targets == NULL) goto failed;
3947 for (i = 0; i < len; i++) {
3948 expr_ty value;
3949 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3950 if (res != 0) goto failed;
3951 asdl_seq_SET(targets, i, value);
3952 }
3953 Py_XDECREF(tmp);
3954 tmp = NULL;
3955 } else {
3956 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Delete");
3957 return 1;
3958 }
3959 *out = Delete(targets, lineno, col_offset, arena);
3960 if (*out == NULL) goto failed;
3961 return 0;
3962 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003963 isinstance = PyObject_IsInstance(obj, (PyObject*)Assign_type);
3964 if (isinstance == -1) {
3965 return 1;
3966 }
3967 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003968 asdl_seq* targets;
3969 expr_ty value;
3970
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003971 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003972 int res;
3973 Py_ssize_t len;
3974 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003975 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003976 if (tmp == NULL) goto failed;
3977 if (!PyList_Check(tmp)) {
3978 PyErr_Format(PyExc_TypeError, "Assign field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3979 goto failed;
3980 }
3981 len = PyList_GET_SIZE(tmp);
3982 targets = asdl_seq_new(len, arena);
3983 if (targets == NULL) goto failed;
3984 for (i = 0; i < len; i++) {
3985 expr_ty value;
3986 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3987 if (res != 0) goto failed;
3988 asdl_seq_SET(targets, i, value);
3989 }
3990 Py_XDECREF(tmp);
3991 tmp = NULL;
3992 } else {
3993 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Assign");
3994 return 1;
3995 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003996 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003997 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003998 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003999 if (tmp == NULL) goto failed;
4000 res = obj2ast_expr(tmp, &value, arena);
4001 if (res != 0) goto failed;
4002 Py_XDECREF(tmp);
4003 tmp = NULL;
4004 } else {
4005 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Assign");
4006 return 1;
4007 }
4008 *out = Assign(targets, value, lineno, col_offset, arena);
4009 if (*out == NULL) goto failed;
4010 return 0;
4011 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004012 isinstance = PyObject_IsInstance(obj, (PyObject*)AugAssign_type);
4013 if (isinstance == -1) {
4014 return 1;
4015 }
4016 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004017 expr_ty target;
4018 operator_ty op;
4019 expr_ty value;
4020
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004021 if (_PyObject_HasAttrId(obj, &PyId_target)) {
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_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004024 if (tmp == NULL) goto failed;
4025 res = obj2ast_expr(tmp, &target, arena);
4026 if (res != 0) goto failed;
4027 Py_XDECREF(tmp);
4028 tmp = NULL;
4029 } else {
4030 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from AugAssign");
4031 return 1;
4032 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004033 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004034 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004035 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004036 if (tmp == NULL) goto failed;
4037 res = obj2ast_operator(tmp, &op, arena);
4038 if (res != 0) goto failed;
4039 Py_XDECREF(tmp);
4040 tmp = NULL;
4041 } else {
4042 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from AugAssign");
4043 return 1;
4044 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004045 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004046 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004047 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004048 if (tmp == NULL) goto failed;
4049 res = obj2ast_expr(tmp, &value, arena);
4050 if (res != 0) goto failed;
4051 Py_XDECREF(tmp);
4052 tmp = NULL;
4053 } else {
4054 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from AugAssign");
4055 return 1;
4056 }
4057 *out = AugAssign(target, op, value, lineno, col_offset, arena);
4058 if (*out == NULL) goto failed;
4059 return 0;
4060 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004061 isinstance = PyObject_IsInstance(obj, (PyObject*)For_type);
4062 if (isinstance == -1) {
4063 return 1;
4064 }
4065 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004066 expr_ty target;
4067 expr_ty iter;
4068 asdl_seq* body;
4069 asdl_seq* orelse;
4070
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004071 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004072 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004073 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004074 if (tmp == NULL) goto failed;
4075 res = obj2ast_expr(tmp, &target, arena);
4076 if (res != 0) goto failed;
4077 Py_XDECREF(tmp);
4078 tmp = NULL;
4079 } else {
4080 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from For");
4081 return 1;
4082 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004083 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004084 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004085 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004086 if (tmp == NULL) goto failed;
4087 res = obj2ast_expr(tmp, &iter, arena);
4088 if (res != 0) goto failed;
4089 Py_XDECREF(tmp);
4090 tmp = NULL;
4091 } else {
4092 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from For");
4093 return 1;
4094 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004095 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004096 int res;
4097 Py_ssize_t len;
4098 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004099 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004100 if (tmp == NULL) goto failed;
4101 if (!PyList_Check(tmp)) {
4102 PyErr_Format(PyExc_TypeError, "For field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4103 goto failed;
4104 }
4105 len = PyList_GET_SIZE(tmp);
4106 body = asdl_seq_new(len, arena);
4107 if (body == NULL) goto failed;
4108 for (i = 0; i < len; i++) {
4109 stmt_ty value;
4110 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4111 if (res != 0) goto failed;
4112 asdl_seq_SET(body, i, value);
4113 }
4114 Py_XDECREF(tmp);
4115 tmp = NULL;
4116 } else {
4117 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from For");
4118 return 1;
4119 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004120 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
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_orelse);
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 \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4128 goto failed;
4129 }
4130 len = PyList_GET_SIZE(tmp);
4131 orelse = asdl_seq_new(len, arena);
4132 if (orelse == 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(orelse, i, value);
4138 }
4139 Py_XDECREF(tmp);
4140 tmp = NULL;
4141 } else {
4142 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from For");
4143 return 1;
4144 }
4145 *out = For(target, iter, body, orelse, lineno, col_offset,
4146 arena);
4147 if (*out == NULL) goto failed;
4148 return 0;
4149 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004150 isinstance = PyObject_IsInstance(obj, (PyObject*)While_type);
4151 if (isinstance == -1) {
4152 return 1;
4153 }
4154 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004155 expr_ty test;
4156 asdl_seq* body;
4157 asdl_seq* orelse;
4158
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004159 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004160 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004161 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004162 if (tmp == NULL) goto failed;
4163 res = obj2ast_expr(tmp, &test, arena);
4164 if (res != 0) goto failed;
4165 Py_XDECREF(tmp);
4166 tmp = NULL;
4167 } else {
4168 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from While");
4169 return 1;
4170 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004171 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004172 int res;
4173 Py_ssize_t len;
4174 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004175 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004176 if (tmp == NULL) goto failed;
4177 if (!PyList_Check(tmp)) {
4178 PyErr_Format(PyExc_TypeError, "While field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4179 goto failed;
4180 }
4181 len = PyList_GET_SIZE(tmp);
4182 body = asdl_seq_new(len, arena);
4183 if (body == NULL) goto failed;
4184 for (i = 0; i < len; i++) {
4185 stmt_ty value;
4186 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4187 if (res != 0) goto failed;
4188 asdl_seq_SET(body, i, value);
4189 }
4190 Py_XDECREF(tmp);
4191 tmp = NULL;
4192 } else {
4193 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from While");
4194 return 1;
4195 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004196 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
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_orelse);
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 \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4204 goto failed;
4205 }
4206 len = PyList_GET_SIZE(tmp);
4207 orelse = asdl_seq_new(len, arena);
4208 if (orelse == 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(orelse, i, value);
4214 }
4215 Py_XDECREF(tmp);
4216 tmp = NULL;
4217 } else {
4218 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from While");
4219 return 1;
4220 }
4221 *out = While(test, body, orelse, lineno, col_offset, arena);
4222 if (*out == NULL) goto failed;
4223 return 0;
4224 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004225 isinstance = PyObject_IsInstance(obj, (PyObject*)If_type);
4226 if (isinstance == -1) {
4227 return 1;
4228 }
4229 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004230 expr_ty test;
4231 asdl_seq* body;
4232 asdl_seq* orelse;
4233
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004234 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004235 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004236 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004237 if (tmp == NULL) goto failed;
4238 res = obj2ast_expr(tmp, &test, arena);
4239 if (res != 0) goto failed;
4240 Py_XDECREF(tmp);
4241 tmp = NULL;
4242 } else {
4243 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from If");
4244 return 1;
4245 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004246 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004247 int res;
4248 Py_ssize_t len;
4249 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004250 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004251 if (tmp == NULL) goto failed;
4252 if (!PyList_Check(tmp)) {
4253 PyErr_Format(PyExc_TypeError, "If field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4254 goto failed;
4255 }
4256 len = PyList_GET_SIZE(tmp);
4257 body = asdl_seq_new(len, arena);
4258 if (body == NULL) goto failed;
4259 for (i = 0; i < len; i++) {
4260 stmt_ty value;
4261 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4262 if (res != 0) goto failed;
4263 asdl_seq_SET(body, i, value);
4264 }
4265 Py_XDECREF(tmp);
4266 tmp = NULL;
4267 } else {
4268 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from If");
4269 return 1;
4270 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004271 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
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_orelse);
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 \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4279 goto failed;
4280 }
4281 len = PyList_GET_SIZE(tmp);
4282 orelse = asdl_seq_new(len, arena);
4283 if (orelse == 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(orelse, i, value);
4289 }
4290 Py_XDECREF(tmp);
4291 tmp = NULL;
4292 } else {
4293 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from If");
4294 return 1;
4295 }
4296 *out = If(test, body, orelse, lineno, col_offset, arena);
4297 if (*out == NULL) goto failed;
4298 return 0;
4299 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004300 isinstance = PyObject_IsInstance(obj, (PyObject*)With_type);
4301 if (isinstance == -1) {
4302 return 1;
4303 }
4304 if (isinstance) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004305 asdl_seq* items;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004306 asdl_seq* body;
4307
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004308 if (_PyObject_HasAttrId(obj, &PyId_items)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004309 int res;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004310 Py_ssize_t len;
4311 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004312 tmp = _PyObject_GetAttrId(obj, &PyId_items);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004313 if (tmp == NULL) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004314 if (!PyList_Check(tmp)) {
4315 PyErr_Format(PyExc_TypeError, "With field \"items\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4316 goto failed;
4317 }
4318 len = PyList_GET_SIZE(tmp);
4319 items = asdl_seq_new(len, arena);
4320 if (items == NULL) goto failed;
4321 for (i = 0; i < len; i++) {
4322 withitem_ty value;
4323 res = obj2ast_withitem(PyList_GET_ITEM(tmp, i), &value, arena);
4324 if (res != 0) goto failed;
4325 asdl_seq_SET(items, i, value);
4326 }
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004327 Py_XDECREF(tmp);
4328 tmp = NULL;
4329 } else {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004330 PyErr_SetString(PyExc_TypeError, "required field \"items\" missing from With");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004331 return 1;
4332 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004333 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004334 int res;
4335 Py_ssize_t len;
4336 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004337 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004338 if (tmp == NULL) goto failed;
4339 if (!PyList_Check(tmp)) {
4340 PyErr_Format(PyExc_TypeError, "With field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4341 goto failed;
4342 }
4343 len = PyList_GET_SIZE(tmp);
4344 body = asdl_seq_new(len, arena);
4345 if (body == NULL) goto failed;
4346 for (i = 0; i < len; i++) {
4347 stmt_ty value;
4348 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4349 if (res != 0) goto failed;
4350 asdl_seq_SET(body, i, value);
4351 }
4352 Py_XDECREF(tmp);
4353 tmp = NULL;
4354 } else {
4355 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from With");
4356 return 1;
4357 }
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004358 *out = With(items, body, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004359 if (*out == NULL) goto failed;
4360 return 0;
4361 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004362 isinstance = PyObject_IsInstance(obj, (PyObject*)Raise_type);
4363 if (isinstance == -1) {
4364 return 1;
4365 }
4366 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004367 expr_ty exc;
4368 expr_ty cause;
4369
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004370 if (_PyObject_HasAttrId(obj, &PyId_exc)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004371 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004372 tmp = _PyObject_GetAttrId(obj, &PyId_exc);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004373 if (tmp == NULL) goto failed;
4374 res = obj2ast_expr(tmp, &exc, arena);
4375 if (res != 0) goto failed;
4376 Py_XDECREF(tmp);
4377 tmp = NULL;
4378 } else {
4379 exc = NULL;
4380 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004381 if (_PyObject_HasAttrId(obj, &PyId_cause)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004382 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004383 tmp = _PyObject_GetAttrId(obj, &PyId_cause);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004384 if (tmp == NULL) goto failed;
4385 res = obj2ast_expr(tmp, &cause, arena);
4386 if (res != 0) goto failed;
4387 Py_XDECREF(tmp);
4388 tmp = NULL;
4389 } else {
4390 cause = NULL;
4391 }
4392 *out = Raise(exc, cause, lineno, col_offset, arena);
4393 if (*out == NULL) goto failed;
4394 return 0;
4395 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004396 isinstance = PyObject_IsInstance(obj, (PyObject*)Try_type);
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004397 if (isinstance == -1) {
4398 return 1;
4399 }
4400 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004401 asdl_seq* body;
4402 asdl_seq* handlers;
4403 asdl_seq* orelse;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004404 asdl_seq* finalbody;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004405
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004406 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004407 int res;
4408 Py_ssize_t len;
4409 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004410 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004411 if (tmp == NULL) goto failed;
4412 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004413 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 +00004414 goto failed;
4415 }
4416 len = PyList_GET_SIZE(tmp);
4417 body = asdl_seq_new(len, arena);
4418 if (body == NULL) goto failed;
4419 for (i = 0; i < len; i++) {
4420 stmt_ty value;
4421 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4422 if (res != 0) goto failed;
4423 asdl_seq_SET(body, i, value);
4424 }
4425 Py_XDECREF(tmp);
4426 tmp = NULL;
4427 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004428 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004429 return 1;
4430 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004431 if (_PyObject_HasAttrId(obj, &PyId_handlers)) {
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_handlers);
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 \"handlers\" 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 handlers = asdl_seq_new(len, arena);
4443 if (handlers == NULL) goto failed;
4444 for (i = 0; i < len; i++) {
4445 excepthandler_ty value;
4446 res = obj2ast_excepthandler(PyList_GET_ITEM(tmp, i), &value, arena);
4447 if (res != 0) goto failed;
4448 asdl_seq_SET(handlers, 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 \"handlers\" 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_orelse)) {
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_orelse);
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 \"orelse\" 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 orelse = asdl_seq_new(len, arena);
4468 if (orelse == NULL) goto failed;
4469 for (i = 0; i < len; i++) {
4470 stmt_ty value;
4471 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4472 if (res != 0) goto failed;
4473 asdl_seq_SET(orelse, 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 \"orelse\" 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_finalbody)) {
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_finalbody);
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 \"finalbody\" 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 finalbody = asdl_seq_new(len, arena);
4493 if (finalbody == 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(finalbody, 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 \"finalbody\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004504 return 1;
4505 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004506 *out = Try(body, handlers, orelse, finalbody, lineno,
4507 col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004508 if (*out == NULL) goto failed;
4509 return 0;
4510 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004511 isinstance = PyObject_IsInstance(obj, (PyObject*)Assert_type);
4512 if (isinstance == -1) {
4513 return 1;
4514 }
4515 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004516 expr_ty test;
4517 expr_ty msg;
4518
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004519 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004520 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004521 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004522 if (tmp == NULL) goto failed;
4523 res = obj2ast_expr(tmp, &test, arena);
4524 if (res != 0) goto failed;
4525 Py_XDECREF(tmp);
4526 tmp = NULL;
4527 } else {
4528 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from Assert");
4529 return 1;
4530 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004531 if (_PyObject_HasAttrId(obj, &PyId_msg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004532 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004533 tmp = _PyObject_GetAttrId(obj, &PyId_msg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004534 if (tmp == NULL) goto failed;
4535 res = obj2ast_expr(tmp, &msg, arena);
4536 if (res != 0) goto failed;
4537 Py_XDECREF(tmp);
4538 tmp = NULL;
4539 } else {
4540 msg = NULL;
4541 }
4542 *out = Assert(test, msg, lineno, col_offset, arena);
4543 if (*out == NULL) goto failed;
4544 return 0;
4545 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004546 isinstance = PyObject_IsInstance(obj, (PyObject*)Import_type);
4547 if (isinstance == -1) {
4548 return 1;
4549 }
4550 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004551 asdl_seq* names;
4552
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004553 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004554 int res;
4555 Py_ssize_t len;
4556 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004557 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004558 if (tmp == NULL) goto failed;
4559 if (!PyList_Check(tmp)) {
4560 PyErr_Format(PyExc_TypeError, "Import field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4561 goto failed;
4562 }
4563 len = PyList_GET_SIZE(tmp);
4564 names = asdl_seq_new(len, arena);
4565 if (names == NULL) goto failed;
4566 for (i = 0; i < len; i++) {
4567 alias_ty value;
4568 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4569 if (res != 0) goto failed;
4570 asdl_seq_SET(names, i, value);
4571 }
4572 Py_XDECREF(tmp);
4573 tmp = NULL;
4574 } else {
4575 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Import");
4576 return 1;
4577 }
4578 *out = Import(names, lineno, col_offset, arena);
4579 if (*out == NULL) goto failed;
4580 return 0;
4581 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004582 isinstance = PyObject_IsInstance(obj, (PyObject*)ImportFrom_type);
4583 if (isinstance == -1) {
4584 return 1;
4585 }
4586 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004587 identifier module;
4588 asdl_seq* names;
4589 int level;
4590
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004591 if (_PyObject_HasAttrId(obj, &PyId_module)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004592 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004593 tmp = _PyObject_GetAttrId(obj, &PyId_module);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004594 if (tmp == NULL) goto failed;
4595 res = obj2ast_identifier(tmp, &module, arena);
4596 if (res != 0) goto failed;
4597 Py_XDECREF(tmp);
4598 tmp = NULL;
4599 } else {
Benjamin Peterson78565b22009-06-28 19:19:51 +00004600 module = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004601 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004602 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004603 int res;
4604 Py_ssize_t len;
4605 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004606 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004607 if (tmp == NULL) goto failed;
4608 if (!PyList_Check(tmp)) {
4609 PyErr_Format(PyExc_TypeError, "ImportFrom field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4610 goto failed;
4611 }
4612 len = PyList_GET_SIZE(tmp);
4613 names = asdl_seq_new(len, arena);
4614 if (names == NULL) goto failed;
4615 for (i = 0; i < len; i++) {
4616 alias_ty value;
4617 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4618 if (res != 0) goto failed;
4619 asdl_seq_SET(names, i, value);
4620 }
4621 Py_XDECREF(tmp);
4622 tmp = NULL;
4623 } else {
4624 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from ImportFrom");
4625 return 1;
4626 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004627 if (_PyObject_HasAttrId(obj, &PyId_level)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004628 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004629 tmp = _PyObject_GetAttrId(obj, &PyId_level);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004630 if (tmp == NULL) goto failed;
4631 res = obj2ast_int(tmp, &level, arena);
4632 if (res != 0) goto failed;
4633 Py_XDECREF(tmp);
4634 tmp = NULL;
4635 } else {
4636 level = 0;
4637 }
4638 *out = ImportFrom(module, names, level, lineno, col_offset,
4639 arena);
4640 if (*out == NULL) goto failed;
4641 return 0;
4642 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004643 isinstance = PyObject_IsInstance(obj, (PyObject*)Global_type);
4644 if (isinstance == -1) {
4645 return 1;
4646 }
4647 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004648 asdl_seq* names;
4649
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004650 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004651 int res;
4652 Py_ssize_t len;
4653 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004654 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004655 if (tmp == NULL) goto failed;
4656 if (!PyList_Check(tmp)) {
4657 PyErr_Format(PyExc_TypeError, "Global field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4658 goto failed;
4659 }
4660 len = PyList_GET_SIZE(tmp);
4661 names = asdl_seq_new(len, arena);
4662 if (names == NULL) goto failed;
4663 for (i = 0; i < len; i++) {
4664 identifier value;
4665 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4666 if (res != 0) goto failed;
4667 asdl_seq_SET(names, i, value);
4668 }
4669 Py_XDECREF(tmp);
4670 tmp = NULL;
4671 } else {
4672 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Global");
4673 return 1;
4674 }
4675 *out = Global(names, lineno, col_offset, arena);
4676 if (*out == NULL) goto failed;
4677 return 0;
4678 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004679 isinstance = PyObject_IsInstance(obj, (PyObject*)Nonlocal_type);
4680 if (isinstance == -1) {
4681 return 1;
4682 }
4683 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004684 asdl_seq* names;
4685
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004686 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004687 int res;
4688 Py_ssize_t len;
4689 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004690 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004691 if (tmp == NULL) goto failed;
4692 if (!PyList_Check(tmp)) {
4693 PyErr_Format(PyExc_TypeError, "Nonlocal field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4694 goto failed;
4695 }
4696 len = PyList_GET_SIZE(tmp);
4697 names = asdl_seq_new(len, arena);
4698 if (names == NULL) goto failed;
4699 for (i = 0; i < len; i++) {
4700 identifier value;
4701 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4702 if (res != 0) goto failed;
4703 asdl_seq_SET(names, i, value);
4704 }
4705 Py_XDECREF(tmp);
4706 tmp = NULL;
4707 } else {
4708 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Nonlocal");
4709 return 1;
4710 }
4711 *out = Nonlocal(names, lineno, col_offset, arena);
4712 if (*out == NULL) goto failed;
4713 return 0;
4714 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004715 isinstance = PyObject_IsInstance(obj, (PyObject*)Expr_type);
4716 if (isinstance == -1) {
4717 return 1;
4718 }
4719 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004720 expr_ty value;
4721
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004722 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004723 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004724 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004725 if (tmp == NULL) goto failed;
4726 res = obj2ast_expr(tmp, &value, arena);
4727 if (res != 0) goto failed;
4728 Py_XDECREF(tmp);
4729 tmp = NULL;
4730 } else {
4731 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Expr");
4732 return 1;
4733 }
4734 *out = Expr(value, lineno, col_offset, arena);
4735 if (*out == NULL) goto failed;
4736 return 0;
4737 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004738 isinstance = PyObject_IsInstance(obj, (PyObject*)Pass_type);
4739 if (isinstance == -1) {
4740 return 1;
4741 }
4742 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004743
4744 *out = Pass(lineno, col_offset, arena);
4745 if (*out == NULL) goto failed;
4746 return 0;
4747 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004748 isinstance = PyObject_IsInstance(obj, (PyObject*)Break_type);
4749 if (isinstance == -1) {
4750 return 1;
4751 }
4752 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004753
4754 *out = Break(lineno, col_offset, arena);
4755 if (*out == NULL) goto failed;
4756 return 0;
4757 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004758 isinstance = PyObject_IsInstance(obj, (PyObject*)Continue_type);
4759 if (isinstance == -1) {
4760 return 1;
4761 }
4762 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004763
4764 *out = Continue(lineno, col_offset, arena);
4765 if (*out == NULL) goto failed;
4766 return 0;
4767 }
4768
Benjamin Peterson5b066812010-11-20 01:38:49 +00004769 PyErr_Format(PyExc_TypeError, "expected some sort of stmt, but got %R", obj);
4770 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00004771 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004772 return 1;
4773}
4774
4775int
4776obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena)
4777{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004778 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004779
Benjamin Petersond8f65972010-11-20 04:31:07 +00004780 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004781 int lineno;
4782 int col_offset;
4783
4784 if (obj == Py_None) {
4785 *out = NULL;
4786 return 0;
4787 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004788 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004789 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004790 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004791 if (tmp == NULL) goto failed;
4792 res = obj2ast_int(tmp, &lineno, arena);
4793 if (res != 0) goto failed;
4794 Py_XDECREF(tmp);
4795 tmp = NULL;
4796 } else {
4797 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from expr");
4798 return 1;
4799 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004800 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004801 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004802 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004803 if (tmp == NULL) goto failed;
4804 res = obj2ast_int(tmp, &col_offset, arena);
4805 if (res != 0) goto failed;
4806 Py_XDECREF(tmp);
4807 tmp = NULL;
4808 } else {
4809 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from expr");
4810 return 1;
4811 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004812 isinstance = PyObject_IsInstance(obj, (PyObject*)BoolOp_type);
4813 if (isinstance == -1) {
4814 return 1;
4815 }
4816 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004817 boolop_ty op;
4818 asdl_seq* values;
4819
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004820 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004821 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004822 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004823 if (tmp == NULL) goto failed;
4824 res = obj2ast_boolop(tmp, &op, arena);
4825 if (res != 0) goto failed;
4826 Py_XDECREF(tmp);
4827 tmp = NULL;
4828 } else {
4829 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BoolOp");
4830 return 1;
4831 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004832 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004833 int res;
4834 Py_ssize_t len;
4835 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004836 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004837 if (tmp == NULL) goto failed;
4838 if (!PyList_Check(tmp)) {
4839 PyErr_Format(PyExc_TypeError, "BoolOp field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4840 goto failed;
4841 }
4842 len = PyList_GET_SIZE(tmp);
4843 values = asdl_seq_new(len, arena);
4844 if (values == NULL) goto failed;
4845 for (i = 0; i < len; i++) {
4846 expr_ty value;
4847 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4848 if (res != 0) goto failed;
4849 asdl_seq_SET(values, i, value);
4850 }
4851 Py_XDECREF(tmp);
4852 tmp = NULL;
4853 } else {
4854 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from BoolOp");
4855 return 1;
4856 }
4857 *out = BoolOp(op, values, lineno, col_offset, arena);
4858 if (*out == NULL) goto failed;
4859 return 0;
4860 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004861 isinstance = PyObject_IsInstance(obj, (PyObject*)BinOp_type);
4862 if (isinstance == -1) {
4863 return 1;
4864 }
4865 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004866 expr_ty left;
4867 operator_ty op;
4868 expr_ty right;
4869
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004870 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004871 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004872 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004873 if (tmp == NULL) goto failed;
4874 res = obj2ast_expr(tmp, &left, arena);
4875 if (res != 0) goto failed;
4876 Py_XDECREF(tmp);
4877 tmp = NULL;
4878 } else {
4879 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from BinOp");
4880 return 1;
4881 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004882 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004883 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004884 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004885 if (tmp == NULL) goto failed;
4886 res = obj2ast_operator(tmp, &op, arena);
4887 if (res != 0) goto failed;
4888 Py_XDECREF(tmp);
4889 tmp = NULL;
4890 } else {
4891 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BinOp");
4892 return 1;
4893 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004894 if (_PyObject_HasAttrId(obj, &PyId_right)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004895 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004896 tmp = _PyObject_GetAttrId(obj, &PyId_right);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004897 if (tmp == NULL) goto failed;
4898 res = obj2ast_expr(tmp, &right, arena);
4899 if (res != 0) goto failed;
4900 Py_XDECREF(tmp);
4901 tmp = NULL;
4902 } else {
4903 PyErr_SetString(PyExc_TypeError, "required field \"right\" missing from BinOp");
4904 return 1;
4905 }
4906 *out = BinOp(left, op, right, lineno, col_offset, arena);
4907 if (*out == NULL) goto failed;
4908 return 0;
4909 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004910 isinstance = PyObject_IsInstance(obj, (PyObject*)UnaryOp_type);
4911 if (isinstance == -1) {
4912 return 1;
4913 }
4914 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004915 unaryop_ty op;
4916 expr_ty operand;
4917
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004918 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004919 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004920 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004921 if (tmp == NULL) goto failed;
4922 res = obj2ast_unaryop(tmp, &op, arena);
4923 if (res != 0) goto failed;
4924 Py_XDECREF(tmp);
4925 tmp = NULL;
4926 } else {
4927 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from UnaryOp");
4928 return 1;
4929 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004930 if (_PyObject_HasAttrId(obj, &PyId_operand)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004931 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004932 tmp = _PyObject_GetAttrId(obj, &PyId_operand);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004933 if (tmp == NULL) goto failed;
4934 res = obj2ast_expr(tmp, &operand, arena);
4935 if (res != 0) goto failed;
4936 Py_XDECREF(tmp);
4937 tmp = NULL;
4938 } else {
4939 PyErr_SetString(PyExc_TypeError, "required field \"operand\" missing from UnaryOp");
4940 return 1;
4941 }
4942 *out = UnaryOp(op, operand, lineno, col_offset, arena);
4943 if (*out == NULL) goto failed;
4944 return 0;
4945 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004946 isinstance = PyObject_IsInstance(obj, (PyObject*)Lambda_type);
4947 if (isinstance == -1) {
4948 return 1;
4949 }
4950 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004951 arguments_ty args;
4952 expr_ty body;
4953
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004954 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004955 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004956 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004957 if (tmp == NULL) goto failed;
4958 res = obj2ast_arguments(tmp, &args, arena);
4959 if (res != 0) goto failed;
4960 Py_XDECREF(tmp);
4961 tmp = NULL;
4962 } else {
4963 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Lambda");
4964 return 1;
4965 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004966 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004967 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004968 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004969 if (tmp == NULL) goto failed;
4970 res = obj2ast_expr(tmp, &body, arena);
4971 if (res != 0) goto failed;
4972 Py_XDECREF(tmp);
4973 tmp = NULL;
4974 } else {
4975 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Lambda");
4976 return 1;
4977 }
4978 *out = Lambda(args, body, lineno, col_offset, arena);
4979 if (*out == NULL) goto failed;
4980 return 0;
4981 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004982 isinstance = PyObject_IsInstance(obj, (PyObject*)IfExp_type);
4983 if (isinstance == -1) {
4984 return 1;
4985 }
4986 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004987 expr_ty test;
4988 expr_ty body;
4989 expr_ty orelse;
4990
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004991 if (_PyObject_HasAttrId(obj, &PyId_test)) {
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_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004994 if (tmp == NULL) goto failed;
4995 res = obj2ast_expr(tmp, &test, arena);
4996 if (res != 0) goto failed;
4997 Py_XDECREF(tmp);
4998 tmp = NULL;
4999 } else {
5000 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from IfExp");
5001 return 1;
5002 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005003 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005004 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005005 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005006 if (tmp == NULL) goto failed;
5007 res = obj2ast_expr(tmp, &body, arena);
5008 if (res != 0) goto failed;
5009 Py_XDECREF(tmp);
5010 tmp = NULL;
5011 } else {
5012 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from IfExp");
5013 return 1;
5014 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005015 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005016 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005017 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005018 if (tmp == NULL) goto failed;
5019 res = obj2ast_expr(tmp, &orelse, arena);
5020 if (res != 0) goto failed;
5021 Py_XDECREF(tmp);
5022 tmp = NULL;
5023 } else {
5024 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from IfExp");
5025 return 1;
5026 }
5027 *out = IfExp(test, body, orelse, lineno, col_offset, arena);
5028 if (*out == NULL) goto failed;
5029 return 0;
5030 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005031 isinstance = PyObject_IsInstance(obj, (PyObject*)Dict_type);
5032 if (isinstance == -1) {
5033 return 1;
5034 }
5035 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005036 asdl_seq* keys;
5037 asdl_seq* values;
5038
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005039 if (_PyObject_HasAttrId(obj, &PyId_keys)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005040 int res;
5041 Py_ssize_t len;
5042 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005043 tmp = _PyObject_GetAttrId(obj, &PyId_keys);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005044 if (tmp == NULL) goto failed;
5045 if (!PyList_Check(tmp)) {
5046 PyErr_Format(PyExc_TypeError, "Dict field \"keys\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5047 goto failed;
5048 }
5049 len = PyList_GET_SIZE(tmp);
5050 keys = asdl_seq_new(len, arena);
5051 if (keys == NULL) goto failed;
5052 for (i = 0; i < len; i++) {
5053 expr_ty value;
5054 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5055 if (res != 0) goto failed;
5056 asdl_seq_SET(keys, i, value);
5057 }
5058 Py_XDECREF(tmp);
5059 tmp = NULL;
5060 } else {
5061 PyErr_SetString(PyExc_TypeError, "required field \"keys\" missing from Dict");
5062 return 1;
5063 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005064 if (_PyObject_HasAttrId(obj, &PyId_values)) {
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_values);
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 \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5072 goto failed;
5073 }
5074 len = PyList_GET_SIZE(tmp);
5075 values = asdl_seq_new(len, arena);
5076 if (values == 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(values, i, value);
5082 }
5083 Py_XDECREF(tmp);
5084 tmp = NULL;
5085 } else {
5086 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from Dict");
5087 return 1;
5088 }
5089 *out = Dict(keys, values, lineno, col_offset, arena);
5090 if (*out == NULL) goto failed;
5091 return 0;
5092 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005093 isinstance = PyObject_IsInstance(obj, (PyObject*)Set_type);
5094 if (isinstance == -1) {
5095 return 1;
5096 }
5097 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005098 asdl_seq* elts;
5099
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005100 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005101 int res;
5102 Py_ssize_t len;
5103 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005104 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005105 if (tmp == NULL) goto failed;
5106 if (!PyList_Check(tmp)) {
5107 PyErr_Format(PyExc_TypeError, "Set field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5108 goto failed;
5109 }
5110 len = PyList_GET_SIZE(tmp);
5111 elts = asdl_seq_new(len, arena);
5112 if (elts == NULL) goto failed;
5113 for (i = 0; i < len; i++) {
5114 expr_ty value;
5115 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5116 if (res != 0) goto failed;
5117 asdl_seq_SET(elts, i, value);
5118 }
5119 Py_XDECREF(tmp);
5120 tmp = NULL;
5121 } else {
5122 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Set");
5123 return 1;
5124 }
5125 *out = Set(elts, lineno, col_offset, arena);
5126 if (*out == NULL) goto failed;
5127 return 0;
5128 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005129 isinstance = PyObject_IsInstance(obj, (PyObject*)ListComp_type);
5130 if (isinstance == -1) {
5131 return 1;
5132 }
5133 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005134 expr_ty elt;
5135 asdl_seq* generators;
5136
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005137 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005138 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005139 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005140 if (tmp == NULL) goto failed;
5141 res = obj2ast_expr(tmp, &elt, arena);
5142 if (res != 0) goto failed;
5143 Py_XDECREF(tmp);
5144 tmp = NULL;
5145 } else {
5146 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from ListComp");
5147 return 1;
5148 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005149 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005150 int res;
5151 Py_ssize_t len;
5152 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005153 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005154 if (tmp == NULL) goto failed;
5155 if (!PyList_Check(tmp)) {
5156 PyErr_Format(PyExc_TypeError, "ListComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5157 goto failed;
5158 }
5159 len = PyList_GET_SIZE(tmp);
5160 generators = asdl_seq_new(len, arena);
5161 if (generators == NULL) goto failed;
5162 for (i = 0; i < len; i++) {
5163 comprehension_ty value;
5164 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5165 if (res != 0) goto failed;
5166 asdl_seq_SET(generators, i, value);
5167 }
5168 Py_XDECREF(tmp);
5169 tmp = NULL;
5170 } else {
5171 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from ListComp");
5172 return 1;
5173 }
5174 *out = ListComp(elt, generators, lineno, col_offset, arena);
5175 if (*out == NULL) goto failed;
5176 return 0;
5177 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005178 isinstance = PyObject_IsInstance(obj, (PyObject*)SetComp_type);
5179 if (isinstance == -1) {
5180 return 1;
5181 }
5182 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005183 expr_ty elt;
5184 asdl_seq* generators;
5185
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005186 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005187 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005188 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005189 if (tmp == NULL) goto failed;
5190 res = obj2ast_expr(tmp, &elt, arena);
5191 if (res != 0) goto failed;
5192 Py_XDECREF(tmp);
5193 tmp = NULL;
5194 } else {
5195 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from SetComp");
5196 return 1;
5197 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005198 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005199 int res;
5200 Py_ssize_t len;
5201 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005202 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005203 if (tmp == NULL) goto failed;
5204 if (!PyList_Check(tmp)) {
5205 PyErr_Format(PyExc_TypeError, "SetComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5206 goto failed;
5207 }
5208 len = PyList_GET_SIZE(tmp);
5209 generators = asdl_seq_new(len, arena);
5210 if (generators == NULL) goto failed;
5211 for (i = 0; i < len; i++) {
5212 comprehension_ty value;
5213 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5214 if (res != 0) goto failed;
5215 asdl_seq_SET(generators, i, value);
5216 }
5217 Py_XDECREF(tmp);
5218 tmp = NULL;
5219 } else {
5220 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from SetComp");
5221 return 1;
5222 }
5223 *out = SetComp(elt, generators, lineno, col_offset, arena);
5224 if (*out == NULL) goto failed;
5225 return 0;
5226 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005227 isinstance = PyObject_IsInstance(obj, (PyObject*)DictComp_type);
5228 if (isinstance == -1) {
5229 return 1;
5230 }
5231 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005232 expr_ty key;
5233 expr_ty value;
5234 asdl_seq* generators;
5235
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005236 if (_PyObject_HasAttrId(obj, &PyId_key)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005237 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005238 tmp = _PyObject_GetAttrId(obj, &PyId_key);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005239 if (tmp == NULL) goto failed;
5240 res = obj2ast_expr(tmp, &key, arena);
5241 if (res != 0) goto failed;
5242 Py_XDECREF(tmp);
5243 tmp = NULL;
5244 } else {
5245 PyErr_SetString(PyExc_TypeError, "required field \"key\" missing from DictComp");
5246 return 1;
5247 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005248 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005249 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005250 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005251 if (tmp == NULL) goto failed;
5252 res = obj2ast_expr(tmp, &value, arena);
5253 if (res != 0) goto failed;
5254 Py_XDECREF(tmp);
5255 tmp = NULL;
5256 } else {
5257 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from DictComp");
5258 return 1;
5259 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005260 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005261 int res;
5262 Py_ssize_t len;
5263 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005264 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005265 if (tmp == NULL) goto failed;
5266 if (!PyList_Check(tmp)) {
5267 PyErr_Format(PyExc_TypeError, "DictComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5268 goto failed;
5269 }
5270 len = PyList_GET_SIZE(tmp);
5271 generators = asdl_seq_new(len, arena);
5272 if (generators == NULL) goto failed;
5273 for (i = 0; i < len; i++) {
5274 comprehension_ty value;
5275 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5276 if (res != 0) goto failed;
5277 asdl_seq_SET(generators, i, value);
5278 }
5279 Py_XDECREF(tmp);
5280 tmp = NULL;
5281 } else {
5282 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from DictComp");
5283 return 1;
5284 }
5285 *out = DictComp(key, value, generators, lineno, col_offset,
5286 arena);
5287 if (*out == NULL) goto failed;
5288 return 0;
5289 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005290 isinstance = PyObject_IsInstance(obj, (PyObject*)GeneratorExp_type);
5291 if (isinstance == -1) {
5292 return 1;
5293 }
5294 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005295 expr_ty elt;
5296 asdl_seq* generators;
5297
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005298 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005299 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005300 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005301 if (tmp == NULL) goto failed;
5302 res = obj2ast_expr(tmp, &elt, arena);
5303 if (res != 0) goto failed;
5304 Py_XDECREF(tmp);
5305 tmp = NULL;
5306 } else {
5307 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from GeneratorExp");
5308 return 1;
5309 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005310 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005311 int res;
5312 Py_ssize_t len;
5313 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005314 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005315 if (tmp == NULL) goto failed;
5316 if (!PyList_Check(tmp)) {
5317 PyErr_Format(PyExc_TypeError, "GeneratorExp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5318 goto failed;
5319 }
5320 len = PyList_GET_SIZE(tmp);
5321 generators = asdl_seq_new(len, arena);
5322 if (generators == NULL) goto failed;
5323 for (i = 0; i < len; i++) {
5324 comprehension_ty value;
5325 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5326 if (res != 0) goto failed;
5327 asdl_seq_SET(generators, i, value);
5328 }
5329 Py_XDECREF(tmp);
5330 tmp = NULL;
5331 } else {
5332 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from GeneratorExp");
5333 return 1;
5334 }
5335 *out = GeneratorExp(elt, generators, lineno, col_offset, arena);
5336 if (*out == NULL) goto failed;
5337 return 0;
5338 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005339 isinstance = PyObject_IsInstance(obj, (PyObject*)Yield_type);
5340 if (isinstance == -1) {
5341 return 1;
5342 }
5343 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005344 expr_ty value;
5345
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005346 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005347 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005348 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005349 if (tmp == NULL) goto failed;
5350 res = obj2ast_expr(tmp, &value, arena);
5351 if (res != 0) goto failed;
5352 Py_XDECREF(tmp);
5353 tmp = NULL;
5354 } else {
5355 value = NULL;
5356 }
5357 *out = Yield(value, lineno, col_offset, arena);
5358 if (*out == NULL) goto failed;
5359 return 0;
5360 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005361 isinstance = PyObject_IsInstance(obj, (PyObject*)Compare_type);
5362 if (isinstance == -1) {
5363 return 1;
5364 }
5365 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005366 expr_ty left;
5367 asdl_int_seq* ops;
5368 asdl_seq* comparators;
5369
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005370 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005371 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005372 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005373 if (tmp == NULL) goto failed;
5374 res = obj2ast_expr(tmp, &left, arena);
5375 if (res != 0) goto failed;
5376 Py_XDECREF(tmp);
5377 tmp = NULL;
5378 } else {
5379 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from Compare");
5380 return 1;
5381 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005382 if (_PyObject_HasAttrId(obj, &PyId_ops)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005383 int res;
5384 Py_ssize_t len;
5385 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005386 tmp = _PyObject_GetAttrId(obj, &PyId_ops);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005387 if (tmp == NULL) goto failed;
5388 if (!PyList_Check(tmp)) {
5389 PyErr_Format(PyExc_TypeError, "Compare field \"ops\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5390 goto failed;
5391 }
5392 len = PyList_GET_SIZE(tmp);
5393 ops = asdl_int_seq_new(len, arena);
5394 if (ops == NULL) goto failed;
5395 for (i = 0; i < len; i++) {
5396 cmpop_ty value;
5397 res = obj2ast_cmpop(PyList_GET_ITEM(tmp, i), &value, arena);
5398 if (res != 0) goto failed;
5399 asdl_seq_SET(ops, i, value);
5400 }
5401 Py_XDECREF(tmp);
5402 tmp = NULL;
5403 } else {
5404 PyErr_SetString(PyExc_TypeError, "required field \"ops\" missing from Compare");
5405 return 1;
5406 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005407 if (_PyObject_HasAttrId(obj, &PyId_comparators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005408 int res;
5409 Py_ssize_t len;
5410 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005411 tmp = _PyObject_GetAttrId(obj, &PyId_comparators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005412 if (tmp == NULL) goto failed;
5413 if (!PyList_Check(tmp)) {
5414 PyErr_Format(PyExc_TypeError, "Compare field \"comparators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5415 goto failed;
5416 }
5417 len = PyList_GET_SIZE(tmp);
5418 comparators = asdl_seq_new(len, arena);
5419 if (comparators == NULL) goto failed;
5420 for (i = 0; i < len; i++) {
5421 expr_ty value;
5422 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5423 if (res != 0) goto failed;
5424 asdl_seq_SET(comparators, i, value);
5425 }
5426 Py_XDECREF(tmp);
5427 tmp = NULL;
5428 } else {
5429 PyErr_SetString(PyExc_TypeError, "required field \"comparators\" missing from Compare");
5430 return 1;
5431 }
5432 *out = Compare(left, ops, comparators, lineno, col_offset,
5433 arena);
5434 if (*out == NULL) goto failed;
5435 return 0;
5436 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005437 isinstance = PyObject_IsInstance(obj, (PyObject*)Call_type);
5438 if (isinstance == -1) {
5439 return 1;
5440 }
5441 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005442 expr_ty func;
5443 asdl_seq* args;
5444 asdl_seq* keywords;
5445 expr_ty starargs;
5446 expr_ty kwargs;
5447
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005448 if (_PyObject_HasAttrId(obj, &PyId_func)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005449 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005450 tmp = _PyObject_GetAttrId(obj, &PyId_func);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005451 if (tmp == NULL) goto failed;
5452 res = obj2ast_expr(tmp, &func, arena);
5453 if (res != 0) goto failed;
5454 Py_XDECREF(tmp);
5455 tmp = NULL;
5456 } else {
5457 PyErr_SetString(PyExc_TypeError, "required field \"func\" missing from Call");
5458 return 1;
5459 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005460 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005461 int res;
5462 Py_ssize_t len;
5463 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005464 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005465 if (tmp == NULL) goto failed;
5466 if (!PyList_Check(tmp)) {
5467 PyErr_Format(PyExc_TypeError, "Call field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5468 goto failed;
5469 }
5470 len = PyList_GET_SIZE(tmp);
5471 args = asdl_seq_new(len, arena);
5472 if (args == NULL) goto failed;
5473 for (i = 0; i < len; i++) {
5474 expr_ty value;
5475 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5476 if (res != 0) goto failed;
5477 asdl_seq_SET(args, i, value);
5478 }
5479 Py_XDECREF(tmp);
5480 tmp = NULL;
5481 } else {
5482 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Call");
5483 return 1;
5484 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005485 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005486 int res;
5487 Py_ssize_t len;
5488 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005489 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005490 if (tmp == NULL) goto failed;
5491 if (!PyList_Check(tmp)) {
5492 PyErr_Format(PyExc_TypeError, "Call field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5493 goto failed;
5494 }
5495 len = PyList_GET_SIZE(tmp);
5496 keywords = asdl_seq_new(len, arena);
5497 if (keywords == NULL) goto failed;
5498 for (i = 0; i < len; i++) {
5499 keyword_ty value;
5500 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
5501 if (res != 0) goto failed;
5502 asdl_seq_SET(keywords, i, value);
5503 }
5504 Py_XDECREF(tmp);
5505 tmp = NULL;
5506 } else {
5507 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from Call");
5508 return 1;
5509 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005510 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005511 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005512 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005513 if (tmp == NULL) goto failed;
5514 res = obj2ast_expr(tmp, &starargs, arena);
5515 if (res != 0) goto failed;
5516 Py_XDECREF(tmp);
5517 tmp = NULL;
5518 } else {
5519 starargs = NULL;
5520 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005521 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005522 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005523 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005524 if (tmp == NULL) goto failed;
5525 res = obj2ast_expr(tmp, &kwargs, arena);
5526 if (res != 0) goto failed;
5527 Py_XDECREF(tmp);
5528 tmp = NULL;
5529 } else {
5530 kwargs = NULL;
5531 }
5532 *out = Call(func, args, keywords, starargs, kwargs, lineno,
5533 col_offset, arena);
5534 if (*out == NULL) goto failed;
5535 return 0;
5536 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005537 isinstance = PyObject_IsInstance(obj, (PyObject*)Num_type);
5538 if (isinstance == -1) {
5539 return 1;
5540 }
5541 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005542 object n;
5543
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005544 if (_PyObject_HasAttrId(obj, &PyId_n)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005545 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005546 tmp = _PyObject_GetAttrId(obj, &PyId_n);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005547 if (tmp == NULL) goto failed;
5548 res = obj2ast_object(tmp, &n, arena);
5549 if (res != 0) goto failed;
5550 Py_XDECREF(tmp);
5551 tmp = NULL;
5552 } else {
5553 PyErr_SetString(PyExc_TypeError, "required field \"n\" missing from Num");
5554 return 1;
5555 }
5556 *out = Num(n, lineno, col_offset, arena);
5557 if (*out == NULL) goto failed;
5558 return 0;
5559 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005560 isinstance = PyObject_IsInstance(obj, (PyObject*)Str_type);
5561 if (isinstance == -1) {
5562 return 1;
5563 }
5564 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005565 string s;
5566
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005567 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005568 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005569 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005570 if (tmp == NULL) goto failed;
5571 res = obj2ast_string(tmp, &s, arena);
5572 if (res != 0) goto failed;
5573 Py_XDECREF(tmp);
5574 tmp = NULL;
5575 } else {
5576 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Str");
5577 return 1;
5578 }
5579 *out = Str(s, lineno, col_offset, arena);
5580 if (*out == NULL) goto failed;
5581 return 0;
5582 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005583 isinstance = PyObject_IsInstance(obj, (PyObject*)Bytes_type);
5584 if (isinstance == -1) {
5585 return 1;
5586 }
5587 if (isinstance) {
Benjamin Petersone2498412011-08-09 16:08:39 -05005588 bytes s;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005589
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005590 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005591 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005592 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005593 if (tmp == NULL) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05005594 res = obj2ast_bytes(tmp, &s, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005595 if (res != 0) goto failed;
5596 Py_XDECREF(tmp);
5597 tmp = NULL;
5598 } else {
5599 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Bytes");
5600 return 1;
5601 }
5602 *out = Bytes(s, lineno, col_offset, arena);
5603 if (*out == NULL) goto failed;
5604 return 0;
5605 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005606 isinstance = PyObject_IsInstance(obj, (PyObject*)Ellipsis_type);
5607 if (isinstance == -1) {
5608 return 1;
5609 }
5610 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005611
5612 *out = Ellipsis(lineno, col_offset, arena);
5613 if (*out == NULL) goto failed;
5614 return 0;
5615 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005616 isinstance = PyObject_IsInstance(obj, (PyObject*)Attribute_type);
5617 if (isinstance == -1) {
5618 return 1;
5619 }
5620 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005621 expr_ty value;
5622 identifier attr;
5623 expr_context_ty ctx;
5624
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005625 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005626 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005627 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005628 if (tmp == NULL) goto failed;
5629 res = obj2ast_expr(tmp, &value, arena);
5630 if (res != 0) goto failed;
5631 Py_XDECREF(tmp);
5632 tmp = NULL;
5633 } else {
5634 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Attribute");
5635 return 1;
5636 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005637 if (_PyObject_HasAttrId(obj, &PyId_attr)) {
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_attr);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005640 if (tmp == NULL) goto failed;
5641 res = obj2ast_identifier(tmp, &attr, arena);
5642 if (res != 0) goto failed;
5643 Py_XDECREF(tmp);
5644 tmp = NULL;
5645 } else {
5646 PyErr_SetString(PyExc_TypeError, "required field \"attr\" missing from Attribute");
5647 return 1;
5648 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005649 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005650 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005651 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005652 if (tmp == NULL) goto failed;
5653 res = obj2ast_expr_context(tmp, &ctx, arena);
5654 if (res != 0) goto failed;
5655 Py_XDECREF(tmp);
5656 tmp = NULL;
5657 } else {
5658 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Attribute");
5659 return 1;
5660 }
5661 *out = Attribute(value, attr, ctx, lineno, col_offset, arena);
5662 if (*out == NULL) goto failed;
5663 return 0;
5664 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005665 isinstance = PyObject_IsInstance(obj, (PyObject*)Subscript_type);
5666 if (isinstance == -1) {
5667 return 1;
5668 }
5669 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005670 expr_ty value;
5671 slice_ty slice;
5672 expr_context_ty ctx;
5673
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005674 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005675 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005676 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005677 if (tmp == NULL) goto failed;
5678 res = obj2ast_expr(tmp, &value, arena);
5679 if (res != 0) goto failed;
5680 Py_XDECREF(tmp);
5681 tmp = NULL;
5682 } else {
5683 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Subscript");
5684 return 1;
5685 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005686 if (_PyObject_HasAttrId(obj, &PyId_slice)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005687 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005688 tmp = _PyObject_GetAttrId(obj, &PyId_slice);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005689 if (tmp == NULL) goto failed;
5690 res = obj2ast_slice(tmp, &slice, arena);
5691 if (res != 0) goto failed;
5692 Py_XDECREF(tmp);
5693 tmp = NULL;
5694 } else {
5695 PyErr_SetString(PyExc_TypeError, "required field \"slice\" missing from Subscript");
5696 return 1;
5697 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005698 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005699 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005700 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005701 if (tmp == NULL) goto failed;
5702 res = obj2ast_expr_context(tmp, &ctx, arena);
5703 if (res != 0) goto failed;
5704 Py_XDECREF(tmp);
5705 tmp = NULL;
5706 } else {
5707 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Subscript");
5708 return 1;
5709 }
5710 *out = Subscript(value, slice, ctx, lineno, col_offset, arena);
5711 if (*out == NULL) goto failed;
5712 return 0;
5713 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005714 isinstance = PyObject_IsInstance(obj, (PyObject*)Starred_type);
5715 if (isinstance == -1) {
5716 return 1;
5717 }
5718 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005719 expr_ty value;
5720 expr_context_ty ctx;
5721
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005722 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005723 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005724 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005725 if (tmp == NULL) goto failed;
5726 res = obj2ast_expr(tmp, &value, arena);
5727 if (res != 0) goto failed;
5728 Py_XDECREF(tmp);
5729 tmp = NULL;
5730 } else {
5731 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Starred");
5732 return 1;
5733 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005734 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005735 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005736 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005737 if (tmp == NULL) goto failed;
5738 res = obj2ast_expr_context(tmp, &ctx, arena);
5739 if (res != 0) goto failed;
5740 Py_XDECREF(tmp);
5741 tmp = NULL;
5742 } else {
5743 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Starred");
5744 return 1;
5745 }
5746 *out = Starred(value, ctx, lineno, col_offset, arena);
5747 if (*out == NULL) goto failed;
5748 return 0;
5749 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005750 isinstance = PyObject_IsInstance(obj, (PyObject*)Name_type);
5751 if (isinstance == -1) {
5752 return 1;
5753 }
5754 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005755 identifier id;
5756 expr_context_ty ctx;
5757
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005758 if (_PyObject_HasAttrId(obj, &PyId_id)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005759 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005760 tmp = _PyObject_GetAttrId(obj, &PyId_id);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005761 if (tmp == NULL) goto failed;
5762 res = obj2ast_identifier(tmp, &id, arena);
5763 if (res != 0) goto failed;
5764 Py_XDECREF(tmp);
5765 tmp = NULL;
5766 } else {
5767 PyErr_SetString(PyExc_TypeError, "required field \"id\" missing from Name");
5768 return 1;
5769 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005770 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005771 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005772 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005773 if (tmp == NULL) goto failed;
5774 res = obj2ast_expr_context(tmp, &ctx, arena);
5775 if (res != 0) goto failed;
5776 Py_XDECREF(tmp);
5777 tmp = NULL;
5778 } else {
5779 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Name");
5780 return 1;
5781 }
5782 *out = Name(id, ctx, lineno, col_offset, arena);
5783 if (*out == NULL) goto failed;
5784 return 0;
5785 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005786 isinstance = PyObject_IsInstance(obj, (PyObject*)List_type);
5787 if (isinstance == -1) {
5788 return 1;
5789 }
5790 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005791 asdl_seq* elts;
5792 expr_context_ty ctx;
5793
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005794 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005795 int res;
5796 Py_ssize_t len;
5797 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005798 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005799 if (tmp == NULL) goto failed;
5800 if (!PyList_Check(tmp)) {
5801 PyErr_Format(PyExc_TypeError, "List field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5802 goto failed;
5803 }
5804 len = PyList_GET_SIZE(tmp);
5805 elts = asdl_seq_new(len, arena);
5806 if (elts == NULL) goto failed;
5807 for (i = 0; i < len; i++) {
5808 expr_ty value;
5809 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5810 if (res != 0) goto failed;
5811 asdl_seq_SET(elts, i, value);
5812 }
5813 Py_XDECREF(tmp);
5814 tmp = NULL;
5815 } else {
5816 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from List");
5817 return 1;
5818 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005819 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005820 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005821 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005822 if (tmp == NULL) goto failed;
5823 res = obj2ast_expr_context(tmp, &ctx, arena);
5824 if (res != 0) goto failed;
5825 Py_XDECREF(tmp);
5826 tmp = NULL;
5827 } else {
5828 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from List");
5829 return 1;
5830 }
5831 *out = List(elts, ctx, lineno, col_offset, arena);
5832 if (*out == NULL) goto failed;
5833 return 0;
5834 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005835 isinstance = PyObject_IsInstance(obj, (PyObject*)Tuple_type);
5836 if (isinstance == -1) {
5837 return 1;
5838 }
5839 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005840 asdl_seq* elts;
5841 expr_context_ty ctx;
5842
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005843 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005844 int res;
5845 Py_ssize_t len;
5846 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005847 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005848 if (tmp == NULL) goto failed;
5849 if (!PyList_Check(tmp)) {
5850 PyErr_Format(PyExc_TypeError, "Tuple field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5851 goto failed;
5852 }
5853 len = PyList_GET_SIZE(tmp);
5854 elts = asdl_seq_new(len, arena);
5855 if (elts == NULL) goto failed;
5856 for (i = 0; i < len; i++) {
5857 expr_ty value;
5858 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5859 if (res != 0) goto failed;
5860 asdl_seq_SET(elts, i, value);
5861 }
5862 Py_XDECREF(tmp);
5863 tmp = NULL;
5864 } else {
5865 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Tuple");
5866 return 1;
5867 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005868 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005869 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005870 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005871 if (tmp == NULL) goto failed;
5872 res = obj2ast_expr_context(tmp, &ctx, arena);
5873 if (res != 0) goto failed;
5874 Py_XDECREF(tmp);
5875 tmp = NULL;
5876 } else {
5877 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Tuple");
5878 return 1;
5879 }
5880 *out = Tuple(elts, ctx, lineno, col_offset, arena);
5881 if (*out == NULL) goto failed;
5882 return 0;
5883 }
5884
Benjamin Peterson5b066812010-11-20 01:38:49 +00005885 PyErr_Format(PyExc_TypeError, "expected some sort of expr, but got %R", obj);
5886 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00005887 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005888 return 1;
5889}
5890
5891int
5892obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena* arena)
5893{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005894 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005895
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005896 isinstance = PyObject_IsInstance(obj, (PyObject *)Load_type);
5897 if (isinstance == -1) {
5898 return 1;
5899 }
5900 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005901 *out = Load;
5902 return 0;
5903 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005904 isinstance = PyObject_IsInstance(obj, (PyObject *)Store_type);
5905 if (isinstance == -1) {
5906 return 1;
5907 }
5908 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005909 *out = Store;
5910 return 0;
5911 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005912 isinstance = PyObject_IsInstance(obj, (PyObject *)Del_type);
5913 if (isinstance == -1) {
5914 return 1;
5915 }
5916 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005917 *out = Del;
5918 return 0;
5919 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005920 isinstance = PyObject_IsInstance(obj, (PyObject *)AugLoad_type);
5921 if (isinstance == -1) {
5922 return 1;
5923 }
5924 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005925 *out = AugLoad;
5926 return 0;
5927 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005928 isinstance = PyObject_IsInstance(obj, (PyObject *)AugStore_type);
5929 if (isinstance == -1) {
5930 return 1;
5931 }
5932 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005933 *out = AugStore;
5934 return 0;
5935 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005936 isinstance = PyObject_IsInstance(obj, (PyObject *)Param_type);
5937 if (isinstance == -1) {
5938 return 1;
5939 }
5940 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005941 *out = Param;
5942 return 0;
5943 }
5944
Benjamin Peterson5b066812010-11-20 01:38:49 +00005945 PyErr_Format(PyExc_TypeError, "expected some sort of expr_context, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005946 return 1;
5947}
5948
5949int
5950obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena)
5951{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005952 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005953
Benjamin Petersond8f65972010-11-20 04:31:07 +00005954 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005955
5956 if (obj == Py_None) {
5957 *out = NULL;
5958 return 0;
5959 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005960 isinstance = PyObject_IsInstance(obj, (PyObject*)Slice_type);
5961 if (isinstance == -1) {
5962 return 1;
5963 }
5964 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005965 expr_ty lower;
5966 expr_ty upper;
5967 expr_ty step;
5968
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005969 if (_PyObject_HasAttrId(obj, &PyId_lower)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005970 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005971 tmp = _PyObject_GetAttrId(obj, &PyId_lower);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005972 if (tmp == NULL) goto failed;
5973 res = obj2ast_expr(tmp, &lower, arena);
5974 if (res != 0) goto failed;
5975 Py_XDECREF(tmp);
5976 tmp = NULL;
5977 } else {
5978 lower = NULL;
5979 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005980 if (_PyObject_HasAttrId(obj, &PyId_upper)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005981 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005982 tmp = _PyObject_GetAttrId(obj, &PyId_upper);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005983 if (tmp == NULL) goto failed;
5984 res = obj2ast_expr(tmp, &upper, arena);
5985 if (res != 0) goto failed;
5986 Py_XDECREF(tmp);
5987 tmp = NULL;
5988 } else {
5989 upper = NULL;
5990 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005991 if (_PyObject_HasAttrId(obj, &PyId_step)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005992 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005993 tmp = _PyObject_GetAttrId(obj, &PyId_step);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005994 if (tmp == NULL) goto failed;
5995 res = obj2ast_expr(tmp, &step, arena);
5996 if (res != 0) goto failed;
5997 Py_XDECREF(tmp);
5998 tmp = NULL;
5999 } else {
6000 step = NULL;
6001 }
6002 *out = Slice(lower, upper, step, arena);
6003 if (*out == NULL) goto failed;
6004 return 0;
6005 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006006 isinstance = PyObject_IsInstance(obj, (PyObject*)ExtSlice_type);
6007 if (isinstance == -1) {
6008 return 1;
6009 }
6010 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006011 asdl_seq* dims;
6012
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006013 if (_PyObject_HasAttrId(obj, &PyId_dims)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006014 int res;
6015 Py_ssize_t len;
6016 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006017 tmp = _PyObject_GetAttrId(obj, &PyId_dims);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006018 if (tmp == NULL) goto failed;
6019 if (!PyList_Check(tmp)) {
6020 PyErr_Format(PyExc_TypeError, "ExtSlice field \"dims\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6021 goto failed;
6022 }
6023 len = PyList_GET_SIZE(tmp);
6024 dims = asdl_seq_new(len, arena);
6025 if (dims == NULL) goto failed;
6026 for (i = 0; i < len; i++) {
6027 slice_ty value;
6028 res = obj2ast_slice(PyList_GET_ITEM(tmp, i), &value, arena);
6029 if (res != 0) goto failed;
6030 asdl_seq_SET(dims, i, value);
6031 }
6032 Py_XDECREF(tmp);
6033 tmp = NULL;
6034 } else {
6035 PyErr_SetString(PyExc_TypeError, "required field \"dims\" missing from ExtSlice");
6036 return 1;
6037 }
6038 *out = ExtSlice(dims, arena);
6039 if (*out == NULL) goto failed;
6040 return 0;
6041 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006042 isinstance = PyObject_IsInstance(obj, (PyObject*)Index_type);
6043 if (isinstance == -1) {
6044 return 1;
6045 }
6046 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006047 expr_ty value;
6048
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006049 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006050 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006051 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006052 if (tmp == NULL) goto failed;
6053 res = obj2ast_expr(tmp, &value, arena);
6054 if (res != 0) goto failed;
6055 Py_XDECREF(tmp);
6056 tmp = NULL;
6057 } else {
6058 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Index");
6059 return 1;
6060 }
6061 *out = Index(value, arena);
6062 if (*out == NULL) goto failed;
6063 return 0;
6064 }
6065
Benjamin Peterson5b066812010-11-20 01:38:49 +00006066 PyErr_Format(PyExc_TypeError, "expected some sort of slice, but got %R", obj);
6067 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006068 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006069 return 1;
6070}
6071
6072int
6073obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena)
6074{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006075 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006076
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006077 isinstance = PyObject_IsInstance(obj, (PyObject *)And_type);
6078 if (isinstance == -1) {
6079 return 1;
6080 }
6081 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006082 *out = And;
6083 return 0;
6084 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006085 isinstance = PyObject_IsInstance(obj, (PyObject *)Or_type);
6086 if (isinstance == -1) {
6087 return 1;
6088 }
6089 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006090 *out = Or;
6091 return 0;
6092 }
6093
Benjamin Peterson5b066812010-11-20 01:38:49 +00006094 PyErr_Format(PyExc_TypeError, "expected some sort of boolop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006095 return 1;
6096}
6097
6098int
6099obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena)
6100{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006101 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006102
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006103 isinstance = PyObject_IsInstance(obj, (PyObject *)Add_type);
6104 if (isinstance == -1) {
6105 return 1;
6106 }
6107 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006108 *out = Add;
6109 return 0;
6110 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006111 isinstance = PyObject_IsInstance(obj, (PyObject *)Sub_type);
6112 if (isinstance == -1) {
6113 return 1;
6114 }
6115 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006116 *out = Sub;
6117 return 0;
6118 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006119 isinstance = PyObject_IsInstance(obj, (PyObject *)Mult_type);
6120 if (isinstance == -1) {
6121 return 1;
6122 }
6123 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006124 *out = Mult;
6125 return 0;
6126 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006127 isinstance = PyObject_IsInstance(obj, (PyObject *)Div_type);
6128 if (isinstance == -1) {
6129 return 1;
6130 }
6131 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006132 *out = Div;
6133 return 0;
6134 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006135 isinstance = PyObject_IsInstance(obj, (PyObject *)Mod_type);
6136 if (isinstance == -1) {
6137 return 1;
6138 }
6139 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006140 *out = Mod;
6141 return 0;
6142 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006143 isinstance = PyObject_IsInstance(obj, (PyObject *)Pow_type);
6144 if (isinstance == -1) {
6145 return 1;
6146 }
6147 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006148 *out = Pow;
6149 return 0;
6150 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006151 isinstance = PyObject_IsInstance(obj, (PyObject *)LShift_type);
6152 if (isinstance == -1) {
6153 return 1;
6154 }
6155 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006156 *out = LShift;
6157 return 0;
6158 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006159 isinstance = PyObject_IsInstance(obj, (PyObject *)RShift_type);
6160 if (isinstance == -1) {
6161 return 1;
6162 }
6163 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006164 *out = RShift;
6165 return 0;
6166 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006167 isinstance = PyObject_IsInstance(obj, (PyObject *)BitOr_type);
6168 if (isinstance == -1) {
6169 return 1;
6170 }
6171 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006172 *out = BitOr;
6173 return 0;
6174 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006175 isinstance = PyObject_IsInstance(obj, (PyObject *)BitXor_type);
6176 if (isinstance == -1) {
6177 return 1;
6178 }
6179 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006180 *out = BitXor;
6181 return 0;
6182 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006183 isinstance = PyObject_IsInstance(obj, (PyObject *)BitAnd_type);
6184 if (isinstance == -1) {
6185 return 1;
6186 }
6187 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006188 *out = BitAnd;
6189 return 0;
6190 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006191 isinstance = PyObject_IsInstance(obj, (PyObject *)FloorDiv_type);
6192 if (isinstance == -1) {
6193 return 1;
6194 }
6195 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006196 *out = FloorDiv;
6197 return 0;
6198 }
6199
Benjamin Peterson5b066812010-11-20 01:38:49 +00006200 PyErr_Format(PyExc_TypeError, "expected some sort of operator, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006201 return 1;
6202}
6203
6204int
6205obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena)
6206{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006207 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006208
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006209 isinstance = PyObject_IsInstance(obj, (PyObject *)Invert_type);
6210 if (isinstance == -1) {
6211 return 1;
6212 }
6213 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006214 *out = Invert;
6215 return 0;
6216 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006217 isinstance = PyObject_IsInstance(obj, (PyObject *)Not_type);
6218 if (isinstance == -1) {
6219 return 1;
6220 }
6221 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006222 *out = Not;
6223 return 0;
6224 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006225 isinstance = PyObject_IsInstance(obj, (PyObject *)UAdd_type);
6226 if (isinstance == -1) {
6227 return 1;
6228 }
6229 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006230 *out = UAdd;
6231 return 0;
6232 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006233 isinstance = PyObject_IsInstance(obj, (PyObject *)USub_type);
6234 if (isinstance == -1) {
6235 return 1;
6236 }
6237 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006238 *out = USub;
6239 return 0;
6240 }
6241
Benjamin Peterson5b066812010-11-20 01:38:49 +00006242 PyErr_Format(PyExc_TypeError, "expected some sort of unaryop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006243 return 1;
6244}
6245
6246int
6247obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena)
6248{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006249 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006250
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006251 isinstance = PyObject_IsInstance(obj, (PyObject *)Eq_type);
6252 if (isinstance == -1) {
6253 return 1;
6254 }
6255 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006256 *out = Eq;
6257 return 0;
6258 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006259 isinstance = PyObject_IsInstance(obj, (PyObject *)NotEq_type);
6260 if (isinstance == -1) {
6261 return 1;
6262 }
6263 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006264 *out = NotEq;
6265 return 0;
6266 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006267 isinstance = PyObject_IsInstance(obj, (PyObject *)Lt_type);
6268 if (isinstance == -1) {
6269 return 1;
6270 }
6271 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006272 *out = Lt;
6273 return 0;
6274 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006275 isinstance = PyObject_IsInstance(obj, (PyObject *)LtE_type);
6276 if (isinstance == -1) {
6277 return 1;
6278 }
6279 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006280 *out = LtE;
6281 return 0;
6282 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006283 isinstance = PyObject_IsInstance(obj, (PyObject *)Gt_type);
6284 if (isinstance == -1) {
6285 return 1;
6286 }
6287 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006288 *out = Gt;
6289 return 0;
6290 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006291 isinstance = PyObject_IsInstance(obj, (PyObject *)GtE_type);
6292 if (isinstance == -1) {
6293 return 1;
6294 }
6295 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006296 *out = GtE;
6297 return 0;
6298 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006299 isinstance = PyObject_IsInstance(obj, (PyObject *)Is_type);
6300 if (isinstance == -1) {
6301 return 1;
6302 }
6303 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006304 *out = Is;
6305 return 0;
6306 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006307 isinstance = PyObject_IsInstance(obj, (PyObject *)IsNot_type);
6308 if (isinstance == -1) {
6309 return 1;
6310 }
6311 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006312 *out = IsNot;
6313 return 0;
6314 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006315 isinstance = PyObject_IsInstance(obj, (PyObject *)In_type);
6316 if (isinstance == -1) {
6317 return 1;
6318 }
6319 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006320 *out = In;
6321 return 0;
6322 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006323 isinstance = PyObject_IsInstance(obj, (PyObject *)NotIn_type);
6324 if (isinstance == -1) {
6325 return 1;
6326 }
6327 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006328 *out = NotIn;
6329 return 0;
6330 }
6331
Benjamin Peterson5b066812010-11-20 01:38:49 +00006332 PyErr_Format(PyExc_TypeError, "expected some sort of cmpop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006333 return 1;
6334}
6335
6336int
6337obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena* arena)
6338{
6339 PyObject* tmp = NULL;
6340 expr_ty target;
6341 expr_ty iter;
6342 asdl_seq* ifs;
6343
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006344 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006345 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006346 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006347 if (tmp == NULL) goto failed;
6348 res = obj2ast_expr(tmp, &target, arena);
6349 if (res != 0) goto failed;
6350 Py_XDECREF(tmp);
6351 tmp = NULL;
6352 } else {
6353 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from comprehension");
6354 return 1;
6355 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006356 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006357 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006358 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006359 if (tmp == NULL) goto failed;
6360 res = obj2ast_expr(tmp, &iter, arena);
6361 if (res != 0) goto failed;
6362 Py_XDECREF(tmp);
6363 tmp = NULL;
6364 } else {
6365 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from comprehension");
6366 return 1;
6367 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006368 if (_PyObject_HasAttrId(obj, &PyId_ifs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006369 int res;
6370 Py_ssize_t len;
6371 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006372 tmp = _PyObject_GetAttrId(obj, &PyId_ifs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006373 if (tmp == NULL) goto failed;
6374 if (!PyList_Check(tmp)) {
6375 PyErr_Format(PyExc_TypeError, "comprehension field \"ifs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6376 goto failed;
6377 }
6378 len = PyList_GET_SIZE(tmp);
6379 ifs = asdl_seq_new(len, arena);
6380 if (ifs == NULL) goto failed;
6381 for (i = 0; i < len; i++) {
6382 expr_ty value;
6383 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6384 if (res != 0) goto failed;
6385 asdl_seq_SET(ifs, i, value);
6386 }
6387 Py_XDECREF(tmp);
6388 tmp = NULL;
6389 } else {
6390 PyErr_SetString(PyExc_TypeError, "required field \"ifs\" missing from comprehension");
6391 return 1;
6392 }
6393 *out = comprehension(target, iter, ifs, arena);
6394 return 0;
6395failed:
6396 Py_XDECREF(tmp);
6397 return 1;
6398}
6399
6400int
6401obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena* arena)
6402{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006403 int isinstance;
Neal Norwitzad74aa82008-03-31 05:14:30 +00006404
Benjamin Petersond8f65972010-11-20 04:31:07 +00006405 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006406 int lineno;
6407 int col_offset;
6408
Neal Norwitzad74aa82008-03-31 05:14:30 +00006409 if (obj == Py_None) {
6410 *out = NULL;
6411 return 0;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006412 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006413 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006414 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006415 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006416 if (tmp == NULL) goto failed;
6417 res = obj2ast_int(tmp, &lineno, arena);
6418 if (res != 0) goto failed;
6419 Py_XDECREF(tmp);
6420 tmp = NULL;
6421 } else {
6422 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from excepthandler");
6423 return 1;
6424 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006425 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006426 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006427 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006428 if (tmp == NULL) goto failed;
6429 res = obj2ast_int(tmp, &col_offset, arena);
6430 if (res != 0) goto failed;
6431 Py_XDECREF(tmp);
6432 tmp = NULL;
6433 } else {
6434 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from excepthandler");
6435 return 1;
6436 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006437 isinstance = PyObject_IsInstance(obj, (PyObject*)ExceptHandler_type);
6438 if (isinstance == -1) {
6439 return 1;
6440 }
6441 if (isinstance) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006442 expr_ty type;
6443 identifier name;
6444 asdl_seq* body;
6445
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006446 if (_PyObject_HasAttrId(obj, &PyId_type)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006447 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006448 tmp = _PyObject_GetAttrId(obj, &PyId_type);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006449 if (tmp == NULL) goto failed;
6450 res = obj2ast_expr(tmp, &type, arena);
6451 if (res != 0) goto failed;
6452 Py_XDECREF(tmp);
6453 tmp = NULL;
6454 } else {
6455 type = NULL;
6456 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006457 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006458 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006459 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006460 if (tmp == NULL) goto failed;
6461 res = obj2ast_identifier(tmp, &name, arena);
6462 if (res != 0) goto failed;
6463 Py_XDECREF(tmp);
6464 tmp = NULL;
6465 } else {
6466 name = NULL;
6467 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006468 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006469 int res;
6470 Py_ssize_t len;
6471 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006472 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006473 if (tmp == NULL) goto failed;
6474 if (!PyList_Check(tmp)) {
6475 PyErr_Format(PyExc_TypeError, "ExceptHandler field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6476 goto failed;
6477 }
6478 len = PyList_GET_SIZE(tmp);
6479 body = asdl_seq_new(len, arena);
6480 if (body == NULL) goto failed;
6481 for (i = 0; i < len; i++) {
6482 stmt_ty value;
6483 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
6484 if (res != 0) goto failed;
6485 asdl_seq_SET(body, i, value);
6486 }
6487 Py_XDECREF(tmp);
6488 tmp = NULL;
6489 } else {
6490 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ExceptHandler");
6491 return 1;
6492 }
6493 *out = ExceptHandler(type, name, body, lineno, col_offset,
6494 arena);
6495 if (*out == NULL) goto failed;
6496 return 0;
6497 }
6498
Benjamin Peterson5b066812010-11-20 01:38:49 +00006499 PyErr_Format(PyExc_TypeError, "expected some sort of excepthandler, but got %R", obj);
6500 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006501 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006502 return 1;
6503}
6504
6505int
6506obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena)
6507{
6508 PyObject* tmp = NULL;
6509 asdl_seq* args;
6510 identifier vararg;
6511 expr_ty varargannotation;
6512 asdl_seq* kwonlyargs;
6513 identifier kwarg;
6514 expr_ty kwargannotation;
6515 asdl_seq* defaults;
6516 asdl_seq* kw_defaults;
6517
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006518 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006519 int res;
6520 Py_ssize_t len;
6521 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006522 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006523 if (tmp == NULL) goto failed;
6524 if (!PyList_Check(tmp)) {
6525 PyErr_Format(PyExc_TypeError, "arguments field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6526 goto failed;
6527 }
6528 len = PyList_GET_SIZE(tmp);
6529 args = asdl_seq_new(len, arena);
6530 if (args == NULL) goto failed;
6531 for (i = 0; i < len; i++) {
6532 arg_ty value;
6533 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6534 if (res != 0) goto failed;
6535 asdl_seq_SET(args, i, value);
6536 }
6537 Py_XDECREF(tmp);
6538 tmp = NULL;
6539 } else {
6540 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from arguments");
6541 return 1;
6542 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006543 if (_PyObject_HasAttrId(obj, &PyId_vararg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006544 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006545 tmp = _PyObject_GetAttrId(obj, &PyId_vararg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006546 if (tmp == NULL) goto failed;
6547 res = obj2ast_identifier(tmp, &vararg, arena);
6548 if (res != 0) goto failed;
6549 Py_XDECREF(tmp);
6550 tmp = NULL;
6551 } else {
6552 vararg = NULL;
6553 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006554 if (_PyObject_HasAttrId(obj, &PyId_varargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006555 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006556 tmp = _PyObject_GetAttrId(obj, &PyId_varargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006557 if (tmp == NULL) goto failed;
6558 res = obj2ast_expr(tmp, &varargannotation, arena);
6559 if (res != 0) goto failed;
6560 Py_XDECREF(tmp);
6561 tmp = NULL;
6562 } else {
6563 varargannotation = NULL;
6564 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006565 if (_PyObject_HasAttrId(obj, &PyId_kwonlyargs)) {
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_kwonlyargs);
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 \"kwonlyargs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6573 goto failed;
6574 }
6575 len = PyList_GET_SIZE(tmp);
6576 kwonlyargs = asdl_seq_new(len, arena);
6577 if (kwonlyargs == 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(kwonlyargs, i, value);
6583 }
6584 Py_XDECREF(tmp);
6585 tmp = NULL;
6586 } else {
6587 PyErr_SetString(PyExc_TypeError, "required field \"kwonlyargs\" missing from arguments");
6588 return 1;
6589 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006590 if (_PyObject_HasAttrId(obj, &PyId_kwarg)) {
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_kwarg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006593 if (tmp == NULL) goto failed;
6594 res = obj2ast_identifier(tmp, &kwarg, arena);
6595 if (res != 0) goto failed;
6596 Py_XDECREF(tmp);
6597 tmp = NULL;
6598 } else {
6599 kwarg = NULL;
6600 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006601 if (_PyObject_HasAttrId(obj, &PyId_kwargannotation)) {
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_kwargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006604 if (tmp == NULL) goto failed;
6605 res = obj2ast_expr(tmp, &kwargannotation, arena);
6606 if (res != 0) goto failed;
6607 Py_XDECREF(tmp);
6608 tmp = NULL;
6609 } else {
6610 kwargannotation = NULL;
6611 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006612 if (_PyObject_HasAttrId(obj, &PyId_defaults)) {
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_defaults);
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 \"defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6620 goto failed;
6621 }
6622 len = PyList_GET_SIZE(tmp);
6623 defaults = asdl_seq_new(len, arena);
6624 if (defaults == NULL) goto failed;
6625 for (i = 0; i < len; i++) {
6626 expr_ty value;
6627 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6628 if (res != 0) goto failed;
6629 asdl_seq_SET(defaults, i, value);
6630 }
6631 Py_XDECREF(tmp);
6632 tmp = NULL;
6633 } else {
6634 PyErr_SetString(PyExc_TypeError, "required field \"defaults\" missing from arguments");
6635 return 1;
6636 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006637 if (_PyObject_HasAttrId(obj, &PyId_kw_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006638 int res;
6639 Py_ssize_t len;
6640 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006641 tmp = _PyObject_GetAttrId(obj, &PyId_kw_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006642 if (tmp == NULL) goto failed;
6643 if (!PyList_Check(tmp)) {
6644 PyErr_Format(PyExc_TypeError, "arguments field \"kw_defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6645 goto failed;
6646 }
6647 len = PyList_GET_SIZE(tmp);
6648 kw_defaults = asdl_seq_new(len, arena);
6649 if (kw_defaults == NULL) goto failed;
6650 for (i = 0; i < len; i++) {
6651 expr_ty value;
6652 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6653 if (res != 0) goto failed;
6654 asdl_seq_SET(kw_defaults, i, value);
6655 }
6656 Py_XDECREF(tmp);
6657 tmp = NULL;
6658 } else {
6659 PyErr_SetString(PyExc_TypeError, "required field \"kw_defaults\" missing from arguments");
6660 return 1;
6661 }
6662 *out = arguments(args, vararg, varargannotation, kwonlyargs, kwarg,
6663 kwargannotation, defaults, kw_defaults, arena);
6664 return 0;
6665failed:
6666 Py_XDECREF(tmp);
6667 return 1;
6668}
6669
6670int
6671obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena)
6672{
6673 PyObject* tmp = NULL;
6674 identifier arg;
6675 expr_ty annotation;
6676
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006677 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006678 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006679 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006680 if (tmp == NULL) goto failed;
6681 res = obj2ast_identifier(tmp, &arg, arena);
6682 if (res != 0) goto failed;
6683 Py_XDECREF(tmp);
6684 tmp = NULL;
6685 } else {
6686 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from arg");
6687 return 1;
6688 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006689 if (_PyObject_HasAttrId(obj, &PyId_annotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006690 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006691 tmp = _PyObject_GetAttrId(obj, &PyId_annotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006692 if (tmp == NULL) goto failed;
6693 res = obj2ast_expr(tmp, &annotation, arena);
6694 if (res != 0) goto failed;
6695 Py_XDECREF(tmp);
6696 tmp = NULL;
6697 } else {
6698 annotation = NULL;
6699 }
6700 *out = arg(arg, annotation, arena);
6701 return 0;
6702failed:
6703 Py_XDECREF(tmp);
6704 return 1;
6705}
6706
6707int
6708obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena)
6709{
6710 PyObject* tmp = NULL;
6711 identifier arg;
6712 expr_ty value;
6713
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006714 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006715 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006716 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006717 if (tmp == NULL) goto failed;
6718 res = obj2ast_identifier(tmp, &arg, arena);
6719 if (res != 0) goto failed;
6720 Py_XDECREF(tmp);
6721 tmp = NULL;
6722 } else {
6723 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from keyword");
6724 return 1;
6725 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006726 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006727 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006728 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006729 if (tmp == NULL) goto failed;
6730 res = obj2ast_expr(tmp, &value, arena);
6731 if (res != 0) goto failed;
6732 Py_XDECREF(tmp);
6733 tmp = NULL;
6734 } else {
6735 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from keyword");
6736 return 1;
6737 }
6738 *out = keyword(arg, value, arena);
6739 return 0;
6740failed:
6741 Py_XDECREF(tmp);
6742 return 1;
6743}
6744
6745int
6746obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena)
6747{
6748 PyObject* tmp = NULL;
6749 identifier name;
6750 identifier asname;
6751
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006752 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006753 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006754 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006755 if (tmp == NULL) goto failed;
6756 res = obj2ast_identifier(tmp, &name, arena);
6757 if (res != 0) goto failed;
6758 Py_XDECREF(tmp);
6759 tmp = NULL;
6760 } else {
6761 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from alias");
6762 return 1;
6763 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006764 if (_PyObject_HasAttrId(obj, &PyId_asname)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006765 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006766 tmp = _PyObject_GetAttrId(obj, &PyId_asname);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006767 if (tmp == NULL) goto failed;
6768 res = obj2ast_identifier(tmp, &asname, arena);
6769 if (res != 0) goto failed;
6770 Py_XDECREF(tmp);
6771 tmp = NULL;
6772 } else {
6773 asname = NULL;
6774 }
6775 *out = alias(name, asname, arena);
6776 return 0;
6777failed:
6778 Py_XDECREF(tmp);
6779 return 1;
6780}
6781
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006782int
6783obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena)
6784{
6785 PyObject* tmp = NULL;
6786 expr_ty context_expr;
6787 expr_ty optional_vars;
6788
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006789 if (_PyObject_HasAttrId(obj, &PyId_context_expr)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006790 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006791 tmp = _PyObject_GetAttrId(obj, &PyId_context_expr);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006792 if (tmp == NULL) goto failed;
6793 res = obj2ast_expr(tmp, &context_expr, arena);
6794 if (res != 0) goto failed;
6795 Py_XDECREF(tmp);
6796 tmp = NULL;
6797 } else {
6798 PyErr_SetString(PyExc_TypeError, "required field \"context_expr\" missing from withitem");
6799 return 1;
6800 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006801 if (_PyObject_HasAttrId(obj, &PyId_optional_vars)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006802 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006803 tmp = _PyObject_GetAttrId(obj, &PyId_optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006804 if (tmp == NULL) goto failed;
6805 res = obj2ast_expr(tmp, &optional_vars, arena);
6806 if (res != 0) goto failed;
6807 Py_XDECREF(tmp);
6808 tmp = NULL;
6809 } else {
6810 optional_vars = NULL;
6811 }
6812 *out = withitem(context_expr, optional_vars, arena);
6813 return 0;
6814failed:
6815 Py_XDECREF(tmp);
6816 return 1;
6817}
6818
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006819
Martin v. Löwis1a214512008-06-11 05:26:20 +00006820static struct PyModuleDef _astmodule = {
6821 PyModuleDef_HEAD_INIT, "_ast"
6822};
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006823PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00006824PyInit__ast(void)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006825{
6826 PyObject *m, *d;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006827 if (!init_types()) return NULL;
6828 m = PyModule_Create(&_astmodule);
6829 if (!m) return NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006830 d = PyModule_GetDict(m);
Martin v. Löwis1a214512008-06-11 05:26:20 +00006831 if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return
6832 NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006833 if (PyModule_AddIntConstant(m, "PyCF_ONLY_AST", PyCF_ONLY_AST) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006834 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006835 if (PyDict_SetItemString(d, "mod", (PyObject*)mod_type) < 0) return
6836 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006837 if (PyDict_SetItemString(d, "Module", (PyObject*)Module_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006838 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006839 if (PyDict_SetItemString(d, "Interactive", (PyObject*)Interactive_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006840 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006841 if (PyDict_SetItemString(d, "Expression", (PyObject*)Expression_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006842 0) return NULL;
6843 if (PyDict_SetItemString(d, "Suite", (PyObject*)Suite_type) < 0) return
6844 NULL;
6845 if (PyDict_SetItemString(d, "stmt", (PyObject*)stmt_type) < 0) return
6846 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006847 if (PyDict_SetItemString(d, "FunctionDef", (PyObject*)FunctionDef_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006848 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006849 if (PyDict_SetItemString(d, "ClassDef", (PyObject*)ClassDef_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006850 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006851 if (PyDict_SetItemString(d, "Return", (PyObject*)Return_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006852 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006853 if (PyDict_SetItemString(d, "Delete", (PyObject*)Delete_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006854 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006855 if (PyDict_SetItemString(d, "Assign", (PyObject*)Assign_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006856 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006857 if (PyDict_SetItemString(d, "AugAssign", (PyObject*)AugAssign_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006858 0) return NULL;
6859 if (PyDict_SetItemString(d, "For", (PyObject*)For_type) < 0) return
6860 NULL;
6861 if (PyDict_SetItemString(d, "While", (PyObject*)While_type) < 0) return
6862 NULL;
6863 if (PyDict_SetItemString(d, "If", (PyObject*)If_type) < 0) return NULL;
6864 if (PyDict_SetItemString(d, "With", (PyObject*)With_type) < 0) return
6865 NULL;
6866 if (PyDict_SetItemString(d, "Raise", (PyObject*)Raise_type) < 0) return
6867 NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05006868 if (PyDict_SetItemString(d, "Try", (PyObject*)Try_type) < 0) return
6869 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006870 if (PyDict_SetItemString(d, "Assert", (PyObject*)Assert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006871 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006872 if (PyDict_SetItemString(d, "Import", (PyObject*)Import_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006873 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006874 if (PyDict_SetItemString(d, "ImportFrom", (PyObject*)ImportFrom_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006875 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006876 if (PyDict_SetItemString(d, "Global", (PyObject*)Global_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006877 return NULL;
Jeremy Hylton81e95022007-02-27 06:50:52 +00006878 if (PyDict_SetItemString(d, "Nonlocal", (PyObject*)Nonlocal_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006879 return NULL;
6880 if (PyDict_SetItemString(d, "Expr", (PyObject*)Expr_type) < 0) return
6881 NULL;
6882 if (PyDict_SetItemString(d, "Pass", (PyObject*)Pass_type) < 0) return
6883 NULL;
6884 if (PyDict_SetItemString(d, "Break", (PyObject*)Break_type) < 0) return
6885 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006886 if (PyDict_SetItemString(d, "Continue", (PyObject*)Continue_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006887 return NULL;
6888 if (PyDict_SetItemString(d, "expr", (PyObject*)expr_type) < 0) return
6889 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006890 if (PyDict_SetItemString(d, "BoolOp", (PyObject*)BoolOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006891 return NULL;
6892 if (PyDict_SetItemString(d, "BinOp", (PyObject*)BinOp_type) < 0) return
6893 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006894 if (PyDict_SetItemString(d, "UnaryOp", (PyObject*)UnaryOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006895 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006896 if (PyDict_SetItemString(d, "Lambda", (PyObject*)Lambda_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006897 return NULL;
6898 if (PyDict_SetItemString(d, "IfExp", (PyObject*)IfExp_type) < 0) return
6899 NULL;
6900 if (PyDict_SetItemString(d, "Dict", (PyObject*)Dict_type) < 0) return
6901 NULL;
6902 if (PyDict_SetItemString(d, "Set", (PyObject*)Set_type) < 0) return
6903 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006904 if (PyDict_SetItemString(d, "ListComp", (PyObject*)ListComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006905 return NULL;
Nick Coghlan650f0d02007-04-15 12:05:43 +00006906 if (PyDict_SetItemString(d, "SetComp", (PyObject*)SetComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006907 return NULL;
Guido van Rossum992d4a32007-07-11 13:09:30 +00006908 if (PyDict_SetItemString(d, "DictComp", (PyObject*)DictComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006909 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006910 if (PyDict_SetItemString(d, "GeneratorExp",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006911 (PyObject*)GeneratorExp_type) < 0) return NULL;
6912 if (PyDict_SetItemString(d, "Yield", (PyObject*)Yield_type) < 0) return
6913 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006914 if (PyDict_SetItemString(d, "Compare", (PyObject*)Compare_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006915 return NULL;
6916 if (PyDict_SetItemString(d, "Call", (PyObject*)Call_type) < 0) return
6917 NULL;
6918 if (PyDict_SetItemString(d, "Num", (PyObject*)Num_type) < 0) return
6919 NULL;
6920 if (PyDict_SetItemString(d, "Str", (PyObject*)Str_type) < 0) return
6921 NULL;
6922 if (PyDict_SetItemString(d, "Bytes", (PyObject*)Bytes_type) < 0) return
6923 NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00006924 if (PyDict_SetItemString(d, "Ellipsis", (PyObject*)Ellipsis_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006925 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006926 if (PyDict_SetItemString(d, "Attribute", (PyObject*)Attribute_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006927 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006928 if (PyDict_SetItemString(d, "Subscript", (PyObject*)Subscript_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006929 0) return NULL;
Guido van Rossum0368b722007-05-11 16:50:42 +00006930 if (PyDict_SetItemString(d, "Starred", (PyObject*)Starred_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006931 return NULL;
6932 if (PyDict_SetItemString(d, "Name", (PyObject*)Name_type) < 0) return
6933 NULL;
6934 if (PyDict_SetItemString(d, "List", (PyObject*)List_type) < 0) return
6935 NULL;
6936 if (PyDict_SetItemString(d, "Tuple", (PyObject*)Tuple_type) < 0) return
6937 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006938 if (PyDict_SetItemString(d, "expr_context",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006939 (PyObject*)expr_context_type) < 0) return NULL;
6940 if (PyDict_SetItemString(d, "Load", (PyObject*)Load_type) < 0) return
6941 NULL;
6942 if (PyDict_SetItemString(d, "Store", (PyObject*)Store_type) < 0) return
6943 NULL;
6944 if (PyDict_SetItemString(d, "Del", (PyObject*)Del_type) < 0) return
6945 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006946 if (PyDict_SetItemString(d, "AugLoad", (PyObject*)AugLoad_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006947 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006948 if (PyDict_SetItemString(d, "AugStore", (PyObject*)AugStore_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006949 return NULL;
6950 if (PyDict_SetItemString(d, "Param", (PyObject*)Param_type) < 0) return
6951 NULL;
6952 if (PyDict_SetItemString(d, "slice", (PyObject*)slice_type) < 0) return
6953 NULL;
6954 if (PyDict_SetItemString(d, "Slice", (PyObject*)Slice_type) < 0) return
6955 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006956 if (PyDict_SetItemString(d, "ExtSlice", (PyObject*)ExtSlice_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006957 return NULL;
6958 if (PyDict_SetItemString(d, "Index", (PyObject*)Index_type) < 0) return
6959 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006960 if (PyDict_SetItemString(d, "boolop", (PyObject*)boolop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006961 return NULL;
6962 if (PyDict_SetItemString(d, "And", (PyObject*)And_type) < 0) return
6963 NULL;
6964 if (PyDict_SetItemString(d, "Or", (PyObject*)Or_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006965 if (PyDict_SetItemString(d, "operator", (PyObject*)operator_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006966 return NULL;
6967 if (PyDict_SetItemString(d, "Add", (PyObject*)Add_type) < 0) return
6968 NULL;
6969 if (PyDict_SetItemString(d, "Sub", (PyObject*)Sub_type) < 0) return
6970 NULL;
6971 if (PyDict_SetItemString(d, "Mult", (PyObject*)Mult_type) < 0) return
6972 NULL;
6973 if (PyDict_SetItemString(d, "Div", (PyObject*)Div_type) < 0) return
6974 NULL;
6975 if (PyDict_SetItemString(d, "Mod", (PyObject*)Mod_type) < 0) return
6976 NULL;
6977 if (PyDict_SetItemString(d, "Pow", (PyObject*)Pow_type) < 0) return
6978 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006979 if (PyDict_SetItemString(d, "LShift", (PyObject*)LShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006980 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006981 if (PyDict_SetItemString(d, "RShift", (PyObject*)RShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006982 return NULL;
6983 if (PyDict_SetItemString(d, "BitOr", (PyObject*)BitOr_type) < 0) return
6984 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006985 if (PyDict_SetItemString(d, "BitXor", (PyObject*)BitXor_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006986 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006987 if (PyDict_SetItemString(d, "BitAnd", (PyObject*)BitAnd_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006988 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006989 if (PyDict_SetItemString(d, "FloorDiv", (PyObject*)FloorDiv_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006990 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006991 if (PyDict_SetItemString(d, "unaryop", (PyObject*)unaryop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006992 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006993 if (PyDict_SetItemString(d, "Invert", (PyObject*)Invert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006994 return NULL;
6995 if (PyDict_SetItemString(d, "Not", (PyObject*)Not_type) < 0) return
6996 NULL;
6997 if (PyDict_SetItemString(d, "UAdd", (PyObject*)UAdd_type) < 0) return
6998 NULL;
6999 if (PyDict_SetItemString(d, "USub", (PyObject*)USub_type) < 0) return
7000 NULL;
7001 if (PyDict_SetItemString(d, "cmpop", (PyObject*)cmpop_type) < 0) return
7002 NULL;
7003 if (PyDict_SetItemString(d, "Eq", (PyObject*)Eq_type) < 0) return NULL;
7004 if (PyDict_SetItemString(d, "NotEq", (PyObject*)NotEq_type) < 0) return
7005 NULL;
7006 if (PyDict_SetItemString(d, "Lt", (PyObject*)Lt_type) < 0) return NULL;
7007 if (PyDict_SetItemString(d, "LtE", (PyObject*)LtE_type) < 0) return
7008 NULL;
7009 if (PyDict_SetItemString(d, "Gt", (PyObject*)Gt_type) < 0) return NULL;
7010 if (PyDict_SetItemString(d, "GtE", (PyObject*)GtE_type) < 0) return
7011 NULL;
7012 if (PyDict_SetItemString(d, "Is", (PyObject*)Is_type) < 0) return NULL;
7013 if (PyDict_SetItemString(d, "IsNot", (PyObject*)IsNot_type) < 0) return
7014 NULL;
7015 if (PyDict_SetItemString(d, "In", (PyObject*)In_type) < 0) return NULL;
7016 if (PyDict_SetItemString(d, "NotIn", (PyObject*)NotIn_type) < 0) return
7017 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007018 if (PyDict_SetItemString(d, "comprehension",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007019 (PyObject*)comprehension_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007020 if (PyDict_SetItemString(d, "excepthandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007021 (PyObject*)excepthandler_type) < 0) return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00007022 if (PyDict_SetItemString(d, "ExceptHandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007023 (PyObject*)ExceptHandler_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007024 if (PyDict_SetItemString(d, "arguments", (PyObject*)arguments_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007025 0) return NULL;
7026 if (PyDict_SetItemString(d, "arg", (PyObject*)arg_type) < 0) return
7027 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007028 if (PyDict_SetItemString(d, "keyword", (PyObject*)keyword_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007029 return NULL;
7030 if (PyDict_SetItemString(d, "alias", (PyObject*)alias_type) < 0) return
7031 NULL;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05007032 if (PyDict_SetItemString(d, "withitem", (PyObject*)withitem_type) < 0)
7033 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00007034 return m;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00007035}
7036
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007037
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007038PyObject* PyAST_mod2obj(mod_ty t)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007039{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007040 init_types();
7041 return ast2obj_mod(t);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007042}
Martin v. Löwis5b222132007-06-10 09:51:05 +00007043
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007044/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
7045mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007046{
7047 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007048 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
7049 (PyObject*)Interactive_type};
7050 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007051 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007052 assert(0 <= mode && mode <= 2);
7053
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007054 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007055
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007056 isinstance = PyObject_IsInstance(ast, req_type[mode]);
7057 if (isinstance == -1)
7058 return NULL;
7059 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007060 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
7061 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007062 return NULL;
7063 }
7064 if (obj2ast_mod(ast, &res, arena) != 0)
7065 return NULL;
7066 else
7067 return res;
7068}
7069
7070int PyAST_Check(PyObject* obj)
7071{
7072 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00007073 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007074}
7075
Martin v. Löwis5b222132007-06-10 09:51:05 +00007076