blob: 805f2b8af118d93f44e1f1cabd3bdec0960a1413 [file] [log] [blame]
Thomas Wouterscf297e42007-02-23 15:07:44 +00001/* File automatically generated by Parser/asdl_c.py. */
2
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -07003#include <stddef.h>
4
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00005#include "Python.h"
6#include "Python-ast.h"
7
Neal Norwitz207c9f32008-03-31 04:42:11 +00008static PyTypeObject AST_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +00009static PyTypeObject *mod_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +000010static PyObject* ast2obj_mod(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +000011static PyTypeObject *Module_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020012_Py_IDENTIFIER(body);
Neal Norwitz53d960c2006-02-28 22:47:29 +000013static char *Module_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000014 "body",
15};
Neal Norwitz53d960c2006-02-28 22:47:29 +000016static PyTypeObject *Interactive_type;
17static char *Interactive_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000018 "body",
19};
Neal Norwitz53d960c2006-02-28 22:47:29 +000020static PyTypeObject *Expression_type;
21static char *Expression_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000022 "body",
23};
Neal Norwitz53d960c2006-02-28 22:47:29 +000024static PyTypeObject *Suite_type;
25static char *Suite_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000026 "body",
27};
Neal Norwitz53d960c2006-02-28 22:47:29 +000028static PyTypeObject *stmt_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020029_Py_IDENTIFIER(lineno);
30_Py_IDENTIFIER(col_offset);
Neal Norwitz53d960c2006-02-28 22:47:29 +000031static char *stmt_attributes[] = {
Martin v. Löwis577b5b92006-02-27 15:23:19 +000032 "lineno",
Martin v. Löwis49c5da12006-03-01 22:49:05 +000033 "col_offset",
Martin v. Löwis577b5b92006-02-27 15:23:19 +000034};
Martin v. Löwisbd260da2006-02-26 19:42:26 +000035static PyObject* ast2obj_stmt(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +000036static PyTypeObject *FunctionDef_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020037_Py_IDENTIFIER(name);
38_Py_IDENTIFIER(args);
39_Py_IDENTIFIER(decorator_list);
40_Py_IDENTIFIER(returns);
Neal Norwitz53d960c2006-02-28 22:47:29 +000041static char *FunctionDef_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000042 "name",
43 "args",
44 "body",
Guido van Rossumd59da4b2007-05-22 18:11:13 +000045 "decorator_list",
Neal Norwitzc1505362006-12-28 06:47:50 +000046 "returns",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000047};
Neal Norwitz53d960c2006-02-28 22:47:29 +000048static PyTypeObject *ClassDef_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020049_Py_IDENTIFIER(bases);
50_Py_IDENTIFIER(keywords);
51_Py_IDENTIFIER(starargs);
52_Py_IDENTIFIER(kwargs);
Neal Norwitz53d960c2006-02-28 22:47:29 +000053static char *ClassDef_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000054 "name",
55 "bases",
Guido van Rossum52cc1d82007-03-18 15:41:51 +000056 "keywords",
57 "starargs",
58 "kwargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000059 "body",
Guido van Rossumd59da4b2007-05-22 18:11:13 +000060 "decorator_list",
Martin v. Löwisbd260da2006-02-26 19:42:26 +000061};
Neal Norwitz53d960c2006-02-28 22:47:29 +000062static PyTypeObject *Return_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020063_Py_IDENTIFIER(value);
Neal Norwitz53d960c2006-02-28 22:47:29 +000064static char *Return_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000065 "value",
66};
Neal Norwitz53d960c2006-02-28 22:47:29 +000067static PyTypeObject *Delete_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020068_Py_IDENTIFIER(targets);
Neal Norwitz53d960c2006-02-28 22:47:29 +000069static char *Delete_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000070 "targets",
71};
Neal Norwitz53d960c2006-02-28 22:47:29 +000072static PyTypeObject *Assign_type;
73static char *Assign_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000074 "targets",
75 "value",
76};
Neal Norwitz53d960c2006-02-28 22:47:29 +000077static PyTypeObject *AugAssign_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020078_Py_IDENTIFIER(target);
79_Py_IDENTIFIER(op);
Neal Norwitz53d960c2006-02-28 22:47:29 +000080static char *AugAssign_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000081 "target",
82 "op",
83 "value",
84};
Neal Norwitz53d960c2006-02-28 22:47:29 +000085static PyTypeObject *For_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020086_Py_IDENTIFIER(iter);
87_Py_IDENTIFIER(orelse);
Neal Norwitz53d960c2006-02-28 22:47:29 +000088static char *For_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000089 "target",
90 "iter",
91 "body",
92 "orelse",
93};
Neal Norwitz53d960c2006-02-28 22:47:29 +000094static PyTypeObject *While_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +020095_Py_IDENTIFIER(test);
Neal Norwitz53d960c2006-02-28 22:47:29 +000096static char *While_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +000097 "test",
98 "body",
99 "orelse",
100};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000101static PyTypeObject *If_type;
102static char *If_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000103 "test",
104 "body",
105 "orelse",
106};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000107static PyTypeObject *With_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200108_Py_IDENTIFIER(items);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000109static char *With_fields[]={
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500110 "items",
Guido van Rossumc2e20742006-02-27 22:32:47 +0000111 "body",
112};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000113static PyTypeObject *Raise_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200114_Py_IDENTIFIER(exc);
115_Py_IDENTIFIER(cause);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000116static char *Raise_fields[]={
Collin Winter828f04a2007-08-31 00:04:24 +0000117 "exc",
118 "cause",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000119};
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500120static PyTypeObject *Try_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200121_Py_IDENTIFIER(handlers);
122_Py_IDENTIFIER(finalbody);
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500123static char *Try_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000124 "body",
125 "handlers",
126 "orelse",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000127 "finalbody",
128};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000129static PyTypeObject *Assert_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200130_Py_IDENTIFIER(msg);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000131static char *Assert_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000132 "test",
133 "msg",
134};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000135static PyTypeObject *Import_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200136_Py_IDENTIFIER(names);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000137static char *Import_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000138 "names",
139};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000140static PyTypeObject *ImportFrom_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200141_Py_IDENTIFIER(module);
142_Py_IDENTIFIER(level);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000143static char *ImportFrom_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000144 "module",
145 "names",
Thomas Woutersf7f438b2006-02-28 16:09:29 +0000146 "level",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000147};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000148static PyTypeObject *Global_type;
149static char *Global_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000150 "names",
151};
Jeremy Hylton81e95022007-02-27 06:50:52 +0000152static PyTypeObject *Nonlocal_type;
153static char *Nonlocal_fields[]={
154 "names",
155};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000156static PyTypeObject *Expr_type;
157static char *Expr_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000158 "value",
159};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000160static PyTypeObject *Pass_type;
161static PyTypeObject *Break_type;
162static PyTypeObject *Continue_type;
163static PyTypeObject *expr_type;
164static char *expr_attributes[] = {
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000165 "lineno",
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000166 "col_offset",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000167};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000168static PyObject* ast2obj_expr(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000169static PyTypeObject *BoolOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200170_Py_IDENTIFIER(values);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000171static char *BoolOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000172 "op",
173 "values",
174};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000175static PyTypeObject *BinOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200176_Py_IDENTIFIER(left);
177_Py_IDENTIFIER(right);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000178static char *BinOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000179 "left",
180 "op",
181 "right",
182};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000183static PyTypeObject *UnaryOp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200184_Py_IDENTIFIER(operand);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000185static char *UnaryOp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000186 "op",
187 "operand",
188};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000189static PyTypeObject *Lambda_type;
190static char *Lambda_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000191 "args",
192 "body",
193};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000194static PyTypeObject *IfExp_type;
195static char *IfExp_fields[]={
Thomas Woutersdca3b9c2006-02-27 00:24:13 +0000196 "test",
197 "body",
198 "orelse",
199};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000200static PyTypeObject *Dict_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200201_Py_IDENTIFIER(keys);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000202static char *Dict_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000203 "keys",
204 "values",
205};
Guido van Rossum86e58e22006-08-28 15:27:34 +0000206static PyTypeObject *Set_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200207_Py_IDENTIFIER(elts);
Guido van Rossum86e58e22006-08-28 15:27:34 +0000208static char *Set_fields[]={
209 "elts",
210};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000211static PyTypeObject *ListComp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200212_Py_IDENTIFIER(elt);
213_Py_IDENTIFIER(generators);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000214static char *ListComp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000215 "elt",
216 "generators",
217};
Nick Coghlan650f0d02007-04-15 12:05:43 +0000218static PyTypeObject *SetComp_type;
219static char *SetComp_fields[]={
220 "elt",
221 "generators",
222};
Guido van Rossum992d4a32007-07-11 13:09:30 +0000223static PyTypeObject *DictComp_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200224_Py_IDENTIFIER(key);
Guido van Rossum992d4a32007-07-11 13:09:30 +0000225static char *DictComp_fields[]={
226 "key",
227 "value",
228 "generators",
229};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000230static PyTypeObject *GeneratorExp_type;
231static char *GeneratorExp_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000232 "elt",
233 "generators",
234};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000235static PyTypeObject *Yield_type;
236static char *Yield_fields[]={
Benjamin Peterson527c6222012-01-14 08:58:23 -0500237 "value",
238};
239static PyTypeObject *YieldFrom_type;
240static char *YieldFrom_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000241 "value",
242};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000243static PyTypeObject *Compare_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200244_Py_IDENTIFIER(ops);
245_Py_IDENTIFIER(comparators);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000246static char *Compare_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000247 "left",
248 "ops",
249 "comparators",
250};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000251static PyTypeObject *Call_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200252_Py_IDENTIFIER(func);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000253static char *Call_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000254 "func",
255 "args",
256 "keywords",
257 "starargs",
258 "kwargs",
259};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000260static PyTypeObject *Num_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200261_Py_IDENTIFIER(n);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000262static char *Num_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000263 "n",
264};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000265static PyTypeObject *Str_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200266_Py_IDENTIFIER(s);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000267static char *Str_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000268 "s",
269};
Thomas Wouters00e41de2007-02-23 19:56:57 +0000270static PyTypeObject *Bytes_type;
271static char *Bytes_fields[]={
272 "s",
273};
Georg Brandl52318d62006-09-06 07:06:08 +0000274static PyTypeObject *Ellipsis_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +0000275static PyTypeObject *Attribute_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200276_Py_IDENTIFIER(attr);
277_Py_IDENTIFIER(ctx);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000278static char *Attribute_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000279 "value",
280 "attr",
281 "ctx",
282};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000283static PyTypeObject *Subscript_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200284_Py_IDENTIFIER(slice);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000285static char *Subscript_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000286 "value",
287 "slice",
288 "ctx",
289};
Guido van Rossum0368b722007-05-11 16:50:42 +0000290static PyTypeObject *Starred_type;
291static char *Starred_fields[]={
292 "value",
293 "ctx",
294};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000295static PyTypeObject *Name_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200296_Py_IDENTIFIER(id);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000297static char *Name_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000298 "id",
299 "ctx",
300};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000301static PyTypeObject *List_type;
302static char *List_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000303 "elts",
304 "ctx",
305};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000306static PyTypeObject *Tuple_type;
307static char *Tuple_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000308 "elts",
309 "ctx",
310};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000311static PyTypeObject *expr_context_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000312static PyObject *Load_singleton, *Store_singleton, *Del_singleton,
313*AugLoad_singleton, *AugStore_singleton, *Param_singleton;
314static PyObject* ast2obj_expr_context(expr_context_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000315static PyTypeObject *Load_type;
316static PyTypeObject *Store_type;
317static PyTypeObject *Del_type;
318static PyTypeObject *AugLoad_type;
319static PyTypeObject *AugStore_type;
320static PyTypeObject *Param_type;
321static PyTypeObject *slice_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000322static PyObject* ast2obj_slice(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000323static PyTypeObject *Slice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200324_Py_IDENTIFIER(lower);
325_Py_IDENTIFIER(upper);
326_Py_IDENTIFIER(step);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000327static char *Slice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000328 "lower",
329 "upper",
330 "step",
331};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000332static PyTypeObject *ExtSlice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200333_Py_IDENTIFIER(dims);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000334static char *ExtSlice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000335 "dims",
336};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000337static PyTypeObject *Index_type;
338static char *Index_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000339 "value",
340};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000341static PyTypeObject *boolop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000342static PyObject *And_singleton, *Or_singleton;
343static PyObject* ast2obj_boolop(boolop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000344static PyTypeObject *And_type;
345static PyTypeObject *Or_type;
346static PyTypeObject *operator_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000347static PyObject *Add_singleton, *Sub_singleton, *Mult_singleton,
348*Div_singleton, *Mod_singleton, *Pow_singleton, *LShift_singleton,
349*RShift_singleton, *BitOr_singleton, *BitXor_singleton, *BitAnd_singleton,
350*FloorDiv_singleton;
351static PyObject* ast2obj_operator(operator_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000352static PyTypeObject *Add_type;
353static PyTypeObject *Sub_type;
354static PyTypeObject *Mult_type;
355static PyTypeObject *Div_type;
356static PyTypeObject *Mod_type;
357static PyTypeObject *Pow_type;
358static PyTypeObject *LShift_type;
359static PyTypeObject *RShift_type;
360static PyTypeObject *BitOr_type;
361static PyTypeObject *BitXor_type;
362static PyTypeObject *BitAnd_type;
363static PyTypeObject *FloorDiv_type;
364static PyTypeObject *unaryop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000365static PyObject *Invert_singleton, *Not_singleton, *UAdd_singleton,
366*USub_singleton;
367static PyObject* ast2obj_unaryop(unaryop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000368static PyTypeObject *Invert_type;
369static PyTypeObject *Not_type;
370static PyTypeObject *UAdd_type;
371static PyTypeObject *USub_type;
372static PyTypeObject *cmpop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000373static PyObject *Eq_singleton, *NotEq_singleton, *Lt_singleton, *LtE_singleton,
374*Gt_singleton, *GtE_singleton, *Is_singleton, *IsNot_singleton, *In_singleton,
375*NotIn_singleton;
376static PyObject* ast2obj_cmpop(cmpop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000377static PyTypeObject *Eq_type;
378static PyTypeObject *NotEq_type;
379static PyTypeObject *Lt_type;
380static PyTypeObject *LtE_type;
381static PyTypeObject *Gt_type;
382static PyTypeObject *GtE_type;
383static PyTypeObject *Is_type;
384static PyTypeObject *IsNot_type;
385static PyTypeObject *In_type;
386static PyTypeObject *NotIn_type;
387static PyTypeObject *comprehension_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000388static PyObject* ast2obj_comprehension(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200389_Py_IDENTIFIER(ifs);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000390static char *comprehension_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000391 "target",
392 "iter",
393 "ifs",
394};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000395static PyTypeObject *excepthandler_type;
Neal Norwitzad74aa82008-03-31 05:14:30 +0000396static char *excepthandler_attributes[] = {
397 "lineno",
398 "col_offset",
399};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000400static PyObject* ast2obj_excepthandler(void*);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000401static PyTypeObject *ExceptHandler_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200402_Py_IDENTIFIER(type);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000403static char *ExceptHandler_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000404 "type",
405 "name",
406 "body",
407};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000408static PyTypeObject *arguments_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000409static PyObject* ast2obj_arguments(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200410_Py_IDENTIFIER(vararg);
411_Py_IDENTIFIER(varargannotation);
412_Py_IDENTIFIER(kwonlyargs);
413_Py_IDENTIFIER(kwarg);
414_Py_IDENTIFIER(kwargannotation);
415_Py_IDENTIFIER(defaults);
416_Py_IDENTIFIER(kw_defaults);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000417static char *arguments_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000418 "args",
419 "vararg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000420 "varargannotation",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000421 "kwonlyargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000422 "kwarg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000423 "kwargannotation",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000424 "defaults",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000425 "kw_defaults",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000426};
Neal Norwitzc1505362006-12-28 06:47:50 +0000427static PyTypeObject *arg_type;
428static PyObject* ast2obj_arg(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200429_Py_IDENTIFIER(arg);
430_Py_IDENTIFIER(annotation);
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000431static char *arg_fields[]={
Neal Norwitzc1505362006-12-28 06:47:50 +0000432 "arg",
433 "annotation",
434};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000435static PyTypeObject *keyword_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000436static PyObject* ast2obj_keyword(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000437static char *keyword_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000438 "arg",
439 "value",
440};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000441static PyTypeObject *alias_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000442static PyObject* ast2obj_alias(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200443_Py_IDENTIFIER(asname);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000444static char *alias_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000445 "name",
446 "asname",
447};
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500448static PyTypeObject *withitem_type;
449static PyObject* ast2obj_withitem(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200450_Py_IDENTIFIER(context_expr);
451_Py_IDENTIFIER(optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500452static char *withitem_fields[]={
453 "context_expr",
454 "optional_vars",
455};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000456
457
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700458typedef struct {
Victor Stinner45e50de2012-03-13 01:17:31 +0100459 PyObject_HEAD
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700460 PyObject *dict;
461} AST_object;
462
Benjamin Peterson1767e022012-03-14 21:50:29 -0500463static void
464ast_dealloc(AST_object *self)
465{
466 Py_CLEAR(self->dict);
Antoine Pitrou50750742012-07-08 12:43:32 +0200467 Py_TYPE(self)->tp_free(self);
Benjamin Peterson1767e022012-03-14 21:50:29 -0500468}
469
Neal Norwitz207c9f32008-03-31 04:42:11 +0000470static int
Benjamin Peterson81071762012-07-08 11:03:46 -0700471ast_traverse(AST_object *self, visitproc visit, void *arg)
472{
473 Py_VISIT(self->dict);
474 return 0;
475}
476
477static void
478ast_clear(AST_object *self)
479{
480 Py_CLEAR(self->dict);
481}
482
483static int
Neal Norwitz207c9f32008-03-31 04:42:11 +0000484ast_type_init(PyObject *self, PyObject *args, PyObject *kw)
485{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200486 _Py_IDENTIFIER(_fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000487 Py_ssize_t i, numfields = 0;
488 int res = -1;
489 PyObject *key, *value, *fields;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200490 fields = _PyObject_GetAttrId((PyObject*)Py_TYPE(self), &PyId__fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000491 if (!fields)
492 PyErr_Clear();
493 if (fields) {
494 numfields = PySequence_Size(fields);
495 if (numfields == -1)
496 goto cleanup;
497 }
498 res = 0; /* if no error occurs, this stays 0 to the end */
499 if (PyTuple_GET_SIZE(args) > 0) {
500 if (numfields != PyTuple_GET_SIZE(args)) {
501 PyErr_Format(PyExc_TypeError, "%.400s constructor takes %s"
Amaury Forgeot d'Arc245c70b2008-09-10 22:24:24 +0000502 "%zd positional argument%s",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000503 Py_TYPE(self)->tp_name,
504 numfields == 0 ? "" : "either 0 or ",
505 numfields, numfields == 1 ? "" : "s");
506 res = -1;
507 goto cleanup;
508 }
509 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
510 /* cannot be reached when fields is NULL */
511 PyObject *name = PySequence_GetItem(fields, i);
512 if (!name) {
513 res = -1;
514 goto cleanup;
515 }
516 res = PyObject_SetAttr(self, name, PyTuple_GET_ITEM(args, i));
517 Py_DECREF(name);
518 if (res < 0)
519 goto cleanup;
520 }
521 }
522 if (kw) {
523 i = 0; /* needed by PyDict_Next */
524 while (PyDict_Next(kw, &i, &key, &value)) {
525 res = PyObject_SetAttr(self, key, value);
526 if (res < 0)
527 goto cleanup;
528 }
529 }
530 cleanup:
531 Py_XDECREF(fields);
532 return res;
533}
534
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000535/* Pickling support */
536static PyObject *
537ast_type_reduce(PyObject *self, PyObject *unused)
538{
539 PyObject *res;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200540 _Py_IDENTIFIER(__dict__);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200541 PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__);
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000542 if (dict == NULL) {
543 if (PyErr_ExceptionMatches(PyExc_AttributeError))
544 PyErr_Clear();
545 else
546 return NULL;
547 }
548 if (dict) {
549 res = Py_BuildValue("O()O", Py_TYPE(self), dict);
550 Py_DECREF(dict);
551 return res;
552 }
553 return Py_BuildValue("O()", Py_TYPE(self));
554}
555
556static PyMethodDef ast_type_methods[] = {
557 {"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
558 {NULL}
559};
560
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700561static PyGetSetDef ast_type_getsets[] = {
562 {"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict},
563 {NULL}
564};
565
Neal Norwitz207c9f32008-03-31 04:42:11 +0000566static PyTypeObject AST_type = {
567 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000568 "_ast.AST",
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700569 sizeof(AST_object),
Neal Norwitz207c9f32008-03-31 04:42:11 +0000570 0,
Benjamin Peterson1767e022012-03-14 21:50:29 -0500571 (destructor)ast_dealloc, /* tp_dealloc */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000572 0, /* tp_print */
573 0, /* tp_getattr */
574 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000575 0, /* tp_reserved */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000576 0, /* tp_repr */
577 0, /* tp_as_number */
578 0, /* tp_as_sequence */
579 0, /* tp_as_mapping */
580 0, /* tp_hash */
581 0, /* tp_call */
582 0, /* tp_str */
583 PyObject_GenericGetAttr, /* tp_getattro */
584 PyObject_GenericSetAttr, /* tp_setattro */
585 0, /* tp_as_buffer */
Benjamin Peterson81071762012-07-08 11:03:46 -0700586 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /* tp_flags */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000587 0, /* tp_doc */
Benjamin Peterson81071762012-07-08 11:03:46 -0700588 (traverseproc)ast_traverse, /* tp_traverse */
589 (inquiry)ast_clear, /* tp_clear */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000590 0, /* tp_richcompare */
591 0, /* tp_weaklistoffset */
592 0, /* tp_iter */
593 0, /* tp_iternext */
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000594 ast_type_methods, /* tp_methods */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000595 0, /* tp_members */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700596 ast_type_getsets, /* tp_getset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000597 0, /* tp_base */
598 0, /* tp_dict */
599 0, /* tp_descr_get */
600 0, /* tp_descr_set */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700601 offsetof(AST_object, dict),/* tp_dictoffset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000602 (initproc)ast_type_init, /* tp_init */
603 PyType_GenericAlloc, /* tp_alloc */
604 PyType_GenericNew, /* tp_new */
Benjamin Peterson81071762012-07-08 11:03:46 -0700605 PyObject_GC_Del, /* tp_free */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000606};
607
608
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000609static PyTypeObject* make_type(char *type, PyTypeObject* base, char**fields, int num_fields)
610{
611 PyObject *fnames, *result;
612 int i;
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000613 fnames = PyTuple_New(num_fields);
614 if (!fnames) return NULL;
615 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000616 PyObject *field = PyUnicode_FromString(fields[i]);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000617 if (!field) {
618 Py_DECREF(fnames);
619 return NULL;
620 }
621 PyTuple_SET_ITEM(fnames, i, field);
622 }
Victor Stinner7eeb5b52010-06-07 19:57:46 +0000623 result = PyObject_CallFunction((PyObject*)&PyType_Type, "s(O){sOss}",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000624 type, base, "_fields", fnames, "__module__", "_ast");
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000625 Py_DECREF(fnames);
626 return (PyTypeObject*)result;
627}
628
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000629static int add_attributes(PyTypeObject* type, char**attrs, int num_fields)
630{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000631 int i, result;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200632 _Py_IDENTIFIER(_attributes);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000633 PyObject *s, *l = PyTuple_New(num_fields);
Benjamin Peterson3e5cd1d2010-06-27 21:45:24 +0000634 if (!l)
635 return 0;
636 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000637 s = PyUnicode_FromString(attrs[i]);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000638 if (!s) {
639 Py_DECREF(l);
640 return 0;
641 }
Neal Norwitz207c9f32008-03-31 04:42:11 +0000642 PyTuple_SET_ITEM(l, i, s);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000643 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200644 result = _PyObject_SetAttrId((PyObject*)type, &PyId__attributes, l) >= 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000645 Py_DECREF(l);
646 return result;
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000647}
648
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000649/* Conversion AST -> Python */
650
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000651static PyObject* ast2obj_list(asdl_seq *seq, PyObject* (*func)(void*))
652{
Benjamin Peterson77fa9372012-05-15 10:10:27 -0700653 Py_ssize_t i, n = asdl_seq_LEN(seq);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000654 PyObject *result = PyList_New(n);
655 PyObject *value;
656 if (!result)
657 return NULL;
658 for (i = 0; i < n; i++) {
659 value = func(asdl_seq_GET(seq, i));
660 if (!value) {
661 Py_DECREF(result);
662 return NULL;
663 }
664 PyList_SET_ITEM(result, i, value);
665 }
666 return result;
667}
668
669static PyObject* ast2obj_object(void *o)
670{
671 if (!o)
672 o = Py_None;
673 Py_INCREF((PyObject*)o);
674 return (PyObject*)o;
675}
676#define ast2obj_identifier ast2obj_object
677#define ast2obj_string ast2obj_object
Benjamin Petersone2498412011-08-09 16:08:39 -0500678#define ast2obj_bytes ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000679
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000680static PyObject* ast2obj_int(long b)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000681{
Christian Heimes217cfd12007-12-02 14:31:20 +0000682 return PyLong_FromLong(b);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000683}
684
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000685/* Conversion Python -> AST */
686
687static int obj2ast_object(PyObject* obj, PyObject** out, PyArena* arena)
688{
689 if (obj == Py_None)
690 obj = NULL;
691 if (obj)
692 PyArena_AddPyObject(arena, obj);
693 Py_XINCREF(obj);
694 *out = obj;
695 return 0;
696}
697
Benjamin Peterson180e6352011-07-22 11:09:07 -0500698static int obj2ast_identifier(PyObject* obj, PyObject** out, PyArena* arena)
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500699{
Benjamin Peterson180e6352011-07-22 11:09:07 -0500700 if (!PyUnicode_CheckExact(obj) && obj != Py_None) {
701 PyErr_SetString(PyExc_TypeError, "AST identifier must be of type str");
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500702 return 1;
703 }
704 return obj2ast_object(obj, out, arena);
705}
706
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500707static int obj2ast_string(PyObject* obj, PyObject** out, PyArena* arena)
708{
Benjamin Peterson0224d4e2011-08-31 22:13:03 -0400709 if (!PyUnicode_CheckExact(obj) && !PyBytes_CheckExact(obj)) {
Benjamin Peterson180e6352011-07-22 11:09:07 -0500710 PyErr_SetString(PyExc_TypeError, "AST string must be of type str");
711 return 1;
712 }
713 return obj2ast_object(obj, out, arena);
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500714}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000715
Benjamin Petersone2498412011-08-09 16:08:39 -0500716static int obj2ast_bytes(PyObject* obj, PyObject** out, PyArena* arena)
717{
718 if (!PyBytes_CheckExact(obj)) {
719 PyErr_SetString(PyExc_TypeError, "AST bytes must be of type bytes");
720 return 1;
721 }
722 return obj2ast_object(obj, out, arena);
723}
724
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000725static int obj2ast_int(PyObject* obj, int* out, PyArena* arena)
726{
727 int i;
728 if (!PyLong_Check(obj)) {
Amaury Forgeot d'Arc5e8f8102011-11-22 21:52:30 +0100729 PyErr_Format(PyExc_ValueError, "invalid integer value: %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000730 return 1;
731 }
732
733 i = (int)PyLong_AsLong(obj);
734 if (i == -1 && PyErr_Occurred())
735 return 1;
736 *out = i;
737 return 0;
738}
739
Benjamin Peterson1a6e0d02008-10-25 15:49:17 +0000740static int add_ast_fields(void)
Benjamin Petersonce825f12008-10-24 23:11:02 +0000741{
742 PyObject *empty_tuple, *d;
743 if (PyType_Ready(&AST_type) < 0)
744 return -1;
745 d = AST_type.tp_dict;
746 empty_tuple = PyTuple_New(0);
747 if (!empty_tuple ||
748 PyDict_SetItemString(d, "_fields", empty_tuple) < 0 ||
749 PyDict_SetItemString(d, "_attributes", empty_tuple) < 0) {
750 Py_XDECREF(empty_tuple);
751 return -1;
752 }
753 Py_DECREF(empty_tuple);
754 return 0;
755}
756
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000757
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000758static int init_types(void)
759{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000760 static int initialized;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000761 if (initialized) return 1;
Benjamin Petersonce825f12008-10-24 23:11:02 +0000762 if (add_ast_fields() < 0) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000763 mod_type = make_type("mod", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000764 if (!mod_type) return 0;
765 if (!add_attributes(mod_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000766 Module_type = make_type("Module", mod_type, Module_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000767 if (!Module_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000768 Interactive_type = make_type("Interactive", mod_type,
769 Interactive_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000770 if (!Interactive_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000771 Expression_type = make_type("Expression", mod_type, Expression_fields,
772 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000773 if (!Expression_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000774 Suite_type = make_type("Suite", mod_type, Suite_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000775 if (!Suite_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000776 stmt_type = make_type("stmt", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000777 if (!stmt_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000778 if (!add_attributes(stmt_type, stmt_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000779 FunctionDef_type = make_type("FunctionDef", stmt_type,
Neal Norwitzc1505362006-12-28 06:47:50 +0000780 FunctionDef_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000781 if (!FunctionDef_type) return 0;
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000782 ClassDef_type = make_type("ClassDef", stmt_type, ClassDef_fields, 7);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000783 if (!ClassDef_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000784 Return_type = make_type("Return", stmt_type, Return_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000785 if (!Return_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000786 Delete_type = make_type("Delete", stmt_type, Delete_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000787 if (!Delete_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000788 Assign_type = make_type("Assign", stmt_type, Assign_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000789 if (!Assign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000790 AugAssign_type = make_type("AugAssign", stmt_type, AugAssign_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000791 if (!AugAssign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000792 For_type = make_type("For", stmt_type, For_fields, 4);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000793 if (!For_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000794 While_type = make_type("While", stmt_type, While_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000795 if (!While_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000796 If_type = make_type("If", stmt_type, If_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000797 if (!If_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500798 With_type = make_type("With", stmt_type, With_fields, 2);
Guido van Rossumc2e20742006-02-27 22:32:47 +0000799 if (!With_type) return 0;
Collin Winter828f04a2007-08-31 00:04:24 +0000800 Raise_type = make_type("Raise", stmt_type, Raise_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000801 if (!Raise_type) return 0;
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500802 Try_type = make_type("Try", stmt_type, Try_fields, 4);
803 if (!Try_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000804 Assert_type = make_type("Assert", stmt_type, Assert_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000805 if (!Assert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000806 Import_type = make_type("Import", stmt_type, Import_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000807 if (!Import_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000808 ImportFrom_type = make_type("ImportFrom", stmt_type, ImportFrom_fields,
Thomas Woutersf7f438b2006-02-28 16:09:29 +0000809 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000810 if (!ImportFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000811 Global_type = make_type("Global", stmt_type, Global_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000812 if (!Global_type) return 0;
Jeremy Hylton81e95022007-02-27 06:50:52 +0000813 Nonlocal_type = make_type("Nonlocal", stmt_type, Nonlocal_fields, 1);
814 if (!Nonlocal_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000815 Expr_type = make_type("Expr", stmt_type, Expr_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000816 if (!Expr_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000817 Pass_type = make_type("Pass", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000818 if (!Pass_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000819 Break_type = make_type("Break", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000820 if (!Break_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000821 Continue_type = make_type("Continue", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000822 if (!Continue_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000823 expr_type = make_type("expr", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000824 if (!expr_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000825 if (!add_attributes(expr_type, expr_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000826 BoolOp_type = make_type("BoolOp", expr_type, BoolOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000827 if (!BoolOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000828 BinOp_type = make_type("BinOp", expr_type, BinOp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000829 if (!BinOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000830 UnaryOp_type = make_type("UnaryOp", expr_type, UnaryOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000831 if (!UnaryOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000832 Lambda_type = make_type("Lambda", expr_type, Lambda_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000833 if (!Lambda_type) return 0;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +0000834 IfExp_type = make_type("IfExp", expr_type, IfExp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000835 if (!IfExp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000836 Dict_type = make_type("Dict", expr_type, Dict_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000837 if (!Dict_type) return 0;
Guido van Rossum86e58e22006-08-28 15:27:34 +0000838 Set_type = make_type("Set", expr_type, Set_fields, 1);
839 if (!Set_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000840 ListComp_type = make_type("ListComp", expr_type, ListComp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000841 if (!ListComp_type) return 0;
Nick Coghlan650f0d02007-04-15 12:05:43 +0000842 SetComp_type = make_type("SetComp", expr_type, SetComp_fields, 2);
843 if (!SetComp_type) return 0;
Guido van Rossum992d4a32007-07-11 13:09:30 +0000844 DictComp_type = make_type("DictComp", expr_type, DictComp_fields, 3);
845 if (!DictComp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000846 GeneratorExp_type = make_type("GeneratorExp", expr_type,
847 GeneratorExp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000848 if (!GeneratorExp_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500849 Yield_type = make_type("Yield", expr_type, Yield_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000850 if (!Yield_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500851 YieldFrom_type = make_type("YieldFrom", expr_type, YieldFrom_fields, 1);
852 if (!YieldFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000853 Compare_type = make_type("Compare", expr_type, Compare_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000854 if (!Compare_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000855 Call_type = make_type("Call", expr_type, Call_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000856 if (!Call_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000857 Num_type = make_type("Num", expr_type, Num_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000858 if (!Num_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000859 Str_type = make_type("Str", expr_type, Str_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000860 if (!Str_type) return 0;
Thomas Wouters00e41de2007-02-23 19:56:57 +0000861 Bytes_type = make_type("Bytes", expr_type, Bytes_fields, 1);
862 if (!Bytes_type) return 0;
Georg Brandl52318d62006-09-06 07:06:08 +0000863 Ellipsis_type = make_type("Ellipsis", expr_type, NULL, 0);
864 if (!Ellipsis_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000865 Attribute_type = make_type("Attribute", expr_type, Attribute_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000866 if (!Attribute_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000867 Subscript_type = make_type("Subscript", expr_type, Subscript_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000868 if (!Subscript_type) return 0;
Guido van Rossum0368b722007-05-11 16:50:42 +0000869 Starred_type = make_type("Starred", expr_type, Starred_fields, 2);
870 if (!Starred_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000871 Name_type = make_type("Name", expr_type, Name_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000872 if (!Name_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000873 List_type = make_type("List", expr_type, List_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000874 if (!List_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000875 Tuple_type = make_type("Tuple", expr_type, Tuple_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000876 if (!Tuple_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000877 expr_context_type = make_type("expr_context", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000878 if (!expr_context_type) return 0;
879 if (!add_attributes(expr_context_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000880 Load_type = make_type("Load", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000881 if (!Load_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000882 Load_singleton = PyType_GenericNew(Load_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000883 if (!Load_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000884 Store_type = make_type("Store", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000885 if (!Store_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000886 Store_singleton = PyType_GenericNew(Store_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000887 if (!Store_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000888 Del_type = make_type("Del", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000889 if (!Del_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000890 Del_singleton = PyType_GenericNew(Del_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000891 if (!Del_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000892 AugLoad_type = make_type("AugLoad", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000893 if (!AugLoad_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000894 AugLoad_singleton = PyType_GenericNew(AugLoad_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000895 if (!AugLoad_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000896 AugStore_type = make_type("AugStore", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000897 if (!AugStore_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000898 AugStore_singleton = PyType_GenericNew(AugStore_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000899 if (!AugStore_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000900 Param_type = make_type("Param", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000901 if (!Param_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000902 Param_singleton = PyType_GenericNew(Param_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000903 if (!Param_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000904 slice_type = make_type("slice", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000905 if (!slice_type) return 0;
906 if (!add_attributes(slice_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000907 Slice_type = make_type("Slice", slice_type, Slice_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000908 if (!Slice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000909 ExtSlice_type = make_type("ExtSlice", slice_type, ExtSlice_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000910 if (!ExtSlice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000911 Index_type = make_type("Index", slice_type, Index_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000912 if (!Index_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000913 boolop_type = make_type("boolop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000914 if (!boolop_type) return 0;
915 if (!add_attributes(boolop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000916 And_type = make_type("And", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000917 if (!And_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000918 And_singleton = PyType_GenericNew(And_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000919 if (!And_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000920 Or_type = make_type("Or", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000921 if (!Or_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000922 Or_singleton = PyType_GenericNew(Or_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000923 if (!Or_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000924 operator_type = make_type("operator", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000925 if (!operator_type) return 0;
926 if (!add_attributes(operator_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000927 Add_type = make_type("Add", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000928 if (!Add_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000929 Add_singleton = PyType_GenericNew(Add_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000930 if (!Add_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000931 Sub_type = make_type("Sub", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000932 if (!Sub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000933 Sub_singleton = PyType_GenericNew(Sub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000934 if (!Sub_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000935 Mult_type = make_type("Mult", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000936 if (!Mult_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000937 Mult_singleton = PyType_GenericNew(Mult_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000938 if (!Mult_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000939 Div_type = make_type("Div", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000940 if (!Div_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000941 Div_singleton = PyType_GenericNew(Div_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000942 if (!Div_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000943 Mod_type = make_type("Mod", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000944 if (!Mod_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000945 Mod_singleton = PyType_GenericNew(Mod_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000946 if (!Mod_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000947 Pow_type = make_type("Pow", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000948 if (!Pow_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000949 Pow_singleton = PyType_GenericNew(Pow_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000950 if (!Pow_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000951 LShift_type = make_type("LShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000952 if (!LShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000953 LShift_singleton = PyType_GenericNew(LShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000954 if (!LShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000955 RShift_type = make_type("RShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000956 if (!RShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000957 RShift_singleton = PyType_GenericNew(RShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000958 if (!RShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000959 BitOr_type = make_type("BitOr", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000960 if (!BitOr_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000961 BitOr_singleton = PyType_GenericNew(BitOr_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000962 if (!BitOr_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000963 BitXor_type = make_type("BitXor", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000964 if (!BitXor_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000965 BitXor_singleton = PyType_GenericNew(BitXor_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000966 if (!BitXor_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000967 BitAnd_type = make_type("BitAnd", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000968 if (!BitAnd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000969 BitAnd_singleton = PyType_GenericNew(BitAnd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000970 if (!BitAnd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000971 FloorDiv_type = make_type("FloorDiv", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000972 if (!FloorDiv_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000973 FloorDiv_singleton = PyType_GenericNew(FloorDiv_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000974 if (!FloorDiv_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000975 unaryop_type = make_type("unaryop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000976 if (!unaryop_type) return 0;
977 if (!add_attributes(unaryop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000978 Invert_type = make_type("Invert", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000979 if (!Invert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000980 Invert_singleton = PyType_GenericNew(Invert_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000981 if (!Invert_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000982 Not_type = make_type("Not", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000983 if (!Not_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000984 Not_singleton = PyType_GenericNew(Not_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000985 if (!Not_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000986 UAdd_type = make_type("UAdd", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000987 if (!UAdd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000988 UAdd_singleton = PyType_GenericNew(UAdd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000989 if (!UAdd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000990 USub_type = make_type("USub", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000991 if (!USub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000992 USub_singleton = PyType_GenericNew(USub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000993 if (!USub_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000994 cmpop_type = make_type("cmpop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000995 if (!cmpop_type) return 0;
996 if (!add_attributes(cmpop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000997 Eq_type = make_type("Eq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000998 if (!Eq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000999 Eq_singleton = PyType_GenericNew(Eq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001000 if (!Eq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001001 NotEq_type = make_type("NotEq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001002 if (!NotEq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001003 NotEq_singleton = PyType_GenericNew(NotEq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001004 if (!NotEq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001005 Lt_type = make_type("Lt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001006 if (!Lt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001007 Lt_singleton = PyType_GenericNew(Lt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001008 if (!Lt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001009 LtE_type = make_type("LtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001010 if (!LtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001011 LtE_singleton = PyType_GenericNew(LtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001012 if (!LtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001013 Gt_type = make_type("Gt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001014 if (!Gt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001015 Gt_singleton = PyType_GenericNew(Gt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001016 if (!Gt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001017 GtE_type = make_type("GtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001018 if (!GtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001019 GtE_singleton = PyType_GenericNew(GtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001020 if (!GtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001021 Is_type = make_type("Is", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001022 if (!Is_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001023 Is_singleton = PyType_GenericNew(Is_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001024 if (!Is_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001025 IsNot_type = make_type("IsNot", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001026 if (!IsNot_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001027 IsNot_singleton = PyType_GenericNew(IsNot_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001028 if (!IsNot_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001029 In_type = make_type("In", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001030 if (!In_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001031 In_singleton = PyType_GenericNew(In_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001032 if (!In_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001033 NotIn_type = make_type("NotIn", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001034 if (!NotIn_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001035 NotIn_singleton = PyType_GenericNew(NotIn_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001036 if (!NotIn_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001037 comprehension_type = make_type("comprehension", &AST_type,
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001038 comprehension_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001039 if (!comprehension_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001040 excepthandler_type = make_type("excepthandler", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001041 if (!excepthandler_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001042 if (!add_attributes(excepthandler_type, excepthandler_attributes, 2))
1043 return 0;
1044 ExceptHandler_type = make_type("ExceptHandler", excepthandler_type,
1045 ExceptHandler_fields, 3);
1046 if (!ExceptHandler_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001047 arguments_type = make_type("arguments", &AST_type, arguments_fields, 8);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001048 if (!arguments_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001049 arg_type = make_type("arg", &AST_type, arg_fields, 2);
Neal Norwitzc1505362006-12-28 06:47:50 +00001050 if (!arg_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001051 keyword_type = make_type("keyword", &AST_type, keyword_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001052 if (!keyword_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001053 alias_type = make_type("alias", &AST_type, alias_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001054 if (!alias_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001055 withitem_type = make_type("withitem", &AST_type, withitem_fields, 2);
1056 if (!withitem_type) return 0;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001057 initialized = 1;
1058 return 1;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001059}
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001060
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001061static int obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena);
1062static int obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena);
1063static int obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena);
1064static int obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena*
1065 arena);
1066static int obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena);
1067static int obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena);
1068static int obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena);
1069static int obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena);
1070static int obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena);
1071static int obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena*
1072 arena);
1073static int obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena*
1074 arena);
1075static int obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena);
1076static int obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena);
1077static int obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena);
1078static int obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001079static int obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001080
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001081mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001082Module(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001083{
1084 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001085 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001086 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001087 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001088 p->kind = Module_kind;
1089 p->v.Module.body = body;
1090 return p;
1091}
1092
1093mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001094Interactive(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001095{
1096 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001097 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001098 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001099 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001100 p->kind = Interactive_kind;
1101 p->v.Interactive.body = body;
1102 return p;
1103}
1104
1105mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001106Expression(expr_ty body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001107{
1108 mod_ty p;
1109 if (!body) {
1110 PyErr_SetString(PyExc_ValueError,
1111 "field body is required for Expression");
1112 return NULL;
1113 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001114 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001115 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001116 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001117 p->kind = Expression_kind;
1118 p->v.Expression.body = body;
1119 return p;
1120}
1121
1122mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001123Suite(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001124{
1125 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001126 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001127 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001128 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001129 p->kind = Suite_kind;
1130 p->v.Suite.body = body;
1131 return p;
1132}
1133
1134stmt_ty
1135FunctionDef(identifier name, arguments_ty args, asdl_seq * body, asdl_seq *
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001136 decorator_list, expr_ty returns, int lineno, int col_offset,
1137 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001138{
1139 stmt_ty p;
1140 if (!name) {
1141 PyErr_SetString(PyExc_ValueError,
1142 "field name is required for FunctionDef");
1143 return NULL;
1144 }
1145 if (!args) {
1146 PyErr_SetString(PyExc_ValueError,
1147 "field args is required for FunctionDef");
1148 return NULL;
1149 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001150 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001151 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001152 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001153 p->kind = FunctionDef_kind;
1154 p->v.FunctionDef.name = name;
1155 p->v.FunctionDef.args = args;
1156 p->v.FunctionDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001157 p->v.FunctionDef.decorator_list = decorator_list;
Neal Norwitzc1505362006-12-28 06:47:50 +00001158 p->v.FunctionDef.returns = returns;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001159 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001160 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001161 return p;
1162}
1163
1164stmt_ty
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001165ClassDef(identifier name, asdl_seq * bases, asdl_seq * keywords, expr_ty
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001166 starargs, expr_ty kwargs, asdl_seq * body, asdl_seq * decorator_list,
1167 int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001168{
1169 stmt_ty p;
1170 if (!name) {
1171 PyErr_SetString(PyExc_ValueError,
1172 "field name is required for ClassDef");
1173 return NULL;
1174 }
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 = ClassDef_kind;
1179 p->v.ClassDef.name = name;
1180 p->v.ClassDef.bases = bases;
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001181 p->v.ClassDef.keywords = keywords;
1182 p->v.ClassDef.starargs = starargs;
1183 p->v.ClassDef.kwargs = kwargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001184 p->v.ClassDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001185 p->v.ClassDef.decorator_list = decorator_list;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001186 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001187 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001188 return p;
1189}
1190
1191stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001192Return(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001193{
1194 stmt_ty p;
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 = Return_kind;
1199 p->v.Return.value = value;
1200 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001201 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001202 return p;
1203}
1204
1205stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001206Delete(asdl_seq * targets, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001207{
1208 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001209 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001210 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001211 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001212 p->kind = Delete_kind;
1213 p->v.Delete.targets = targets;
1214 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001215 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001216 return p;
1217}
1218
1219stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001220Assign(asdl_seq * targets, expr_ty value, int lineno, int col_offset, PyArena
1221 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001222{
1223 stmt_ty p;
1224 if (!value) {
1225 PyErr_SetString(PyExc_ValueError,
1226 "field value is required for Assign");
1227 return NULL;
1228 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001229 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001230 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001231 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001232 p->kind = Assign_kind;
1233 p->v.Assign.targets = targets;
1234 p->v.Assign.value = value;
1235 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001236 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001237 return p;
1238}
1239
1240stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001241AugAssign(expr_ty target, operator_ty op, expr_ty value, int lineno, int
1242 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001243{
1244 stmt_ty p;
1245 if (!target) {
1246 PyErr_SetString(PyExc_ValueError,
1247 "field target is required for AugAssign");
1248 return NULL;
1249 }
1250 if (!op) {
1251 PyErr_SetString(PyExc_ValueError,
1252 "field op is required for AugAssign");
1253 return NULL;
1254 }
1255 if (!value) {
1256 PyErr_SetString(PyExc_ValueError,
1257 "field value is required for AugAssign");
1258 return NULL;
1259 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001260 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001261 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001262 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001263 p->kind = AugAssign_kind;
1264 p->v.AugAssign.target = target;
1265 p->v.AugAssign.op = op;
1266 p->v.AugAssign.value = value;
1267 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001268 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001269 return p;
1270}
1271
1272stmt_ty
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001273For(expr_ty target, expr_ty iter, asdl_seq * body, asdl_seq * orelse, int
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001274 lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001275{
1276 stmt_ty p;
1277 if (!target) {
1278 PyErr_SetString(PyExc_ValueError,
1279 "field target is required for For");
1280 return NULL;
1281 }
1282 if (!iter) {
1283 PyErr_SetString(PyExc_ValueError,
1284 "field iter is required for For");
1285 return NULL;
1286 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001287 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001288 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001289 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001290 p->kind = For_kind;
1291 p->v.For.target = target;
1292 p->v.For.iter = iter;
1293 p->v.For.body = body;
1294 p->v.For.orelse = orelse;
1295 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001296 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001297 return p;
1298}
1299
1300stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001301While(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1302 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001303{
1304 stmt_ty p;
1305 if (!test) {
1306 PyErr_SetString(PyExc_ValueError,
1307 "field test is required for While");
1308 return NULL;
1309 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001310 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001311 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001312 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001313 p->kind = While_kind;
1314 p->v.While.test = test;
1315 p->v.While.body = body;
1316 p->v.While.orelse = orelse;
1317 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001318 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001319 return p;
1320}
1321
1322stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001323If(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1324 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001325{
1326 stmt_ty p;
1327 if (!test) {
1328 PyErr_SetString(PyExc_ValueError,
1329 "field test is required for If");
1330 return NULL;
1331 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001332 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001333 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001334 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001335 p->kind = If_kind;
1336 p->v.If.test = test;
1337 p->v.If.body = body;
1338 p->v.If.orelse = orelse;
1339 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001340 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001341 return p;
1342}
1343
1344stmt_ty
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001345With(asdl_seq * items, asdl_seq * body, int lineno, int col_offset, PyArena
1346 *arena)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001347{
1348 stmt_ty p;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001349 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001350 if (!p)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001351 return NULL;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001352 p->kind = With_kind;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001353 p->v.With.items = items;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001354 p->v.With.body = body;
1355 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001356 p->col_offset = col_offset;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001357 return p;
1358}
1359
1360stmt_ty
Collin Winter828f04a2007-08-31 00:04:24 +00001361Raise(expr_ty exc, expr_ty cause, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001362{
1363 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001364 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001365 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001366 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001367 p->kind = Raise_kind;
Collin Winter828f04a2007-08-31 00:04:24 +00001368 p->v.Raise.exc = exc;
1369 p->v.Raise.cause = cause;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001370 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001371 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001372 return p;
1373}
1374
1375stmt_ty
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001376Try(asdl_seq * body, asdl_seq * handlers, asdl_seq * orelse, asdl_seq *
1377 finalbody, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001378{
1379 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001380 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001381 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001382 return NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001383 p->kind = Try_kind;
1384 p->v.Try.body = body;
1385 p->v.Try.handlers = handlers;
1386 p->v.Try.orelse = orelse;
1387 p->v.Try.finalbody = finalbody;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001388 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 +00001394Assert(expr_ty test, expr_ty msg, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001395{
1396 stmt_ty p;
1397 if (!test) {
1398 PyErr_SetString(PyExc_ValueError,
1399 "field test is required for Assert");
1400 return NULL;
1401 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001402 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001403 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001404 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001405 p->kind = Assert_kind;
1406 p->v.Assert.test = test;
1407 p->v.Assert.msg = msg;
1408 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001409 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001410 return p;
1411}
1412
1413stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001414Import(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001415{
1416 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001417 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001418 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001419 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001420 p->kind = Import_kind;
1421 p->v.Import.names = names;
1422 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001423 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001424 return p;
1425}
1426
1427stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001428ImportFrom(identifier module, asdl_seq * names, int level, int lineno, int
1429 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001430{
1431 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001432 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001433 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001434 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001435 p->kind = ImportFrom_kind;
1436 p->v.ImportFrom.module = module;
1437 p->v.ImportFrom.names = names;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001438 p->v.ImportFrom.level = level;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001439 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001440 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001441 return p;
1442}
1443
1444stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001445Global(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001446{
1447 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001448 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001449 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001450 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001451 p->kind = Global_kind;
1452 p->v.Global.names = names;
1453 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001454 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001455 return p;
1456}
1457
1458stmt_ty
Jeremy Hylton81e95022007-02-27 06:50:52 +00001459Nonlocal(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
1460{
1461 stmt_ty p;
1462 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
1463 if (!p)
1464 return NULL;
1465 p->kind = Nonlocal_kind;
1466 p->v.Nonlocal.names = names;
1467 p->lineno = lineno;
1468 p->col_offset = col_offset;
1469 return p;
1470}
1471
1472stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001473Expr(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001474{
1475 stmt_ty p;
1476 if (!value) {
1477 PyErr_SetString(PyExc_ValueError,
1478 "field value is required for Expr");
1479 return NULL;
1480 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001481 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001482 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001483 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001484 p->kind = Expr_kind;
1485 p->v.Expr.value = value;
1486 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001487 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001488 return p;
1489}
1490
1491stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001492Pass(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001493{
1494 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001495 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001496 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001497 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001498 p->kind = Pass_kind;
1499 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001500 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001501 return p;
1502}
1503
1504stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001505Break(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001506{
1507 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001508 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001509 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001510 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001511 p->kind = Break_kind;
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
1517stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001518Continue(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001519{
1520 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001521 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001522 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001523 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001524 p->kind = Continue_kind;
1525 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001526 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001527 return p;
1528}
1529
1530expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001531BoolOp(boolop_ty op, asdl_seq * values, int lineno, int col_offset, PyArena
1532 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001533{
1534 expr_ty p;
1535 if (!op) {
1536 PyErr_SetString(PyExc_ValueError,
1537 "field op is required for BoolOp");
1538 return NULL;
1539 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001540 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001541 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001542 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001543 p->kind = BoolOp_kind;
1544 p->v.BoolOp.op = op;
1545 p->v.BoolOp.values = values;
1546 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001547 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001548 return p;
1549}
1550
1551expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001552BinOp(expr_ty left, operator_ty op, expr_ty right, int lineno, int col_offset,
1553 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001554{
1555 expr_ty p;
1556 if (!left) {
1557 PyErr_SetString(PyExc_ValueError,
1558 "field left is required for BinOp");
1559 return NULL;
1560 }
1561 if (!op) {
1562 PyErr_SetString(PyExc_ValueError,
1563 "field op is required for BinOp");
1564 return NULL;
1565 }
1566 if (!right) {
1567 PyErr_SetString(PyExc_ValueError,
1568 "field right is required for BinOp");
1569 return NULL;
1570 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001571 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001572 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001573 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001574 p->kind = BinOp_kind;
1575 p->v.BinOp.left = left;
1576 p->v.BinOp.op = op;
1577 p->v.BinOp.right = right;
1578 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001579 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001580 return p;
1581}
1582
1583expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001584UnaryOp(unaryop_ty op, expr_ty operand, int lineno, int col_offset, PyArena
1585 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001586{
1587 expr_ty p;
1588 if (!op) {
1589 PyErr_SetString(PyExc_ValueError,
1590 "field op is required for UnaryOp");
1591 return NULL;
1592 }
1593 if (!operand) {
1594 PyErr_SetString(PyExc_ValueError,
1595 "field operand is required for UnaryOp");
1596 return NULL;
1597 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001598 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001599 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001600 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001601 p->kind = UnaryOp_kind;
1602 p->v.UnaryOp.op = op;
1603 p->v.UnaryOp.operand = operand;
1604 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001605 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001606 return p;
1607}
1608
1609expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001610Lambda(arguments_ty args, expr_ty body, int lineno, int col_offset, PyArena
1611 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001612{
1613 expr_ty p;
1614 if (!args) {
1615 PyErr_SetString(PyExc_ValueError,
1616 "field args is required for Lambda");
1617 return NULL;
1618 }
1619 if (!body) {
1620 PyErr_SetString(PyExc_ValueError,
1621 "field body is required for Lambda");
1622 return NULL;
1623 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001624 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001625 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001626 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001627 p->kind = Lambda_kind;
1628 p->v.Lambda.args = args;
1629 p->v.Lambda.body = body;
1630 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001631 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001632 return p;
1633}
1634
1635expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001636IfExp(expr_ty test, expr_ty body, expr_ty orelse, int lineno, int col_offset,
1637 PyArena *arena)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001638{
1639 expr_ty p;
1640 if (!test) {
1641 PyErr_SetString(PyExc_ValueError,
1642 "field test is required for IfExp");
1643 return NULL;
1644 }
1645 if (!body) {
1646 PyErr_SetString(PyExc_ValueError,
1647 "field body is required for IfExp");
1648 return NULL;
1649 }
1650 if (!orelse) {
1651 PyErr_SetString(PyExc_ValueError,
1652 "field orelse is required for IfExp");
1653 return NULL;
1654 }
1655 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001656 if (!p)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001657 return NULL;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001658 p->kind = IfExp_kind;
1659 p->v.IfExp.test = test;
1660 p->v.IfExp.body = body;
1661 p->v.IfExp.orelse = orelse;
1662 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001663 p->col_offset = col_offset;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001664 return p;
1665}
1666
1667expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001668Dict(asdl_seq * keys, asdl_seq * values, int lineno, int col_offset, PyArena
1669 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001670{
1671 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001672 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001673 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001674 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001675 p->kind = Dict_kind;
1676 p->v.Dict.keys = keys;
1677 p->v.Dict.values = values;
1678 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001679 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001680 return p;
1681}
1682
1683expr_ty
Guido van Rossum86e58e22006-08-28 15:27:34 +00001684Set(asdl_seq * elts, int lineno, int col_offset, PyArena *arena)
1685{
1686 expr_ty p;
1687 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001688 if (!p)
Guido van Rossum86e58e22006-08-28 15:27:34 +00001689 return NULL;
Guido van Rossum86e58e22006-08-28 15:27:34 +00001690 p->kind = Set_kind;
1691 p->v.Set.elts = elts;
1692 p->lineno = lineno;
1693 p->col_offset = col_offset;
1694 return p;
1695}
1696
1697expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001698ListComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1699 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001700{
1701 expr_ty p;
1702 if (!elt) {
1703 PyErr_SetString(PyExc_ValueError,
1704 "field elt is required for ListComp");
1705 return NULL;
1706 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001707 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001708 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001709 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001710 p->kind = ListComp_kind;
1711 p->v.ListComp.elt = elt;
1712 p->v.ListComp.generators = generators;
1713 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001714 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001715 return p;
1716}
1717
1718expr_ty
Nick Coghlan650f0d02007-04-15 12:05:43 +00001719SetComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset, PyArena
1720 *arena)
1721{
1722 expr_ty p;
1723 if (!elt) {
1724 PyErr_SetString(PyExc_ValueError,
1725 "field elt is required for SetComp");
1726 return NULL;
1727 }
1728 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1729 if (!p)
1730 return NULL;
1731 p->kind = SetComp_kind;
1732 p->v.SetComp.elt = elt;
1733 p->v.SetComp.generators = generators;
1734 p->lineno = lineno;
1735 p->col_offset = col_offset;
1736 return p;
1737}
1738
1739expr_ty
Guido van Rossum992d4a32007-07-11 13:09:30 +00001740DictComp(expr_ty key, expr_ty value, asdl_seq * generators, int lineno, int
1741 col_offset, PyArena *arena)
1742{
1743 expr_ty p;
1744 if (!key) {
1745 PyErr_SetString(PyExc_ValueError,
1746 "field key is required for DictComp");
1747 return NULL;
1748 }
1749 if (!value) {
1750 PyErr_SetString(PyExc_ValueError,
1751 "field value is required for DictComp");
1752 return NULL;
1753 }
1754 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1755 if (!p)
1756 return NULL;
1757 p->kind = DictComp_kind;
1758 p->v.DictComp.key = key;
1759 p->v.DictComp.value = value;
1760 p->v.DictComp.generators = generators;
1761 p->lineno = lineno;
1762 p->col_offset = col_offset;
1763 return p;
1764}
1765
1766expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001767GeneratorExp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1768 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001769{
1770 expr_ty p;
1771 if (!elt) {
1772 PyErr_SetString(PyExc_ValueError,
1773 "field elt is required for GeneratorExp");
1774 return NULL;
1775 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001776 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001777 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001778 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001779 p->kind = GeneratorExp_kind;
1780 p->v.GeneratorExp.elt = elt;
1781 p->v.GeneratorExp.generators = generators;
1782 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001783 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001784 return p;
1785}
1786
1787expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001788Yield(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001789{
1790 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001791 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001792 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001793 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001794 p->kind = Yield_kind;
1795 p->v.Yield.value = value;
1796 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001797 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001798 return p;
1799}
1800
1801expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001802YieldFrom(expr_ty value, int lineno, int col_offset, PyArena *arena)
1803{
1804 expr_ty p;
1805 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1806 if (!p)
1807 return NULL;
1808 p->kind = YieldFrom_kind;
1809 p->v.YieldFrom.value = value;
1810 p->lineno = lineno;
1811 p->col_offset = col_offset;
1812 return p;
1813}
1814
1815expr_ty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001816Compare(expr_ty left, asdl_int_seq * ops, asdl_seq * comparators, int lineno,
1817 int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001818{
1819 expr_ty p;
1820 if (!left) {
1821 PyErr_SetString(PyExc_ValueError,
1822 "field left is required for Compare");
1823 return NULL;
1824 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001825 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001826 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001827 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001828 p->kind = Compare_kind;
1829 p->v.Compare.left = left;
1830 p->v.Compare.ops = ops;
1831 p->v.Compare.comparators = comparators;
1832 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001833 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001834 return p;
1835}
1836
1837expr_ty
1838Call(expr_ty func, asdl_seq * args, asdl_seq * keywords, expr_ty starargs,
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001839 expr_ty kwargs, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001840{
1841 expr_ty p;
1842 if (!func) {
1843 PyErr_SetString(PyExc_ValueError,
1844 "field func is required for Call");
1845 return NULL;
1846 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001847 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001848 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001849 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001850 p->kind = Call_kind;
1851 p->v.Call.func = func;
1852 p->v.Call.args = args;
1853 p->v.Call.keywords = keywords;
1854 p->v.Call.starargs = starargs;
1855 p->v.Call.kwargs = kwargs;
1856 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001857 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001858 return p;
1859}
1860
1861expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001862Num(object n, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001863{
1864 expr_ty p;
1865 if (!n) {
1866 PyErr_SetString(PyExc_ValueError,
1867 "field n is required for Num");
1868 return NULL;
1869 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001870 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001871 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001872 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001873 p->kind = Num_kind;
1874 p->v.Num.n = n;
1875 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001876 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001877 return p;
1878}
1879
1880expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001881Str(string s, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001882{
1883 expr_ty p;
1884 if (!s) {
1885 PyErr_SetString(PyExc_ValueError,
1886 "field s is required for Str");
1887 return NULL;
1888 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001889 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001890 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001891 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001892 p->kind = Str_kind;
1893 p->v.Str.s = s;
1894 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001895 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001896 return p;
1897}
1898
1899expr_ty
Benjamin Petersone2498412011-08-09 16:08:39 -05001900Bytes(bytes s, int lineno, int col_offset, PyArena *arena)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001901{
1902 expr_ty p;
1903 if (!s) {
1904 PyErr_SetString(PyExc_ValueError,
1905 "field s is required for Bytes");
1906 return NULL;
1907 }
1908 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001909 if (!p)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001910 return NULL;
Thomas Wouters00e41de2007-02-23 19:56:57 +00001911 p->kind = Bytes_kind;
1912 p->v.Bytes.s = s;
1913 p->lineno = lineno;
1914 p->col_offset = col_offset;
1915 return p;
1916}
1917
1918expr_ty
Georg Brandl52318d62006-09-06 07:06:08 +00001919Ellipsis(int lineno, int col_offset, PyArena *arena)
1920{
1921 expr_ty p;
1922 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001923 if (!p)
Georg Brandl52318d62006-09-06 07:06:08 +00001924 return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00001925 p->kind = Ellipsis_kind;
1926 p->lineno = lineno;
1927 p->col_offset = col_offset;
1928 return p;
1929}
1930
1931expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001932Attribute(expr_ty value, identifier attr, expr_context_ty ctx, int lineno, int
1933 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001934{
1935 expr_ty p;
1936 if (!value) {
1937 PyErr_SetString(PyExc_ValueError,
1938 "field value is required for Attribute");
1939 return NULL;
1940 }
1941 if (!attr) {
1942 PyErr_SetString(PyExc_ValueError,
1943 "field attr is required for Attribute");
1944 return NULL;
1945 }
1946 if (!ctx) {
1947 PyErr_SetString(PyExc_ValueError,
1948 "field ctx is required for Attribute");
1949 return NULL;
1950 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001951 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001952 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001953 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001954 p->kind = Attribute_kind;
1955 p->v.Attribute.value = value;
1956 p->v.Attribute.attr = attr;
1957 p->v.Attribute.ctx = ctx;
1958 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001959 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001960 return p;
1961}
1962
1963expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001964Subscript(expr_ty value, slice_ty slice, expr_context_ty ctx, int lineno, int
1965 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001966{
1967 expr_ty p;
1968 if (!value) {
1969 PyErr_SetString(PyExc_ValueError,
1970 "field value is required for Subscript");
1971 return NULL;
1972 }
1973 if (!slice) {
1974 PyErr_SetString(PyExc_ValueError,
1975 "field slice is required for Subscript");
1976 return NULL;
1977 }
1978 if (!ctx) {
1979 PyErr_SetString(PyExc_ValueError,
1980 "field ctx is required for Subscript");
1981 return NULL;
1982 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001983 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001984 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001985 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001986 p->kind = Subscript_kind;
1987 p->v.Subscript.value = value;
1988 p->v.Subscript.slice = slice;
1989 p->v.Subscript.ctx = ctx;
1990 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001991 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001992 return p;
1993}
1994
1995expr_ty
Guido van Rossum0368b722007-05-11 16:50:42 +00001996Starred(expr_ty value, expr_context_ty ctx, int lineno, int col_offset, PyArena
1997 *arena)
1998{
1999 expr_ty p;
2000 if (!value) {
2001 PyErr_SetString(PyExc_ValueError,
2002 "field value is required for Starred");
2003 return NULL;
2004 }
2005 if (!ctx) {
2006 PyErr_SetString(PyExc_ValueError,
2007 "field ctx is required for Starred");
2008 return NULL;
2009 }
2010 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
2011 if (!p)
2012 return NULL;
2013 p->kind = Starred_kind;
2014 p->v.Starred.value = value;
2015 p->v.Starred.ctx = ctx;
2016 p->lineno = lineno;
2017 p->col_offset = col_offset;
2018 return p;
2019}
2020
2021expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002022Name(identifier id, expr_context_ty ctx, int lineno, int col_offset, PyArena
2023 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002024{
2025 expr_ty p;
2026 if (!id) {
2027 PyErr_SetString(PyExc_ValueError,
2028 "field id is required for Name");
2029 return NULL;
2030 }
2031 if (!ctx) {
2032 PyErr_SetString(PyExc_ValueError,
2033 "field ctx is required for Name");
2034 return NULL;
2035 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002036 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002037 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002038 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002039 p->kind = Name_kind;
2040 p->v.Name.id = id;
2041 p->v.Name.ctx = ctx;
2042 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002043 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002044 return p;
2045}
2046
2047expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002048List(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2049 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002050{
2051 expr_ty p;
2052 if (!ctx) {
2053 PyErr_SetString(PyExc_ValueError,
2054 "field ctx is required for List");
2055 return NULL;
2056 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002057 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002058 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002059 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002060 p->kind = List_kind;
2061 p->v.List.elts = elts;
2062 p->v.List.ctx = ctx;
2063 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002064 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002065 return p;
2066}
2067
2068expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002069Tuple(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2070 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002071{
2072 expr_ty p;
2073 if (!ctx) {
2074 PyErr_SetString(PyExc_ValueError,
2075 "field ctx is required for Tuple");
2076 return NULL;
2077 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002078 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002079 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002080 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002081 p->kind = Tuple_kind;
2082 p->v.Tuple.elts = elts;
2083 p->v.Tuple.ctx = ctx;
2084 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002085 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002086 return p;
2087}
2088
2089slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002090Slice(expr_ty lower, expr_ty upper, expr_ty step, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002091{
2092 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002093 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002094 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002095 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002096 p->kind = Slice_kind;
2097 p->v.Slice.lower = lower;
2098 p->v.Slice.upper = upper;
2099 p->v.Slice.step = step;
2100 return p;
2101}
2102
2103slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002104ExtSlice(asdl_seq * dims, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002105{
2106 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002107 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002108 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002109 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002110 p->kind = ExtSlice_kind;
2111 p->v.ExtSlice.dims = dims;
2112 return p;
2113}
2114
2115slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002116Index(expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002117{
2118 slice_ty p;
2119 if (!value) {
2120 PyErr_SetString(PyExc_ValueError,
2121 "field value is required for Index");
2122 return NULL;
2123 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002124 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002125 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002126 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002127 p->kind = Index_kind;
2128 p->v.Index.value = value;
2129 return p;
2130}
2131
2132comprehension_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002133comprehension(expr_ty target, expr_ty iter, asdl_seq * ifs, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002134{
2135 comprehension_ty p;
2136 if (!target) {
2137 PyErr_SetString(PyExc_ValueError,
2138 "field target is required for comprehension");
2139 return NULL;
2140 }
2141 if (!iter) {
2142 PyErr_SetString(PyExc_ValueError,
2143 "field iter is required for comprehension");
2144 return NULL;
2145 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002146 p = (comprehension_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002147 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002148 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002149 p->target = target;
2150 p->iter = iter;
2151 p->ifs = ifs;
2152 return p;
2153}
2154
2155excepthandler_ty
Neal Norwitzad74aa82008-03-31 05:14:30 +00002156ExceptHandler(expr_ty type, identifier name, asdl_seq * body, int lineno, int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002157 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002158{
2159 excepthandler_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002160 p = (excepthandler_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002161 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002162 return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00002163 p->kind = ExceptHandler_kind;
2164 p->v.ExceptHandler.type = type;
2165 p->v.ExceptHandler.name = name;
2166 p->v.ExceptHandler.body = body;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002167 p->lineno = lineno;
2168 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002169 return p;
2170}
2171
2172arguments_ty
Neal Norwitzc1505362006-12-28 06:47:50 +00002173arguments(asdl_seq * args, identifier vararg, expr_ty varargannotation,
2174 asdl_seq * kwonlyargs, identifier kwarg, expr_ty kwargannotation,
2175 asdl_seq * defaults, asdl_seq * kw_defaults, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002176{
2177 arguments_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002178 p = (arguments_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002179 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002180 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002181 p->args = args;
2182 p->vararg = vararg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002183 p->varargannotation = varargannotation;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002184 p->kwonlyargs = kwonlyargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002185 p->kwarg = kwarg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002186 p->kwargannotation = kwargannotation;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002187 p->defaults = defaults;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002188 p->kw_defaults = kw_defaults;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002189 return p;
2190}
2191
Neal Norwitzc1505362006-12-28 06:47:50 +00002192arg_ty
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002193arg(identifier arg, expr_ty annotation, PyArena *arena)
Neal Norwitzc1505362006-12-28 06:47:50 +00002194{
2195 arg_ty p;
2196 if (!arg) {
2197 PyErr_SetString(PyExc_ValueError,
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002198 "field arg is required for arg");
Neal Norwitzc1505362006-12-28 06:47:50 +00002199 return NULL;
2200 }
2201 p = (arg_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002202 if (!p)
Neal Norwitzc1505362006-12-28 06:47:50 +00002203 return NULL;
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002204 p->arg = arg;
2205 p->annotation = annotation;
Neal Norwitzc1505362006-12-28 06:47:50 +00002206 return p;
2207}
2208
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002209keyword_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002210keyword(identifier arg, expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002211{
2212 keyword_ty p;
2213 if (!arg) {
2214 PyErr_SetString(PyExc_ValueError,
2215 "field arg is required for keyword");
2216 return NULL;
2217 }
2218 if (!value) {
2219 PyErr_SetString(PyExc_ValueError,
2220 "field value is required for keyword");
2221 return NULL;
2222 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002223 p = (keyword_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002224 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002225 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002226 p->arg = arg;
2227 p->value = value;
2228 return p;
2229}
2230
2231alias_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002232alias(identifier name, identifier asname, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002233{
2234 alias_ty p;
2235 if (!name) {
2236 PyErr_SetString(PyExc_ValueError,
2237 "field name is required for alias");
2238 return NULL;
2239 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002240 p = (alias_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002241 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002242 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002243 p->name = name;
2244 p->asname = asname;
2245 return p;
2246}
2247
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002248withitem_ty
2249withitem(expr_ty context_expr, expr_ty optional_vars, PyArena *arena)
2250{
2251 withitem_ty p;
2252 if (!context_expr) {
2253 PyErr_SetString(PyExc_ValueError,
2254 "field context_expr is required for withitem");
2255 return NULL;
2256 }
2257 p = (withitem_ty)PyArena_Malloc(arena, sizeof(*p));
2258 if (!p)
2259 return NULL;
2260 p->context_expr = context_expr;
2261 p->optional_vars = optional_vars;
2262 return p;
2263}
2264
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002265
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002266PyObject*
2267ast2obj_mod(void* _o)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002268{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002269 mod_ty o = (mod_ty)_o;
2270 PyObject *result = NULL, *value = NULL;
2271 if (!o) {
2272 Py_INCREF(Py_None);
2273 return Py_None;
2274 }
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002275
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002276 switch (o->kind) {
2277 case Module_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002278 result = PyType_GenericNew(Module_type, NULL, NULL);
2279 if (!result) goto failed;
2280 value = ast2obj_list(o->v.Module.body, ast2obj_stmt);
2281 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002282 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002283 goto failed;
2284 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002285 break;
2286 case Interactive_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002287 result = PyType_GenericNew(Interactive_type, NULL, NULL);
2288 if (!result) goto failed;
2289 value = ast2obj_list(o->v.Interactive.body, ast2obj_stmt);
2290 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002291 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002292 goto failed;
2293 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002294 break;
2295 case Expression_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002296 result = PyType_GenericNew(Expression_type, NULL, NULL);
2297 if (!result) goto failed;
2298 value = ast2obj_expr(o->v.Expression.body);
2299 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002300 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002301 goto failed;
2302 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002303 break;
2304 case Suite_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002305 result = PyType_GenericNew(Suite_type, NULL, NULL);
2306 if (!result) goto failed;
2307 value = ast2obj_list(o->v.Suite.body, ast2obj_stmt);
2308 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002309 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002310 goto failed;
2311 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002312 break;
2313 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002314 return result;
2315failed:
2316 Py_XDECREF(value);
2317 Py_XDECREF(result);
2318 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002319}
2320
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002321PyObject*
2322ast2obj_stmt(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002323{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002324 stmt_ty o = (stmt_ty)_o;
2325 PyObject *result = NULL, *value = NULL;
2326 if (!o) {
2327 Py_INCREF(Py_None);
2328 return Py_None;
2329 }
2330
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002331 switch (o->kind) {
2332 case FunctionDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002333 result = PyType_GenericNew(FunctionDef_type, NULL, NULL);
2334 if (!result) goto failed;
2335 value = ast2obj_identifier(o->v.FunctionDef.name);
2336 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002337 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002338 goto failed;
2339 Py_DECREF(value);
2340 value = ast2obj_arguments(o->v.FunctionDef.args);
2341 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002342 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002343 goto failed;
2344 Py_DECREF(value);
2345 value = ast2obj_list(o->v.FunctionDef.body, ast2obj_stmt);
2346 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002347 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002348 goto failed;
2349 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002350 value = ast2obj_list(o->v.FunctionDef.decorator_list,
2351 ast2obj_expr);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002352 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002353 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002354 -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002355 goto failed;
2356 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00002357 value = ast2obj_expr(o->v.FunctionDef.returns);
2358 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002359 if (_PyObject_SetAttrId(result, &PyId_returns, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00002360 goto failed;
2361 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002362 break;
2363 case ClassDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002364 result = PyType_GenericNew(ClassDef_type, NULL, NULL);
2365 if (!result) goto failed;
2366 value = ast2obj_identifier(o->v.ClassDef.name);
2367 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002368 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002369 goto failed;
2370 Py_DECREF(value);
2371 value = ast2obj_list(o->v.ClassDef.bases, ast2obj_expr);
2372 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002373 if (_PyObject_SetAttrId(result, &PyId_bases, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002374 goto failed;
2375 Py_DECREF(value);
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002376 value = ast2obj_list(o->v.ClassDef.keywords, ast2obj_keyword);
2377 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002378 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002379 goto failed;
2380 Py_DECREF(value);
2381 value = ast2obj_expr(o->v.ClassDef.starargs);
2382 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002383 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002384 goto failed;
2385 Py_DECREF(value);
2386 value = ast2obj_expr(o->v.ClassDef.kwargs);
2387 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002388 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002389 goto failed;
2390 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002391 value = ast2obj_list(o->v.ClassDef.body, ast2obj_stmt);
2392 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002393 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002394 goto failed;
2395 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002396 value = ast2obj_list(o->v.ClassDef.decorator_list,
2397 ast2obj_expr);
2398 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002399 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002400 -1)
2401 goto failed;
2402 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002403 break;
2404 case Return_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002405 result = PyType_GenericNew(Return_type, NULL, NULL);
2406 if (!result) goto failed;
2407 value = ast2obj_expr(o->v.Return.value);
2408 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002409 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002410 goto failed;
2411 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002412 break;
2413 case Delete_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002414 result = PyType_GenericNew(Delete_type, NULL, NULL);
2415 if (!result) goto failed;
2416 value = ast2obj_list(o->v.Delete.targets, ast2obj_expr);
2417 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002418 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002419 goto failed;
2420 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002421 break;
2422 case Assign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002423 result = PyType_GenericNew(Assign_type, NULL, NULL);
2424 if (!result) goto failed;
2425 value = ast2obj_list(o->v.Assign.targets, ast2obj_expr);
2426 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002427 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002428 goto failed;
2429 Py_DECREF(value);
2430 value = ast2obj_expr(o->v.Assign.value);
2431 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002432 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002433 goto failed;
2434 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002435 break;
2436 case AugAssign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002437 result = PyType_GenericNew(AugAssign_type, NULL, NULL);
2438 if (!result) goto failed;
2439 value = ast2obj_expr(o->v.AugAssign.target);
2440 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002441 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002442 goto failed;
2443 Py_DECREF(value);
2444 value = ast2obj_operator(o->v.AugAssign.op);
2445 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002446 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002447 goto failed;
2448 Py_DECREF(value);
2449 value = ast2obj_expr(o->v.AugAssign.value);
2450 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002451 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002452 goto failed;
2453 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002454 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002455 case For_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002456 result = PyType_GenericNew(For_type, NULL, NULL);
2457 if (!result) goto failed;
2458 value = ast2obj_expr(o->v.For.target);
2459 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002460 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002461 goto failed;
2462 Py_DECREF(value);
2463 value = ast2obj_expr(o->v.For.iter);
2464 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002465 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002466 goto failed;
2467 Py_DECREF(value);
2468 value = ast2obj_list(o->v.For.body, ast2obj_stmt);
2469 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002470 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002471 goto failed;
2472 Py_DECREF(value);
2473 value = ast2obj_list(o->v.For.orelse, ast2obj_stmt);
2474 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002475 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002476 goto failed;
2477 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002478 break;
2479 case While_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002480 result = PyType_GenericNew(While_type, NULL, NULL);
2481 if (!result) goto failed;
2482 value = ast2obj_expr(o->v.While.test);
2483 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002484 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002485 goto failed;
2486 Py_DECREF(value);
2487 value = ast2obj_list(o->v.While.body, ast2obj_stmt);
2488 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002489 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002490 goto failed;
2491 Py_DECREF(value);
2492 value = ast2obj_list(o->v.While.orelse, ast2obj_stmt);
2493 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002494 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002495 goto failed;
2496 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002497 break;
2498 case If_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002499 result = PyType_GenericNew(If_type, NULL, NULL);
2500 if (!result) goto failed;
2501 value = ast2obj_expr(o->v.If.test);
2502 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002503 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002504 goto failed;
2505 Py_DECREF(value);
2506 value = ast2obj_list(o->v.If.body, ast2obj_stmt);
2507 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002508 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002509 goto failed;
2510 Py_DECREF(value);
2511 value = ast2obj_list(o->v.If.orelse, ast2obj_stmt);
2512 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002513 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002514 goto failed;
2515 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002516 break;
Guido van Rossumc2e20742006-02-27 22:32:47 +00002517 case With_kind:
2518 result = PyType_GenericNew(With_type, NULL, NULL);
2519 if (!result) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002520 value = ast2obj_list(o->v.With.items, ast2obj_withitem);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002521 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002522 if (_PyObject_SetAttrId(result, &PyId_items, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002523 goto failed;
2524 Py_DECREF(value);
2525 value = ast2obj_list(o->v.With.body, ast2obj_stmt);
2526 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002527 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002528 goto failed;
2529 Py_DECREF(value);
2530 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002531 case Raise_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002532 result = PyType_GenericNew(Raise_type, NULL, NULL);
2533 if (!result) goto failed;
Collin Winter828f04a2007-08-31 00:04:24 +00002534 value = ast2obj_expr(o->v.Raise.exc);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002535 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002536 if (_PyObject_SetAttrId(result, &PyId_exc, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002537 goto failed;
2538 Py_DECREF(value);
Collin Winter828f04a2007-08-31 00:04:24 +00002539 value = ast2obj_expr(o->v.Raise.cause);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002540 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002541 if (_PyObject_SetAttrId(result, &PyId_cause, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002542 goto failed;
2543 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002544 break;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002545 case Try_kind:
2546 result = PyType_GenericNew(Try_type, NULL, NULL);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002547 if (!result) goto failed;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002548 value = ast2obj_list(o->v.Try.body, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002549 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002550 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002551 goto failed;
2552 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002553 value = ast2obj_list(o->v.Try.handlers, ast2obj_excepthandler);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002554 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002555 if (_PyObject_SetAttrId(result, &PyId_handlers, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002556 goto failed;
2557 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002558 value = ast2obj_list(o->v.Try.orelse, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002559 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002560 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002561 goto failed;
2562 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002563 value = ast2obj_list(o->v.Try.finalbody, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002564 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002565 if (_PyObject_SetAttrId(result, &PyId_finalbody, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002566 goto failed;
2567 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002568 break;
2569 case Assert_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002570 result = PyType_GenericNew(Assert_type, NULL, NULL);
2571 if (!result) goto failed;
2572 value = ast2obj_expr(o->v.Assert.test);
2573 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002574 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002575 goto failed;
2576 Py_DECREF(value);
2577 value = ast2obj_expr(o->v.Assert.msg);
2578 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002579 if (_PyObject_SetAttrId(result, &PyId_msg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002580 goto failed;
2581 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002582 break;
2583 case Import_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002584 result = PyType_GenericNew(Import_type, NULL, NULL);
2585 if (!result) goto failed;
2586 value = ast2obj_list(o->v.Import.names, ast2obj_alias);
2587 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002588 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002589 goto failed;
2590 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002591 break;
2592 case ImportFrom_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002593 result = PyType_GenericNew(ImportFrom_type, NULL, NULL);
2594 if (!result) goto failed;
2595 value = ast2obj_identifier(o->v.ImportFrom.module);
2596 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002597 if (_PyObject_SetAttrId(result, &PyId_module, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002598 goto failed;
2599 Py_DECREF(value);
2600 value = ast2obj_list(o->v.ImportFrom.names, ast2obj_alias);
2601 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002602 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002603 goto failed;
2604 Py_DECREF(value);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002605 value = ast2obj_int(o->v.ImportFrom.level);
2606 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002607 if (_PyObject_SetAttrId(result, &PyId_level, value) == -1)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002608 goto failed;
2609 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002610 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002611 case Global_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002612 result = PyType_GenericNew(Global_type, NULL, NULL);
2613 if (!result) goto failed;
2614 value = ast2obj_list(o->v.Global.names, ast2obj_identifier);
2615 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002616 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002617 goto failed;
2618 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002619 break;
Jeremy Hylton81e95022007-02-27 06:50:52 +00002620 case Nonlocal_kind:
2621 result = PyType_GenericNew(Nonlocal_type, NULL, NULL);
2622 if (!result) goto failed;
2623 value = ast2obj_list(o->v.Nonlocal.names, ast2obj_identifier);
2624 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002625 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Jeremy Hylton81e95022007-02-27 06:50:52 +00002626 goto failed;
2627 Py_DECREF(value);
2628 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002629 case Expr_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002630 result = PyType_GenericNew(Expr_type, NULL, NULL);
2631 if (!result) goto failed;
2632 value = ast2obj_expr(o->v.Expr.value);
2633 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002634 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002635 goto failed;
2636 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002637 break;
2638 case Pass_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002639 result = PyType_GenericNew(Pass_type, NULL, NULL);
2640 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002641 break;
2642 case Break_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002643 result = PyType_GenericNew(Break_type, NULL, NULL);
2644 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002645 break;
2646 case Continue_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002647 result = PyType_GenericNew(Continue_type, NULL, NULL);
2648 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002649 break;
2650 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002651 value = ast2obj_int(o->lineno);
2652 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002653 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002654 goto failed;
2655 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002656 value = ast2obj_int(o->col_offset);
2657 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002658 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002659 goto failed;
2660 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002661 return result;
2662failed:
2663 Py_XDECREF(value);
2664 Py_XDECREF(result);
2665 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002666}
2667
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002668PyObject*
2669ast2obj_expr(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002670{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002671 expr_ty o = (expr_ty)_o;
2672 PyObject *result = NULL, *value = NULL;
2673 if (!o) {
2674 Py_INCREF(Py_None);
2675 return Py_None;
2676 }
2677
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002678 switch (o->kind) {
2679 case BoolOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002680 result = PyType_GenericNew(BoolOp_type, NULL, NULL);
2681 if (!result) goto failed;
2682 value = ast2obj_boolop(o->v.BoolOp.op);
2683 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002684 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002685 goto failed;
2686 Py_DECREF(value);
2687 value = ast2obj_list(o->v.BoolOp.values, ast2obj_expr);
2688 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002689 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002690 goto failed;
2691 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002692 break;
2693 case BinOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002694 result = PyType_GenericNew(BinOp_type, NULL, NULL);
2695 if (!result) goto failed;
2696 value = ast2obj_expr(o->v.BinOp.left);
2697 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002698 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002699 goto failed;
2700 Py_DECREF(value);
2701 value = ast2obj_operator(o->v.BinOp.op);
2702 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002703 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002704 goto failed;
2705 Py_DECREF(value);
2706 value = ast2obj_expr(o->v.BinOp.right);
2707 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002708 if (_PyObject_SetAttrId(result, &PyId_right, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002709 goto failed;
2710 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002711 break;
2712 case UnaryOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002713 result = PyType_GenericNew(UnaryOp_type, NULL, NULL);
2714 if (!result) goto failed;
2715 value = ast2obj_unaryop(o->v.UnaryOp.op);
2716 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002717 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002718 goto failed;
2719 Py_DECREF(value);
2720 value = ast2obj_expr(o->v.UnaryOp.operand);
2721 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002722 if (_PyObject_SetAttrId(result, &PyId_operand, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002723 goto failed;
2724 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002725 break;
2726 case Lambda_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002727 result = PyType_GenericNew(Lambda_type, NULL, NULL);
2728 if (!result) goto failed;
2729 value = ast2obj_arguments(o->v.Lambda.args);
2730 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002731 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002732 goto failed;
2733 Py_DECREF(value);
2734 value = ast2obj_expr(o->v.Lambda.body);
2735 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002736 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002737 goto failed;
2738 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002739 break;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002740 case IfExp_kind:
2741 result = PyType_GenericNew(IfExp_type, NULL, NULL);
2742 if (!result) goto failed;
2743 value = ast2obj_expr(o->v.IfExp.test);
2744 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002745 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002746 goto failed;
2747 Py_DECREF(value);
2748 value = ast2obj_expr(o->v.IfExp.body);
2749 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002750 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002751 goto failed;
2752 Py_DECREF(value);
2753 value = ast2obj_expr(o->v.IfExp.orelse);
2754 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002755 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002756 goto failed;
2757 Py_DECREF(value);
2758 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002759 case Dict_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002760 result = PyType_GenericNew(Dict_type, NULL, NULL);
2761 if (!result) goto failed;
2762 value = ast2obj_list(o->v.Dict.keys, ast2obj_expr);
2763 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002764 if (_PyObject_SetAttrId(result, &PyId_keys, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002765 goto failed;
2766 Py_DECREF(value);
2767 value = ast2obj_list(o->v.Dict.values, ast2obj_expr);
2768 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002769 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002770 goto failed;
2771 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002772 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002773 case Set_kind:
2774 result = PyType_GenericNew(Set_type, NULL, NULL);
2775 if (!result) goto failed;
2776 value = ast2obj_list(o->v.Set.elts, ast2obj_expr);
2777 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002778 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Guido van Rossum86e58e22006-08-28 15:27:34 +00002779 goto failed;
2780 Py_DECREF(value);
2781 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002782 case ListComp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002783 result = PyType_GenericNew(ListComp_type, NULL, NULL);
2784 if (!result) goto failed;
2785 value = ast2obj_expr(o->v.ListComp.elt);
2786 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002787 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002788 goto failed;
2789 Py_DECREF(value);
2790 value = ast2obj_list(o->v.ListComp.generators,
2791 ast2obj_comprehension);
2792 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002793 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002794 goto failed;
2795 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002796 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00002797 case SetComp_kind:
2798 result = PyType_GenericNew(SetComp_type, NULL, NULL);
2799 if (!result) goto failed;
2800 value = ast2obj_expr(o->v.SetComp.elt);
2801 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002802 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002803 goto failed;
2804 Py_DECREF(value);
2805 value = ast2obj_list(o->v.SetComp.generators,
2806 ast2obj_comprehension);
2807 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002808 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002809 goto failed;
2810 Py_DECREF(value);
2811 break;
Guido van Rossum992d4a32007-07-11 13:09:30 +00002812 case DictComp_kind:
2813 result = PyType_GenericNew(DictComp_type, NULL, NULL);
2814 if (!result) goto failed;
2815 value = ast2obj_expr(o->v.DictComp.key);
2816 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002817 if (_PyObject_SetAttrId(result, &PyId_key, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002818 goto failed;
2819 Py_DECREF(value);
2820 value = ast2obj_expr(o->v.DictComp.value);
2821 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002822 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002823 goto failed;
2824 Py_DECREF(value);
2825 value = ast2obj_list(o->v.DictComp.generators,
2826 ast2obj_comprehension);
2827 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002828 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002829 goto failed;
2830 Py_DECREF(value);
2831 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002832 case GeneratorExp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002833 result = PyType_GenericNew(GeneratorExp_type, NULL, NULL);
2834 if (!result) goto failed;
2835 value = ast2obj_expr(o->v.GeneratorExp.elt);
2836 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002837 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002838 goto failed;
2839 Py_DECREF(value);
2840 value = ast2obj_list(o->v.GeneratorExp.generators,
2841 ast2obj_comprehension);
2842 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002843 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002844 goto failed;
2845 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002846 break;
2847 case Yield_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002848 result = PyType_GenericNew(Yield_type, NULL, NULL);
2849 if (!result) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002850 value = ast2obj_expr(o->v.Yield.value);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002851 if (!value) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002852 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002853 goto failed;
2854 Py_DECREF(value);
Benjamin Peterson527c6222012-01-14 08:58:23 -05002855 break;
2856 case YieldFrom_kind:
2857 result = PyType_GenericNew(YieldFrom_type, NULL, NULL);
2858 if (!result) goto failed;
2859 value = ast2obj_expr(o->v.YieldFrom.value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002860 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002861 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002862 goto failed;
2863 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002864 break;
2865 case Compare_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002866 result = PyType_GenericNew(Compare_type, NULL, NULL);
2867 if (!result) goto failed;
2868 value = ast2obj_expr(o->v.Compare.left);
2869 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002870 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002871 goto failed;
2872 Py_DECREF(value);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002873 {
Benjamin Peterson77fa9372012-05-15 10:10:27 -07002874 Py_ssize_t i, n = asdl_seq_LEN(o->v.Compare.ops);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002875 value = PyList_New(n);
2876 if (!value) goto failed;
2877 for(i = 0; i < n; i++)
2878 PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(o->v.Compare.ops, i)));
2879 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002880 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002881 if (_PyObject_SetAttrId(result, &PyId_ops, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002882 goto failed;
2883 Py_DECREF(value);
2884 value = ast2obj_list(o->v.Compare.comparators, ast2obj_expr);
2885 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002886 if (_PyObject_SetAttrId(result, &PyId_comparators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002887 goto failed;
2888 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002889 break;
2890 case Call_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002891 result = PyType_GenericNew(Call_type, NULL, NULL);
2892 if (!result) goto failed;
2893 value = ast2obj_expr(o->v.Call.func);
2894 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002895 if (_PyObject_SetAttrId(result, &PyId_func, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002896 goto failed;
2897 Py_DECREF(value);
2898 value = ast2obj_list(o->v.Call.args, ast2obj_expr);
2899 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002900 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002901 goto failed;
2902 Py_DECREF(value);
2903 value = ast2obj_list(o->v.Call.keywords, ast2obj_keyword);
2904 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002905 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002906 goto failed;
2907 Py_DECREF(value);
2908 value = ast2obj_expr(o->v.Call.starargs);
2909 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002910 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002911 goto failed;
2912 Py_DECREF(value);
2913 value = ast2obj_expr(o->v.Call.kwargs);
2914 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002915 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002916 goto failed;
2917 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002918 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002919 case Num_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002920 result = PyType_GenericNew(Num_type, NULL, NULL);
2921 if (!result) goto failed;
2922 value = ast2obj_object(o->v.Num.n);
2923 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002924 if (_PyObject_SetAttrId(result, &PyId_n, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002925 goto failed;
2926 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002927 break;
2928 case Str_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002929 result = PyType_GenericNew(Str_type, NULL, NULL);
2930 if (!result) goto failed;
2931 value = ast2obj_string(o->v.Str.s);
2932 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002933 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002934 goto failed;
2935 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002936 break;
Thomas Wouters00e41de2007-02-23 19:56:57 +00002937 case Bytes_kind:
2938 result = PyType_GenericNew(Bytes_type, NULL, NULL);
2939 if (!result) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05002940 value = ast2obj_bytes(o->v.Bytes.s);
Thomas Wouters00e41de2007-02-23 19:56:57 +00002941 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002942 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Thomas Wouters00e41de2007-02-23 19:56:57 +00002943 goto failed;
2944 Py_DECREF(value);
2945 break;
Georg Brandl52318d62006-09-06 07:06:08 +00002946 case Ellipsis_kind:
2947 result = PyType_GenericNew(Ellipsis_type, NULL, NULL);
2948 if (!result) goto failed;
2949 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002950 case Attribute_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002951 result = PyType_GenericNew(Attribute_type, NULL, NULL);
2952 if (!result) goto failed;
2953 value = ast2obj_expr(o->v.Attribute.value);
2954 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002955 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002956 goto failed;
2957 Py_DECREF(value);
2958 value = ast2obj_identifier(o->v.Attribute.attr);
2959 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002960 if (_PyObject_SetAttrId(result, &PyId_attr, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002961 goto failed;
2962 Py_DECREF(value);
2963 value = ast2obj_expr_context(o->v.Attribute.ctx);
2964 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002965 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002966 goto failed;
2967 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002968 break;
2969 case Subscript_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002970 result = PyType_GenericNew(Subscript_type, NULL, NULL);
2971 if (!result) goto failed;
2972 value = ast2obj_expr(o->v.Subscript.value);
2973 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002974 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002975 goto failed;
2976 Py_DECREF(value);
2977 value = ast2obj_slice(o->v.Subscript.slice);
2978 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002979 if (_PyObject_SetAttrId(result, &PyId_slice, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002980 goto failed;
2981 Py_DECREF(value);
2982 value = ast2obj_expr_context(o->v.Subscript.ctx);
2983 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002984 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002985 goto failed;
2986 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002987 break;
Guido van Rossum0368b722007-05-11 16:50:42 +00002988 case Starred_kind:
2989 result = PyType_GenericNew(Starred_type, NULL, NULL);
2990 if (!result) goto failed;
2991 value = ast2obj_expr(o->v.Starred.value);
2992 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002993 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002994 goto failed;
2995 Py_DECREF(value);
2996 value = ast2obj_expr_context(o->v.Starred.ctx);
2997 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002998 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002999 goto failed;
3000 Py_DECREF(value);
3001 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003002 case Name_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003003 result = PyType_GenericNew(Name_type, NULL, NULL);
3004 if (!result) goto failed;
3005 value = ast2obj_identifier(o->v.Name.id);
3006 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003007 if (_PyObject_SetAttrId(result, &PyId_id, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003008 goto failed;
3009 Py_DECREF(value);
3010 value = ast2obj_expr_context(o->v.Name.ctx);
3011 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003012 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003013 goto failed;
3014 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003015 break;
3016 case List_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003017 result = PyType_GenericNew(List_type, NULL, NULL);
3018 if (!result) goto failed;
3019 value = ast2obj_list(o->v.List.elts, ast2obj_expr);
3020 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003021 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003022 goto failed;
3023 Py_DECREF(value);
3024 value = ast2obj_expr_context(o->v.List.ctx);
3025 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003026 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003027 goto failed;
3028 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003029 break;
3030 case Tuple_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003031 result = PyType_GenericNew(Tuple_type, NULL, NULL);
3032 if (!result) goto failed;
3033 value = ast2obj_list(o->v.Tuple.elts, ast2obj_expr);
3034 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003035 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003036 goto failed;
3037 Py_DECREF(value);
3038 value = ast2obj_expr_context(o->v.Tuple.ctx);
3039 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003040 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003041 goto failed;
3042 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003043 break;
3044 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00003045 value = ast2obj_int(o->lineno);
3046 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003047 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003048 goto failed;
3049 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00003050 value = ast2obj_int(o->col_offset);
3051 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003052 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003053 goto failed;
3054 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003055 return result;
3056failed:
3057 Py_XDECREF(value);
3058 Py_XDECREF(result);
3059 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003060}
3061
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003062PyObject* ast2obj_expr_context(expr_context_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003063{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003064 switch(o) {
3065 case Load:
3066 Py_INCREF(Load_singleton);
3067 return Load_singleton;
3068 case Store:
3069 Py_INCREF(Store_singleton);
3070 return Store_singleton;
3071 case Del:
3072 Py_INCREF(Del_singleton);
3073 return Del_singleton;
3074 case AugLoad:
3075 Py_INCREF(AugLoad_singleton);
3076 return AugLoad_singleton;
3077 case AugStore:
3078 Py_INCREF(AugStore_singleton);
3079 return AugStore_singleton;
3080 case Param:
3081 Py_INCREF(Param_singleton);
3082 return Param_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003083 default:
3084 /* should never happen, but just in case ... */
3085 PyErr_Format(PyExc_SystemError, "unknown expr_context found");
3086 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003087 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003088}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003089PyObject*
3090ast2obj_slice(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003091{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003092 slice_ty o = (slice_ty)_o;
3093 PyObject *result = NULL, *value = NULL;
3094 if (!o) {
3095 Py_INCREF(Py_None);
3096 return Py_None;
3097 }
3098
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003099 switch (o->kind) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003100 case Slice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003101 result = PyType_GenericNew(Slice_type, NULL, NULL);
3102 if (!result) goto failed;
3103 value = ast2obj_expr(o->v.Slice.lower);
3104 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003105 if (_PyObject_SetAttrId(result, &PyId_lower, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003106 goto failed;
3107 Py_DECREF(value);
3108 value = ast2obj_expr(o->v.Slice.upper);
3109 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003110 if (_PyObject_SetAttrId(result, &PyId_upper, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003111 goto failed;
3112 Py_DECREF(value);
3113 value = ast2obj_expr(o->v.Slice.step);
3114 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003115 if (_PyObject_SetAttrId(result, &PyId_step, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003116 goto failed;
3117 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003118 break;
3119 case ExtSlice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003120 result = PyType_GenericNew(ExtSlice_type, NULL, NULL);
3121 if (!result) goto failed;
3122 value = ast2obj_list(o->v.ExtSlice.dims, ast2obj_slice);
3123 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003124 if (_PyObject_SetAttrId(result, &PyId_dims, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003125 goto failed;
3126 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003127 break;
3128 case Index_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003129 result = PyType_GenericNew(Index_type, NULL, NULL);
3130 if (!result) goto failed;
3131 value = ast2obj_expr(o->v.Index.value);
3132 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003133 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003134 goto failed;
3135 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003136 break;
3137 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003138 return result;
3139failed:
3140 Py_XDECREF(value);
3141 Py_XDECREF(result);
3142 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003143}
3144
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003145PyObject* ast2obj_boolop(boolop_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003146{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003147 switch(o) {
3148 case And:
3149 Py_INCREF(And_singleton);
3150 return And_singleton;
3151 case Or:
3152 Py_INCREF(Or_singleton);
3153 return Or_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003154 default:
3155 /* should never happen, but just in case ... */
3156 PyErr_Format(PyExc_SystemError, "unknown boolop found");
3157 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003158 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003159}
3160PyObject* ast2obj_operator(operator_ty o)
3161{
3162 switch(o) {
3163 case Add:
3164 Py_INCREF(Add_singleton);
3165 return Add_singleton;
3166 case Sub:
3167 Py_INCREF(Sub_singleton);
3168 return Sub_singleton;
3169 case Mult:
3170 Py_INCREF(Mult_singleton);
3171 return Mult_singleton;
3172 case Div:
3173 Py_INCREF(Div_singleton);
3174 return Div_singleton;
3175 case Mod:
3176 Py_INCREF(Mod_singleton);
3177 return Mod_singleton;
3178 case Pow:
3179 Py_INCREF(Pow_singleton);
3180 return Pow_singleton;
3181 case LShift:
3182 Py_INCREF(LShift_singleton);
3183 return LShift_singleton;
3184 case RShift:
3185 Py_INCREF(RShift_singleton);
3186 return RShift_singleton;
3187 case BitOr:
3188 Py_INCREF(BitOr_singleton);
3189 return BitOr_singleton;
3190 case BitXor:
3191 Py_INCREF(BitXor_singleton);
3192 return BitXor_singleton;
3193 case BitAnd:
3194 Py_INCREF(BitAnd_singleton);
3195 return BitAnd_singleton;
3196 case FloorDiv:
3197 Py_INCREF(FloorDiv_singleton);
3198 return FloorDiv_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003199 default:
3200 /* should never happen, but just in case ... */
3201 PyErr_Format(PyExc_SystemError, "unknown operator found");
3202 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003203 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003204}
3205PyObject* ast2obj_unaryop(unaryop_ty o)
3206{
3207 switch(o) {
3208 case Invert:
3209 Py_INCREF(Invert_singleton);
3210 return Invert_singleton;
3211 case Not:
3212 Py_INCREF(Not_singleton);
3213 return Not_singleton;
3214 case UAdd:
3215 Py_INCREF(UAdd_singleton);
3216 return UAdd_singleton;
3217 case USub:
3218 Py_INCREF(USub_singleton);
3219 return USub_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003220 default:
3221 /* should never happen, but just in case ... */
3222 PyErr_Format(PyExc_SystemError, "unknown unaryop found");
3223 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003224 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003225}
3226PyObject* ast2obj_cmpop(cmpop_ty o)
3227{
3228 switch(o) {
3229 case Eq:
3230 Py_INCREF(Eq_singleton);
3231 return Eq_singleton;
3232 case NotEq:
3233 Py_INCREF(NotEq_singleton);
3234 return NotEq_singleton;
3235 case Lt:
3236 Py_INCREF(Lt_singleton);
3237 return Lt_singleton;
3238 case LtE:
3239 Py_INCREF(LtE_singleton);
3240 return LtE_singleton;
3241 case Gt:
3242 Py_INCREF(Gt_singleton);
3243 return Gt_singleton;
3244 case GtE:
3245 Py_INCREF(GtE_singleton);
3246 return GtE_singleton;
3247 case Is:
3248 Py_INCREF(Is_singleton);
3249 return Is_singleton;
3250 case IsNot:
3251 Py_INCREF(IsNot_singleton);
3252 return IsNot_singleton;
3253 case In:
3254 Py_INCREF(In_singleton);
3255 return In_singleton;
3256 case NotIn:
3257 Py_INCREF(NotIn_singleton);
3258 return NotIn_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003259 default:
3260 /* should never happen, but just in case ... */
3261 PyErr_Format(PyExc_SystemError, "unknown cmpop found");
3262 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003263 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003264}
3265PyObject*
3266ast2obj_comprehension(void* _o)
3267{
3268 comprehension_ty o = (comprehension_ty)_o;
3269 PyObject *result = NULL, *value = NULL;
3270 if (!o) {
3271 Py_INCREF(Py_None);
3272 return Py_None;
3273 }
3274
3275 result = PyType_GenericNew(comprehension_type, NULL, NULL);
3276 if (!result) return NULL;
3277 value = ast2obj_expr(o->target);
3278 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003279 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003280 goto failed;
3281 Py_DECREF(value);
3282 value = ast2obj_expr(o->iter);
3283 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003284 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003285 goto failed;
3286 Py_DECREF(value);
3287 value = ast2obj_list(o->ifs, ast2obj_expr);
3288 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003289 if (_PyObject_SetAttrId(result, &PyId_ifs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003290 goto failed;
3291 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003292 return result;
3293failed:
3294 Py_XDECREF(value);
3295 Py_XDECREF(result);
3296 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003297}
3298
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003299PyObject*
3300ast2obj_excepthandler(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003301{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003302 excepthandler_ty o = (excepthandler_ty)_o;
3303 PyObject *result = NULL, *value = NULL;
3304 if (!o) {
3305 Py_INCREF(Py_None);
3306 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003307 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003308
Neal Norwitzad74aa82008-03-31 05:14:30 +00003309 switch (o->kind) {
3310 case ExceptHandler_kind:
3311 result = PyType_GenericNew(ExceptHandler_type, NULL, NULL);
3312 if (!result) goto failed;
3313 value = ast2obj_expr(o->v.ExceptHandler.type);
3314 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003315 if (_PyObject_SetAttrId(result, &PyId_type, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003316 goto failed;
3317 Py_DECREF(value);
3318 value = ast2obj_identifier(o->v.ExceptHandler.name);
3319 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003320 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003321 goto failed;
3322 Py_DECREF(value);
3323 value = ast2obj_list(o->v.ExceptHandler.body, ast2obj_stmt);
3324 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003325 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003326 goto failed;
3327 Py_DECREF(value);
3328 break;
3329 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003330 value = ast2obj_int(o->lineno);
3331 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003332 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003333 goto failed;
3334 Py_DECREF(value);
3335 value = ast2obj_int(o->col_offset);
3336 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003337 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003338 goto failed;
3339 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003340 return result;
3341failed:
3342 Py_XDECREF(value);
3343 Py_XDECREF(result);
3344 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003345}
3346
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003347PyObject*
3348ast2obj_arguments(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003349{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003350 arguments_ty o = (arguments_ty)_o;
3351 PyObject *result = NULL, *value = NULL;
3352 if (!o) {
3353 Py_INCREF(Py_None);
3354 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003355 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003356
3357 result = PyType_GenericNew(arguments_type, NULL, NULL);
3358 if (!result) return NULL;
Neal Norwitzc1505362006-12-28 06:47:50 +00003359 value = ast2obj_list(o->args, ast2obj_arg);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003360 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003361 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003362 goto failed;
3363 Py_DECREF(value);
3364 value = ast2obj_identifier(o->vararg);
3365 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003366 if (_PyObject_SetAttrId(result, &PyId_vararg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003367 goto failed;
3368 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003369 value = ast2obj_expr(o->varargannotation);
3370 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003371 if (_PyObject_SetAttrId(result, &PyId_varargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003372 goto failed;
3373 Py_DECREF(value);
3374 value = ast2obj_list(o->kwonlyargs, ast2obj_arg);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003375 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003376 if (_PyObject_SetAttrId(result, &PyId_kwonlyargs, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003377 goto failed;
3378 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003379 value = ast2obj_identifier(o->kwarg);
3380 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003381 if (_PyObject_SetAttrId(result, &PyId_kwarg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003382 goto failed;
3383 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003384 value = ast2obj_expr(o->kwargannotation);
3385 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003386 if (_PyObject_SetAttrId(result, &PyId_kwargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003387 goto failed;
3388 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003389 value = ast2obj_list(o->defaults, ast2obj_expr);
3390 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003391 if (_PyObject_SetAttrId(result, &PyId_defaults, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003392 goto failed;
3393 Py_DECREF(value);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003394 value = ast2obj_list(o->kw_defaults, ast2obj_expr);
3395 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003396 if (_PyObject_SetAttrId(result, &PyId_kw_defaults, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003397 goto failed;
3398 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003399 return result;
3400failed:
3401 Py_XDECREF(value);
3402 Py_XDECREF(result);
3403 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003404}
3405
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003406PyObject*
Neal Norwitzc1505362006-12-28 06:47:50 +00003407ast2obj_arg(void* _o)
3408{
3409 arg_ty o = (arg_ty)_o;
3410 PyObject *result = NULL, *value = NULL;
3411 if (!o) {
3412 Py_INCREF(Py_None);
3413 return Py_None;
3414 }
3415
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003416 result = PyType_GenericNew(arg_type, NULL, NULL);
3417 if (!result) return NULL;
3418 value = ast2obj_identifier(o->arg);
3419 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003420 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003421 goto failed;
3422 Py_DECREF(value);
3423 value = ast2obj_expr(o->annotation);
3424 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003425 if (_PyObject_SetAttrId(result, &PyId_annotation, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003426 goto failed;
3427 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003428 return result;
3429failed:
3430 Py_XDECREF(value);
3431 Py_XDECREF(result);
3432 return NULL;
3433}
3434
3435PyObject*
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003436ast2obj_keyword(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003437{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003438 keyword_ty o = (keyword_ty)_o;
3439 PyObject *result = NULL, *value = NULL;
3440 if (!o) {
3441 Py_INCREF(Py_None);
3442 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003443 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003444
3445 result = PyType_GenericNew(keyword_type, NULL, NULL);
3446 if (!result) return NULL;
3447 value = ast2obj_identifier(o->arg);
3448 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003449 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003450 goto failed;
3451 Py_DECREF(value);
3452 value = ast2obj_expr(o->value);
3453 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003454 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003455 goto failed;
3456 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003457 return result;
3458failed:
3459 Py_XDECREF(value);
3460 Py_XDECREF(result);
3461 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003462}
3463
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003464PyObject*
3465ast2obj_alias(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003466{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003467 alias_ty o = (alias_ty)_o;
3468 PyObject *result = NULL, *value = NULL;
3469 if (!o) {
3470 Py_INCREF(Py_None);
3471 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003472 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003473
3474 result = PyType_GenericNew(alias_type, NULL, NULL);
3475 if (!result) return NULL;
3476 value = ast2obj_identifier(o->name);
3477 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003478 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003479 goto failed;
3480 Py_DECREF(value);
3481 value = ast2obj_identifier(o->asname);
3482 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003483 if (_PyObject_SetAttrId(result, &PyId_asname, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003484 goto failed;
3485 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003486 return result;
3487failed:
3488 Py_XDECREF(value);
3489 Py_XDECREF(result);
3490 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003491}
3492
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003493PyObject*
3494ast2obj_withitem(void* _o)
3495{
3496 withitem_ty o = (withitem_ty)_o;
3497 PyObject *result = NULL, *value = NULL;
3498 if (!o) {
3499 Py_INCREF(Py_None);
3500 return Py_None;
3501 }
3502
3503 result = PyType_GenericNew(withitem_type, NULL, NULL);
3504 if (!result) return NULL;
3505 value = ast2obj_expr(o->context_expr);
3506 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003507 if (_PyObject_SetAttrId(result, &PyId_context_expr, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003508 goto failed;
3509 Py_DECREF(value);
3510 value = ast2obj_expr(o->optional_vars);
3511 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003512 if (_PyObject_SetAttrId(result, &PyId_optional_vars, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003513 goto failed;
3514 Py_DECREF(value);
3515 return result;
3516failed:
3517 Py_XDECREF(value);
3518 Py_XDECREF(result);
3519 return NULL;
3520}
3521
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003522
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003523int
3524obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena)
3525{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003526 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003527
Benjamin Petersond8f65972010-11-20 04:31:07 +00003528 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003529
3530 if (obj == Py_None) {
3531 *out = NULL;
3532 return 0;
3533 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003534 isinstance = PyObject_IsInstance(obj, (PyObject*)Module_type);
3535 if (isinstance == -1) {
3536 return 1;
3537 }
3538 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003539 asdl_seq* body;
3540
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003541 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003542 int res;
3543 Py_ssize_t len;
3544 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003545 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003546 if (tmp == NULL) goto failed;
3547 if (!PyList_Check(tmp)) {
3548 PyErr_Format(PyExc_TypeError, "Module field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3549 goto failed;
3550 }
3551 len = PyList_GET_SIZE(tmp);
3552 body = asdl_seq_new(len, arena);
3553 if (body == NULL) goto failed;
3554 for (i = 0; i < len; i++) {
3555 stmt_ty value;
3556 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3557 if (res != 0) goto failed;
3558 asdl_seq_SET(body, i, value);
3559 }
3560 Py_XDECREF(tmp);
3561 tmp = NULL;
3562 } else {
3563 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Module");
3564 return 1;
3565 }
3566 *out = Module(body, arena);
3567 if (*out == NULL) goto failed;
3568 return 0;
3569 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003570 isinstance = PyObject_IsInstance(obj, (PyObject*)Interactive_type);
3571 if (isinstance == -1) {
3572 return 1;
3573 }
3574 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003575 asdl_seq* body;
3576
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003577 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003578 int res;
3579 Py_ssize_t len;
3580 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003581 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003582 if (tmp == NULL) goto failed;
3583 if (!PyList_Check(tmp)) {
3584 PyErr_Format(PyExc_TypeError, "Interactive field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3585 goto failed;
3586 }
3587 len = PyList_GET_SIZE(tmp);
3588 body = asdl_seq_new(len, arena);
3589 if (body == NULL) goto failed;
3590 for (i = 0; i < len; i++) {
3591 stmt_ty value;
3592 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3593 if (res != 0) goto failed;
3594 asdl_seq_SET(body, i, value);
3595 }
3596 Py_XDECREF(tmp);
3597 tmp = NULL;
3598 } else {
3599 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Interactive");
3600 return 1;
3601 }
3602 *out = Interactive(body, arena);
3603 if (*out == NULL) goto failed;
3604 return 0;
3605 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003606 isinstance = PyObject_IsInstance(obj, (PyObject*)Expression_type);
3607 if (isinstance == -1) {
3608 return 1;
3609 }
3610 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003611 expr_ty body;
3612
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003613 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003614 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003615 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003616 if (tmp == NULL) goto failed;
3617 res = obj2ast_expr(tmp, &body, arena);
3618 if (res != 0) goto failed;
3619 Py_XDECREF(tmp);
3620 tmp = NULL;
3621 } else {
3622 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Expression");
3623 return 1;
3624 }
3625 *out = Expression(body, arena);
3626 if (*out == NULL) goto failed;
3627 return 0;
3628 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003629 isinstance = PyObject_IsInstance(obj, (PyObject*)Suite_type);
3630 if (isinstance == -1) {
3631 return 1;
3632 }
3633 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003634 asdl_seq* body;
3635
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003636 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003637 int res;
3638 Py_ssize_t len;
3639 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003640 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003641 if (tmp == NULL) goto failed;
3642 if (!PyList_Check(tmp)) {
3643 PyErr_Format(PyExc_TypeError, "Suite field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3644 goto failed;
3645 }
3646 len = PyList_GET_SIZE(tmp);
3647 body = asdl_seq_new(len, arena);
3648 if (body == NULL) goto failed;
3649 for (i = 0; i < len; i++) {
3650 stmt_ty value;
3651 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3652 if (res != 0) goto failed;
3653 asdl_seq_SET(body, i, value);
3654 }
3655 Py_XDECREF(tmp);
3656 tmp = NULL;
3657 } else {
3658 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Suite");
3659 return 1;
3660 }
3661 *out = Suite(body, arena);
3662 if (*out == NULL) goto failed;
3663 return 0;
3664 }
3665
Benjamin Peterson5b066812010-11-20 01:38:49 +00003666 PyErr_Format(PyExc_TypeError, "expected some sort of mod, but got %R", obj);
3667 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00003668 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003669 return 1;
3670}
3671
3672int
3673obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena)
3674{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003675 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003676
Benjamin Petersond8f65972010-11-20 04:31:07 +00003677 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003678 int lineno;
3679 int col_offset;
3680
3681 if (obj == Py_None) {
3682 *out = NULL;
3683 return 0;
3684 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003685 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003686 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003687 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003688 if (tmp == NULL) goto failed;
3689 res = obj2ast_int(tmp, &lineno, arena);
3690 if (res != 0) goto failed;
3691 Py_XDECREF(tmp);
3692 tmp = NULL;
3693 } else {
3694 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from stmt");
3695 return 1;
3696 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003697 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003698 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003699 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003700 if (tmp == NULL) goto failed;
3701 res = obj2ast_int(tmp, &col_offset, arena);
3702 if (res != 0) goto failed;
3703 Py_XDECREF(tmp);
3704 tmp = NULL;
3705 } else {
3706 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from stmt");
3707 return 1;
3708 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003709 isinstance = PyObject_IsInstance(obj, (PyObject*)FunctionDef_type);
3710 if (isinstance == -1) {
3711 return 1;
3712 }
3713 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003714 identifier name;
3715 arguments_ty args;
3716 asdl_seq* body;
3717 asdl_seq* decorator_list;
3718 expr_ty returns;
3719
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003720 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003721 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003722 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003723 if (tmp == NULL) goto failed;
3724 res = obj2ast_identifier(tmp, &name, arena);
3725 if (res != 0) goto failed;
3726 Py_XDECREF(tmp);
3727 tmp = NULL;
3728 } else {
3729 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from FunctionDef");
3730 return 1;
3731 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003732 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003733 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003734 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003735 if (tmp == NULL) goto failed;
3736 res = obj2ast_arguments(tmp, &args, arena);
3737 if (res != 0) goto failed;
3738 Py_XDECREF(tmp);
3739 tmp = NULL;
3740 } else {
3741 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from FunctionDef");
3742 return 1;
3743 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003744 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003745 int res;
3746 Py_ssize_t len;
3747 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003748 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003749 if (tmp == NULL) goto failed;
3750 if (!PyList_Check(tmp)) {
3751 PyErr_Format(PyExc_TypeError, "FunctionDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3752 goto failed;
3753 }
3754 len = PyList_GET_SIZE(tmp);
3755 body = asdl_seq_new(len, arena);
3756 if (body == NULL) goto failed;
3757 for (i = 0; i < len; i++) {
3758 stmt_ty value;
3759 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3760 if (res != 0) goto failed;
3761 asdl_seq_SET(body, i, value);
3762 }
3763 Py_XDECREF(tmp);
3764 tmp = NULL;
3765 } else {
3766 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from FunctionDef");
3767 return 1;
3768 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003769 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003770 int res;
3771 Py_ssize_t len;
3772 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003773 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003774 if (tmp == NULL) goto failed;
3775 if (!PyList_Check(tmp)) {
3776 PyErr_Format(PyExc_TypeError, "FunctionDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3777 goto failed;
3778 }
3779 len = PyList_GET_SIZE(tmp);
3780 decorator_list = asdl_seq_new(len, arena);
3781 if (decorator_list == NULL) goto failed;
3782 for (i = 0; i < len; i++) {
3783 expr_ty value;
3784 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3785 if (res != 0) goto failed;
3786 asdl_seq_SET(decorator_list, i, value);
3787 }
3788 Py_XDECREF(tmp);
3789 tmp = NULL;
3790 } else {
3791 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from FunctionDef");
3792 return 1;
3793 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003794 if (_PyObject_HasAttrId(obj, &PyId_returns)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003795 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003796 tmp = _PyObject_GetAttrId(obj, &PyId_returns);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003797 if (tmp == NULL) goto failed;
3798 res = obj2ast_expr(tmp, &returns, arena);
3799 if (res != 0) goto failed;
3800 Py_XDECREF(tmp);
3801 tmp = NULL;
3802 } else {
3803 returns = NULL;
3804 }
3805 *out = FunctionDef(name, args, body, decorator_list, returns,
3806 lineno, col_offset, arena);
3807 if (*out == NULL) goto failed;
3808 return 0;
3809 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003810 isinstance = PyObject_IsInstance(obj, (PyObject*)ClassDef_type);
3811 if (isinstance == -1) {
3812 return 1;
3813 }
3814 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003815 identifier name;
3816 asdl_seq* bases;
3817 asdl_seq* keywords;
3818 expr_ty starargs;
3819 expr_ty kwargs;
3820 asdl_seq* body;
3821 asdl_seq* decorator_list;
3822
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003823 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003824 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003825 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003826 if (tmp == NULL) goto failed;
3827 res = obj2ast_identifier(tmp, &name, arena);
3828 if (res != 0) goto failed;
3829 Py_XDECREF(tmp);
3830 tmp = NULL;
3831 } else {
3832 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from ClassDef");
3833 return 1;
3834 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003835 if (_PyObject_HasAttrId(obj, &PyId_bases)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003836 int res;
3837 Py_ssize_t len;
3838 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003839 tmp = _PyObject_GetAttrId(obj, &PyId_bases);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003840 if (tmp == NULL) goto failed;
3841 if (!PyList_Check(tmp)) {
3842 PyErr_Format(PyExc_TypeError, "ClassDef field \"bases\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3843 goto failed;
3844 }
3845 len = PyList_GET_SIZE(tmp);
3846 bases = asdl_seq_new(len, arena);
3847 if (bases == NULL) goto failed;
3848 for (i = 0; i < len; i++) {
3849 expr_ty value;
3850 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3851 if (res != 0) goto failed;
3852 asdl_seq_SET(bases, i, value);
3853 }
3854 Py_XDECREF(tmp);
3855 tmp = NULL;
3856 } else {
3857 PyErr_SetString(PyExc_TypeError, "required field \"bases\" missing from ClassDef");
3858 return 1;
3859 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003860 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003861 int res;
3862 Py_ssize_t len;
3863 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003864 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003865 if (tmp == NULL) goto failed;
3866 if (!PyList_Check(tmp)) {
3867 PyErr_Format(PyExc_TypeError, "ClassDef field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3868 goto failed;
3869 }
3870 len = PyList_GET_SIZE(tmp);
3871 keywords = asdl_seq_new(len, arena);
3872 if (keywords == NULL) goto failed;
3873 for (i = 0; i < len; i++) {
3874 keyword_ty value;
3875 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
3876 if (res != 0) goto failed;
3877 asdl_seq_SET(keywords, i, value);
3878 }
3879 Py_XDECREF(tmp);
3880 tmp = NULL;
3881 } else {
3882 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from ClassDef");
3883 return 1;
3884 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003885 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003886 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003887 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003888 if (tmp == NULL) goto failed;
3889 res = obj2ast_expr(tmp, &starargs, arena);
3890 if (res != 0) goto failed;
3891 Py_XDECREF(tmp);
3892 tmp = NULL;
3893 } else {
3894 starargs = NULL;
3895 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003896 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003897 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003898 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003899 if (tmp == NULL) goto failed;
3900 res = obj2ast_expr(tmp, &kwargs, arena);
3901 if (res != 0) goto failed;
3902 Py_XDECREF(tmp);
3903 tmp = NULL;
3904 } else {
3905 kwargs = NULL;
3906 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003907 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003908 int res;
3909 Py_ssize_t len;
3910 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003911 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003912 if (tmp == NULL) goto failed;
3913 if (!PyList_Check(tmp)) {
3914 PyErr_Format(PyExc_TypeError, "ClassDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3915 goto failed;
3916 }
3917 len = PyList_GET_SIZE(tmp);
3918 body = asdl_seq_new(len, arena);
3919 if (body == NULL) goto failed;
3920 for (i = 0; i < len; i++) {
3921 stmt_ty value;
3922 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3923 if (res != 0) goto failed;
3924 asdl_seq_SET(body, i, value);
3925 }
3926 Py_XDECREF(tmp);
3927 tmp = NULL;
3928 } else {
3929 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ClassDef");
3930 return 1;
3931 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003932 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003933 int res;
3934 Py_ssize_t len;
3935 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003936 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003937 if (tmp == NULL) goto failed;
3938 if (!PyList_Check(tmp)) {
3939 PyErr_Format(PyExc_TypeError, "ClassDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3940 goto failed;
3941 }
3942 len = PyList_GET_SIZE(tmp);
3943 decorator_list = asdl_seq_new(len, arena);
3944 if (decorator_list == NULL) goto failed;
3945 for (i = 0; i < len; i++) {
3946 expr_ty value;
3947 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3948 if (res != 0) goto failed;
3949 asdl_seq_SET(decorator_list, i, value);
3950 }
3951 Py_XDECREF(tmp);
3952 tmp = NULL;
3953 } else {
3954 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from ClassDef");
3955 return 1;
3956 }
3957 *out = ClassDef(name, bases, keywords, starargs, kwargs, body,
3958 decorator_list, lineno, col_offset, arena);
3959 if (*out == NULL) goto failed;
3960 return 0;
3961 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003962 isinstance = PyObject_IsInstance(obj, (PyObject*)Return_type);
3963 if (isinstance == -1) {
3964 return 1;
3965 }
3966 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003967 expr_ty value;
3968
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003969 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003970 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003971 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003972 if (tmp == NULL) goto failed;
3973 res = obj2ast_expr(tmp, &value, arena);
3974 if (res != 0) goto failed;
3975 Py_XDECREF(tmp);
3976 tmp = NULL;
3977 } else {
3978 value = NULL;
3979 }
3980 *out = Return(value, lineno, col_offset, arena);
3981 if (*out == NULL) goto failed;
3982 return 0;
3983 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003984 isinstance = PyObject_IsInstance(obj, (PyObject*)Delete_type);
3985 if (isinstance == -1) {
3986 return 1;
3987 }
3988 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003989 asdl_seq* targets;
3990
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003991 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003992 int res;
3993 Py_ssize_t len;
3994 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003995 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003996 if (tmp == NULL) goto failed;
3997 if (!PyList_Check(tmp)) {
3998 PyErr_Format(PyExc_TypeError, "Delete field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3999 goto failed;
4000 }
4001 len = PyList_GET_SIZE(tmp);
4002 targets = asdl_seq_new(len, arena);
4003 if (targets == NULL) goto failed;
4004 for (i = 0; i < len; i++) {
4005 expr_ty value;
4006 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4007 if (res != 0) goto failed;
4008 asdl_seq_SET(targets, i, value);
4009 }
4010 Py_XDECREF(tmp);
4011 tmp = NULL;
4012 } else {
4013 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Delete");
4014 return 1;
4015 }
4016 *out = Delete(targets, lineno, col_offset, arena);
4017 if (*out == NULL) goto failed;
4018 return 0;
4019 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004020 isinstance = PyObject_IsInstance(obj, (PyObject*)Assign_type);
4021 if (isinstance == -1) {
4022 return 1;
4023 }
4024 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004025 asdl_seq* targets;
4026 expr_ty value;
4027
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004028 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004029 int res;
4030 Py_ssize_t len;
4031 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004032 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004033 if (tmp == NULL) goto failed;
4034 if (!PyList_Check(tmp)) {
4035 PyErr_Format(PyExc_TypeError, "Assign field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4036 goto failed;
4037 }
4038 len = PyList_GET_SIZE(tmp);
4039 targets = asdl_seq_new(len, arena);
4040 if (targets == NULL) goto failed;
4041 for (i = 0; i < len; i++) {
4042 expr_ty value;
4043 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4044 if (res != 0) goto failed;
4045 asdl_seq_SET(targets, i, value);
4046 }
4047 Py_XDECREF(tmp);
4048 tmp = NULL;
4049 } else {
4050 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Assign");
4051 return 1;
4052 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004053 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004054 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004055 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004056 if (tmp == NULL) goto failed;
4057 res = obj2ast_expr(tmp, &value, arena);
4058 if (res != 0) goto failed;
4059 Py_XDECREF(tmp);
4060 tmp = NULL;
4061 } else {
4062 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Assign");
4063 return 1;
4064 }
4065 *out = Assign(targets, value, lineno, col_offset, arena);
4066 if (*out == NULL) goto failed;
4067 return 0;
4068 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004069 isinstance = PyObject_IsInstance(obj, (PyObject*)AugAssign_type);
4070 if (isinstance == -1) {
4071 return 1;
4072 }
4073 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004074 expr_ty target;
4075 operator_ty op;
4076 expr_ty value;
4077
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004078 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004079 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004080 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004081 if (tmp == NULL) goto failed;
4082 res = obj2ast_expr(tmp, &target, arena);
4083 if (res != 0) goto failed;
4084 Py_XDECREF(tmp);
4085 tmp = NULL;
4086 } else {
4087 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from AugAssign");
4088 return 1;
4089 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004090 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004091 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004092 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004093 if (tmp == NULL) goto failed;
4094 res = obj2ast_operator(tmp, &op, arena);
4095 if (res != 0) goto failed;
4096 Py_XDECREF(tmp);
4097 tmp = NULL;
4098 } else {
4099 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from AugAssign");
4100 return 1;
4101 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004102 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004103 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004104 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004105 if (tmp == NULL) goto failed;
4106 res = obj2ast_expr(tmp, &value, arena);
4107 if (res != 0) goto failed;
4108 Py_XDECREF(tmp);
4109 tmp = NULL;
4110 } else {
4111 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from AugAssign");
4112 return 1;
4113 }
4114 *out = AugAssign(target, op, value, lineno, col_offset, arena);
4115 if (*out == NULL) goto failed;
4116 return 0;
4117 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004118 isinstance = PyObject_IsInstance(obj, (PyObject*)For_type);
4119 if (isinstance == -1) {
4120 return 1;
4121 }
4122 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004123 expr_ty target;
4124 expr_ty iter;
4125 asdl_seq* body;
4126 asdl_seq* orelse;
4127
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004128 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004129 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004130 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004131 if (tmp == NULL) goto failed;
4132 res = obj2ast_expr(tmp, &target, arena);
4133 if (res != 0) goto failed;
4134 Py_XDECREF(tmp);
4135 tmp = NULL;
4136 } else {
4137 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from For");
4138 return 1;
4139 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004140 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004141 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004142 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004143 if (tmp == NULL) goto failed;
4144 res = obj2ast_expr(tmp, &iter, arena);
4145 if (res != 0) goto failed;
4146 Py_XDECREF(tmp);
4147 tmp = NULL;
4148 } else {
4149 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from For");
4150 return 1;
4151 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004152 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004153 int res;
4154 Py_ssize_t len;
4155 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004156 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004157 if (tmp == NULL) goto failed;
4158 if (!PyList_Check(tmp)) {
4159 PyErr_Format(PyExc_TypeError, "For field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4160 goto failed;
4161 }
4162 len = PyList_GET_SIZE(tmp);
4163 body = asdl_seq_new(len, arena);
4164 if (body == NULL) goto failed;
4165 for (i = 0; i < len; i++) {
4166 stmt_ty value;
4167 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4168 if (res != 0) goto failed;
4169 asdl_seq_SET(body, i, value);
4170 }
4171 Py_XDECREF(tmp);
4172 tmp = NULL;
4173 } else {
4174 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from For");
4175 return 1;
4176 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004177 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004178 int res;
4179 Py_ssize_t len;
4180 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004181 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004182 if (tmp == NULL) goto failed;
4183 if (!PyList_Check(tmp)) {
4184 PyErr_Format(PyExc_TypeError, "For field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4185 goto failed;
4186 }
4187 len = PyList_GET_SIZE(tmp);
4188 orelse = asdl_seq_new(len, arena);
4189 if (orelse == NULL) goto failed;
4190 for (i = 0; i < len; i++) {
4191 stmt_ty value;
4192 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4193 if (res != 0) goto failed;
4194 asdl_seq_SET(orelse, i, value);
4195 }
4196 Py_XDECREF(tmp);
4197 tmp = NULL;
4198 } else {
4199 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from For");
4200 return 1;
4201 }
4202 *out = For(target, iter, body, orelse, lineno, col_offset,
4203 arena);
4204 if (*out == NULL) goto failed;
4205 return 0;
4206 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004207 isinstance = PyObject_IsInstance(obj, (PyObject*)While_type);
4208 if (isinstance == -1) {
4209 return 1;
4210 }
4211 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004212 expr_ty test;
4213 asdl_seq* body;
4214 asdl_seq* orelse;
4215
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004216 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004217 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004218 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004219 if (tmp == NULL) goto failed;
4220 res = obj2ast_expr(tmp, &test, arena);
4221 if (res != 0) goto failed;
4222 Py_XDECREF(tmp);
4223 tmp = NULL;
4224 } else {
4225 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from While");
4226 return 1;
4227 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004228 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004229 int res;
4230 Py_ssize_t len;
4231 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004232 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004233 if (tmp == NULL) goto failed;
4234 if (!PyList_Check(tmp)) {
4235 PyErr_Format(PyExc_TypeError, "While field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4236 goto failed;
4237 }
4238 len = PyList_GET_SIZE(tmp);
4239 body = asdl_seq_new(len, arena);
4240 if (body == NULL) goto failed;
4241 for (i = 0; i < len; i++) {
4242 stmt_ty value;
4243 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4244 if (res != 0) goto failed;
4245 asdl_seq_SET(body, i, value);
4246 }
4247 Py_XDECREF(tmp);
4248 tmp = NULL;
4249 } else {
4250 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from While");
4251 return 1;
4252 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004253 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004254 int res;
4255 Py_ssize_t len;
4256 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004257 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004258 if (tmp == NULL) goto failed;
4259 if (!PyList_Check(tmp)) {
4260 PyErr_Format(PyExc_TypeError, "While field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4261 goto failed;
4262 }
4263 len = PyList_GET_SIZE(tmp);
4264 orelse = asdl_seq_new(len, arena);
4265 if (orelse == NULL) goto failed;
4266 for (i = 0; i < len; i++) {
4267 stmt_ty value;
4268 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4269 if (res != 0) goto failed;
4270 asdl_seq_SET(orelse, i, value);
4271 }
4272 Py_XDECREF(tmp);
4273 tmp = NULL;
4274 } else {
4275 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from While");
4276 return 1;
4277 }
4278 *out = While(test, body, orelse, lineno, col_offset, arena);
4279 if (*out == NULL) goto failed;
4280 return 0;
4281 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004282 isinstance = PyObject_IsInstance(obj, (PyObject*)If_type);
4283 if (isinstance == -1) {
4284 return 1;
4285 }
4286 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004287 expr_ty test;
4288 asdl_seq* body;
4289 asdl_seq* orelse;
4290
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004291 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004292 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004293 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004294 if (tmp == NULL) goto failed;
4295 res = obj2ast_expr(tmp, &test, arena);
4296 if (res != 0) goto failed;
4297 Py_XDECREF(tmp);
4298 tmp = NULL;
4299 } else {
4300 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from If");
4301 return 1;
4302 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004303 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004304 int res;
4305 Py_ssize_t len;
4306 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004307 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004308 if (tmp == NULL) goto failed;
4309 if (!PyList_Check(tmp)) {
4310 PyErr_Format(PyExc_TypeError, "If field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4311 goto failed;
4312 }
4313 len = PyList_GET_SIZE(tmp);
4314 body = asdl_seq_new(len, arena);
4315 if (body == NULL) goto failed;
4316 for (i = 0; i < len; i++) {
4317 stmt_ty value;
4318 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4319 if (res != 0) goto failed;
4320 asdl_seq_SET(body, i, value);
4321 }
4322 Py_XDECREF(tmp);
4323 tmp = NULL;
4324 } else {
4325 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from If");
4326 return 1;
4327 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004328 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004329 int res;
4330 Py_ssize_t len;
4331 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004332 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004333 if (tmp == NULL) goto failed;
4334 if (!PyList_Check(tmp)) {
4335 PyErr_Format(PyExc_TypeError, "If field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4336 goto failed;
4337 }
4338 len = PyList_GET_SIZE(tmp);
4339 orelse = asdl_seq_new(len, arena);
4340 if (orelse == NULL) goto failed;
4341 for (i = 0; i < len; i++) {
4342 stmt_ty value;
4343 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4344 if (res != 0) goto failed;
4345 asdl_seq_SET(orelse, i, value);
4346 }
4347 Py_XDECREF(tmp);
4348 tmp = NULL;
4349 } else {
4350 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from If");
4351 return 1;
4352 }
4353 *out = If(test, body, orelse, lineno, col_offset, arena);
4354 if (*out == NULL) goto failed;
4355 return 0;
4356 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004357 isinstance = PyObject_IsInstance(obj, (PyObject*)With_type);
4358 if (isinstance == -1) {
4359 return 1;
4360 }
4361 if (isinstance) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004362 asdl_seq* items;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004363 asdl_seq* body;
4364
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004365 if (_PyObject_HasAttrId(obj, &PyId_items)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004366 int res;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004367 Py_ssize_t len;
4368 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004369 tmp = _PyObject_GetAttrId(obj, &PyId_items);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004370 if (tmp == NULL) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004371 if (!PyList_Check(tmp)) {
4372 PyErr_Format(PyExc_TypeError, "With field \"items\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4373 goto failed;
4374 }
4375 len = PyList_GET_SIZE(tmp);
4376 items = asdl_seq_new(len, arena);
4377 if (items == NULL) goto failed;
4378 for (i = 0; i < len; i++) {
4379 withitem_ty value;
4380 res = obj2ast_withitem(PyList_GET_ITEM(tmp, i), &value, arena);
4381 if (res != 0) goto failed;
4382 asdl_seq_SET(items, i, value);
4383 }
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004384 Py_XDECREF(tmp);
4385 tmp = NULL;
4386 } else {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004387 PyErr_SetString(PyExc_TypeError, "required field \"items\" missing from With");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004388 return 1;
4389 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004390 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004391 int res;
4392 Py_ssize_t len;
4393 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004394 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004395 if (tmp == NULL) goto failed;
4396 if (!PyList_Check(tmp)) {
4397 PyErr_Format(PyExc_TypeError, "With field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4398 goto failed;
4399 }
4400 len = PyList_GET_SIZE(tmp);
4401 body = asdl_seq_new(len, arena);
4402 if (body == NULL) goto failed;
4403 for (i = 0; i < len; i++) {
4404 stmt_ty value;
4405 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4406 if (res != 0) goto failed;
4407 asdl_seq_SET(body, i, value);
4408 }
4409 Py_XDECREF(tmp);
4410 tmp = NULL;
4411 } else {
4412 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from With");
4413 return 1;
4414 }
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004415 *out = With(items, body, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004416 if (*out == NULL) goto failed;
4417 return 0;
4418 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004419 isinstance = PyObject_IsInstance(obj, (PyObject*)Raise_type);
4420 if (isinstance == -1) {
4421 return 1;
4422 }
4423 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004424 expr_ty exc;
4425 expr_ty cause;
4426
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004427 if (_PyObject_HasAttrId(obj, &PyId_exc)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004428 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004429 tmp = _PyObject_GetAttrId(obj, &PyId_exc);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004430 if (tmp == NULL) goto failed;
4431 res = obj2ast_expr(tmp, &exc, arena);
4432 if (res != 0) goto failed;
4433 Py_XDECREF(tmp);
4434 tmp = NULL;
4435 } else {
4436 exc = NULL;
4437 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004438 if (_PyObject_HasAttrId(obj, &PyId_cause)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004439 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004440 tmp = _PyObject_GetAttrId(obj, &PyId_cause);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004441 if (tmp == NULL) goto failed;
4442 res = obj2ast_expr(tmp, &cause, arena);
4443 if (res != 0) goto failed;
4444 Py_XDECREF(tmp);
4445 tmp = NULL;
4446 } else {
4447 cause = NULL;
4448 }
4449 *out = Raise(exc, cause, lineno, col_offset, arena);
4450 if (*out == NULL) goto failed;
4451 return 0;
4452 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004453 isinstance = PyObject_IsInstance(obj, (PyObject*)Try_type);
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004454 if (isinstance == -1) {
4455 return 1;
4456 }
4457 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004458 asdl_seq* body;
4459 asdl_seq* handlers;
4460 asdl_seq* orelse;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004461 asdl_seq* finalbody;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004462
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004463 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004464 int res;
4465 Py_ssize_t len;
4466 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004467 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004468 if (tmp == NULL) goto failed;
4469 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004470 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 +00004471 goto failed;
4472 }
4473 len = PyList_GET_SIZE(tmp);
4474 body = asdl_seq_new(len, arena);
4475 if (body == NULL) goto failed;
4476 for (i = 0; i < len; i++) {
4477 stmt_ty value;
4478 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4479 if (res != 0) goto failed;
4480 asdl_seq_SET(body, i, value);
4481 }
4482 Py_XDECREF(tmp);
4483 tmp = NULL;
4484 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004485 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004486 return 1;
4487 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004488 if (_PyObject_HasAttrId(obj, &PyId_handlers)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004489 int res;
4490 Py_ssize_t len;
4491 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004492 tmp = _PyObject_GetAttrId(obj, &PyId_handlers);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004493 if (tmp == NULL) goto failed;
4494 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004495 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 +00004496 goto failed;
4497 }
4498 len = PyList_GET_SIZE(tmp);
4499 handlers = asdl_seq_new(len, arena);
4500 if (handlers == NULL) goto failed;
4501 for (i = 0; i < len; i++) {
4502 excepthandler_ty value;
4503 res = obj2ast_excepthandler(PyList_GET_ITEM(tmp, i), &value, arena);
4504 if (res != 0) goto failed;
4505 asdl_seq_SET(handlers, i, value);
4506 }
4507 Py_XDECREF(tmp);
4508 tmp = NULL;
4509 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004510 PyErr_SetString(PyExc_TypeError, "required field \"handlers\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004511 return 1;
4512 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004513 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004514 int res;
4515 Py_ssize_t len;
4516 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004517 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004518 if (tmp == NULL) goto failed;
4519 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004520 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 +00004521 goto failed;
4522 }
4523 len = PyList_GET_SIZE(tmp);
4524 orelse = asdl_seq_new(len, arena);
4525 if (orelse == NULL) goto failed;
4526 for (i = 0; i < len; i++) {
4527 stmt_ty value;
4528 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4529 if (res != 0) goto failed;
4530 asdl_seq_SET(orelse, i, value);
4531 }
4532 Py_XDECREF(tmp);
4533 tmp = NULL;
4534 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004535 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004536 return 1;
4537 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004538 if (_PyObject_HasAttrId(obj, &PyId_finalbody)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004539 int res;
4540 Py_ssize_t len;
4541 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004542 tmp = _PyObject_GetAttrId(obj, &PyId_finalbody);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004543 if (tmp == NULL) goto failed;
4544 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004545 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 +00004546 goto failed;
4547 }
4548 len = PyList_GET_SIZE(tmp);
4549 finalbody = asdl_seq_new(len, arena);
4550 if (finalbody == NULL) goto failed;
4551 for (i = 0; i < len; i++) {
4552 stmt_ty value;
4553 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4554 if (res != 0) goto failed;
4555 asdl_seq_SET(finalbody, i, value);
4556 }
4557 Py_XDECREF(tmp);
4558 tmp = NULL;
4559 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004560 PyErr_SetString(PyExc_TypeError, "required field \"finalbody\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004561 return 1;
4562 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004563 *out = Try(body, handlers, orelse, finalbody, lineno,
4564 col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004565 if (*out == NULL) goto failed;
4566 return 0;
4567 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004568 isinstance = PyObject_IsInstance(obj, (PyObject*)Assert_type);
4569 if (isinstance == -1) {
4570 return 1;
4571 }
4572 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004573 expr_ty test;
4574 expr_ty msg;
4575
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004576 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004577 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004578 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004579 if (tmp == NULL) goto failed;
4580 res = obj2ast_expr(tmp, &test, arena);
4581 if (res != 0) goto failed;
4582 Py_XDECREF(tmp);
4583 tmp = NULL;
4584 } else {
4585 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from Assert");
4586 return 1;
4587 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004588 if (_PyObject_HasAttrId(obj, &PyId_msg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004589 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004590 tmp = _PyObject_GetAttrId(obj, &PyId_msg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004591 if (tmp == NULL) goto failed;
4592 res = obj2ast_expr(tmp, &msg, arena);
4593 if (res != 0) goto failed;
4594 Py_XDECREF(tmp);
4595 tmp = NULL;
4596 } else {
4597 msg = NULL;
4598 }
4599 *out = Assert(test, msg, lineno, col_offset, arena);
4600 if (*out == NULL) goto failed;
4601 return 0;
4602 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004603 isinstance = PyObject_IsInstance(obj, (PyObject*)Import_type);
4604 if (isinstance == -1) {
4605 return 1;
4606 }
4607 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004608 asdl_seq* names;
4609
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004610 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004611 int res;
4612 Py_ssize_t len;
4613 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004614 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004615 if (tmp == NULL) goto failed;
4616 if (!PyList_Check(tmp)) {
4617 PyErr_Format(PyExc_TypeError, "Import field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4618 goto failed;
4619 }
4620 len = PyList_GET_SIZE(tmp);
4621 names = asdl_seq_new(len, arena);
4622 if (names == NULL) goto failed;
4623 for (i = 0; i < len; i++) {
4624 alias_ty value;
4625 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4626 if (res != 0) goto failed;
4627 asdl_seq_SET(names, i, value);
4628 }
4629 Py_XDECREF(tmp);
4630 tmp = NULL;
4631 } else {
4632 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Import");
4633 return 1;
4634 }
4635 *out = Import(names, lineno, col_offset, arena);
4636 if (*out == NULL) goto failed;
4637 return 0;
4638 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004639 isinstance = PyObject_IsInstance(obj, (PyObject*)ImportFrom_type);
4640 if (isinstance == -1) {
4641 return 1;
4642 }
4643 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004644 identifier module;
4645 asdl_seq* names;
4646 int level;
4647
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004648 if (_PyObject_HasAttrId(obj, &PyId_module)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004649 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004650 tmp = _PyObject_GetAttrId(obj, &PyId_module);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004651 if (tmp == NULL) goto failed;
4652 res = obj2ast_identifier(tmp, &module, arena);
4653 if (res != 0) goto failed;
4654 Py_XDECREF(tmp);
4655 tmp = NULL;
4656 } else {
Benjamin Peterson78565b22009-06-28 19:19:51 +00004657 module = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004658 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004659 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004660 int res;
4661 Py_ssize_t len;
4662 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004663 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004664 if (tmp == NULL) goto failed;
4665 if (!PyList_Check(tmp)) {
4666 PyErr_Format(PyExc_TypeError, "ImportFrom field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4667 goto failed;
4668 }
4669 len = PyList_GET_SIZE(tmp);
4670 names = asdl_seq_new(len, arena);
4671 if (names == NULL) goto failed;
4672 for (i = 0; i < len; i++) {
4673 alias_ty value;
4674 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4675 if (res != 0) goto failed;
4676 asdl_seq_SET(names, i, value);
4677 }
4678 Py_XDECREF(tmp);
4679 tmp = NULL;
4680 } else {
4681 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from ImportFrom");
4682 return 1;
4683 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004684 if (_PyObject_HasAttrId(obj, &PyId_level)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004685 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004686 tmp = _PyObject_GetAttrId(obj, &PyId_level);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004687 if (tmp == NULL) goto failed;
4688 res = obj2ast_int(tmp, &level, arena);
4689 if (res != 0) goto failed;
4690 Py_XDECREF(tmp);
4691 tmp = NULL;
4692 } else {
4693 level = 0;
4694 }
4695 *out = ImportFrom(module, names, level, lineno, col_offset,
4696 arena);
4697 if (*out == NULL) goto failed;
4698 return 0;
4699 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004700 isinstance = PyObject_IsInstance(obj, (PyObject*)Global_type);
4701 if (isinstance == -1) {
4702 return 1;
4703 }
4704 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004705 asdl_seq* names;
4706
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004707 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004708 int res;
4709 Py_ssize_t len;
4710 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004711 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004712 if (tmp == NULL) goto failed;
4713 if (!PyList_Check(tmp)) {
4714 PyErr_Format(PyExc_TypeError, "Global field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4715 goto failed;
4716 }
4717 len = PyList_GET_SIZE(tmp);
4718 names = asdl_seq_new(len, arena);
4719 if (names == NULL) goto failed;
4720 for (i = 0; i < len; i++) {
4721 identifier value;
4722 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4723 if (res != 0) goto failed;
4724 asdl_seq_SET(names, i, value);
4725 }
4726 Py_XDECREF(tmp);
4727 tmp = NULL;
4728 } else {
4729 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Global");
4730 return 1;
4731 }
4732 *out = Global(names, lineno, col_offset, arena);
4733 if (*out == NULL) goto failed;
4734 return 0;
4735 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004736 isinstance = PyObject_IsInstance(obj, (PyObject*)Nonlocal_type);
4737 if (isinstance == -1) {
4738 return 1;
4739 }
4740 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004741 asdl_seq* names;
4742
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004743 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004744 int res;
4745 Py_ssize_t len;
4746 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004747 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004748 if (tmp == NULL) goto failed;
4749 if (!PyList_Check(tmp)) {
4750 PyErr_Format(PyExc_TypeError, "Nonlocal field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4751 goto failed;
4752 }
4753 len = PyList_GET_SIZE(tmp);
4754 names = asdl_seq_new(len, arena);
4755 if (names == NULL) goto failed;
4756 for (i = 0; i < len; i++) {
4757 identifier value;
4758 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4759 if (res != 0) goto failed;
4760 asdl_seq_SET(names, i, value);
4761 }
4762 Py_XDECREF(tmp);
4763 tmp = NULL;
4764 } else {
4765 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Nonlocal");
4766 return 1;
4767 }
4768 *out = Nonlocal(names, lineno, col_offset, arena);
4769 if (*out == NULL) goto failed;
4770 return 0;
4771 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004772 isinstance = PyObject_IsInstance(obj, (PyObject*)Expr_type);
4773 if (isinstance == -1) {
4774 return 1;
4775 }
4776 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004777 expr_ty value;
4778
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004779 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004780 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004781 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004782 if (tmp == NULL) goto failed;
4783 res = obj2ast_expr(tmp, &value, arena);
4784 if (res != 0) goto failed;
4785 Py_XDECREF(tmp);
4786 tmp = NULL;
4787 } else {
4788 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Expr");
4789 return 1;
4790 }
4791 *out = Expr(value, lineno, col_offset, arena);
4792 if (*out == NULL) goto failed;
4793 return 0;
4794 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004795 isinstance = PyObject_IsInstance(obj, (PyObject*)Pass_type);
4796 if (isinstance == -1) {
4797 return 1;
4798 }
4799 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004800
4801 *out = Pass(lineno, col_offset, arena);
4802 if (*out == NULL) goto failed;
4803 return 0;
4804 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004805 isinstance = PyObject_IsInstance(obj, (PyObject*)Break_type);
4806 if (isinstance == -1) {
4807 return 1;
4808 }
4809 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004810
4811 *out = Break(lineno, col_offset, arena);
4812 if (*out == NULL) goto failed;
4813 return 0;
4814 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004815 isinstance = PyObject_IsInstance(obj, (PyObject*)Continue_type);
4816 if (isinstance == -1) {
4817 return 1;
4818 }
4819 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004820
4821 *out = Continue(lineno, col_offset, arena);
4822 if (*out == NULL) goto failed;
4823 return 0;
4824 }
4825
Benjamin Peterson5b066812010-11-20 01:38:49 +00004826 PyErr_Format(PyExc_TypeError, "expected some sort of stmt, but got %R", obj);
4827 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00004828 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004829 return 1;
4830}
4831
4832int
4833obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena)
4834{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004835 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004836
Benjamin Petersond8f65972010-11-20 04:31:07 +00004837 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004838 int lineno;
4839 int col_offset;
4840
4841 if (obj == Py_None) {
4842 *out = NULL;
4843 return 0;
4844 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004845 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004846 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004847 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004848 if (tmp == NULL) goto failed;
4849 res = obj2ast_int(tmp, &lineno, arena);
4850 if (res != 0) goto failed;
4851 Py_XDECREF(tmp);
4852 tmp = NULL;
4853 } else {
4854 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from expr");
4855 return 1;
4856 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004857 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004858 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004859 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004860 if (tmp == NULL) goto failed;
4861 res = obj2ast_int(tmp, &col_offset, arena);
4862 if (res != 0) goto failed;
4863 Py_XDECREF(tmp);
4864 tmp = NULL;
4865 } else {
4866 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from expr");
4867 return 1;
4868 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004869 isinstance = PyObject_IsInstance(obj, (PyObject*)BoolOp_type);
4870 if (isinstance == -1) {
4871 return 1;
4872 }
4873 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004874 boolop_ty op;
4875 asdl_seq* values;
4876
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004877 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004878 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004879 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004880 if (tmp == NULL) goto failed;
4881 res = obj2ast_boolop(tmp, &op, arena);
4882 if (res != 0) goto failed;
4883 Py_XDECREF(tmp);
4884 tmp = NULL;
4885 } else {
4886 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BoolOp");
4887 return 1;
4888 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004889 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004890 int res;
4891 Py_ssize_t len;
4892 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004893 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004894 if (tmp == NULL) goto failed;
4895 if (!PyList_Check(tmp)) {
4896 PyErr_Format(PyExc_TypeError, "BoolOp field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4897 goto failed;
4898 }
4899 len = PyList_GET_SIZE(tmp);
4900 values = asdl_seq_new(len, arena);
4901 if (values == NULL) goto failed;
4902 for (i = 0; i < len; i++) {
4903 expr_ty value;
4904 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4905 if (res != 0) goto failed;
4906 asdl_seq_SET(values, i, value);
4907 }
4908 Py_XDECREF(tmp);
4909 tmp = NULL;
4910 } else {
4911 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from BoolOp");
4912 return 1;
4913 }
4914 *out = BoolOp(op, values, lineno, col_offset, arena);
4915 if (*out == NULL) goto failed;
4916 return 0;
4917 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004918 isinstance = PyObject_IsInstance(obj, (PyObject*)BinOp_type);
4919 if (isinstance == -1) {
4920 return 1;
4921 }
4922 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004923 expr_ty left;
4924 operator_ty op;
4925 expr_ty right;
4926
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004927 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004928 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004929 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004930 if (tmp == NULL) goto failed;
4931 res = obj2ast_expr(tmp, &left, arena);
4932 if (res != 0) goto failed;
4933 Py_XDECREF(tmp);
4934 tmp = NULL;
4935 } else {
4936 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from BinOp");
4937 return 1;
4938 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004939 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004940 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004941 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004942 if (tmp == NULL) goto failed;
4943 res = obj2ast_operator(tmp, &op, arena);
4944 if (res != 0) goto failed;
4945 Py_XDECREF(tmp);
4946 tmp = NULL;
4947 } else {
4948 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BinOp");
4949 return 1;
4950 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004951 if (_PyObject_HasAttrId(obj, &PyId_right)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004952 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004953 tmp = _PyObject_GetAttrId(obj, &PyId_right);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004954 if (tmp == NULL) goto failed;
4955 res = obj2ast_expr(tmp, &right, arena);
4956 if (res != 0) goto failed;
4957 Py_XDECREF(tmp);
4958 tmp = NULL;
4959 } else {
4960 PyErr_SetString(PyExc_TypeError, "required field \"right\" missing from BinOp");
4961 return 1;
4962 }
4963 *out = BinOp(left, op, right, lineno, col_offset, arena);
4964 if (*out == NULL) goto failed;
4965 return 0;
4966 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004967 isinstance = PyObject_IsInstance(obj, (PyObject*)UnaryOp_type);
4968 if (isinstance == -1) {
4969 return 1;
4970 }
4971 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004972 unaryop_ty op;
4973 expr_ty operand;
4974
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004975 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004976 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004977 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004978 if (tmp == NULL) goto failed;
4979 res = obj2ast_unaryop(tmp, &op, arena);
4980 if (res != 0) goto failed;
4981 Py_XDECREF(tmp);
4982 tmp = NULL;
4983 } else {
4984 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from UnaryOp");
4985 return 1;
4986 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004987 if (_PyObject_HasAttrId(obj, &PyId_operand)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004988 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004989 tmp = _PyObject_GetAttrId(obj, &PyId_operand);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004990 if (tmp == NULL) goto failed;
4991 res = obj2ast_expr(tmp, &operand, arena);
4992 if (res != 0) goto failed;
4993 Py_XDECREF(tmp);
4994 tmp = NULL;
4995 } else {
4996 PyErr_SetString(PyExc_TypeError, "required field \"operand\" missing from UnaryOp");
4997 return 1;
4998 }
4999 *out = UnaryOp(op, operand, lineno, col_offset, arena);
5000 if (*out == NULL) goto failed;
5001 return 0;
5002 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005003 isinstance = PyObject_IsInstance(obj, (PyObject*)Lambda_type);
5004 if (isinstance == -1) {
5005 return 1;
5006 }
5007 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005008 arguments_ty args;
5009 expr_ty body;
5010
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005011 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005012 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005013 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005014 if (tmp == NULL) goto failed;
5015 res = obj2ast_arguments(tmp, &args, arena);
5016 if (res != 0) goto failed;
5017 Py_XDECREF(tmp);
5018 tmp = NULL;
5019 } else {
5020 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Lambda");
5021 return 1;
5022 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005023 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005024 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005025 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005026 if (tmp == NULL) goto failed;
5027 res = obj2ast_expr(tmp, &body, arena);
5028 if (res != 0) goto failed;
5029 Py_XDECREF(tmp);
5030 tmp = NULL;
5031 } else {
5032 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Lambda");
5033 return 1;
5034 }
5035 *out = Lambda(args, body, lineno, col_offset, arena);
5036 if (*out == NULL) goto failed;
5037 return 0;
5038 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005039 isinstance = PyObject_IsInstance(obj, (PyObject*)IfExp_type);
5040 if (isinstance == -1) {
5041 return 1;
5042 }
5043 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005044 expr_ty test;
5045 expr_ty body;
5046 expr_ty orelse;
5047
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005048 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005049 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005050 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005051 if (tmp == NULL) goto failed;
5052 res = obj2ast_expr(tmp, &test, arena);
5053 if (res != 0) goto failed;
5054 Py_XDECREF(tmp);
5055 tmp = NULL;
5056 } else {
5057 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from IfExp");
5058 return 1;
5059 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005060 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005061 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005062 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005063 if (tmp == NULL) goto failed;
5064 res = obj2ast_expr(tmp, &body, arena);
5065 if (res != 0) goto failed;
5066 Py_XDECREF(tmp);
5067 tmp = NULL;
5068 } else {
5069 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from IfExp");
5070 return 1;
5071 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005072 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005073 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005074 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005075 if (tmp == NULL) goto failed;
5076 res = obj2ast_expr(tmp, &orelse, arena);
5077 if (res != 0) goto failed;
5078 Py_XDECREF(tmp);
5079 tmp = NULL;
5080 } else {
5081 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from IfExp");
5082 return 1;
5083 }
5084 *out = IfExp(test, body, orelse, lineno, col_offset, arena);
5085 if (*out == NULL) goto failed;
5086 return 0;
5087 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005088 isinstance = PyObject_IsInstance(obj, (PyObject*)Dict_type);
5089 if (isinstance == -1) {
5090 return 1;
5091 }
5092 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005093 asdl_seq* keys;
5094 asdl_seq* values;
5095
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005096 if (_PyObject_HasAttrId(obj, &PyId_keys)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005097 int res;
5098 Py_ssize_t len;
5099 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005100 tmp = _PyObject_GetAttrId(obj, &PyId_keys);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005101 if (tmp == NULL) goto failed;
5102 if (!PyList_Check(tmp)) {
5103 PyErr_Format(PyExc_TypeError, "Dict field \"keys\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5104 goto failed;
5105 }
5106 len = PyList_GET_SIZE(tmp);
5107 keys = asdl_seq_new(len, arena);
5108 if (keys == NULL) goto failed;
5109 for (i = 0; i < len; i++) {
5110 expr_ty value;
5111 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5112 if (res != 0) goto failed;
5113 asdl_seq_SET(keys, i, value);
5114 }
5115 Py_XDECREF(tmp);
5116 tmp = NULL;
5117 } else {
5118 PyErr_SetString(PyExc_TypeError, "required field \"keys\" missing from Dict");
5119 return 1;
5120 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005121 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005122 int res;
5123 Py_ssize_t len;
5124 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005125 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005126 if (tmp == NULL) goto failed;
5127 if (!PyList_Check(tmp)) {
5128 PyErr_Format(PyExc_TypeError, "Dict field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5129 goto failed;
5130 }
5131 len = PyList_GET_SIZE(tmp);
5132 values = asdl_seq_new(len, arena);
5133 if (values == NULL) goto failed;
5134 for (i = 0; i < len; i++) {
5135 expr_ty value;
5136 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5137 if (res != 0) goto failed;
5138 asdl_seq_SET(values, i, value);
5139 }
5140 Py_XDECREF(tmp);
5141 tmp = NULL;
5142 } else {
5143 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from Dict");
5144 return 1;
5145 }
5146 *out = Dict(keys, values, lineno, col_offset, arena);
5147 if (*out == NULL) goto failed;
5148 return 0;
5149 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005150 isinstance = PyObject_IsInstance(obj, (PyObject*)Set_type);
5151 if (isinstance == -1) {
5152 return 1;
5153 }
5154 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005155 asdl_seq* elts;
5156
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005157 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005158 int res;
5159 Py_ssize_t len;
5160 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005161 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005162 if (tmp == NULL) goto failed;
5163 if (!PyList_Check(tmp)) {
5164 PyErr_Format(PyExc_TypeError, "Set field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5165 goto failed;
5166 }
5167 len = PyList_GET_SIZE(tmp);
5168 elts = asdl_seq_new(len, arena);
5169 if (elts == NULL) goto failed;
5170 for (i = 0; i < len; i++) {
5171 expr_ty value;
5172 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5173 if (res != 0) goto failed;
5174 asdl_seq_SET(elts, i, value);
5175 }
5176 Py_XDECREF(tmp);
5177 tmp = NULL;
5178 } else {
5179 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Set");
5180 return 1;
5181 }
5182 *out = Set(elts, lineno, col_offset, arena);
5183 if (*out == NULL) goto failed;
5184 return 0;
5185 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005186 isinstance = PyObject_IsInstance(obj, (PyObject*)ListComp_type);
5187 if (isinstance == -1) {
5188 return 1;
5189 }
5190 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005191 expr_ty elt;
5192 asdl_seq* generators;
5193
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005194 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005195 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005196 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005197 if (tmp == NULL) goto failed;
5198 res = obj2ast_expr(tmp, &elt, arena);
5199 if (res != 0) goto failed;
5200 Py_XDECREF(tmp);
5201 tmp = NULL;
5202 } else {
5203 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from ListComp");
5204 return 1;
5205 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005206 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005207 int res;
5208 Py_ssize_t len;
5209 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005210 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005211 if (tmp == NULL) goto failed;
5212 if (!PyList_Check(tmp)) {
5213 PyErr_Format(PyExc_TypeError, "ListComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5214 goto failed;
5215 }
5216 len = PyList_GET_SIZE(tmp);
5217 generators = asdl_seq_new(len, arena);
5218 if (generators == NULL) goto failed;
5219 for (i = 0; i < len; i++) {
5220 comprehension_ty value;
5221 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5222 if (res != 0) goto failed;
5223 asdl_seq_SET(generators, i, value);
5224 }
5225 Py_XDECREF(tmp);
5226 tmp = NULL;
5227 } else {
5228 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from ListComp");
5229 return 1;
5230 }
5231 *out = ListComp(elt, generators, lineno, col_offset, arena);
5232 if (*out == NULL) goto failed;
5233 return 0;
5234 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005235 isinstance = PyObject_IsInstance(obj, (PyObject*)SetComp_type);
5236 if (isinstance == -1) {
5237 return 1;
5238 }
5239 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005240 expr_ty elt;
5241 asdl_seq* generators;
5242
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005243 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005244 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005245 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005246 if (tmp == NULL) goto failed;
5247 res = obj2ast_expr(tmp, &elt, arena);
5248 if (res != 0) goto failed;
5249 Py_XDECREF(tmp);
5250 tmp = NULL;
5251 } else {
5252 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from SetComp");
5253 return 1;
5254 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005255 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005256 int res;
5257 Py_ssize_t len;
5258 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005259 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005260 if (tmp == NULL) goto failed;
5261 if (!PyList_Check(tmp)) {
5262 PyErr_Format(PyExc_TypeError, "SetComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5263 goto failed;
5264 }
5265 len = PyList_GET_SIZE(tmp);
5266 generators = asdl_seq_new(len, arena);
5267 if (generators == NULL) goto failed;
5268 for (i = 0; i < len; i++) {
5269 comprehension_ty value;
5270 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5271 if (res != 0) goto failed;
5272 asdl_seq_SET(generators, i, value);
5273 }
5274 Py_XDECREF(tmp);
5275 tmp = NULL;
5276 } else {
5277 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from SetComp");
5278 return 1;
5279 }
5280 *out = SetComp(elt, generators, lineno, col_offset, arena);
5281 if (*out == NULL) goto failed;
5282 return 0;
5283 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005284 isinstance = PyObject_IsInstance(obj, (PyObject*)DictComp_type);
5285 if (isinstance == -1) {
5286 return 1;
5287 }
5288 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005289 expr_ty key;
5290 expr_ty value;
5291 asdl_seq* generators;
5292
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005293 if (_PyObject_HasAttrId(obj, &PyId_key)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005294 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005295 tmp = _PyObject_GetAttrId(obj, &PyId_key);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005296 if (tmp == NULL) goto failed;
5297 res = obj2ast_expr(tmp, &key, arena);
5298 if (res != 0) goto failed;
5299 Py_XDECREF(tmp);
5300 tmp = NULL;
5301 } else {
5302 PyErr_SetString(PyExc_TypeError, "required field \"key\" missing from DictComp");
5303 return 1;
5304 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005305 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005306 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005307 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005308 if (tmp == NULL) goto failed;
5309 res = obj2ast_expr(tmp, &value, arena);
5310 if (res != 0) goto failed;
5311 Py_XDECREF(tmp);
5312 tmp = NULL;
5313 } else {
5314 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from DictComp");
5315 return 1;
5316 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005317 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005318 int res;
5319 Py_ssize_t len;
5320 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005321 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005322 if (tmp == NULL) goto failed;
5323 if (!PyList_Check(tmp)) {
5324 PyErr_Format(PyExc_TypeError, "DictComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5325 goto failed;
5326 }
5327 len = PyList_GET_SIZE(tmp);
5328 generators = asdl_seq_new(len, arena);
5329 if (generators == NULL) goto failed;
5330 for (i = 0; i < len; i++) {
5331 comprehension_ty value;
5332 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5333 if (res != 0) goto failed;
5334 asdl_seq_SET(generators, i, value);
5335 }
5336 Py_XDECREF(tmp);
5337 tmp = NULL;
5338 } else {
5339 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from DictComp");
5340 return 1;
5341 }
5342 *out = DictComp(key, value, generators, lineno, col_offset,
5343 arena);
5344 if (*out == NULL) goto failed;
5345 return 0;
5346 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005347 isinstance = PyObject_IsInstance(obj, (PyObject*)GeneratorExp_type);
5348 if (isinstance == -1) {
5349 return 1;
5350 }
5351 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005352 expr_ty elt;
5353 asdl_seq* generators;
5354
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005355 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005356 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005357 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005358 if (tmp == NULL) goto failed;
5359 res = obj2ast_expr(tmp, &elt, arena);
5360 if (res != 0) goto failed;
5361 Py_XDECREF(tmp);
5362 tmp = NULL;
5363 } else {
5364 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from GeneratorExp");
5365 return 1;
5366 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005367 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005368 int res;
5369 Py_ssize_t len;
5370 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005371 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005372 if (tmp == NULL) goto failed;
5373 if (!PyList_Check(tmp)) {
5374 PyErr_Format(PyExc_TypeError, "GeneratorExp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5375 goto failed;
5376 }
5377 len = PyList_GET_SIZE(tmp);
5378 generators = asdl_seq_new(len, arena);
5379 if (generators == NULL) goto failed;
5380 for (i = 0; i < len; i++) {
5381 comprehension_ty value;
5382 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5383 if (res != 0) goto failed;
5384 asdl_seq_SET(generators, i, value);
5385 }
5386 Py_XDECREF(tmp);
5387 tmp = NULL;
5388 } else {
5389 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from GeneratorExp");
5390 return 1;
5391 }
5392 *out = GeneratorExp(elt, generators, lineno, col_offset, arena);
5393 if (*out == NULL) goto failed;
5394 return 0;
5395 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005396 isinstance = PyObject_IsInstance(obj, (PyObject*)Yield_type);
5397 if (isinstance == -1) {
5398 return 1;
5399 }
5400 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005401 expr_ty value;
5402
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005403 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005404 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005405 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005406 if (tmp == NULL) goto failed;
5407 res = obj2ast_expr(tmp, &value, arena);
5408 if (res != 0) goto failed;
5409 Py_XDECREF(tmp);
5410 tmp = NULL;
5411 } else {
5412 value = NULL;
5413 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05005414 *out = Yield(value, lineno, col_offset, arena);
5415 if (*out == NULL) goto failed;
5416 return 0;
5417 }
5418 isinstance = PyObject_IsInstance(obj, (PyObject*)YieldFrom_type);
5419 if (isinstance == -1) {
5420 return 1;
5421 }
5422 if (isinstance) {
5423 expr_ty value;
5424
5425 if (_PyObject_HasAttrId(obj, &PyId_value)) {
5426 int res;
5427 tmp = _PyObject_GetAttrId(obj, &PyId_value);
5428 if (tmp == NULL) goto failed;
5429 res = obj2ast_expr(tmp, &value, arena);
5430 if (res != 0) goto failed;
5431 Py_XDECREF(tmp);
5432 tmp = NULL;
5433 } else {
5434 value = NULL;
5435 }
5436 *out = YieldFrom(value, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005437 if (*out == NULL) goto failed;
5438 return 0;
5439 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005440 isinstance = PyObject_IsInstance(obj, (PyObject*)Compare_type);
5441 if (isinstance == -1) {
5442 return 1;
5443 }
5444 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005445 expr_ty left;
5446 asdl_int_seq* ops;
5447 asdl_seq* comparators;
5448
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005449 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005450 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005451 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005452 if (tmp == NULL) goto failed;
5453 res = obj2ast_expr(tmp, &left, arena);
5454 if (res != 0) goto failed;
5455 Py_XDECREF(tmp);
5456 tmp = NULL;
5457 } else {
5458 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from Compare");
5459 return 1;
5460 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005461 if (_PyObject_HasAttrId(obj, &PyId_ops)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005462 int res;
5463 Py_ssize_t len;
5464 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005465 tmp = _PyObject_GetAttrId(obj, &PyId_ops);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005466 if (tmp == NULL) goto failed;
5467 if (!PyList_Check(tmp)) {
5468 PyErr_Format(PyExc_TypeError, "Compare field \"ops\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5469 goto failed;
5470 }
5471 len = PyList_GET_SIZE(tmp);
5472 ops = asdl_int_seq_new(len, arena);
5473 if (ops == NULL) goto failed;
5474 for (i = 0; i < len; i++) {
5475 cmpop_ty value;
5476 res = obj2ast_cmpop(PyList_GET_ITEM(tmp, i), &value, arena);
5477 if (res != 0) goto failed;
5478 asdl_seq_SET(ops, i, value);
5479 }
5480 Py_XDECREF(tmp);
5481 tmp = NULL;
5482 } else {
5483 PyErr_SetString(PyExc_TypeError, "required field \"ops\" missing from Compare");
5484 return 1;
5485 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005486 if (_PyObject_HasAttrId(obj, &PyId_comparators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005487 int res;
5488 Py_ssize_t len;
5489 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005490 tmp = _PyObject_GetAttrId(obj, &PyId_comparators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005491 if (tmp == NULL) goto failed;
5492 if (!PyList_Check(tmp)) {
5493 PyErr_Format(PyExc_TypeError, "Compare field \"comparators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5494 goto failed;
5495 }
5496 len = PyList_GET_SIZE(tmp);
5497 comparators = asdl_seq_new(len, arena);
5498 if (comparators == NULL) goto failed;
5499 for (i = 0; i < len; i++) {
5500 expr_ty value;
5501 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5502 if (res != 0) goto failed;
5503 asdl_seq_SET(comparators, i, value);
5504 }
5505 Py_XDECREF(tmp);
5506 tmp = NULL;
5507 } else {
5508 PyErr_SetString(PyExc_TypeError, "required field \"comparators\" missing from Compare");
5509 return 1;
5510 }
5511 *out = Compare(left, ops, comparators, lineno, col_offset,
5512 arena);
5513 if (*out == NULL) goto failed;
5514 return 0;
5515 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005516 isinstance = PyObject_IsInstance(obj, (PyObject*)Call_type);
5517 if (isinstance == -1) {
5518 return 1;
5519 }
5520 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005521 expr_ty func;
5522 asdl_seq* args;
5523 asdl_seq* keywords;
5524 expr_ty starargs;
5525 expr_ty kwargs;
5526
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005527 if (_PyObject_HasAttrId(obj, &PyId_func)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005528 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005529 tmp = _PyObject_GetAttrId(obj, &PyId_func);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005530 if (tmp == NULL) goto failed;
5531 res = obj2ast_expr(tmp, &func, arena);
5532 if (res != 0) goto failed;
5533 Py_XDECREF(tmp);
5534 tmp = NULL;
5535 } else {
5536 PyErr_SetString(PyExc_TypeError, "required field \"func\" missing from Call");
5537 return 1;
5538 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005539 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005540 int res;
5541 Py_ssize_t len;
5542 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005543 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005544 if (tmp == NULL) goto failed;
5545 if (!PyList_Check(tmp)) {
5546 PyErr_Format(PyExc_TypeError, "Call field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5547 goto failed;
5548 }
5549 len = PyList_GET_SIZE(tmp);
5550 args = asdl_seq_new(len, arena);
5551 if (args == NULL) goto failed;
5552 for (i = 0; i < len; i++) {
5553 expr_ty value;
5554 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5555 if (res != 0) goto failed;
5556 asdl_seq_SET(args, i, value);
5557 }
5558 Py_XDECREF(tmp);
5559 tmp = NULL;
5560 } else {
5561 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Call");
5562 return 1;
5563 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005564 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005565 int res;
5566 Py_ssize_t len;
5567 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005568 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005569 if (tmp == NULL) goto failed;
5570 if (!PyList_Check(tmp)) {
5571 PyErr_Format(PyExc_TypeError, "Call field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5572 goto failed;
5573 }
5574 len = PyList_GET_SIZE(tmp);
5575 keywords = asdl_seq_new(len, arena);
5576 if (keywords == NULL) goto failed;
5577 for (i = 0; i < len; i++) {
5578 keyword_ty value;
5579 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
5580 if (res != 0) goto failed;
5581 asdl_seq_SET(keywords, i, value);
5582 }
5583 Py_XDECREF(tmp);
5584 tmp = NULL;
5585 } else {
5586 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from Call");
5587 return 1;
5588 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005589 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005590 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005591 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005592 if (tmp == NULL) goto failed;
5593 res = obj2ast_expr(tmp, &starargs, arena);
5594 if (res != 0) goto failed;
5595 Py_XDECREF(tmp);
5596 tmp = NULL;
5597 } else {
5598 starargs = NULL;
5599 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005600 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005601 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005602 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005603 if (tmp == NULL) goto failed;
5604 res = obj2ast_expr(tmp, &kwargs, arena);
5605 if (res != 0) goto failed;
5606 Py_XDECREF(tmp);
5607 tmp = NULL;
5608 } else {
5609 kwargs = NULL;
5610 }
5611 *out = Call(func, args, keywords, starargs, kwargs, lineno,
5612 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*)Num_type);
5617 if (isinstance == -1) {
5618 return 1;
5619 }
5620 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005621 object n;
5622
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005623 if (_PyObject_HasAttrId(obj, &PyId_n)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005624 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005625 tmp = _PyObject_GetAttrId(obj, &PyId_n);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005626 if (tmp == NULL) goto failed;
5627 res = obj2ast_object(tmp, &n, arena);
5628 if (res != 0) goto failed;
5629 Py_XDECREF(tmp);
5630 tmp = NULL;
5631 } else {
5632 PyErr_SetString(PyExc_TypeError, "required field \"n\" missing from Num");
5633 return 1;
5634 }
5635 *out = Num(n, lineno, col_offset, arena);
5636 if (*out == NULL) goto failed;
5637 return 0;
5638 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005639 isinstance = PyObject_IsInstance(obj, (PyObject*)Str_type);
5640 if (isinstance == -1) {
5641 return 1;
5642 }
5643 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005644 string s;
5645
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005646 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005647 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005648 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005649 if (tmp == NULL) goto failed;
5650 res = obj2ast_string(tmp, &s, arena);
5651 if (res != 0) goto failed;
5652 Py_XDECREF(tmp);
5653 tmp = NULL;
5654 } else {
5655 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Str");
5656 return 1;
5657 }
5658 *out = Str(s, lineno, col_offset, arena);
5659 if (*out == NULL) goto failed;
5660 return 0;
5661 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005662 isinstance = PyObject_IsInstance(obj, (PyObject*)Bytes_type);
5663 if (isinstance == -1) {
5664 return 1;
5665 }
5666 if (isinstance) {
Benjamin Petersone2498412011-08-09 16:08:39 -05005667 bytes s;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005668
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005669 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005670 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005671 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005672 if (tmp == NULL) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05005673 res = obj2ast_bytes(tmp, &s, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005674 if (res != 0) goto failed;
5675 Py_XDECREF(tmp);
5676 tmp = NULL;
5677 } else {
5678 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Bytes");
5679 return 1;
5680 }
5681 *out = Bytes(s, lineno, col_offset, arena);
5682 if (*out == NULL) goto failed;
5683 return 0;
5684 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005685 isinstance = PyObject_IsInstance(obj, (PyObject*)Ellipsis_type);
5686 if (isinstance == -1) {
5687 return 1;
5688 }
5689 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005690
5691 *out = Ellipsis(lineno, col_offset, arena);
5692 if (*out == NULL) goto failed;
5693 return 0;
5694 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005695 isinstance = PyObject_IsInstance(obj, (PyObject*)Attribute_type);
5696 if (isinstance == -1) {
5697 return 1;
5698 }
5699 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005700 expr_ty value;
5701 identifier attr;
5702 expr_context_ty ctx;
5703
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005704 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005705 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005706 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005707 if (tmp == NULL) goto failed;
5708 res = obj2ast_expr(tmp, &value, arena);
5709 if (res != 0) goto failed;
5710 Py_XDECREF(tmp);
5711 tmp = NULL;
5712 } else {
5713 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Attribute");
5714 return 1;
5715 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005716 if (_PyObject_HasAttrId(obj, &PyId_attr)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005717 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005718 tmp = _PyObject_GetAttrId(obj, &PyId_attr);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005719 if (tmp == NULL) goto failed;
5720 res = obj2ast_identifier(tmp, &attr, arena);
5721 if (res != 0) goto failed;
5722 Py_XDECREF(tmp);
5723 tmp = NULL;
5724 } else {
5725 PyErr_SetString(PyExc_TypeError, "required field \"attr\" missing from Attribute");
5726 return 1;
5727 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005728 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005729 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005730 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005731 if (tmp == NULL) goto failed;
5732 res = obj2ast_expr_context(tmp, &ctx, arena);
5733 if (res != 0) goto failed;
5734 Py_XDECREF(tmp);
5735 tmp = NULL;
5736 } else {
5737 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Attribute");
5738 return 1;
5739 }
5740 *out = Attribute(value, attr, ctx, lineno, col_offset, arena);
5741 if (*out == NULL) goto failed;
5742 return 0;
5743 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005744 isinstance = PyObject_IsInstance(obj, (PyObject*)Subscript_type);
5745 if (isinstance == -1) {
5746 return 1;
5747 }
5748 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005749 expr_ty value;
5750 slice_ty slice;
5751 expr_context_ty ctx;
5752
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005753 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005754 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005755 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005756 if (tmp == NULL) goto failed;
5757 res = obj2ast_expr(tmp, &value, arena);
5758 if (res != 0) goto failed;
5759 Py_XDECREF(tmp);
5760 tmp = NULL;
5761 } else {
5762 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Subscript");
5763 return 1;
5764 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005765 if (_PyObject_HasAttrId(obj, &PyId_slice)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005766 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005767 tmp = _PyObject_GetAttrId(obj, &PyId_slice);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005768 if (tmp == NULL) goto failed;
5769 res = obj2ast_slice(tmp, &slice, arena);
5770 if (res != 0) goto failed;
5771 Py_XDECREF(tmp);
5772 tmp = NULL;
5773 } else {
5774 PyErr_SetString(PyExc_TypeError, "required field \"slice\" missing from Subscript");
5775 return 1;
5776 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005777 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005778 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005779 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005780 if (tmp == NULL) goto failed;
5781 res = obj2ast_expr_context(tmp, &ctx, arena);
5782 if (res != 0) goto failed;
5783 Py_XDECREF(tmp);
5784 tmp = NULL;
5785 } else {
5786 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Subscript");
5787 return 1;
5788 }
5789 *out = Subscript(value, slice, ctx, lineno, col_offset, arena);
5790 if (*out == NULL) goto failed;
5791 return 0;
5792 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005793 isinstance = PyObject_IsInstance(obj, (PyObject*)Starred_type);
5794 if (isinstance == -1) {
5795 return 1;
5796 }
5797 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005798 expr_ty value;
5799 expr_context_ty ctx;
5800
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005801 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005802 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005803 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005804 if (tmp == NULL) goto failed;
5805 res = obj2ast_expr(tmp, &value, arena);
5806 if (res != 0) goto failed;
5807 Py_XDECREF(tmp);
5808 tmp = NULL;
5809 } else {
5810 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Starred");
5811 return 1;
5812 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005813 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005814 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005815 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005816 if (tmp == NULL) goto failed;
5817 res = obj2ast_expr_context(tmp, &ctx, arena);
5818 if (res != 0) goto failed;
5819 Py_XDECREF(tmp);
5820 tmp = NULL;
5821 } else {
5822 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Starred");
5823 return 1;
5824 }
5825 *out = Starred(value, ctx, lineno, col_offset, arena);
5826 if (*out == NULL) goto failed;
5827 return 0;
5828 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005829 isinstance = PyObject_IsInstance(obj, (PyObject*)Name_type);
5830 if (isinstance == -1) {
5831 return 1;
5832 }
5833 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005834 identifier id;
5835 expr_context_ty ctx;
5836
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005837 if (_PyObject_HasAttrId(obj, &PyId_id)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005838 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005839 tmp = _PyObject_GetAttrId(obj, &PyId_id);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005840 if (tmp == NULL) goto failed;
5841 res = obj2ast_identifier(tmp, &id, arena);
5842 if (res != 0) goto failed;
5843 Py_XDECREF(tmp);
5844 tmp = NULL;
5845 } else {
5846 PyErr_SetString(PyExc_TypeError, "required field \"id\" missing from Name");
5847 return 1;
5848 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005849 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005850 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005851 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005852 if (tmp == NULL) goto failed;
5853 res = obj2ast_expr_context(tmp, &ctx, arena);
5854 if (res != 0) goto failed;
5855 Py_XDECREF(tmp);
5856 tmp = NULL;
5857 } else {
5858 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Name");
5859 return 1;
5860 }
5861 *out = Name(id, ctx, lineno, col_offset, arena);
5862 if (*out == NULL) goto failed;
5863 return 0;
5864 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005865 isinstance = PyObject_IsInstance(obj, (PyObject*)List_type);
5866 if (isinstance == -1) {
5867 return 1;
5868 }
5869 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005870 asdl_seq* elts;
5871 expr_context_ty ctx;
5872
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005873 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005874 int res;
5875 Py_ssize_t len;
5876 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005877 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005878 if (tmp == NULL) goto failed;
5879 if (!PyList_Check(tmp)) {
5880 PyErr_Format(PyExc_TypeError, "List field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5881 goto failed;
5882 }
5883 len = PyList_GET_SIZE(tmp);
5884 elts = asdl_seq_new(len, arena);
5885 if (elts == NULL) goto failed;
5886 for (i = 0; i < len; i++) {
5887 expr_ty value;
5888 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5889 if (res != 0) goto failed;
5890 asdl_seq_SET(elts, i, value);
5891 }
5892 Py_XDECREF(tmp);
5893 tmp = NULL;
5894 } else {
5895 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from List");
5896 return 1;
5897 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005898 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005899 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005900 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005901 if (tmp == NULL) goto failed;
5902 res = obj2ast_expr_context(tmp, &ctx, arena);
5903 if (res != 0) goto failed;
5904 Py_XDECREF(tmp);
5905 tmp = NULL;
5906 } else {
5907 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from List");
5908 return 1;
5909 }
5910 *out = List(elts, ctx, lineno, col_offset, arena);
5911 if (*out == NULL) goto failed;
5912 return 0;
5913 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005914 isinstance = PyObject_IsInstance(obj, (PyObject*)Tuple_type);
5915 if (isinstance == -1) {
5916 return 1;
5917 }
5918 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005919 asdl_seq* elts;
5920 expr_context_ty ctx;
5921
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005922 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005923 int res;
5924 Py_ssize_t len;
5925 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005926 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005927 if (tmp == NULL) goto failed;
5928 if (!PyList_Check(tmp)) {
5929 PyErr_Format(PyExc_TypeError, "Tuple field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5930 goto failed;
5931 }
5932 len = PyList_GET_SIZE(tmp);
5933 elts = asdl_seq_new(len, arena);
5934 if (elts == NULL) goto failed;
5935 for (i = 0; i < len; i++) {
5936 expr_ty value;
5937 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5938 if (res != 0) goto failed;
5939 asdl_seq_SET(elts, i, value);
5940 }
5941 Py_XDECREF(tmp);
5942 tmp = NULL;
5943 } else {
5944 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Tuple");
5945 return 1;
5946 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005947 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005948 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005949 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005950 if (tmp == NULL) goto failed;
5951 res = obj2ast_expr_context(tmp, &ctx, arena);
5952 if (res != 0) goto failed;
5953 Py_XDECREF(tmp);
5954 tmp = NULL;
5955 } else {
5956 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Tuple");
5957 return 1;
5958 }
5959 *out = Tuple(elts, ctx, lineno, col_offset, arena);
5960 if (*out == NULL) goto failed;
5961 return 0;
5962 }
5963
Benjamin Peterson5b066812010-11-20 01:38:49 +00005964 PyErr_Format(PyExc_TypeError, "expected some sort of expr, but got %R", obj);
5965 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00005966 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005967 return 1;
5968}
5969
5970int
5971obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena* arena)
5972{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005973 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005974
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005975 isinstance = PyObject_IsInstance(obj, (PyObject *)Load_type);
5976 if (isinstance == -1) {
5977 return 1;
5978 }
5979 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005980 *out = Load;
5981 return 0;
5982 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005983 isinstance = PyObject_IsInstance(obj, (PyObject *)Store_type);
5984 if (isinstance == -1) {
5985 return 1;
5986 }
5987 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005988 *out = Store;
5989 return 0;
5990 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005991 isinstance = PyObject_IsInstance(obj, (PyObject *)Del_type);
5992 if (isinstance == -1) {
5993 return 1;
5994 }
5995 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005996 *out = Del;
5997 return 0;
5998 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005999 isinstance = PyObject_IsInstance(obj, (PyObject *)AugLoad_type);
6000 if (isinstance == -1) {
6001 return 1;
6002 }
6003 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006004 *out = AugLoad;
6005 return 0;
6006 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006007 isinstance = PyObject_IsInstance(obj, (PyObject *)AugStore_type);
6008 if (isinstance == -1) {
6009 return 1;
6010 }
6011 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006012 *out = AugStore;
6013 return 0;
6014 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006015 isinstance = PyObject_IsInstance(obj, (PyObject *)Param_type);
6016 if (isinstance == -1) {
6017 return 1;
6018 }
6019 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006020 *out = Param;
6021 return 0;
6022 }
6023
Benjamin Peterson5b066812010-11-20 01:38:49 +00006024 PyErr_Format(PyExc_TypeError, "expected some sort of expr_context, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006025 return 1;
6026}
6027
6028int
6029obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena)
6030{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006031 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006032
Benjamin Petersond8f65972010-11-20 04:31:07 +00006033 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006034
6035 if (obj == Py_None) {
6036 *out = NULL;
6037 return 0;
6038 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006039 isinstance = PyObject_IsInstance(obj, (PyObject*)Slice_type);
6040 if (isinstance == -1) {
6041 return 1;
6042 }
6043 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006044 expr_ty lower;
6045 expr_ty upper;
6046 expr_ty step;
6047
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006048 if (_PyObject_HasAttrId(obj, &PyId_lower)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006049 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006050 tmp = _PyObject_GetAttrId(obj, &PyId_lower);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006051 if (tmp == NULL) goto failed;
6052 res = obj2ast_expr(tmp, &lower, arena);
6053 if (res != 0) goto failed;
6054 Py_XDECREF(tmp);
6055 tmp = NULL;
6056 } else {
6057 lower = NULL;
6058 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006059 if (_PyObject_HasAttrId(obj, &PyId_upper)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006060 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006061 tmp = _PyObject_GetAttrId(obj, &PyId_upper);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006062 if (tmp == NULL) goto failed;
6063 res = obj2ast_expr(tmp, &upper, arena);
6064 if (res != 0) goto failed;
6065 Py_XDECREF(tmp);
6066 tmp = NULL;
6067 } else {
6068 upper = NULL;
6069 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006070 if (_PyObject_HasAttrId(obj, &PyId_step)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006071 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006072 tmp = _PyObject_GetAttrId(obj, &PyId_step);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006073 if (tmp == NULL) goto failed;
6074 res = obj2ast_expr(tmp, &step, arena);
6075 if (res != 0) goto failed;
6076 Py_XDECREF(tmp);
6077 tmp = NULL;
6078 } else {
6079 step = NULL;
6080 }
6081 *out = Slice(lower, upper, step, arena);
6082 if (*out == NULL) goto failed;
6083 return 0;
6084 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006085 isinstance = PyObject_IsInstance(obj, (PyObject*)ExtSlice_type);
6086 if (isinstance == -1) {
6087 return 1;
6088 }
6089 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006090 asdl_seq* dims;
6091
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006092 if (_PyObject_HasAttrId(obj, &PyId_dims)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006093 int res;
6094 Py_ssize_t len;
6095 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006096 tmp = _PyObject_GetAttrId(obj, &PyId_dims);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006097 if (tmp == NULL) goto failed;
6098 if (!PyList_Check(tmp)) {
6099 PyErr_Format(PyExc_TypeError, "ExtSlice field \"dims\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6100 goto failed;
6101 }
6102 len = PyList_GET_SIZE(tmp);
6103 dims = asdl_seq_new(len, arena);
6104 if (dims == NULL) goto failed;
6105 for (i = 0; i < len; i++) {
6106 slice_ty value;
6107 res = obj2ast_slice(PyList_GET_ITEM(tmp, i), &value, arena);
6108 if (res != 0) goto failed;
6109 asdl_seq_SET(dims, i, value);
6110 }
6111 Py_XDECREF(tmp);
6112 tmp = NULL;
6113 } else {
6114 PyErr_SetString(PyExc_TypeError, "required field \"dims\" missing from ExtSlice");
6115 return 1;
6116 }
6117 *out = ExtSlice(dims, arena);
6118 if (*out == NULL) goto failed;
6119 return 0;
6120 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006121 isinstance = PyObject_IsInstance(obj, (PyObject*)Index_type);
6122 if (isinstance == -1) {
6123 return 1;
6124 }
6125 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006126 expr_ty value;
6127
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006128 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006129 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006130 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006131 if (tmp == NULL) goto failed;
6132 res = obj2ast_expr(tmp, &value, arena);
6133 if (res != 0) goto failed;
6134 Py_XDECREF(tmp);
6135 tmp = NULL;
6136 } else {
6137 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Index");
6138 return 1;
6139 }
6140 *out = Index(value, arena);
6141 if (*out == NULL) goto failed;
6142 return 0;
6143 }
6144
Benjamin Peterson5b066812010-11-20 01:38:49 +00006145 PyErr_Format(PyExc_TypeError, "expected some sort of slice, but got %R", obj);
6146 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006147 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006148 return 1;
6149}
6150
6151int
6152obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena)
6153{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006154 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006155
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006156 isinstance = PyObject_IsInstance(obj, (PyObject *)And_type);
6157 if (isinstance == -1) {
6158 return 1;
6159 }
6160 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006161 *out = And;
6162 return 0;
6163 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006164 isinstance = PyObject_IsInstance(obj, (PyObject *)Or_type);
6165 if (isinstance == -1) {
6166 return 1;
6167 }
6168 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006169 *out = Or;
6170 return 0;
6171 }
6172
Benjamin Peterson5b066812010-11-20 01:38:49 +00006173 PyErr_Format(PyExc_TypeError, "expected some sort of boolop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006174 return 1;
6175}
6176
6177int
6178obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena)
6179{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006180 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006181
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006182 isinstance = PyObject_IsInstance(obj, (PyObject *)Add_type);
6183 if (isinstance == -1) {
6184 return 1;
6185 }
6186 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006187 *out = Add;
6188 return 0;
6189 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006190 isinstance = PyObject_IsInstance(obj, (PyObject *)Sub_type);
6191 if (isinstance == -1) {
6192 return 1;
6193 }
6194 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006195 *out = Sub;
6196 return 0;
6197 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006198 isinstance = PyObject_IsInstance(obj, (PyObject *)Mult_type);
6199 if (isinstance == -1) {
6200 return 1;
6201 }
6202 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006203 *out = Mult;
6204 return 0;
6205 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006206 isinstance = PyObject_IsInstance(obj, (PyObject *)Div_type);
6207 if (isinstance == -1) {
6208 return 1;
6209 }
6210 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006211 *out = Div;
6212 return 0;
6213 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006214 isinstance = PyObject_IsInstance(obj, (PyObject *)Mod_type);
6215 if (isinstance == -1) {
6216 return 1;
6217 }
6218 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006219 *out = Mod;
6220 return 0;
6221 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006222 isinstance = PyObject_IsInstance(obj, (PyObject *)Pow_type);
6223 if (isinstance == -1) {
6224 return 1;
6225 }
6226 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006227 *out = Pow;
6228 return 0;
6229 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006230 isinstance = PyObject_IsInstance(obj, (PyObject *)LShift_type);
6231 if (isinstance == -1) {
6232 return 1;
6233 }
6234 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006235 *out = LShift;
6236 return 0;
6237 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006238 isinstance = PyObject_IsInstance(obj, (PyObject *)RShift_type);
6239 if (isinstance == -1) {
6240 return 1;
6241 }
6242 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006243 *out = RShift;
6244 return 0;
6245 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006246 isinstance = PyObject_IsInstance(obj, (PyObject *)BitOr_type);
6247 if (isinstance == -1) {
6248 return 1;
6249 }
6250 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006251 *out = BitOr;
6252 return 0;
6253 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006254 isinstance = PyObject_IsInstance(obj, (PyObject *)BitXor_type);
6255 if (isinstance == -1) {
6256 return 1;
6257 }
6258 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006259 *out = BitXor;
6260 return 0;
6261 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006262 isinstance = PyObject_IsInstance(obj, (PyObject *)BitAnd_type);
6263 if (isinstance == -1) {
6264 return 1;
6265 }
6266 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006267 *out = BitAnd;
6268 return 0;
6269 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006270 isinstance = PyObject_IsInstance(obj, (PyObject *)FloorDiv_type);
6271 if (isinstance == -1) {
6272 return 1;
6273 }
6274 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006275 *out = FloorDiv;
6276 return 0;
6277 }
6278
Benjamin Peterson5b066812010-11-20 01:38:49 +00006279 PyErr_Format(PyExc_TypeError, "expected some sort of operator, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006280 return 1;
6281}
6282
6283int
6284obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena)
6285{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006286 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006287
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006288 isinstance = PyObject_IsInstance(obj, (PyObject *)Invert_type);
6289 if (isinstance == -1) {
6290 return 1;
6291 }
6292 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006293 *out = Invert;
6294 return 0;
6295 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006296 isinstance = PyObject_IsInstance(obj, (PyObject *)Not_type);
6297 if (isinstance == -1) {
6298 return 1;
6299 }
6300 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006301 *out = Not;
6302 return 0;
6303 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006304 isinstance = PyObject_IsInstance(obj, (PyObject *)UAdd_type);
6305 if (isinstance == -1) {
6306 return 1;
6307 }
6308 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006309 *out = UAdd;
6310 return 0;
6311 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006312 isinstance = PyObject_IsInstance(obj, (PyObject *)USub_type);
6313 if (isinstance == -1) {
6314 return 1;
6315 }
6316 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006317 *out = USub;
6318 return 0;
6319 }
6320
Benjamin Peterson5b066812010-11-20 01:38:49 +00006321 PyErr_Format(PyExc_TypeError, "expected some sort of unaryop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006322 return 1;
6323}
6324
6325int
6326obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena)
6327{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006328 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006329
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006330 isinstance = PyObject_IsInstance(obj, (PyObject *)Eq_type);
6331 if (isinstance == -1) {
6332 return 1;
6333 }
6334 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006335 *out = Eq;
6336 return 0;
6337 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006338 isinstance = PyObject_IsInstance(obj, (PyObject *)NotEq_type);
6339 if (isinstance == -1) {
6340 return 1;
6341 }
6342 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006343 *out = NotEq;
6344 return 0;
6345 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006346 isinstance = PyObject_IsInstance(obj, (PyObject *)Lt_type);
6347 if (isinstance == -1) {
6348 return 1;
6349 }
6350 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006351 *out = Lt;
6352 return 0;
6353 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006354 isinstance = PyObject_IsInstance(obj, (PyObject *)LtE_type);
6355 if (isinstance == -1) {
6356 return 1;
6357 }
6358 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006359 *out = LtE;
6360 return 0;
6361 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006362 isinstance = PyObject_IsInstance(obj, (PyObject *)Gt_type);
6363 if (isinstance == -1) {
6364 return 1;
6365 }
6366 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006367 *out = Gt;
6368 return 0;
6369 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006370 isinstance = PyObject_IsInstance(obj, (PyObject *)GtE_type);
6371 if (isinstance == -1) {
6372 return 1;
6373 }
6374 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006375 *out = GtE;
6376 return 0;
6377 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006378 isinstance = PyObject_IsInstance(obj, (PyObject *)Is_type);
6379 if (isinstance == -1) {
6380 return 1;
6381 }
6382 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006383 *out = Is;
6384 return 0;
6385 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006386 isinstance = PyObject_IsInstance(obj, (PyObject *)IsNot_type);
6387 if (isinstance == -1) {
6388 return 1;
6389 }
6390 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006391 *out = IsNot;
6392 return 0;
6393 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006394 isinstance = PyObject_IsInstance(obj, (PyObject *)In_type);
6395 if (isinstance == -1) {
6396 return 1;
6397 }
6398 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006399 *out = In;
6400 return 0;
6401 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006402 isinstance = PyObject_IsInstance(obj, (PyObject *)NotIn_type);
6403 if (isinstance == -1) {
6404 return 1;
6405 }
6406 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006407 *out = NotIn;
6408 return 0;
6409 }
6410
Benjamin Peterson5b066812010-11-20 01:38:49 +00006411 PyErr_Format(PyExc_TypeError, "expected some sort of cmpop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006412 return 1;
6413}
6414
6415int
6416obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena* arena)
6417{
6418 PyObject* tmp = NULL;
6419 expr_ty target;
6420 expr_ty iter;
6421 asdl_seq* ifs;
6422
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006423 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006424 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006425 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006426 if (tmp == NULL) goto failed;
6427 res = obj2ast_expr(tmp, &target, arena);
6428 if (res != 0) goto failed;
6429 Py_XDECREF(tmp);
6430 tmp = NULL;
6431 } else {
6432 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from comprehension");
6433 return 1;
6434 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006435 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006436 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006437 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006438 if (tmp == NULL) goto failed;
6439 res = obj2ast_expr(tmp, &iter, arena);
6440 if (res != 0) goto failed;
6441 Py_XDECREF(tmp);
6442 tmp = NULL;
6443 } else {
6444 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from comprehension");
6445 return 1;
6446 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006447 if (_PyObject_HasAttrId(obj, &PyId_ifs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006448 int res;
6449 Py_ssize_t len;
6450 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006451 tmp = _PyObject_GetAttrId(obj, &PyId_ifs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006452 if (tmp == NULL) goto failed;
6453 if (!PyList_Check(tmp)) {
6454 PyErr_Format(PyExc_TypeError, "comprehension field \"ifs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6455 goto failed;
6456 }
6457 len = PyList_GET_SIZE(tmp);
6458 ifs = asdl_seq_new(len, arena);
6459 if (ifs == NULL) goto failed;
6460 for (i = 0; i < len; i++) {
6461 expr_ty value;
6462 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6463 if (res != 0) goto failed;
6464 asdl_seq_SET(ifs, i, value);
6465 }
6466 Py_XDECREF(tmp);
6467 tmp = NULL;
6468 } else {
6469 PyErr_SetString(PyExc_TypeError, "required field \"ifs\" missing from comprehension");
6470 return 1;
6471 }
6472 *out = comprehension(target, iter, ifs, arena);
6473 return 0;
6474failed:
6475 Py_XDECREF(tmp);
6476 return 1;
6477}
6478
6479int
6480obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena* arena)
6481{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006482 int isinstance;
Neal Norwitzad74aa82008-03-31 05:14:30 +00006483
Benjamin Petersond8f65972010-11-20 04:31:07 +00006484 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006485 int lineno;
6486 int col_offset;
6487
Neal Norwitzad74aa82008-03-31 05:14:30 +00006488 if (obj == Py_None) {
6489 *out = NULL;
6490 return 0;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006491 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006492 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006493 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006494 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006495 if (tmp == NULL) goto failed;
6496 res = obj2ast_int(tmp, &lineno, arena);
6497 if (res != 0) goto failed;
6498 Py_XDECREF(tmp);
6499 tmp = NULL;
6500 } else {
6501 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from excepthandler");
6502 return 1;
6503 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006504 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006505 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006506 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006507 if (tmp == NULL) goto failed;
6508 res = obj2ast_int(tmp, &col_offset, arena);
6509 if (res != 0) goto failed;
6510 Py_XDECREF(tmp);
6511 tmp = NULL;
6512 } else {
6513 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from excepthandler");
6514 return 1;
6515 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006516 isinstance = PyObject_IsInstance(obj, (PyObject*)ExceptHandler_type);
6517 if (isinstance == -1) {
6518 return 1;
6519 }
6520 if (isinstance) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006521 expr_ty type;
6522 identifier name;
6523 asdl_seq* body;
6524
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006525 if (_PyObject_HasAttrId(obj, &PyId_type)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006526 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006527 tmp = _PyObject_GetAttrId(obj, &PyId_type);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006528 if (tmp == NULL) goto failed;
6529 res = obj2ast_expr(tmp, &type, arena);
6530 if (res != 0) goto failed;
6531 Py_XDECREF(tmp);
6532 tmp = NULL;
6533 } else {
6534 type = NULL;
6535 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006536 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006537 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006538 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006539 if (tmp == NULL) goto failed;
6540 res = obj2ast_identifier(tmp, &name, arena);
6541 if (res != 0) goto failed;
6542 Py_XDECREF(tmp);
6543 tmp = NULL;
6544 } else {
6545 name = NULL;
6546 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006547 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006548 int res;
6549 Py_ssize_t len;
6550 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006551 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006552 if (tmp == NULL) goto failed;
6553 if (!PyList_Check(tmp)) {
6554 PyErr_Format(PyExc_TypeError, "ExceptHandler field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6555 goto failed;
6556 }
6557 len = PyList_GET_SIZE(tmp);
6558 body = asdl_seq_new(len, arena);
6559 if (body == NULL) goto failed;
6560 for (i = 0; i < len; i++) {
6561 stmt_ty value;
6562 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
6563 if (res != 0) goto failed;
6564 asdl_seq_SET(body, i, value);
6565 }
6566 Py_XDECREF(tmp);
6567 tmp = NULL;
6568 } else {
6569 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ExceptHandler");
6570 return 1;
6571 }
6572 *out = ExceptHandler(type, name, body, lineno, col_offset,
6573 arena);
6574 if (*out == NULL) goto failed;
6575 return 0;
6576 }
6577
Benjamin Peterson5b066812010-11-20 01:38:49 +00006578 PyErr_Format(PyExc_TypeError, "expected some sort of excepthandler, but got %R", obj);
6579 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006580 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006581 return 1;
6582}
6583
6584int
6585obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena)
6586{
6587 PyObject* tmp = NULL;
6588 asdl_seq* args;
6589 identifier vararg;
6590 expr_ty varargannotation;
6591 asdl_seq* kwonlyargs;
6592 identifier kwarg;
6593 expr_ty kwargannotation;
6594 asdl_seq* defaults;
6595 asdl_seq* kw_defaults;
6596
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006597 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006598 int res;
6599 Py_ssize_t len;
6600 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006601 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006602 if (tmp == NULL) goto failed;
6603 if (!PyList_Check(tmp)) {
6604 PyErr_Format(PyExc_TypeError, "arguments field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6605 goto failed;
6606 }
6607 len = PyList_GET_SIZE(tmp);
6608 args = asdl_seq_new(len, arena);
6609 if (args == NULL) goto failed;
6610 for (i = 0; i < len; i++) {
6611 arg_ty value;
6612 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6613 if (res != 0) goto failed;
6614 asdl_seq_SET(args, i, value);
6615 }
6616 Py_XDECREF(tmp);
6617 tmp = NULL;
6618 } else {
6619 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from arguments");
6620 return 1;
6621 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006622 if (_PyObject_HasAttrId(obj, &PyId_vararg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006623 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006624 tmp = _PyObject_GetAttrId(obj, &PyId_vararg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006625 if (tmp == NULL) goto failed;
6626 res = obj2ast_identifier(tmp, &vararg, arena);
6627 if (res != 0) goto failed;
6628 Py_XDECREF(tmp);
6629 tmp = NULL;
6630 } else {
6631 vararg = NULL;
6632 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006633 if (_PyObject_HasAttrId(obj, &PyId_varargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006634 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006635 tmp = _PyObject_GetAttrId(obj, &PyId_varargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006636 if (tmp == NULL) goto failed;
6637 res = obj2ast_expr(tmp, &varargannotation, arena);
6638 if (res != 0) goto failed;
6639 Py_XDECREF(tmp);
6640 tmp = NULL;
6641 } else {
6642 varargannotation = NULL;
6643 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006644 if (_PyObject_HasAttrId(obj, &PyId_kwonlyargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006645 int res;
6646 Py_ssize_t len;
6647 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006648 tmp = _PyObject_GetAttrId(obj, &PyId_kwonlyargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006649 if (tmp == NULL) goto failed;
6650 if (!PyList_Check(tmp)) {
6651 PyErr_Format(PyExc_TypeError, "arguments field \"kwonlyargs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6652 goto failed;
6653 }
6654 len = PyList_GET_SIZE(tmp);
6655 kwonlyargs = asdl_seq_new(len, arena);
6656 if (kwonlyargs == NULL) goto failed;
6657 for (i = 0; i < len; i++) {
6658 arg_ty value;
6659 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6660 if (res != 0) goto failed;
6661 asdl_seq_SET(kwonlyargs, i, value);
6662 }
6663 Py_XDECREF(tmp);
6664 tmp = NULL;
6665 } else {
6666 PyErr_SetString(PyExc_TypeError, "required field \"kwonlyargs\" missing from arguments");
6667 return 1;
6668 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006669 if (_PyObject_HasAttrId(obj, &PyId_kwarg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006670 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006671 tmp = _PyObject_GetAttrId(obj, &PyId_kwarg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006672 if (tmp == NULL) goto failed;
6673 res = obj2ast_identifier(tmp, &kwarg, arena);
6674 if (res != 0) goto failed;
6675 Py_XDECREF(tmp);
6676 tmp = NULL;
6677 } else {
6678 kwarg = NULL;
6679 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006680 if (_PyObject_HasAttrId(obj, &PyId_kwargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006681 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006682 tmp = _PyObject_GetAttrId(obj, &PyId_kwargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006683 if (tmp == NULL) goto failed;
6684 res = obj2ast_expr(tmp, &kwargannotation, arena);
6685 if (res != 0) goto failed;
6686 Py_XDECREF(tmp);
6687 tmp = NULL;
6688 } else {
6689 kwargannotation = NULL;
6690 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006691 if (_PyObject_HasAttrId(obj, &PyId_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006692 int res;
6693 Py_ssize_t len;
6694 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006695 tmp = _PyObject_GetAttrId(obj, &PyId_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006696 if (tmp == NULL) goto failed;
6697 if (!PyList_Check(tmp)) {
6698 PyErr_Format(PyExc_TypeError, "arguments field \"defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6699 goto failed;
6700 }
6701 len = PyList_GET_SIZE(tmp);
6702 defaults = asdl_seq_new(len, arena);
6703 if (defaults == NULL) goto failed;
6704 for (i = 0; i < len; i++) {
6705 expr_ty value;
6706 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6707 if (res != 0) goto failed;
6708 asdl_seq_SET(defaults, i, value);
6709 }
6710 Py_XDECREF(tmp);
6711 tmp = NULL;
6712 } else {
6713 PyErr_SetString(PyExc_TypeError, "required field \"defaults\" missing from arguments");
6714 return 1;
6715 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006716 if (_PyObject_HasAttrId(obj, &PyId_kw_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006717 int res;
6718 Py_ssize_t len;
6719 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006720 tmp = _PyObject_GetAttrId(obj, &PyId_kw_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006721 if (tmp == NULL) goto failed;
6722 if (!PyList_Check(tmp)) {
6723 PyErr_Format(PyExc_TypeError, "arguments field \"kw_defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6724 goto failed;
6725 }
6726 len = PyList_GET_SIZE(tmp);
6727 kw_defaults = asdl_seq_new(len, arena);
6728 if (kw_defaults == NULL) goto failed;
6729 for (i = 0; i < len; i++) {
6730 expr_ty value;
6731 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6732 if (res != 0) goto failed;
6733 asdl_seq_SET(kw_defaults, i, value);
6734 }
6735 Py_XDECREF(tmp);
6736 tmp = NULL;
6737 } else {
6738 PyErr_SetString(PyExc_TypeError, "required field \"kw_defaults\" missing from arguments");
6739 return 1;
6740 }
6741 *out = arguments(args, vararg, varargannotation, kwonlyargs, kwarg,
6742 kwargannotation, defaults, kw_defaults, arena);
6743 return 0;
6744failed:
6745 Py_XDECREF(tmp);
6746 return 1;
6747}
6748
6749int
6750obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena)
6751{
6752 PyObject* tmp = NULL;
6753 identifier arg;
6754 expr_ty annotation;
6755
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006756 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006757 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006758 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006759 if (tmp == NULL) goto failed;
6760 res = obj2ast_identifier(tmp, &arg, arena);
6761 if (res != 0) goto failed;
6762 Py_XDECREF(tmp);
6763 tmp = NULL;
6764 } else {
6765 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from arg");
6766 return 1;
6767 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006768 if (_PyObject_HasAttrId(obj, &PyId_annotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006769 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006770 tmp = _PyObject_GetAttrId(obj, &PyId_annotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006771 if (tmp == NULL) goto failed;
6772 res = obj2ast_expr(tmp, &annotation, arena);
6773 if (res != 0) goto failed;
6774 Py_XDECREF(tmp);
6775 tmp = NULL;
6776 } else {
6777 annotation = NULL;
6778 }
6779 *out = arg(arg, annotation, arena);
6780 return 0;
6781failed:
6782 Py_XDECREF(tmp);
6783 return 1;
6784}
6785
6786int
6787obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena)
6788{
6789 PyObject* tmp = NULL;
6790 identifier arg;
6791 expr_ty value;
6792
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006793 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006794 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006795 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006796 if (tmp == NULL) goto failed;
6797 res = obj2ast_identifier(tmp, &arg, arena);
6798 if (res != 0) goto failed;
6799 Py_XDECREF(tmp);
6800 tmp = NULL;
6801 } else {
6802 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from keyword");
6803 return 1;
6804 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006805 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006806 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006807 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006808 if (tmp == NULL) goto failed;
6809 res = obj2ast_expr(tmp, &value, arena);
6810 if (res != 0) goto failed;
6811 Py_XDECREF(tmp);
6812 tmp = NULL;
6813 } else {
6814 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from keyword");
6815 return 1;
6816 }
6817 *out = keyword(arg, value, arena);
6818 return 0;
6819failed:
6820 Py_XDECREF(tmp);
6821 return 1;
6822}
6823
6824int
6825obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena)
6826{
6827 PyObject* tmp = NULL;
6828 identifier name;
6829 identifier asname;
6830
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006831 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006832 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006833 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006834 if (tmp == NULL) goto failed;
6835 res = obj2ast_identifier(tmp, &name, arena);
6836 if (res != 0) goto failed;
6837 Py_XDECREF(tmp);
6838 tmp = NULL;
6839 } else {
6840 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from alias");
6841 return 1;
6842 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006843 if (_PyObject_HasAttrId(obj, &PyId_asname)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006844 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006845 tmp = _PyObject_GetAttrId(obj, &PyId_asname);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006846 if (tmp == NULL) goto failed;
6847 res = obj2ast_identifier(tmp, &asname, arena);
6848 if (res != 0) goto failed;
6849 Py_XDECREF(tmp);
6850 tmp = NULL;
6851 } else {
6852 asname = NULL;
6853 }
6854 *out = alias(name, asname, arena);
6855 return 0;
6856failed:
6857 Py_XDECREF(tmp);
6858 return 1;
6859}
6860
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006861int
6862obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena)
6863{
6864 PyObject* tmp = NULL;
6865 expr_ty context_expr;
6866 expr_ty optional_vars;
6867
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006868 if (_PyObject_HasAttrId(obj, &PyId_context_expr)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006869 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006870 tmp = _PyObject_GetAttrId(obj, &PyId_context_expr);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006871 if (tmp == NULL) goto failed;
6872 res = obj2ast_expr(tmp, &context_expr, arena);
6873 if (res != 0) goto failed;
6874 Py_XDECREF(tmp);
6875 tmp = NULL;
6876 } else {
6877 PyErr_SetString(PyExc_TypeError, "required field \"context_expr\" missing from withitem");
6878 return 1;
6879 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006880 if (_PyObject_HasAttrId(obj, &PyId_optional_vars)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006881 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006882 tmp = _PyObject_GetAttrId(obj, &PyId_optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006883 if (tmp == NULL) goto failed;
6884 res = obj2ast_expr(tmp, &optional_vars, arena);
6885 if (res != 0) goto failed;
6886 Py_XDECREF(tmp);
6887 tmp = NULL;
6888 } else {
6889 optional_vars = NULL;
6890 }
6891 *out = withitem(context_expr, optional_vars, arena);
6892 return 0;
6893failed:
6894 Py_XDECREF(tmp);
6895 return 1;
6896}
6897
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006898
Martin v. Löwis1a214512008-06-11 05:26:20 +00006899static struct PyModuleDef _astmodule = {
6900 PyModuleDef_HEAD_INIT, "_ast"
6901};
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006902PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00006903PyInit__ast(void)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006904{
6905 PyObject *m, *d;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006906 if (!init_types()) return NULL;
6907 m = PyModule_Create(&_astmodule);
6908 if (!m) return NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006909 d = PyModule_GetDict(m);
Martin v. Löwis1a214512008-06-11 05:26:20 +00006910 if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return
6911 NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006912 if (PyModule_AddIntConstant(m, "PyCF_ONLY_AST", PyCF_ONLY_AST) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006913 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006914 if (PyDict_SetItemString(d, "mod", (PyObject*)mod_type) < 0) return
6915 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006916 if (PyDict_SetItemString(d, "Module", (PyObject*)Module_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006917 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006918 if (PyDict_SetItemString(d, "Interactive", (PyObject*)Interactive_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006919 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006920 if (PyDict_SetItemString(d, "Expression", (PyObject*)Expression_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006921 0) return NULL;
6922 if (PyDict_SetItemString(d, "Suite", (PyObject*)Suite_type) < 0) return
6923 NULL;
6924 if (PyDict_SetItemString(d, "stmt", (PyObject*)stmt_type) < 0) return
6925 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006926 if (PyDict_SetItemString(d, "FunctionDef", (PyObject*)FunctionDef_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, "ClassDef", (PyObject*)ClassDef_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006929 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006930 if (PyDict_SetItemString(d, "Return", (PyObject*)Return_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006931 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006932 if (PyDict_SetItemString(d, "Delete", (PyObject*)Delete_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006933 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006934 if (PyDict_SetItemString(d, "Assign", (PyObject*)Assign_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006935 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006936 if (PyDict_SetItemString(d, "AugAssign", (PyObject*)AugAssign_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006937 0) return NULL;
6938 if (PyDict_SetItemString(d, "For", (PyObject*)For_type) < 0) return
6939 NULL;
6940 if (PyDict_SetItemString(d, "While", (PyObject*)While_type) < 0) return
6941 NULL;
6942 if (PyDict_SetItemString(d, "If", (PyObject*)If_type) < 0) return NULL;
6943 if (PyDict_SetItemString(d, "With", (PyObject*)With_type) < 0) return
6944 NULL;
6945 if (PyDict_SetItemString(d, "Raise", (PyObject*)Raise_type) < 0) return
6946 NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05006947 if (PyDict_SetItemString(d, "Try", (PyObject*)Try_type) < 0) return
6948 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006949 if (PyDict_SetItemString(d, "Assert", (PyObject*)Assert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006950 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006951 if (PyDict_SetItemString(d, "Import", (PyObject*)Import_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006952 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006953 if (PyDict_SetItemString(d, "ImportFrom", (PyObject*)ImportFrom_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006954 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006955 if (PyDict_SetItemString(d, "Global", (PyObject*)Global_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006956 return NULL;
Jeremy Hylton81e95022007-02-27 06:50:52 +00006957 if (PyDict_SetItemString(d, "Nonlocal", (PyObject*)Nonlocal_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006958 return NULL;
6959 if (PyDict_SetItemString(d, "Expr", (PyObject*)Expr_type) < 0) return
6960 NULL;
6961 if (PyDict_SetItemString(d, "Pass", (PyObject*)Pass_type) < 0) return
6962 NULL;
6963 if (PyDict_SetItemString(d, "Break", (PyObject*)Break_type) < 0) return
6964 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006965 if (PyDict_SetItemString(d, "Continue", (PyObject*)Continue_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006966 return NULL;
6967 if (PyDict_SetItemString(d, "expr", (PyObject*)expr_type) < 0) return
6968 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006969 if (PyDict_SetItemString(d, "BoolOp", (PyObject*)BoolOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006970 return NULL;
6971 if (PyDict_SetItemString(d, "BinOp", (PyObject*)BinOp_type) < 0) return
6972 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006973 if (PyDict_SetItemString(d, "UnaryOp", (PyObject*)UnaryOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006974 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006975 if (PyDict_SetItemString(d, "Lambda", (PyObject*)Lambda_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006976 return NULL;
6977 if (PyDict_SetItemString(d, "IfExp", (PyObject*)IfExp_type) < 0) return
6978 NULL;
6979 if (PyDict_SetItemString(d, "Dict", (PyObject*)Dict_type) < 0) return
6980 NULL;
6981 if (PyDict_SetItemString(d, "Set", (PyObject*)Set_type) < 0) return
6982 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006983 if (PyDict_SetItemString(d, "ListComp", (PyObject*)ListComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006984 return NULL;
Nick Coghlan650f0d02007-04-15 12:05:43 +00006985 if (PyDict_SetItemString(d, "SetComp", (PyObject*)SetComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006986 return NULL;
Guido van Rossum992d4a32007-07-11 13:09:30 +00006987 if (PyDict_SetItemString(d, "DictComp", (PyObject*)DictComp_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, "GeneratorExp",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006990 (PyObject*)GeneratorExp_type) < 0) return NULL;
6991 if (PyDict_SetItemString(d, "Yield", (PyObject*)Yield_type) < 0) return
6992 NULL;
Benjamin Peterson527c6222012-01-14 08:58:23 -05006993 if (PyDict_SetItemString(d, "YieldFrom", (PyObject*)YieldFrom_type) <
6994 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006995 if (PyDict_SetItemString(d, "Compare", (PyObject*)Compare_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006996 return NULL;
6997 if (PyDict_SetItemString(d, "Call", (PyObject*)Call_type) < 0) return
6998 NULL;
6999 if (PyDict_SetItemString(d, "Num", (PyObject*)Num_type) < 0) return
7000 NULL;
7001 if (PyDict_SetItemString(d, "Str", (PyObject*)Str_type) < 0) return
7002 NULL;
7003 if (PyDict_SetItemString(d, "Bytes", (PyObject*)Bytes_type) < 0) return
7004 NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00007005 if (PyDict_SetItemString(d, "Ellipsis", (PyObject*)Ellipsis_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007006 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007007 if (PyDict_SetItemString(d, "Attribute", (PyObject*)Attribute_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007008 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007009 if (PyDict_SetItemString(d, "Subscript", (PyObject*)Subscript_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007010 0) return NULL;
Guido van Rossum0368b722007-05-11 16:50:42 +00007011 if (PyDict_SetItemString(d, "Starred", (PyObject*)Starred_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007012 return NULL;
7013 if (PyDict_SetItemString(d, "Name", (PyObject*)Name_type) < 0) return
7014 NULL;
7015 if (PyDict_SetItemString(d, "List", (PyObject*)List_type) < 0) return
7016 NULL;
7017 if (PyDict_SetItemString(d, "Tuple", (PyObject*)Tuple_type) < 0) return
7018 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007019 if (PyDict_SetItemString(d, "expr_context",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007020 (PyObject*)expr_context_type) < 0) return NULL;
7021 if (PyDict_SetItemString(d, "Load", (PyObject*)Load_type) < 0) return
7022 NULL;
7023 if (PyDict_SetItemString(d, "Store", (PyObject*)Store_type) < 0) return
7024 NULL;
7025 if (PyDict_SetItemString(d, "Del", (PyObject*)Del_type) < 0) return
7026 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007027 if (PyDict_SetItemString(d, "AugLoad", (PyObject*)AugLoad_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007028 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007029 if (PyDict_SetItemString(d, "AugStore", (PyObject*)AugStore_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007030 return NULL;
7031 if (PyDict_SetItemString(d, "Param", (PyObject*)Param_type) < 0) return
7032 NULL;
7033 if (PyDict_SetItemString(d, "slice", (PyObject*)slice_type) < 0) return
7034 NULL;
7035 if (PyDict_SetItemString(d, "Slice", (PyObject*)Slice_type) < 0) return
7036 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007037 if (PyDict_SetItemString(d, "ExtSlice", (PyObject*)ExtSlice_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007038 return NULL;
7039 if (PyDict_SetItemString(d, "Index", (PyObject*)Index_type) < 0) return
7040 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007041 if (PyDict_SetItemString(d, "boolop", (PyObject*)boolop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007042 return NULL;
7043 if (PyDict_SetItemString(d, "And", (PyObject*)And_type) < 0) return
7044 NULL;
7045 if (PyDict_SetItemString(d, "Or", (PyObject*)Or_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007046 if (PyDict_SetItemString(d, "operator", (PyObject*)operator_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007047 return NULL;
7048 if (PyDict_SetItemString(d, "Add", (PyObject*)Add_type) < 0) return
7049 NULL;
7050 if (PyDict_SetItemString(d, "Sub", (PyObject*)Sub_type) < 0) return
7051 NULL;
7052 if (PyDict_SetItemString(d, "Mult", (PyObject*)Mult_type) < 0) return
7053 NULL;
7054 if (PyDict_SetItemString(d, "Div", (PyObject*)Div_type) < 0) return
7055 NULL;
7056 if (PyDict_SetItemString(d, "Mod", (PyObject*)Mod_type) < 0) return
7057 NULL;
7058 if (PyDict_SetItemString(d, "Pow", (PyObject*)Pow_type) < 0) return
7059 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007060 if (PyDict_SetItemString(d, "LShift", (PyObject*)LShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007061 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007062 if (PyDict_SetItemString(d, "RShift", (PyObject*)RShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007063 return NULL;
7064 if (PyDict_SetItemString(d, "BitOr", (PyObject*)BitOr_type) < 0) return
7065 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007066 if (PyDict_SetItemString(d, "BitXor", (PyObject*)BitXor_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007067 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007068 if (PyDict_SetItemString(d, "BitAnd", (PyObject*)BitAnd_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007069 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007070 if (PyDict_SetItemString(d, "FloorDiv", (PyObject*)FloorDiv_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007071 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007072 if (PyDict_SetItemString(d, "unaryop", (PyObject*)unaryop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007073 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007074 if (PyDict_SetItemString(d, "Invert", (PyObject*)Invert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007075 return NULL;
7076 if (PyDict_SetItemString(d, "Not", (PyObject*)Not_type) < 0) return
7077 NULL;
7078 if (PyDict_SetItemString(d, "UAdd", (PyObject*)UAdd_type) < 0) return
7079 NULL;
7080 if (PyDict_SetItemString(d, "USub", (PyObject*)USub_type) < 0) return
7081 NULL;
7082 if (PyDict_SetItemString(d, "cmpop", (PyObject*)cmpop_type) < 0) return
7083 NULL;
7084 if (PyDict_SetItemString(d, "Eq", (PyObject*)Eq_type) < 0) return NULL;
7085 if (PyDict_SetItemString(d, "NotEq", (PyObject*)NotEq_type) < 0) return
7086 NULL;
7087 if (PyDict_SetItemString(d, "Lt", (PyObject*)Lt_type) < 0) return NULL;
7088 if (PyDict_SetItemString(d, "LtE", (PyObject*)LtE_type) < 0) return
7089 NULL;
7090 if (PyDict_SetItemString(d, "Gt", (PyObject*)Gt_type) < 0) return NULL;
7091 if (PyDict_SetItemString(d, "GtE", (PyObject*)GtE_type) < 0) return
7092 NULL;
7093 if (PyDict_SetItemString(d, "Is", (PyObject*)Is_type) < 0) return NULL;
7094 if (PyDict_SetItemString(d, "IsNot", (PyObject*)IsNot_type) < 0) return
7095 NULL;
7096 if (PyDict_SetItemString(d, "In", (PyObject*)In_type) < 0) return NULL;
7097 if (PyDict_SetItemString(d, "NotIn", (PyObject*)NotIn_type) < 0) return
7098 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007099 if (PyDict_SetItemString(d, "comprehension",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007100 (PyObject*)comprehension_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007101 if (PyDict_SetItemString(d, "excepthandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007102 (PyObject*)excepthandler_type) < 0) return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00007103 if (PyDict_SetItemString(d, "ExceptHandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007104 (PyObject*)ExceptHandler_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007105 if (PyDict_SetItemString(d, "arguments", (PyObject*)arguments_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007106 0) return NULL;
7107 if (PyDict_SetItemString(d, "arg", (PyObject*)arg_type) < 0) return
7108 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007109 if (PyDict_SetItemString(d, "keyword", (PyObject*)keyword_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007110 return NULL;
7111 if (PyDict_SetItemString(d, "alias", (PyObject*)alias_type) < 0) return
7112 NULL;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05007113 if (PyDict_SetItemString(d, "withitem", (PyObject*)withitem_type) < 0)
7114 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00007115 return m;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00007116}
7117
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007118
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007119PyObject* PyAST_mod2obj(mod_ty t)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007120{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007121 init_types();
7122 return ast2obj_mod(t);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007123}
Martin v. Löwis5b222132007-06-10 09:51:05 +00007124
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007125/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
7126mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007127{
7128 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007129 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
7130 (PyObject*)Interactive_type};
7131 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007132 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007133 assert(0 <= mode && mode <= 2);
7134
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007135 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007136
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007137 isinstance = PyObject_IsInstance(ast, req_type[mode]);
7138 if (isinstance == -1)
7139 return NULL;
7140 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007141 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
7142 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007143 return NULL;
7144 }
7145 if (obj2ast_mod(ast, &res, arena) != 0)
7146 return NULL;
7147 else
7148 return res;
7149}
7150
7151int PyAST_Check(PyObject* obj)
7152{
7153 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00007154 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007155}
7156
Martin v. Löwis5b222132007-06-10 09:51:05 +00007157