blob: e6f1e58252e4511f91d01edada85cd6b66fbacd8 [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;
Mark Dickinsonded35ae2012-11-25 14:36:26 +00001805 if (!value) {
1806 PyErr_SetString(PyExc_ValueError,
1807 "field value is required for YieldFrom");
1808 return NULL;
1809 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05001810 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1811 if (!p)
1812 return NULL;
1813 p->kind = YieldFrom_kind;
1814 p->v.YieldFrom.value = value;
1815 p->lineno = lineno;
1816 p->col_offset = col_offset;
1817 return p;
1818}
1819
1820expr_ty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001821Compare(expr_ty left, asdl_int_seq * ops, asdl_seq * comparators, int lineno,
1822 int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001823{
1824 expr_ty p;
1825 if (!left) {
1826 PyErr_SetString(PyExc_ValueError,
1827 "field left is required for Compare");
1828 return NULL;
1829 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001830 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001831 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001832 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001833 p->kind = Compare_kind;
1834 p->v.Compare.left = left;
1835 p->v.Compare.ops = ops;
1836 p->v.Compare.comparators = comparators;
1837 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001838 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001839 return p;
1840}
1841
1842expr_ty
1843Call(expr_ty func, asdl_seq * args, asdl_seq * keywords, expr_ty starargs,
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001844 expr_ty kwargs, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001845{
1846 expr_ty p;
1847 if (!func) {
1848 PyErr_SetString(PyExc_ValueError,
1849 "field func is required for Call");
1850 return NULL;
1851 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001852 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001853 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001854 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001855 p->kind = Call_kind;
1856 p->v.Call.func = func;
1857 p->v.Call.args = args;
1858 p->v.Call.keywords = keywords;
1859 p->v.Call.starargs = starargs;
1860 p->v.Call.kwargs = kwargs;
1861 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001862 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001863 return p;
1864}
1865
1866expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001867Num(object n, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001868{
1869 expr_ty p;
1870 if (!n) {
1871 PyErr_SetString(PyExc_ValueError,
1872 "field n is required for Num");
1873 return NULL;
1874 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001875 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001876 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001877 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001878 p->kind = Num_kind;
1879 p->v.Num.n = n;
1880 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001881 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001882 return p;
1883}
1884
1885expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001886Str(string s, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001887{
1888 expr_ty p;
1889 if (!s) {
1890 PyErr_SetString(PyExc_ValueError,
1891 "field s is required for Str");
1892 return NULL;
1893 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001894 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001895 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001896 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001897 p->kind = Str_kind;
1898 p->v.Str.s = s;
1899 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001900 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001901 return p;
1902}
1903
1904expr_ty
Benjamin Petersone2498412011-08-09 16:08:39 -05001905Bytes(bytes s, int lineno, int col_offset, PyArena *arena)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001906{
1907 expr_ty p;
1908 if (!s) {
1909 PyErr_SetString(PyExc_ValueError,
1910 "field s is required for Bytes");
1911 return NULL;
1912 }
1913 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001914 if (!p)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001915 return NULL;
Thomas Wouters00e41de2007-02-23 19:56:57 +00001916 p->kind = Bytes_kind;
1917 p->v.Bytes.s = s;
1918 p->lineno = lineno;
1919 p->col_offset = col_offset;
1920 return p;
1921}
1922
1923expr_ty
Georg Brandl52318d62006-09-06 07:06:08 +00001924Ellipsis(int lineno, int col_offset, PyArena *arena)
1925{
1926 expr_ty p;
1927 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001928 if (!p)
Georg Brandl52318d62006-09-06 07:06:08 +00001929 return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00001930 p->kind = Ellipsis_kind;
1931 p->lineno = lineno;
1932 p->col_offset = col_offset;
1933 return p;
1934}
1935
1936expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001937Attribute(expr_ty value, identifier attr, expr_context_ty ctx, int lineno, int
1938 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001939{
1940 expr_ty p;
1941 if (!value) {
1942 PyErr_SetString(PyExc_ValueError,
1943 "field value is required for Attribute");
1944 return NULL;
1945 }
1946 if (!attr) {
1947 PyErr_SetString(PyExc_ValueError,
1948 "field attr is required for Attribute");
1949 return NULL;
1950 }
1951 if (!ctx) {
1952 PyErr_SetString(PyExc_ValueError,
1953 "field ctx is required for Attribute");
1954 return NULL;
1955 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001956 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001957 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001958 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001959 p->kind = Attribute_kind;
1960 p->v.Attribute.value = value;
1961 p->v.Attribute.attr = attr;
1962 p->v.Attribute.ctx = ctx;
1963 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001964 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001965 return p;
1966}
1967
1968expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001969Subscript(expr_ty value, slice_ty slice, expr_context_ty ctx, int lineno, int
1970 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001971{
1972 expr_ty p;
1973 if (!value) {
1974 PyErr_SetString(PyExc_ValueError,
1975 "field value is required for Subscript");
1976 return NULL;
1977 }
1978 if (!slice) {
1979 PyErr_SetString(PyExc_ValueError,
1980 "field slice is required for Subscript");
1981 return NULL;
1982 }
1983 if (!ctx) {
1984 PyErr_SetString(PyExc_ValueError,
1985 "field ctx is required for Subscript");
1986 return NULL;
1987 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001988 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001989 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001990 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001991 p->kind = Subscript_kind;
1992 p->v.Subscript.value = value;
1993 p->v.Subscript.slice = slice;
1994 p->v.Subscript.ctx = ctx;
1995 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001996 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001997 return p;
1998}
1999
2000expr_ty
Guido van Rossum0368b722007-05-11 16:50:42 +00002001Starred(expr_ty value, expr_context_ty ctx, int lineno, int col_offset, PyArena
2002 *arena)
2003{
2004 expr_ty p;
2005 if (!value) {
2006 PyErr_SetString(PyExc_ValueError,
2007 "field value is required for Starred");
2008 return NULL;
2009 }
2010 if (!ctx) {
2011 PyErr_SetString(PyExc_ValueError,
2012 "field ctx is required for Starred");
2013 return NULL;
2014 }
2015 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
2016 if (!p)
2017 return NULL;
2018 p->kind = Starred_kind;
2019 p->v.Starred.value = value;
2020 p->v.Starred.ctx = ctx;
2021 p->lineno = lineno;
2022 p->col_offset = col_offset;
2023 return p;
2024}
2025
2026expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002027Name(identifier id, expr_context_ty ctx, int lineno, int col_offset, PyArena
2028 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002029{
2030 expr_ty p;
2031 if (!id) {
2032 PyErr_SetString(PyExc_ValueError,
2033 "field id is required for Name");
2034 return NULL;
2035 }
2036 if (!ctx) {
2037 PyErr_SetString(PyExc_ValueError,
2038 "field ctx is required for Name");
2039 return NULL;
2040 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002041 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002042 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002043 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002044 p->kind = Name_kind;
2045 p->v.Name.id = id;
2046 p->v.Name.ctx = ctx;
2047 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002048 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002049 return p;
2050}
2051
2052expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002053List(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2054 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002055{
2056 expr_ty p;
2057 if (!ctx) {
2058 PyErr_SetString(PyExc_ValueError,
2059 "field ctx is required for List");
2060 return NULL;
2061 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002062 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002063 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002064 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002065 p->kind = List_kind;
2066 p->v.List.elts = elts;
2067 p->v.List.ctx = ctx;
2068 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002069 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002070 return p;
2071}
2072
2073expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002074Tuple(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2075 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002076{
2077 expr_ty p;
2078 if (!ctx) {
2079 PyErr_SetString(PyExc_ValueError,
2080 "field ctx is required for Tuple");
2081 return NULL;
2082 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002083 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002084 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002085 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002086 p->kind = Tuple_kind;
2087 p->v.Tuple.elts = elts;
2088 p->v.Tuple.ctx = ctx;
2089 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002090 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002091 return p;
2092}
2093
2094slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002095Slice(expr_ty lower, expr_ty upper, expr_ty step, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002096{
2097 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002098 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002099 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002100 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002101 p->kind = Slice_kind;
2102 p->v.Slice.lower = lower;
2103 p->v.Slice.upper = upper;
2104 p->v.Slice.step = step;
2105 return p;
2106}
2107
2108slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002109ExtSlice(asdl_seq * dims, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002110{
2111 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002112 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002113 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002114 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002115 p->kind = ExtSlice_kind;
2116 p->v.ExtSlice.dims = dims;
2117 return p;
2118}
2119
2120slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002121Index(expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002122{
2123 slice_ty p;
2124 if (!value) {
2125 PyErr_SetString(PyExc_ValueError,
2126 "field value is required for Index");
2127 return NULL;
2128 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002129 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002130 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002131 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002132 p->kind = Index_kind;
2133 p->v.Index.value = value;
2134 return p;
2135}
2136
2137comprehension_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002138comprehension(expr_ty target, expr_ty iter, asdl_seq * ifs, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002139{
2140 comprehension_ty p;
2141 if (!target) {
2142 PyErr_SetString(PyExc_ValueError,
2143 "field target is required for comprehension");
2144 return NULL;
2145 }
2146 if (!iter) {
2147 PyErr_SetString(PyExc_ValueError,
2148 "field iter is required for comprehension");
2149 return NULL;
2150 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002151 p = (comprehension_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002152 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002153 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002154 p->target = target;
2155 p->iter = iter;
2156 p->ifs = ifs;
2157 return p;
2158}
2159
2160excepthandler_ty
Neal Norwitzad74aa82008-03-31 05:14:30 +00002161ExceptHandler(expr_ty type, identifier name, asdl_seq * body, int lineno, int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002162 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002163{
2164 excepthandler_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002165 p = (excepthandler_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002166 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002167 return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00002168 p->kind = ExceptHandler_kind;
2169 p->v.ExceptHandler.type = type;
2170 p->v.ExceptHandler.name = name;
2171 p->v.ExceptHandler.body = body;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002172 p->lineno = lineno;
2173 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002174 return p;
2175}
2176
2177arguments_ty
Neal Norwitzc1505362006-12-28 06:47:50 +00002178arguments(asdl_seq * args, identifier vararg, expr_ty varargannotation,
2179 asdl_seq * kwonlyargs, identifier kwarg, expr_ty kwargannotation,
2180 asdl_seq * defaults, asdl_seq * kw_defaults, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002181{
2182 arguments_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002183 p = (arguments_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002184 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002185 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002186 p->args = args;
2187 p->vararg = vararg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002188 p->varargannotation = varargannotation;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002189 p->kwonlyargs = kwonlyargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002190 p->kwarg = kwarg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002191 p->kwargannotation = kwargannotation;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002192 p->defaults = defaults;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002193 p->kw_defaults = kw_defaults;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002194 return p;
2195}
2196
Neal Norwitzc1505362006-12-28 06:47:50 +00002197arg_ty
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002198arg(identifier arg, expr_ty annotation, PyArena *arena)
Neal Norwitzc1505362006-12-28 06:47:50 +00002199{
2200 arg_ty p;
2201 if (!arg) {
2202 PyErr_SetString(PyExc_ValueError,
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002203 "field arg is required for arg");
Neal Norwitzc1505362006-12-28 06:47:50 +00002204 return NULL;
2205 }
2206 p = (arg_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002207 if (!p)
Neal Norwitzc1505362006-12-28 06:47:50 +00002208 return NULL;
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002209 p->arg = arg;
2210 p->annotation = annotation;
Neal Norwitzc1505362006-12-28 06:47:50 +00002211 return p;
2212}
2213
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002214keyword_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002215keyword(identifier arg, expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002216{
2217 keyword_ty p;
2218 if (!arg) {
2219 PyErr_SetString(PyExc_ValueError,
2220 "field arg is required for keyword");
2221 return NULL;
2222 }
2223 if (!value) {
2224 PyErr_SetString(PyExc_ValueError,
2225 "field value is required for keyword");
2226 return NULL;
2227 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002228 p = (keyword_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002229 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002230 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002231 p->arg = arg;
2232 p->value = value;
2233 return p;
2234}
2235
2236alias_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002237alias(identifier name, identifier asname, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002238{
2239 alias_ty p;
2240 if (!name) {
2241 PyErr_SetString(PyExc_ValueError,
2242 "field name is required for alias");
2243 return NULL;
2244 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002245 p = (alias_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002246 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002247 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002248 p->name = name;
2249 p->asname = asname;
2250 return p;
2251}
2252
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002253withitem_ty
2254withitem(expr_ty context_expr, expr_ty optional_vars, PyArena *arena)
2255{
2256 withitem_ty p;
2257 if (!context_expr) {
2258 PyErr_SetString(PyExc_ValueError,
2259 "field context_expr is required for withitem");
2260 return NULL;
2261 }
2262 p = (withitem_ty)PyArena_Malloc(arena, sizeof(*p));
2263 if (!p)
2264 return NULL;
2265 p->context_expr = context_expr;
2266 p->optional_vars = optional_vars;
2267 return p;
2268}
2269
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002270
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002271PyObject*
2272ast2obj_mod(void* _o)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002273{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002274 mod_ty o = (mod_ty)_o;
2275 PyObject *result = NULL, *value = NULL;
2276 if (!o) {
2277 Py_INCREF(Py_None);
2278 return Py_None;
2279 }
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002280
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002281 switch (o->kind) {
2282 case Module_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002283 result = PyType_GenericNew(Module_type, NULL, NULL);
2284 if (!result) goto failed;
2285 value = ast2obj_list(o->v.Module.body, ast2obj_stmt);
2286 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002287 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002288 goto failed;
2289 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002290 break;
2291 case Interactive_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002292 result = PyType_GenericNew(Interactive_type, NULL, NULL);
2293 if (!result) goto failed;
2294 value = ast2obj_list(o->v.Interactive.body, ast2obj_stmt);
2295 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002296 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002297 goto failed;
2298 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002299 break;
2300 case Expression_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002301 result = PyType_GenericNew(Expression_type, NULL, NULL);
2302 if (!result) goto failed;
2303 value = ast2obj_expr(o->v.Expression.body);
2304 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002305 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002306 goto failed;
2307 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002308 break;
2309 case Suite_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002310 result = PyType_GenericNew(Suite_type, NULL, NULL);
2311 if (!result) goto failed;
2312 value = ast2obj_list(o->v.Suite.body, ast2obj_stmt);
2313 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002314 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002315 goto failed;
2316 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002317 break;
2318 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002319 return result;
2320failed:
2321 Py_XDECREF(value);
2322 Py_XDECREF(result);
2323 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002324}
2325
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002326PyObject*
2327ast2obj_stmt(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002328{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002329 stmt_ty o = (stmt_ty)_o;
2330 PyObject *result = NULL, *value = NULL;
2331 if (!o) {
2332 Py_INCREF(Py_None);
2333 return Py_None;
2334 }
2335
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002336 switch (o->kind) {
2337 case FunctionDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002338 result = PyType_GenericNew(FunctionDef_type, NULL, NULL);
2339 if (!result) goto failed;
2340 value = ast2obj_identifier(o->v.FunctionDef.name);
2341 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002342 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002343 goto failed;
2344 Py_DECREF(value);
2345 value = ast2obj_arguments(o->v.FunctionDef.args);
2346 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002347 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002348 goto failed;
2349 Py_DECREF(value);
2350 value = ast2obj_list(o->v.FunctionDef.body, ast2obj_stmt);
2351 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002352 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002353 goto failed;
2354 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002355 value = ast2obj_list(o->v.FunctionDef.decorator_list,
2356 ast2obj_expr);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002357 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002358 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002359 -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002360 goto failed;
2361 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00002362 value = ast2obj_expr(o->v.FunctionDef.returns);
2363 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002364 if (_PyObject_SetAttrId(result, &PyId_returns, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00002365 goto failed;
2366 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002367 break;
2368 case ClassDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002369 result = PyType_GenericNew(ClassDef_type, NULL, NULL);
2370 if (!result) goto failed;
2371 value = ast2obj_identifier(o->v.ClassDef.name);
2372 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002373 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002374 goto failed;
2375 Py_DECREF(value);
2376 value = ast2obj_list(o->v.ClassDef.bases, ast2obj_expr);
2377 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002378 if (_PyObject_SetAttrId(result, &PyId_bases, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002379 goto failed;
2380 Py_DECREF(value);
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002381 value = ast2obj_list(o->v.ClassDef.keywords, ast2obj_keyword);
2382 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002383 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002384 goto failed;
2385 Py_DECREF(value);
2386 value = ast2obj_expr(o->v.ClassDef.starargs);
2387 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002388 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002389 goto failed;
2390 Py_DECREF(value);
2391 value = ast2obj_expr(o->v.ClassDef.kwargs);
2392 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002393 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002394 goto failed;
2395 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002396 value = ast2obj_list(o->v.ClassDef.body, ast2obj_stmt);
2397 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002398 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002399 goto failed;
2400 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002401 value = ast2obj_list(o->v.ClassDef.decorator_list,
2402 ast2obj_expr);
2403 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002404 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002405 -1)
2406 goto failed;
2407 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002408 break;
2409 case Return_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002410 result = PyType_GenericNew(Return_type, NULL, NULL);
2411 if (!result) goto failed;
2412 value = ast2obj_expr(o->v.Return.value);
2413 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002414 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002415 goto failed;
2416 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002417 break;
2418 case Delete_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002419 result = PyType_GenericNew(Delete_type, NULL, NULL);
2420 if (!result) goto failed;
2421 value = ast2obj_list(o->v.Delete.targets, ast2obj_expr);
2422 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002423 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002424 goto failed;
2425 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002426 break;
2427 case Assign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002428 result = PyType_GenericNew(Assign_type, NULL, NULL);
2429 if (!result) goto failed;
2430 value = ast2obj_list(o->v.Assign.targets, ast2obj_expr);
2431 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002432 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002433 goto failed;
2434 Py_DECREF(value);
2435 value = ast2obj_expr(o->v.Assign.value);
2436 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002437 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002438 goto failed;
2439 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002440 break;
2441 case AugAssign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002442 result = PyType_GenericNew(AugAssign_type, NULL, NULL);
2443 if (!result) goto failed;
2444 value = ast2obj_expr(o->v.AugAssign.target);
2445 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002446 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002447 goto failed;
2448 Py_DECREF(value);
2449 value = ast2obj_operator(o->v.AugAssign.op);
2450 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002451 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002452 goto failed;
2453 Py_DECREF(value);
2454 value = ast2obj_expr(o->v.AugAssign.value);
2455 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002456 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002457 goto failed;
2458 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002459 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002460 case For_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002461 result = PyType_GenericNew(For_type, NULL, NULL);
2462 if (!result) goto failed;
2463 value = ast2obj_expr(o->v.For.target);
2464 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002465 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002466 goto failed;
2467 Py_DECREF(value);
2468 value = ast2obj_expr(o->v.For.iter);
2469 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002470 if (_PyObject_SetAttrId(result, &PyId_iter, 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.body, ast2obj_stmt);
2474 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002475 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002476 goto failed;
2477 Py_DECREF(value);
2478 value = ast2obj_list(o->v.For.orelse, ast2obj_stmt);
2479 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002480 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002481 goto failed;
2482 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002483 break;
2484 case While_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002485 result = PyType_GenericNew(While_type, NULL, NULL);
2486 if (!result) goto failed;
2487 value = ast2obj_expr(o->v.While.test);
2488 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002489 if (_PyObject_SetAttrId(result, &PyId_test, 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.body, ast2obj_stmt);
2493 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002494 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002495 goto failed;
2496 Py_DECREF(value);
2497 value = ast2obj_list(o->v.While.orelse, ast2obj_stmt);
2498 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002499 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002500 goto failed;
2501 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002502 break;
2503 case If_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002504 result = PyType_GenericNew(If_type, NULL, NULL);
2505 if (!result) goto failed;
2506 value = ast2obj_expr(o->v.If.test);
2507 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002508 if (_PyObject_SetAttrId(result, &PyId_test, 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.body, ast2obj_stmt);
2512 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002513 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002514 goto failed;
2515 Py_DECREF(value);
2516 value = ast2obj_list(o->v.If.orelse, ast2obj_stmt);
2517 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002518 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002519 goto failed;
2520 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002521 break;
Guido van Rossumc2e20742006-02-27 22:32:47 +00002522 case With_kind:
2523 result = PyType_GenericNew(With_type, NULL, NULL);
2524 if (!result) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002525 value = ast2obj_list(o->v.With.items, ast2obj_withitem);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002526 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002527 if (_PyObject_SetAttrId(result, &PyId_items, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002528 goto failed;
2529 Py_DECREF(value);
2530 value = ast2obj_list(o->v.With.body, ast2obj_stmt);
2531 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002532 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002533 goto failed;
2534 Py_DECREF(value);
2535 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002536 case Raise_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002537 result = PyType_GenericNew(Raise_type, NULL, NULL);
2538 if (!result) goto failed;
Collin Winter828f04a2007-08-31 00:04:24 +00002539 value = ast2obj_expr(o->v.Raise.exc);
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_exc, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002542 goto failed;
2543 Py_DECREF(value);
Collin Winter828f04a2007-08-31 00:04:24 +00002544 value = ast2obj_expr(o->v.Raise.cause);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002545 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002546 if (_PyObject_SetAttrId(result, &PyId_cause, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002547 goto failed;
2548 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002549 break;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002550 case Try_kind:
2551 result = PyType_GenericNew(Try_type, NULL, NULL);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002552 if (!result) goto failed;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002553 value = ast2obj_list(o->v.Try.body, ast2obj_stmt);
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_body, 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.handlers, ast2obj_excepthandler);
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_handlers, 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.orelse, 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_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002566 goto failed;
2567 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002568 value = ast2obj_list(o->v.Try.finalbody, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002569 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002570 if (_PyObject_SetAttrId(result, &PyId_finalbody, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002571 goto failed;
2572 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002573 break;
2574 case Assert_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002575 result = PyType_GenericNew(Assert_type, NULL, NULL);
2576 if (!result) goto failed;
2577 value = ast2obj_expr(o->v.Assert.test);
2578 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002579 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002580 goto failed;
2581 Py_DECREF(value);
2582 value = ast2obj_expr(o->v.Assert.msg);
2583 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002584 if (_PyObject_SetAttrId(result, &PyId_msg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002585 goto failed;
2586 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002587 break;
2588 case Import_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002589 result = PyType_GenericNew(Import_type, NULL, NULL);
2590 if (!result) goto failed;
2591 value = ast2obj_list(o->v.Import.names, ast2obj_alias);
2592 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002593 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002594 goto failed;
2595 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002596 break;
2597 case ImportFrom_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002598 result = PyType_GenericNew(ImportFrom_type, NULL, NULL);
2599 if (!result) goto failed;
2600 value = ast2obj_identifier(o->v.ImportFrom.module);
2601 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002602 if (_PyObject_SetAttrId(result, &PyId_module, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002603 goto failed;
2604 Py_DECREF(value);
2605 value = ast2obj_list(o->v.ImportFrom.names, ast2obj_alias);
2606 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002607 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002608 goto failed;
2609 Py_DECREF(value);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002610 value = ast2obj_int(o->v.ImportFrom.level);
2611 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002612 if (_PyObject_SetAttrId(result, &PyId_level, value) == -1)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002613 goto failed;
2614 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002615 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002616 case Global_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002617 result = PyType_GenericNew(Global_type, NULL, NULL);
2618 if (!result) goto failed;
2619 value = ast2obj_list(o->v.Global.names, ast2obj_identifier);
2620 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002621 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002622 goto failed;
2623 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002624 break;
Jeremy Hylton81e95022007-02-27 06:50:52 +00002625 case Nonlocal_kind:
2626 result = PyType_GenericNew(Nonlocal_type, NULL, NULL);
2627 if (!result) goto failed;
2628 value = ast2obj_list(o->v.Nonlocal.names, ast2obj_identifier);
2629 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002630 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Jeremy Hylton81e95022007-02-27 06:50:52 +00002631 goto failed;
2632 Py_DECREF(value);
2633 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002634 case Expr_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002635 result = PyType_GenericNew(Expr_type, NULL, NULL);
2636 if (!result) goto failed;
2637 value = ast2obj_expr(o->v.Expr.value);
2638 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002639 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002640 goto failed;
2641 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002642 break;
2643 case Pass_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002644 result = PyType_GenericNew(Pass_type, NULL, NULL);
2645 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002646 break;
2647 case Break_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002648 result = PyType_GenericNew(Break_type, NULL, NULL);
2649 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002650 break;
2651 case Continue_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002652 result = PyType_GenericNew(Continue_type, NULL, NULL);
2653 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002654 break;
2655 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002656 value = ast2obj_int(o->lineno);
2657 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002658 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002659 goto failed;
2660 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002661 value = ast2obj_int(o->col_offset);
2662 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002663 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002664 goto failed;
2665 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002666 return result;
2667failed:
2668 Py_XDECREF(value);
2669 Py_XDECREF(result);
2670 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002671}
2672
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002673PyObject*
2674ast2obj_expr(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002675{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002676 expr_ty o = (expr_ty)_o;
2677 PyObject *result = NULL, *value = NULL;
2678 if (!o) {
2679 Py_INCREF(Py_None);
2680 return Py_None;
2681 }
2682
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002683 switch (o->kind) {
2684 case BoolOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002685 result = PyType_GenericNew(BoolOp_type, NULL, NULL);
2686 if (!result) goto failed;
2687 value = ast2obj_boolop(o->v.BoolOp.op);
2688 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002689 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002690 goto failed;
2691 Py_DECREF(value);
2692 value = ast2obj_list(o->v.BoolOp.values, ast2obj_expr);
2693 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002694 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002695 goto failed;
2696 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002697 break;
2698 case BinOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002699 result = PyType_GenericNew(BinOp_type, NULL, NULL);
2700 if (!result) goto failed;
2701 value = ast2obj_expr(o->v.BinOp.left);
2702 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002703 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002704 goto failed;
2705 Py_DECREF(value);
2706 value = ast2obj_operator(o->v.BinOp.op);
2707 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002708 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002709 goto failed;
2710 Py_DECREF(value);
2711 value = ast2obj_expr(o->v.BinOp.right);
2712 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002713 if (_PyObject_SetAttrId(result, &PyId_right, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002714 goto failed;
2715 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002716 break;
2717 case UnaryOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002718 result = PyType_GenericNew(UnaryOp_type, NULL, NULL);
2719 if (!result) goto failed;
2720 value = ast2obj_unaryop(o->v.UnaryOp.op);
2721 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002722 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002723 goto failed;
2724 Py_DECREF(value);
2725 value = ast2obj_expr(o->v.UnaryOp.operand);
2726 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002727 if (_PyObject_SetAttrId(result, &PyId_operand, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002728 goto failed;
2729 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002730 break;
2731 case Lambda_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002732 result = PyType_GenericNew(Lambda_type, NULL, NULL);
2733 if (!result) goto failed;
2734 value = ast2obj_arguments(o->v.Lambda.args);
2735 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002736 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002737 goto failed;
2738 Py_DECREF(value);
2739 value = ast2obj_expr(o->v.Lambda.body);
2740 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002741 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002742 goto failed;
2743 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002744 break;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002745 case IfExp_kind:
2746 result = PyType_GenericNew(IfExp_type, NULL, NULL);
2747 if (!result) goto failed;
2748 value = ast2obj_expr(o->v.IfExp.test);
2749 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002750 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002751 goto failed;
2752 Py_DECREF(value);
2753 value = ast2obj_expr(o->v.IfExp.body);
2754 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002755 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002756 goto failed;
2757 Py_DECREF(value);
2758 value = ast2obj_expr(o->v.IfExp.orelse);
2759 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002760 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002761 goto failed;
2762 Py_DECREF(value);
2763 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002764 case Dict_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002765 result = PyType_GenericNew(Dict_type, NULL, NULL);
2766 if (!result) goto failed;
2767 value = ast2obj_list(o->v.Dict.keys, ast2obj_expr);
2768 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002769 if (_PyObject_SetAttrId(result, &PyId_keys, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002770 goto failed;
2771 Py_DECREF(value);
2772 value = ast2obj_list(o->v.Dict.values, ast2obj_expr);
2773 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002774 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002775 goto failed;
2776 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002777 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002778 case Set_kind:
2779 result = PyType_GenericNew(Set_type, NULL, NULL);
2780 if (!result) goto failed;
2781 value = ast2obj_list(o->v.Set.elts, ast2obj_expr);
2782 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002783 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Guido van Rossum86e58e22006-08-28 15:27:34 +00002784 goto failed;
2785 Py_DECREF(value);
2786 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002787 case ListComp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002788 result = PyType_GenericNew(ListComp_type, NULL, NULL);
2789 if (!result) goto failed;
2790 value = ast2obj_expr(o->v.ListComp.elt);
2791 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002792 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002793 goto failed;
2794 Py_DECREF(value);
2795 value = ast2obj_list(o->v.ListComp.generators,
2796 ast2obj_comprehension);
2797 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002798 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002799 goto failed;
2800 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002801 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00002802 case SetComp_kind:
2803 result = PyType_GenericNew(SetComp_type, NULL, NULL);
2804 if (!result) goto failed;
2805 value = ast2obj_expr(o->v.SetComp.elt);
2806 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002807 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002808 goto failed;
2809 Py_DECREF(value);
2810 value = ast2obj_list(o->v.SetComp.generators,
2811 ast2obj_comprehension);
2812 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002813 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002814 goto failed;
2815 Py_DECREF(value);
2816 break;
Guido van Rossum992d4a32007-07-11 13:09:30 +00002817 case DictComp_kind:
2818 result = PyType_GenericNew(DictComp_type, NULL, NULL);
2819 if (!result) goto failed;
2820 value = ast2obj_expr(o->v.DictComp.key);
2821 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002822 if (_PyObject_SetAttrId(result, &PyId_key, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002823 goto failed;
2824 Py_DECREF(value);
2825 value = ast2obj_expr(o->v.DictComp.value);
2826 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002827 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002828 goto failed;
2829 Py_DECREF(value);
2830 value = ast2obj_list(o->v.DictComp.generators,
2831 ast2obj_comprehension);
2832 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002833 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002834 goto failed;
2835 Py_DECREF(value);
2836 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002837 case GeneratorExp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002838 result = PyType_GenericNew(GeneratorExp_type, NULL, NULL);
2839 if (!result) goto failed;
2840 value = ast2obj_expr(o->v.GeneratorExp.elt);
2841 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002842 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002843 goto failed;
2844 Py_DECREF(value);
2845 value = ast2obj_list(o->v.GeneratorExp.generators,
2846 ast2obj_comprehension);
2847 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002848 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002849 goto failed;
2850 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002851 break;
2852 case Yield_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002853 result = PyType_GenericNew(Yield_type, NULL, NULL);
2854 if (!result) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002855 value = ast2obj_expr(o->v.Yield.value);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002856 if (!value) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002857 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002858 goto failed;
2859 Py_DECREF(value);
Benjamin Peterson527c6222012-01-14 08:58:23 -05002860 break;
2861 case YieldFrom_kind:
2862 result = PyType_GenericNew(YieldFrom_type, NULL, NULL);
2863 if (!result) goto failed;
2864 value = ast2obj_expr(o->v.YieldFrom.value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002865 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002866 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002867 goto failed;
2868 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002869 break;
2870 case Compare_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002871 result = PyType_GenericNew(Compare_type, NULL, NULL);
2872 if (!result) goto failed;
2873 value = ast2obj_expr(o->v.Compare.left);
2874 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002875 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002876 goto failed;
2877 Py_DECREF(value);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002878 {
Benjamin Peterson77fa9372012-05-15 10:10:27 -07002879 Py_ssize_t i, n = asdl_seq_LEN(o->v.Compare.ops);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002880 value = PyList_New(n);
2881 if (!value) goto failed;
2882 for(i = 0; i < n; i++)
2883 PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(o->v.Compare.ops, i)));
2884 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002885 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002886 if (_PyObject_SetAttrId(result, &PyId_ops, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002887 goto failed;
2888 Py_DECREF(value);
2889 value = ast2obj_list(o->v.Compare.comparators, ast2obj_expr);
2890 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002891 if (_PyObject_SetAttrId(result, &PyId_comparators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002892 goto failed;
2893 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002894 break;
2895 case Call_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002896 result = PyType_GenericNew(Call_type, NULL, NULL);
2897 if (!result) goto failed;
2898 value = ast2obj_expr(o->v.Call.func);
2899 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002900 if (_PyObject_SetAttrId(result, &PyId_func, 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.args, ast2obj_expr);
2904 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002905 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002906 goto failed;
2907 Py_DECREF(value);
2908 value = ast2obj_list(o->v.Call.keywords, ast2obj_keyword);
2909 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002910 if (_PyObject_SetAttrId(result, &PyId_keywords, 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.starargs);
2914 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002915 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002916 goto failed;
2917 Py_DECREF(value);
2918 value = ast2obj_expr(o->v.Call.kwargs);
2919 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002920 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002921 goto failed;
2922 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002923 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002924 case Num_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002925 result = PyType_GenericNew(Num_type, NULL, NULL);
2926 if (!result) goto failed;
2927 value = ast2obj_object(o->v.Num.n);
2928 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002929 if (_PyObject_SetAttrId(result, &PyId_n, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002930 goto failed;
2931 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002932 break;
2933 case Str_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002934 result = PyType_GenericNew(Str_type, NULL, NULL);
2935 if (!result) goto failed;
2936 value = ast2obj_string(o->v.Str.s);
2937 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002938 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002939 goto failed;
2940 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002941 break;
Thomas Wouters00e41de2007-02-23 19:56:57 +00002942 case Bytes_kind:
2943 result = PyType_GenericNew(Bytes_type, NULL, NULL);
2944 if (!result) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05002945 value = ast2obj_bytes(o->v.Bytes.s);
Thomas Wouters00e41de2007-02-23 19:56:57 +00002946 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002947 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Thomas Wouters00e41de2007-02-23 19:56:57 +00002948 goto failed;
2949 Py_DECREF(value);
2950 break;
Georg Brandl52318d62006-09-06 07:06:08 +00002951 case Ellipsis_kind:
2952 result = PyType_GenericNew(Ellipsis_type, NULL, NULL);
2953 if (!result) goto failed;
2954 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002955 case Attribute_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002956 result = PyType_GenericNew(Attribute_type, NULL, NULL);
2957 if (!result) goto failed;
2958 value = ast2obj_expr(o->v.Attribute.value);
2959 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002960 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002961 goto failed;
2962 Py_DECREF(value);
2963 value = ast2obj_identifier(o->v.Attribute.attr);
2964 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002965 if (_PyObject_SetAttrId(result, &PyId_attr, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002966 goto failed;
2967 Py_DECREF(value);
2968 value = ast2obj_expr_context(o->v.Attribute.ctx);
2969 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002970 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002971 goto failed;
2972 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002973 break;
2974 case Subscript_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002975 result = PyType_GenericNew(Subscript_type, NULL, NULL);
2976 if (!result) goto failed;
2977 value = ast2obj_expr(o->v.Subscript.value);
2978 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002979 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002980 goto failed;
2981 Py_DECREF(value);
2982 value = ast2obj_slice(o->v.Subscript.slice);
2983 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002984 if (_PyObject_SetAttrId(result, &PyId_slice, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002985 goto failed;
2986 Py_DECREF(value);
2987 value = ast2obj_expr_context(o->v.Subscript.ctx);
2988 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002989 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002990 goto failed;
2991 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002992 break;
Guido van Rossum0368b722007-05-11 16:50:42 +00002993 case Starred_kind:
2994 result = PyType_GenericNew(Starred_type, NULL, NULL);
2995 if (!result) goto failed;
2996 value = ast2obj_expr(o->v.Starred.value);
2997 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002998 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00002999 goto failed;
3000 Py_DECREF(value);
3001 value = ast2obj_expr_context(o->v.Starred.ctx);
3002 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003003 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00003004 goto failed;
3005 Py_DECREF(value);
3006 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003007 case Name_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003008 result = PyType_GenericNew(Name_type, NULL, NULL);
3009 if (!result) goto failed;
3010 value = ast2obj_identifier(o->v.Name.id);
3011 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003012 if (_PyObject_SetAttrId(result, &PyId_id, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003013 goto failed;
3014 Py_DECREF(value);
3015 value = ast2obj_expr_context(o->v.Name.ctx);
3016 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003017 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003018 goto failed;
3019 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003020 break;
3021 case List_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003022 result = PyType_GenericNew(List_type, NULL, NULL);
3023 if (!result) goto failed;
3024 value = ast2obj_list(o->v.List.elts, ast2obj_expr);
3025 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003026 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003027 goto failed;
3028 Py_DECREF(value);
3029 value = ast2obj_expr_context(o->v.List.ctx);
3030 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003031 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003032 goto failed;
3033 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003034 break;
3035 case Tuple_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003036 result = PyType_GenericNew(Tuple_type, NULL, NULL);
3037 if (!result) goto failed;
3038 value = ast2obj_list(o->v.Tuple.elts, ast2obj_expr);
3039 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003040 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003041 goto failed;
3042 Py_DECREF(value);
3043 value = ast2obj_expr_context(o->v.Tuple.ctx);
3044 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003045 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003046 goto failed;
3047 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003048 break;
3049 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00003050 value = ast2obj_int(o->lineno);
3051 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003052 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003053 goto failed;
3054 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00003055 value = ast2obj_int(o->col_offset);
3056 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003057 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003058 goto failed;
3059 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003060 return result;
3061failed:
3062 Py_XDECREF(value);
3063 Py_XDECREF(result);
3064 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003065}
3066
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003067PyObject* ast2obj_expr_context(expr_context_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003068{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003069 switch(o) {
3070 case Load:
3071 Py_INCREF(Load_singleton);
3072 return Load_singleton;
3073 case Store:
3074 Py_INCREF(Store_singleton);
3075 return Store_singleton;
3076 case Del:
3077 Py_INCREF(Del_singleton);
3078 return Del_singleton;
3079 case AugLoad:
3080 Py_INCREF(AugLoad_singleton);
3081 return AugLoad_singleton;
3082 case AugStore:
3083 Py_INCREF(AugStore_singleton);
3084 return AugStore_singleton;
3085 case Param:
3086 Py_INCREF(Param_singleton);
3087 return Param_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003088 default:
3089 /* should never happen, but just in case ... */
3090 PyErr_Format(PyExc_SystemError, "unknown expr_context found");
3091 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003092 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003093}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003094PyObject*
3095ast2obj_slice(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003096{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003097 slice_ty o = (slice_ty)_o;
3098 PyObject *result = NULL, *value = NULL;
3099 if (!o) {
3100 Py_INCREF(Py_None);
3101 return Py_None;
3102 }
3103
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003104 switch (o->kind) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003105 case Slice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003106 result = PyType_GenericNew(Slice_type, NULL, NULL);
3107 if (!result) goto failed;
3108 value = ast2obj_expr(o->v.Slice.lower);
3109 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003110 if (_PyObject_SetAttrId(result, &PyId_lower, 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.upper);
3114 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003115 if (_PyObject_SetAttrId(result, &PyId_upper, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003116 goto failed;
3117 Py_DECREF(value);
3118 value = ast2obj_expr(o->v.Slice.step);
3119 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003120 if (_PyObject_SetAttrId(result, &PyId_step, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003121 goto failed;
3122 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003123 break;
3124 case ExtSlice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003125 result = PyType_GenericNew(ExtSlice_type, NULL, NULL);
3126 if (!result) goto failed;
3127 value = ast2obj_list(o->v.ExtSlice.dims, ast2obj_slice);
3128 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003129 if (_PyObject_SetAttrId(result, &PyId_dims, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003130 goto failed;
3131 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003132 break;
3133 case Index_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003134 result = PyType_GenericNew(Index_type, NULL, NULL);
3135 if (!result) goto failed;
3136 value = ast2obj_expr(o->v.Index.value);
3137 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003138 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003139 goto failed;
3140 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003141 break;
3142 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003143 return result;
3144failed:
3145 Py_XDECREF(value);
3146 Py_XDECREF(result);
3147 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003148}
3149
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003150PyObject* ast2obj_boolop(boolop_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003151{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003152 switch(o) {
3153 case And:
3154 Py_INCREF(And_singleton);
3155 return And_singleton;
3156 case Or:
3157 Py_INCREF(Or_singleton);
3158 return Or_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003159 default:
3160 /* should never happen, but just in case ... */
3161 PyErr_Format(PyExc_SystemError, "unknown boolop found");
3162 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003163 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003164}
3165PyObject* ast2obj_operator(operator_ty o)
3166{
3167 switch(o) {
3168 case Add:
3169 Py_INCREF(Add_singleton);
3170 return Add_singleton;
3171 case Sub:
3172 Py_INCREF(Sub_singleton);
3173 return Sub_singleton;
3174 case Mult:
3175 Py_INCREF(Mult_singleton);
3176 return Mult_singleton;
3177 case Div:
3178 Py_INCREF(Div_singleton);
3179 return Div_singleton;
3180 case Mod:
3181 Py_INCREF(Mod_singleton);
3182 return Mod_singleton;
3183 case Pow:
3184 Py_INCREF(Pow_singleton);
3185 return Pow_singleton;
3186 case LShift:
3187 Py_INCREF(LShift_singleton);
3188 return LShift_singleton;
3189 case RShift:
3190 Py_INCREF(RShift_singleton);
3191 return RShift_singleton;
3192 case BitOr:
3193 Py_INCREF(BitOr_singleton);
3194 return BitOr_singleton;
3195 case BitXor:
3196 Py_INCREF(BitXor_singleton);
3197 return BitXor_singleton;
3198 case BitAnd:
3199 Py_INCREF(BitAnd_singleton);
3200 return BitAnd_singleton;
3201 case FloorDiv:
3202 Py_INCREF(FloorDiv_singleton);
3203 return FloorDiv_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003204 default:
3205 /* should never happen, but just in case ... */
3206 PyErr_Format(PyExc_SystemError, "unknown operator found");
3207 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003208 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003209}
3210PyObject* ast2obj_unaryop(unaryop_ty o)
3211{
3212 switch(o) {
3213 case Invert:
3214 Py_INCREF(Invert_singleton);
3215 return Invert_singleton;
3216 case Not:
3217 Py_INCREF(Not_singleton);
3218 return Not_singleton;
3219 case UAdd:
3220 Py_INCREF(UAdd_singleton);
3221 return UAdd_singleton;
3222 case USub:
3223 Py_INCREF(USub_singleton);
3224 return USub_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003225 default:
3226 /* should never happen, but just in case ... */
3227 PyErr_Format(PyExc_SystemError, "unknown unaryop found");
3228 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003229 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003230}
3231PyObject* ast2obj_cmpop(cmpop_ty o)
3232{
3233 switch(o) {
3234 case Eq:
3235 Py_INCREF(Eq_singleton);
3236 return Eq_singleton;
3237 case NotEq:
3238 Py_INCREF(NotEq_singleton);
3239 return NotEq_singleton;
3240 case Lt:
3241 Py_INCREF(Lt_singleton);
3242 return Lt_singleton;
3243 case LtE:
3244 Py_INCREF(LtE_singleton);
3245 return LtE_singleton;
3246 case Gt:
3247 Py_INCREF(Gt_singleton);
3248 return Gt_singleton;
3249 case GtE:
3250 Py_INCREF(GtE_singleton);
3251 return GtE_singleton;
3252 case Is:
3253 Py_INCREF(Is_singleton);
3254 return Is_singleton;
3255 case IsNot:
3256 Py_INCREF(IsNot_singleton);
3257 return IsNot_singleton;
3258 case In:
3259 Py_INCREF(In_singleton);
3260 return In_singleton;
3261 case NotIn:
3262 Py_INCREF(NotIn_singleton);
3263 return NotIn_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003264 default:
3265 /* should never happen, but just in case ... */
3266 PyErr_Format(PyExc_SystemError, "unknown cmpop found");
3267 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003268 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003269}
3270PyObject*
3271ast2obj_comprehension(void* _o)
3272{
3273 comprehension_ty o = (comprehension_ty)_o;
3274 PyObject *result = NULL, *value = NULL;
3275 if (!o) {
3276 Py_INCREF(Py_None);
3277 return Py_None;
3278 }
3279
3280 result = PyType_GenericNew(comprehension_type, NULL, NULL);
3281 if (!result) return NULL;
3282 value = ast2obj_expr(o->target);
3283 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003284 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003285 goto failed;
3286 Py_DECREF(value);
3287 value = ast2obj_expr(o->iter);
3288 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003289 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003290 goto failed;
3291 Py_DECREF(value);
3292 value = ast2obj_list(o->ifs, ast2obj_expr);
3293 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003294 if (_PyObject_SetAttrId(result, &PyId_ifs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003295 goto failed;
3296 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003297 return result;
3298failed:
3299 Py_XDECREF(value);
3300 Py_XDECREF(result);
3301 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003302}
3303
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003304PyObject*
3305ast2obj_excepthandler(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003306{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003307 excepthandler_ty o = (excepthandler_ty)_o;
3308 PyObject *result = NULL, *value = NULL;
3309 if (!o) {
3310 Py_INCREF(Py_None);
3311 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003312 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003313
Neal Norwitzad74aa82008-03-31 05:14:30 +00003314 switch (o->kind) {
3315 case ExceptHandler_kind:
3316 result = PyType_GenericNew(ExceptHandler_type, NULL, NULL);
3317 if (!result) goto failed;
3318 value = ast2obj_expr(o->v.ExceptHandler.type);
3319 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003320 if (_PyObject_SetAttrId(result, &PyId_type, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003321 goto failed;
3322 Py_DECREF(value);
3323 value = ast2obj_identifier(o->v.ExceptHandler.name);
3324 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003325 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003326 goto failed;
3327 Py_DECREF(value);
3328 value = ast2obj_list(o->v.ExceptHandler.body, ast2obj_stmt);
3329 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003330 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003331 goto failed;
3332 Py_DECREF(value);
3333 break;
3334 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003335 value = ast2obj_int(o->lineno);
3336 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003337 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003338 goto failed;
3339 Py_DECREF(value);
3340 value = ast2obj_int(o->col_offset);
3341 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003342 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003343 goto failed;
3344 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003345 return result;
3346failed:
3347 Py_XDECREF(value);
3348 Py_XDECREF(result);
3349 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003350}
3351
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003352PyObject*
3353ast2obj_arguments(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003354{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003355 arguments_ty o = (arguments_ty)_o;
3356 PyObject *result = NULL, *value = NULL;
3357 if (!o) {
3358 Py_INCREF(Py_None);
3359 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003360 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003361
3362 result = PyType_GenericNew(arguments_type, NULL, NULL);
3363 if (!result) return NULL;
Neal Norwitzc1505362006-12-28 06:47:50 +00003364 value = ast2obj_list(o->args, ast2obj_arg);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003365 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003366 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003367 goto failed;
3368 Py_DECREF(value);
3369 value = ast2obj_identifier(o->vararg);
3370 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003371 if (_PyObject_SetAttrId(result, &PyId_vararg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003372 goto failed;
3373 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003374 value = ast2obj_expr(o->varargannotation);
3375 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003376 if (_PyObject_SetAttrId(result, &PyId_varargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003377 goto failed;
3378 Py_DECREF(value);
3379 value = ast2obj_list(o->kwonlyargs, ast2obj_arg);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003380 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003381 if (_PyObject_SetAttrId(result, &PyId_kwonlyargs, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003382 goto failed;
3383 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003384 value = ast2obj_identifier(o->kwarg);
3385 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003386 if (_PyObject_SetAttrId(result, &PyId_kwarg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003387 goto failed;
3388 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003389 value = ast2obj_expr(o->kwargannotation);
3390 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003391 if (_PyObject_SetAttrId(result, &PyId_kwargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003392 goto failed;
3393 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003394 value = ast2obj_list(o->defaults, ast2obj_expr);
3395 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003396 if (_PyObject_SetAttrId(result, &PyId_defaults, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003397 goto failed;
3398 Py_DECREF(value);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003399 value = ast2obj_list(o->kw_defaults, ast2obj_expr);
3400 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003401 if (_PyObject_SetAttrId(result, &PyId_kw_defaults, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003402 goto failed;
3403 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003404 return result;
3405failed:
3406 Py_XDECREF(value);
3407 Py_XDECREF(result);
3408 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003409}
3410
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003411PyObject*
Neal Norwitzc1505362006-12-28 06:47:50 +00003412ast2obj_arg(void* _o)
3413{
3414 arg_ty o = (arg_ty)_o;
3415 PyObject *result = NULL, *value = NULL;
3416 if (!o) {
3417 Py_INCREF(Py_None);
3418 return Py_None;
3419 }
3420
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003421 result = PyType_GenericNew(arg_type, NULL, NULL);
3422 if (!result) return NULL;
3423 value = ast2obj_identifier(o->arg);
3424 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003425 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003426 goto failed;
3427 Py_DECREF(value);
3428 value = ast2obj_expr(o->annotation);
3429 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003430 if (_PyObject_SetAttrId(result, &PyId_annotation, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003431 goto failed;
3432 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003433 return result;
3434failed:
3435 Py_XDECREF(value);
3436 Py_XDECREF(result);
3437 return NULL;
3438}
3439
3440PyObject*
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003441ast2obj_keyword(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003442{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003443 keyword_ty o = (keyword_ty)_o;
3444 PyObject *result = NULL, *value = NULL;
3445 if (!o) {
3446 Py_INCREF(Py_None);
3447 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003448 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003449
3450 result = PyType_GenericNew(keyword_type, NULL, NULL);
3451 if (!result) return NULL;
3452 value = ast2obj_identifier(o->arg);
3453 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003454 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003455 goto failed;
3456 Py_DECREF(value);
3457 value = ast2obj_expr(o->value);
3458 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003459 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003460 goto failed;
3461 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003462 return result;
3463failed:
3464 Py_XDECREF(value);
3465 Py_XDECREF(result);
3466 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003467}
3468
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003469PyObject*
3470ast2obj_alias(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003471{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003472 alias_ty o = (alias_ty)_o;
3473 PyObject *result = NULL, *value = NULL;
3474 if (!o) {
3475 Py_INCREF(Py_None);
3476 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003477 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003478
3479 result = PyType_GenericNew(alias_type, NULL, NULL);
3480 if (!result) return NULL;
3481 value = ast2obj_identifier(o->name);
3482 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003483 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003484 goto failed;
3485 Py_DECREF(value);
3486 value = ast2obj_identifier(o->asname);
3487 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003488 if (_PyObject_SetAttrId(result, &PyId_asname, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003489 goto failed;
3490 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003491 return result;
3492failed:
3493 Py_XDECREF(value);
3494 Py_XDECREF(result);
3495 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003496}
3497
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003498PyObject*
3499ast2obj_withitem(void* _o)
3500{
3501 withitem_ty o = (withitem_ty)_o;
3502 PyObject *result = NULL, *value = NULL;
3503 if (!o) {
3504 Py_INCREF(Py_None);
3505 return Py_None;
3506 }
3507
3508 result = PyType_GenericNew(withitem_type, NULL, NULL);
3509 if (!result) return NULL;
3510 value = ast2obj_expr(o->context_expr);
3511 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003512 if (_PyObject_SetAttrId(result, &PyId_context_expr, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003513 goto failed;
3514 Py_DECREF(value);
3515 value = ast2obj_expr(o->optional_vars);
3516 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003517 if (_PyObject_SetAttrId(result, &PyId_optional_vars, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003518 goto failed;
3519 Py_DECREF(value);
3520 return result;
3521failed:
3522 Py_XDECREF(value);
3523 Py_XDECREF(result);
3524 return NULL;
3525}
3526
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003527
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003528int
3529obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena)
3530{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003531 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003532
Benjamin Petersond8f65972010-11-20 04:31:07 +00003533 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003534
3535 if (obj == Py_None) {
3536 *out = NULL;
3537 return 0;
3538 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003539 isinstance = PyObject_IsInstance(obj, (PyObject*)Module_type);
3540 if (isinstance == -1) {
3541 return 1;
3542 }
3543 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003544 asdl_seq* body;
3545
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003546 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003547 int res;
3548 Py_ssize_t len;
3549 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003550 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003551 if (tmp == NULL) goto failed;
3552 if (!PyList_Check(tmp)) {
3553 PyErr_Format(PyExc_TypeError, "Module field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3554 goto failed;
3555 }
3556 len = PyList_GET_SIZE(tmp);
3557 body = asdl_seq_new(len, arena);
3558 if (body == NULL) goto failed;
3559 for (i = 0; i < len; i++) {
3560 stmt_ty value;
3561 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3562 if (res != 0) goto failed;
3563 asdl_seq_SET(body, i, value);
3564 }
3565 Py_XDECREF(tmp);
3566 tmp = NULL;
3567 } else {
3568 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Module");
3569 return 1;
3570 }
3571 *out = Module(body, arena);
3572 if (*out == NULL) goto failed;
3573 return 0;
3574 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003575 isinstance = PyObject_IsInstance(obj, (PyObject*)Interactive_type);
3576 if (isinstance == -1) {
3577 return 1;
3578 }
3579 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003580 asdl_seq* body;
3581
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003582 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003583 int res;
3584 Py_ssize_t len;
3585 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003586 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003587 if (tmp == NULL) goto failed;
3588 if (!PyList_Check(tmp)) {
3589 PyErr_Format(PyExc_TypeError, "Interactive field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3590 goto failed;
3591 }
3592 len = PyList_GET_SIZE(tmp);
3593 body = asdl_seq_new(len, arena);
3594 if (body == NULL) goto failed;
3595 for (i = 0; i < len; i++) {
3596 stmt_ty value;
3597 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3598 if (res != 0) goto failed;
3599 asdl_seq_SET(body, i, value);
3600 }
3601 Py_XDECREF(tmp);
3602 tmp = NULL;
3603 } else {
3604 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Interactive");
3605 return 1;
3606 }
3607 *out = Interactive(body, arena);
3608 if (*out == NULL) goto failed;
3609 return 0;
3610 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003611 isinstance = PyObject_IsInstance(obj, (PyObject*)Expression_type);
3612 if (isinstance == -1) {
3613 return 1;
3614 }
3615 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003616 expr_ty body;
3617
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003618 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003619 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003620 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003621 if (tmp == NULL) goto failed;
3622 res = obj2ast_expr(tmp, &body, arena);
3623 if (res != 0) goto failed;
3624 Py_XDECREF(tmp);
3625 tmp = NULL;
3626 } else {
3627 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Expression");
3628 return 1;
3629 }
3630 *out = Expression(body, arena);
3631 if (*out == NULL) goto failed;
3632 return 0;
3633 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003634 isinstance = PyObject_IsInstance(obj, (PyObject*)Suite_type);
3635 if (isinstance == -1) {
3636 return 1;
3637 }
3638 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003639 asdl_seq* body;
3640
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003641 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003642 int res;
3643 Py_ssize_t len;
3644 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003645 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003646 if (tmp == NULL) goto failed;
3647 if (!PyList_Check(tmp)) {
3648 PyErr_Format(PyExc_TypeError, "Suite field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3649 goto failed;
3650 }
3651 len = PyList_GET_SIZE(tmp);
3652 body = asdl_seq_new(len, arena);
3653 if (body == NULL) goto failed;
3654 for (i = 0; i < len; i++) {
3655 stmt_ty value;
3656 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3657 if (res != 0) goto failed;
3658 asdl_seq_SET(body, i, value);
3659 }
3660 Py_XDECREF(tmp);
3661 tmp = NULL;
3662 } else {
3663 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Suite");
3664 return 1;
3665 }
3666 *out = Suite(body, arena);
3667 if (*out == NULL) goto failed;
3668 return 0;
3669 }
3670
Benjamin Peterson5b066812010-11-20 01:38:49 +00003671 PyErr_Format(PyExc_TypeError, "expected some sort of mod, but got %R", obj);
3672 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00003673 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003674 return 1;
3675}
3676
3677int
3678obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena)
3679{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003680 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003681
Benjamin Petersond8f65972010-11-20 04:31:07 +00003682 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003683 int lineno;
3684 int col_offset;
3685
3686 if (obj == Py_None) {
3687 *out = NULL;
3688 return 0;
3689 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003690 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003691 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003692 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003693 if (tmp == NULL) goto failed;
3694 res = obj2ast_int(tmp, &lineno, arena);
3695 if (res != 0) goto failed;
3696 Py_XDECREF(tmp);
3697 tmp = NULL;
3698 } else {
3699 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from stmt");
3700 return 1;
3701 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003702 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003703 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003704 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003705 if (tmp == NULL) goto failed;
3706 res = obj2ast_int(tmp, &col_offset, arena);
3707 if (res != 0) goto failed;
3708 Py_XDECREF(tmp);
3709 tmp = NULL;
3710 } else {
3711 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from stmt");
3712 return 1;
3713 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003714 isinstance = PyObject_IsInstance(obj, (PyObject*)FunctionDef_type);
3715 if (isinstance == -1) {
3716 return 1;
3717 }
3718 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003719 identifier name;
3720 arguments_ty args;
3721 asdl_seq* body;
3722 asdl_seq* decorator_list;
3723 expr_ty returns;
3724
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003725 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003726 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003727 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003728 if (tmp == NULL) goto failed;
3729 res = obj2ast_identifier(tmp, &name, arena);
3730 if (res != 0) goto failed;
3731 Py_XDECREF(tmp);
3732 tmp = NULL;
3733 } else {
3734 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from FunctionDef");
3735 return 1;
3736 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003737 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003738 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003739 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003740 if (tmp == NULL) goto failed;
3741 res = obj2ast_arguments(tmp, &args, arena);
3742 if (res != 0) goto failed;
3743 Py_XDECREF(tmp);
3744 tmp = NULL;
3745 } else {
3746 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from FunctionDef");
3747 return 1;
3748 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003749 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003750 int res;
3751 Py_ssize_t len;
3752 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003753 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003754 if (tmp == NULL) goto failed;
3755 if (!PyList_Check(tmp)) {
3756 PyErr_Format(PyExc_TypeError, "FunctionDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3757 goto failed;
3758 }
3759 len = PyList_GET_SIZE(tmp);
3760 body = asdl_seq_new(len, arena);
3761 if (body == NULL) goto failed;
3762 for (i = 0; i < len; i++) {
3763 stmt_ty value;
3764 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3765 if (res != 0) goto failed;
3766 asdl_seq_SET(body, i, value);
3767 }
3768 Py_XDECREF(tmp);
3769 tmp = NULL;
3770 } else {
3771 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from FunctionDef");
3772 return 1;
3773 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003774 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003775 int res;
3776 Py_ssize_t len;
3777 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003778 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003779 if (tmp == NULL) goto failed;
3780 if (!PyList_Check(tmp)) {
3781 PyErr_Format(PyExc_TypeError, "FunctionDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3782 goto failed;
3783 }
3784 len = PyList_GET_SIZE(tmp);
3785 decorator_list = asdl_seq_new(len, arena);
3786 if (decorator_list == NULL) goto failed;
3787 for (i = 0; i < len; i++) {
3788 expr_ty value;
3789 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3790 if (res != 0) goto failed;
3791 asdl_seq_SET(decorator_list, i, value);
3792 }
3793 Py_XDECREF(tmp);
3794 tmp = NULL;
3795 } else {
3796 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from FunctionDef");
3797 return 1;
3798 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003799 if (_PyObject_HasAttrId(obj, &PyId_returns)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003800 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003801 tmp = _PyObject_GetAttrId(obj, &PyId_returns);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003802 if (tmp == NULL) goto failed;
3803 res = obj2ast_expr(tmp, &returns, arena);
3804 if (res != 0) goto failed;
3805 Py_XDECREF(tmp);
3806 tmp = NULL;
3807 } else {
3808 returns = NULL;
3809 }
3810 *out = FunctionDef(name, args, body, decorator_list, returns,
3811 lineno, col_offset, arena);
3812 if (*out == NULL) goto failed;
3813 return 0;
3814 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003815 isinstance = PyObject_IsInstance(obj, (PyObject*)ClassDef_type);
3816 if (isinstance == -1) {
3817 return 1;
3818 }
3819 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003820 identifier name;
3821 asdl_seq* bases;
3822 asdl_seq* keywords;
3823 expr_ty starargs;
3824 expr_ty kwargs;
3825 asdl_seq* body;
3826 asdl_seq* decorator_list;
3827
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003828 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003829 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003830 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003831 if (tmp == NULL) goto failed;
3832 res = obj2ast_identifier(tmp, &name, arena);
3833 if (res != 0) goto failed;
3834 Py_XDECREF(tmp);
3835 tmp = NULL;
3836 } else {
3837 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from ClassDef");
3838 return 1;
3839 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003840 if (_PyObject_HasAttrId(obj, &PyId_bases)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003841 int res;
3842 Py_ssize_t len;
3843 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003844 tmp = _PyObject_GetAttrId(obj, &PyId_bases);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003845 if (tmp == NULL) goto failed;
3846 if (!PyList_Check(tmp)) {
3847 PyErr_Format(PyExc_TypeError, "ClassDef field \"bases\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3848 goto failed;
3849 }
3850 len = PyList_GET_SIZE(tmp);
3851 bases = asdl_seq_new(len, arena);
3852 if (bases == NULL) goto failed;
3853 for (i = 0; i < len; i++) {
3854 expr_ty value;
3855 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3856 if (res != 0) goto failed;
3857 asdl_seq_SET(bases, i, value);
3858 }
3859 Py_XDECREF(tmp);
3860 tmp = NULL;
3861 } else {
3862 PyErr_SetString(PyExc_TypeError, "required field \"bases\" missing from ClassDef");
3863 return 1;
3864 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003865 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003866 int res;
3867 Py_ssize_t len;
3868 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003869 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003870 if (tmp == NULL) goto failed;
3871 if (!PyList_Check(tmp)) {
3872 PyErr_Format(PyExc_TypeError, "ClassDef field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3873 goto failed;
3874 }
3875 len = PyList_GET_SIZE(tmp);
3876 keywords = asdl_seq_new(len, arena);
3877 if (keywords == NULL) goto failed;
3878 for (i = 0; i < len; i++) {
3879 keyword_ty value;
3880 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
3881 if (res != 0) goto failed;
3882 asdl_seq_SET(keywords, i, value);
3883 }
3884 Py_XDECREF(tmp);
3885 tmp = NULL;
3886 } else {
3887 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from ClassDef");
3888 return 1;
3889 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003890 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003891 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003892 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003893 if (tmp == NULL) goto failed;
3894 res = obj2ast_expr(tmp, &starargs, arena);
3895 if (res != 0) goto failed;
3896 Py_XDECREF(tmp);
3897 tmp = NULL;
3898 } else {
3899 starargs = NULL;
3900 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003901 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003902 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003903 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003904 if (tmp == NULL) goto failed;
3905 res = obj2ast_expr(tmp, &kwargs, arena);
3906 if (res != 0) goto failed;
3907 Py_XDECREF(tmp);
3908 tmp = NULL;
3909 } else {
3910 kwargs = NULL;
3911 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003912 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003913 int res;
3914 Py_ssize_t len;
3915 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003916 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003917 if (tmp == NULL) goto failed;
3918 if (!PyList_Check(tmp)) {
3919 PyErr_Format(PyExc_TypeError, "ClassDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3920 goto failed;
3921 }
3922 len = PyList_GET_SIZE(tmp);
3923 body = asdl_seq_new(len, arena);
3924 if (body == NULL) goto failed;
3925 for (i = 0; i < len; i++) {
3926 stmt_ty value;
3927 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3928 if (res != 0) goto failed;
3929 asdl_seq_SET(body, i, value);
3930 }
3931 Py_XDECREF(tmp);
3932 tmp = NULL;
3933 } else {
3934 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ClassDef");
3935 return 1;
3936 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003937 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003938 int res;
3939 Py_ssize_t len;
3940 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003941 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003942 if (tmp == NULL) goto failed;
3943 if (!PyList_Check(tmp)) {
3944 PyErr_Format(PyExc_TypeError, "ClassDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3945 goto failed;
3946 }
3947 len = PyList_GET_SIZE(tmp);
3948 decorator_list = asdl_seq_new(len, arena);
3949 if (decorator_list == NULL) goto failed;
3950 for (i = 0; i < len; i++) {
3951 expr_ty value;
3952 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3953 if (res != 0) goto failed;
3954 asdl_seq_SET(decorator_list, i, value);
3955 }
3956 Py_XDECREF(tmp);
3957 tmp = NULL;
3958 } else {
3959 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from ClassDef");
3960 return 1;
3961 }
3962 *out = ClassDef(name, bases, keywords, starargs, kwargs, body,
3963 decorator_list, lineno, col_offset, arena);
3964 if (*out == NULL) goto failed;
3965 return 0;
3966 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003967 isinstance = PyObject_IsInstance(obj, (PyObject*)Return_type);
3968 if (isinstance == -1) {
3969 return 1;
3970 }
3971 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003972 expr_ty value;
3973
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003974 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003975 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003976 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003977 if (tmp == NULL) goto failed;
3978 res = obj2ast_expr(tmp, &value, arena);
3979 if (res != 0) goto failed;
3980 Py_XDECREF(tmp);
3981 tmp = NULL;
3982 } else {
3983 value = NULL;
3984 }
3985 *out = Return(value, lineno, col_offset, arena);
3986 if (*out == NULL) goto failed;
3987 return 0;
3988 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003989 isinstance = PyObject_IsInstance(obj, (PyObject*)Delete_type);
3990 if (isinstance == -1) {
3991 return 1;
3992 }
3993 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003994 asdl_seq* targets;
3995
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003996 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003997 int res;
3998 Py_ssize_t len;
3999 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004000 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004001 if (tmp == NULL) goto failed;
4002 if (!PyList_Check(tmp)) {
4003 PyErr_Format(PyExc_TypeError, "Delete field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4004 goto failed;
4005 }
4006 len = PyList_GET_SIZE(tmp);
4007 targets = asdl_seq_new(len, arena);
4008 if (targets == NULL) goto failed;
4009 for (i = 0; i < len; i++) {
4010 expr_ty value;
4011 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4012 if (res != 0) goto failed;
4013 asdl_seq_SET(targets, i, value);
4014 }
4015 Py_XDECREF(tmp);
4016 tmp = NULL;
4017 } else {
4018 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Delete");
4019 return 1;
4020 }
4021 *out = Delete(targets, lineno, col_offset, arena);
4022 if (*out == NULL) goto failed;
4023 return 0;
4024 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004025 isinstance = PyObject_IsInstance(obj, (PyObject*)Assign_type);
4026 if (isinstance == -1) {
4027 return 1;
4028 }
4029 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004030 asdl_seq* targets;
4031 expr_ty value;
4032
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004033 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004034 int res;
4035 Py_ssize_t len;
4036 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004037 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004038 if (tmp == NULL) goto failed;
4039 if (!PyList_Check(tmp)) {
4040 PyErr_Format(PyExc_TypeError, "Assign field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4041 goto failed;
4042 }
4043 len = PyList_GET_SIZE(tmp);
4044 targets = asdl_seq_new(len, arena);
4045 if (targets == NULL) goto failed;
4046 for (i = 0; i < len; i++) {
4047 expr_ty value;
4048 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4049 if (res != 0) goto failed;
4050 asdl_seq_SET(targets, i, value);
4051 }
4052 Py_XDECREF(tmp);
4053 tmp = NULL;
4054 } else {
4055 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Assign");
4056 return 1;
4057 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004058 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004059 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004060 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004061 if (tmp == NULL) goto failed;
4062 res = obj2ast_expr(tmp, &value, arena);
4063 if (res != 0) goto failed;
4064 Py_XDECREF(tmp);
4065 tmp = NULL;
4066 } else {
4067 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Assign");
4068 return 1;
4069 }
4070 *out = Assign(targets, value, lineno, col_offset, arena);
4071 if (*out == NULL) goto failed;
4072 return 0;
4073 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004074 isinstance = PyObject_IsInstance(obj, (PyObject*)AugAssign_type);
4075 if (isinstance == -1) {
4076 return 1;
4077 }
4078 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004079 expr_ty target;
4080 operator_ty op;
4081 expr_ty value;
4082
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004083 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004084 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004085 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004086 if (tmp == NULL) goto failed;
4087 res = obj2ast_expr(tmp, &target, arena);
4088 if (res != 0) goto failed;
4089 Py_XDECREF(tmp);
4090 tmp = NULL;
4091 } else {
4092 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from AugAssign");
4093 return 1;
4094 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004095 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004096 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004097 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004098 if (tmp == NULL) goto failed;
4099 res = obj2ast_operator(tmp, &op, arena);
4100 if (res != 0) goto failed;
4101 Py_XDECREF(tmp);
4102 tmp = NULL;
4103 } else {
4104 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from AugAssign");
4105 return 1;
4106 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004107 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004108 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004109 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004110 if (tmp == NULL) goto failed;
4111 res = obj2ast_expr(tmp, &value, arena);
4112 if (res != 0) goto failed;
4113 Py_XDECREF(tmp);
4114 tmp = NULL;
4115 } else {
4116 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from AugAssign");
4117 return 1;
4118 }
4119 *out = AugAssign(target, op, value, lineno, col_offset, arena);
4120 if (*out == NULL) goto failed;
4121 return 0;
4122 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004123 isinstance = PyObject_IsInstance(obj, (PyObject*)For_type);
4124 if (isinstance == -1) {
4125 return 1;
4126 }
4127 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004128 expr_ty target;
4129 expr_ty iter;
4130 asdl_seq* body;
4131 asdl_seq* orelse;
4132
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004133 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004134 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004135 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004136 if (tmp == NULL) goto failed;
4137 res = obj2ast_expr(tmp, &target, arena);
4138 if (res != 0) goto failed;
4139 Py_XDECREF(tmp);
4140 tmp = NULL;
4141 } else {
4142 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from For");
4143 return 1;
4144 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004145 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004146 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004147 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004148 if (tmp == NULL) goto failed;
4149 res = obj2ast_expr(tmp, &iter, arena);
4150 if (res != 0) goto failed;
4151 Py_XDECREF(tmp);
4152 tmp = NULL;
4153 } else {
4154 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from For");
4155 return 1;
4156 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004157 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004158 int res;
4159 Py_ssize_t len;
4160 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004161 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004162 if (tmp == NULL) goto failed;
4163 if (!PyList_Check(tmp)) {
4164 PyErr_Format(PyExc_TypeError, "For field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4165 goto failed;
4166 }
4167 len = PyList_GET_SIZE(tmp);
4168 body = asdl_seq_new(len, arena);
4169 if (body == NULL) goto failed;
4170 for (i = 0; i < len; i++) {
4171 stmt_ty value;
4172 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4173 if (res != 0) goto failed;
4174 asdl_seq_SET(body, i, value);
4175 }
4176 Py_XDECREF(tmp);
4177 tmp = NULL;
4178 } else {
4179 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from For");
4180 return 1;
4181 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004182 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004183 int res;
4184 Py_ssize_t len;
4185 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004186 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004187 if (tmp == NULL) goto failed;
4188 if (!PyList_Check(tmp)) {
4189 PyErr_Format(PyExc_TypeError, "For field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4190 goto failed;
4191 }
4192 len = PyList_GET_SIZE(tmp);
4193 orelse = asdl_seq_new(len, arena);
4194 if (orelse == NULL) goto failed;
4195 for (i = 0; i < len; i++) {
4196 stmt_ty value;
4197 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4198 if (res != 0) goto failed;
4199 asdl_seq_SET(orelse, i, value);
4200 }
4201 Py_XDECREF(tmp);
4202 tmp = NULL;
4203 } else {
4204 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from For");
4205 return 1;
4206 }
4207 *out = For(target, iter, body, orelse, lineno, col_offset,
4208 arena);
4209 if (*out == NULL) goto failed;
4210 return 0;
4211 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004212 isinstance = PyObject_IsInstance(obj, (PyObject*)While_type);
4213 if (isinstance == -1) {
4214 return 1;
4215 }
4216 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004217 expr_ty test;
4218 asdl_seq* body;
4219 asdl_seq* orelse;
4220
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004221 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004222 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004223 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004224 if (tmp == NULL) goto failed;
4225 res = obj2ast_expr(tmp, &test, arena);
4226 if (res != 0) goto failed;
4227 Py_XDECREF(tmp);
4228 tmp = NULL;
4229 } else {
4230 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from While");
4231 return 1;
4232 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004233 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004234 int res;
4235 Py_ssize_t len;
4236 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004237 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004238 if (tmp == NULL) goto failed;
4239 if (!PyList_Check(tmp)) {
4240 PyErr_Format(PyExc_TypeError, "While field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4241 goto failed;
4242 }
4243 len = PyList_GET_SIZE(tmp);
4244 body = asdl_seq_new(len, arena);
4245 if (body == NULL) goto failed;
4246 for (i = 0; i < len; i++) {
4247 stmt_ty value;
4248 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4249 if (res != 0) goto failed;
4250 asdl_seq_SET(body, i, value);
4251 }
4252 Py_XDECREF(tmp);
4253 tmp = NULL;
4254 } else {
4255 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from While");
4256 return 1;
4257 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004258 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004259 int res;
4260 Py_ssize_t len;
4261 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004262 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004263 if (tmp == NULL) goto failed;
4264 if (!PyList_Check(tmp)) {
4265 PyErr_Format(PyExc_TypeError, "While field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4266 goto failed;
4267 }
4268 len = PyList_GET_SIZE(tmp);
4269 orelse = asdl_seq_new(len, arena);
4270 if (orelse == NULL) goto failed;
4271 for (i = 0; i < len; i++) {
4272 stmt_ty value;
4273 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4274 if (res != 0) goto failed;
4275 asdl_seq_SET(orelse, i, value);
4276 }
4277 Py_XDECREF(tmp);
4278 tmp = NULL;
4279 } else {
4280 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from While");
4281 return 1;
4282 }
4283 *out = While(test, body, orelse, lineno, col_offset, arena);
4284 if (*out == NULL) goto failed;
4285 return 0;
4286 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004287 isinstance = PyObject_IsInstance(obj, (PyObject*)If_type);
4288 if (isinstance == -1) {
4289 return 1;
4290 }
4291 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004292 expr_ty test;
4293 asdl_seq* body;
4294 asdl_seq* orelse;
4295
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004296 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004297 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004298 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004299 if (tmp == NULL) goto failed;
4300 res = obj2ast_expr(tmp, &test, arena);
4301 if (res != 0) goto failed;
4302 Py_XDECREF(tmp);
4303 tmp = NULL;
4304 } else {
4305 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from If");
4306 return 1;
4307 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004308 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004309 int res;
4310 Py_ssize_t len;
4311 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004312 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004313 if (tmp == NULL) goto failed;
4314 if (!PyList_Check(tmp)) {
4315 PyErr_Format(PyExc_TypeError, "If field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4316 goto failed;
4317 }
4318 len = PyList_GET_SIZE(tmp);
4319 body = asdl_seq_new(len, arena);
4320 if (body == NULL) goto failed;
4321 for (i = 0; i < len; i++) {
4322 stmt_ty value;
4323 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4324 if (res != 0) goto failed;
4325 asdl_seq_SET(body, i, value);
4326 }
4327 Py_XDECREF(tmp);
4328 tmp = NULL;
4329 } else {
4330 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from If");
4331 return 1;
4332 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004333 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004334 int res;
4335 Py_ssize_t len;
4336 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004337 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004338 if (tmp == NULL) goto failed;
4339 if (!PyList_Check(tmp)) {
4340 PyErr_Format(PyExc_TypeError, "If field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4341 goto failed;
4342 }
4343 len = PyList_GET_SIZE(tmp);
4344 orelse = asdl_seq_new(len, arena);
4345 if (orelse == NULL) goto failed;
4346 for (i = 0; i < len; i++) {
4347 stmt_ty value;
4348 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4349 if (res != 0) goto failed;
4350 asdl_seq_SET(orelse, i, value);
4351 }
4352 Py_XDECREF(tmp);
4353 tmp = NULL;
4354 } else {
4355 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from If");
4356 return 1;
4357 }
4358 *out = If(test, body, orelse, lineno, col_offset, arena);
4359 if (*out == NULL) goto failed;
4360 return 0;
4361 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004362 isinstance = PyObject_IsInstance(obj, (PyObject*)With_type);
4363 if (isinstance == -1) {
4364 return 1;
4365 }
4366 if (isinstance) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004367 asdl_seq* items;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004368 asdl_seq* body;
4369
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004370 if (_PyObject_HasAttrId(obj, &PyId_items)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004371 int res;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004372 Py_ssize_t len;
4373 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004374 tmp = _PyObject_GetAttrId(obj, &PyId_items);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004375 if (tmp == NULL) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004376 if (!PyList_Check(tmp)) {
4377 PyErr_Format(PyExc_TypeError, "With field \"items\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4378 goto failed;
4379 }
4380 len = PyList_GET_SIZE(tmp);
4381 items = asdl_seq_new(len, arena);
4382 if (items == NULL) goto failed;
4383 for (i = 0; i < len; i++) {
4384 withitem_ty value;
4385 res = obj2ast_withitem(PyList_GET_ITEM(tmp, i), &value, arena);
4386 if (res != 0) goto failed;
4387 asdl_seq_SET(items, i, value);
4388 }
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004389 Py_XDECREF(tmp);
4390 tmp = NULL;
4391 } else {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004392 PyErr_SetString(PyExc_TypeError, "required field \"items\" missing from With");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004393 return 1;
4394 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004395 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004396 int res;
4397 Py_ssize_t len;
4398 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004399 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004400 if (tmp == NULL) goto failed;
4401 if (!PyList_Check(tmp)) {
4402 PyErr_Format(PyExc_TypeError, "With field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4403 goto failed;
4404 }
4405 len = PyList_GET_SIZE(tmp);
4406 body = asdl_seq_new(len, arena);
4407 if (body == NULL) goto failed;
4408 for (i = 0; i < len; i++) {
4409 stmt_ty value;
4410 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4411 if (res != 0) goto failed;
4412 asdl_seq_SET(body, i, value);
4413 }
4414 Py_XDECREF(tmp);
4415 tmp = NULL;
4416 } else {
4417 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from With");
4418 return 1;
4419 }
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004420 *out = With(items, body, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004421 if (*out == NULL) goto failed;
4422 return 0;
4423 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004424 isinstance = PyObject_IsInstance(obj, (PyObject*)Raise_type);
4425 if (isinstance == -1) {
4426 return 1;
4427 }
4428 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004429 expr_ty exc;
4430 expr_ty cause;
4431
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004432 if (_PyObject_HasAttrId(obj, &PyId_exc)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004433 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004434 tmp = _PyObject_GetAttrId(obj, &PyId_exc);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004435 if (tmp == NULL) goto failed;
4436 res = obj2ast_expr(tmp, &exc, arena);
4437 if (res != 0) goto failed;
4438 Py_XDECREF(tmp);
4439 tmp = NULL;
4440 } else {
4441 exc = NULL;
4442 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004443 if (_PyObject_HasAttrId(obj, &PyId_cause)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004444 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004445 tmp = _PyObject_GetAttrId(obj, &PyId_cause);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004446 if (tmp == NULL) goto failed;
4447 res = obj2ast_expr(tmp, &cause, arena);
4448 if (res != 0) goto failed;
4449 Py_XDECREF(tmp);
4450 tmp = NULL;
4451 } else {
4452 cause = NULL;
4453 }
4454 *out = Raise(exc, cause, lineno, col_offset, arena);
4455 if (*out == NULL) goto failed;
4456 return 0;
4457 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004458 isinstance = PyObject_IsInstance(obj, (PyObject*)Try_type);
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004459 if (isinstance == -1) {
4460 return 1;
4461 }
4462 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004463 asdl_seq* body;
4464 asdl_seq* handlers;
4465 asdl_seq* orelse;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004466 asdl_seq* finalbody;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004467
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004468 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004469 int res;
4470 Py_ssize_t len;
4471 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004472 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004473 if (tmp == NULL) goto failed;
4474 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004475 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 +00004476 goto failed;
4477 }
4478 len = PyList_GET_SIZE(tmp);
4479 body = asdl_seq_new(len, arena);
4480 if (body == NULL) goto failed;
4481 for (i = 0; i < len; i++) {
4482 stmt_ty value;
4483 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4484 if (res != 0) goto failed;
4485 asdl_seq_SET(body, i, value);
4486 }
4487 Py_XDECREF(tmp);
4488 tmp = NULL;
4489 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004490 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004491 return 1;
4492 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004493 if (_PyObject_HasAttrId(obj, &PyId_handlers)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004494 int res;
4495 Py_ssize_t len;
4496 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004497 tmp = _PyObject_GetAttrId(obj, &PyId_handlers);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004498 if (tmp == NULL) goto failed;
4499 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004500 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 +00004501 goto failed;
4502 }
4503 len = PyList_GET_SIZE(tmp);
4504 handlers = asdl_seq_new(len, arena);
4505 if (handlers == NULL) goto failed;
4506 for (i = 0; i < len; i++) {
4507 excepthandler_ty value;
4508 res = obj2ast_excepthandler(PyList_GET_ITEM(tmp, i), &value, arena);
4509 if (res != 0) goto failed;
4510 asdl_seq_SET(handlers, i, value);
4511 }
4512 Py_XDECREF(tmp);
4513 tmp = NULL;
4514 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004515 PyErr_SetString(PyExc_TypeError, "required field \"handlers\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004516 return 1;
4517 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004518 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004519 int res;
4520 Py_ssize_t len;
4521 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004522 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004523 if (tmp == NULL) goto failed;
4524 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004525 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 +00004526 goto failed;
4527 }
4528 len = PyList_GET_SIZE(tmp);
4529 orelse = asdl_seq_new(len, arena);
4530 if (orelse == NULL) goto failed;
4531 for (i = 0; i < len; i++) {
4532 stmt_ty value;
4533 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4534 if (res != 0) goto failed;
4535 asdl_seq_SET(orelse, i, value);
4536 }
4537 Py_XDECREF(tmp);
4538 tmp = NULL;
4539 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004540 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004541 return 1;
4542 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004543 if (_PyObject_HasAttrId(obj, &PyId_finalbody)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004544 int res;
4545 Py_ssize_t len;
4546 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004547 tmp = _PyObject_GetAttrId(obj, &PyId_finalbody);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004548 if (tmp == NULL) goto failed;
4549 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004550 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 +00004551 goto failed;
4552 }
4553 len = PyList_GET_SIZE(tmp);
4554 finalbody = asdl_seq_new(len, arena);
4555 if (finalbody == NULL) goto failed;
4556 for (i = 0; i < len; i++) {
4557 stmt_ty value;
4558 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4559 if (res != 0) goto failed;
4560 asdl_seq_SET(finalbody, i, value);
4561 }
4562 Py_XDECREF(tmp);
4563 tmp = NULL;
4564 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004565 PyErr_SetString(PyExc_TypeError, "required field \"finalbody\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004566 return 1;
4567 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004568 *out = Try(body, handlers, orelse, finalbody, lineno,
4569 col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004570 if (*out == NULL) goto failed;
4571 return 0;
4572 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004573 isinstance = PyObject_IsInstance(obj, (PyObject*)Assert_type);
4574 if (isinstance == -1) {
4575 return 1;
4576 }
4577 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004578 expr_ty test;
4579 expr_ty msg;
4580
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004581 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004582 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004583 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004584 if (tmp == NULL) goto failed;
4585 res = obj2ast_expr(tmp, &test, arena);
4586 if (res != 0) goto failed;
4587 Py_XDECREF(tmp);
4588 tmp = NULL;
4589 } else {
4590 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from Assert");
4591 return 1;
4592 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004593 if (_PyObject_HasAttrId(obj, &PyId_msg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004594 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004595 tmp = _PyObject_GetAttrId(obj, &PyId_msg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004596 if (tmp == NULL) goto failed;
4597 res = obj2ast_expr(tmp, &msg, arena);
4598 if (res != 0) goto failed;
4599 Py_XDECREF(tmp);
4600 tmp = NULL;
4601 } else {
4602 msg = NULL;
4603 }
4604 *out = Assert(test, msg, lineno, col_offset, arena);
4605 if (*out == NULL) goto failed;
4606 return 0;
4607 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004608 isinstance = PyObject_IsInstance(obj, (PyObject*)Import_type);
4609 if (isinstance == -1) {
4610 return 1;
4611 }
4612 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004613 asdl_seq* names;
4614
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004615 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004616 int res;
4617 Py_ssize_t len;
4618 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004619 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004620 if (tmp == NULL) goto failed;
4621 if (!PyList_Check(tmp)) {
4622 PyErr_Format(PyExc_TypeError, "Import field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4623 goto failed;
4624 }
4625 len = PyList_GET_SIZE(tmp);
4626 names = asdl_seq_new(len, arena);
4627 if (names == NULL) goto failed;
4628 for (i = 0; i < len; i++) {
4629 alias_ty value;
4630 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4631 if (res != 0) goto failed;
4632 asdl_seq_SET(names, i, value);
4633 }
4634 Py_XDECREF(tmp);
4635 tmp = NULL;
4636 } else {
4637 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Import");
4638 return 1;
4639 }
4640 *out = Import(names, lineno, col_offset, arena);
4641 if (*out == NULL) goto failed;
4642 return 0;
4643 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004644 isinstance = PyObject_IsInstance(obj, (PyObject*)ImportFrom_type);
4645 if (isinstance == -1) {
4646 return 1;
4647 }
4648 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004649 identifier module;
4650 asdl_seq* names;
4651 int level;
4652
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004653 if (_PyObject_HasAttrId(obj, &PyId_module)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004654 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004655 tmp = _PyObject_GetAttrId(obj, &PyId_module);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004656 if (tmp == NULL) goto failed;
4657 res = obj2ast_identifier(tmp, &module, arena);
4658 if (res != 0) goto failed;
4659 Py_XDECREF(tmp);
4660 tmp = NULL;
4661 } else {
Benjamin Peterson78565b22009-06-28 19:19:51 +00004662 module = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004663 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004664 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004665 int res;
4666 Py_ssize_t len;
4667 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004668 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004669 if (tmp == NULL) goto failed;
4670 if (!PyList_Check(tmp)) {
4671 PyErr_Format(PyExc_TypeError, "ImportFrom field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4672 goto failed;
4673 }
4674 len = PyList_GET_SIZE(tmp);
4675 names = asdl_seq_new(len, arena);
4676 if (names == NULL) goto failed;
4677 for (i = 0; i < len; i++) {
4678 alias_ty value;
4679 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4680 if (res != 0) goto failed;
4681 asdl_seq_SET(names, i, value);
4682 }
4683 Py_XDECREF(tmp);
4684 tmp = NULL;
4685 } else {
4686 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from ImportFrom");
4687 return 1;
4688 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004689 if (_PyObject_HasAttrId(obj, &PyId_level)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004690 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004691 tmp = _PyObject_GetAttrId(obj, &PyId_level);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004692 if (tmp == NULL) goto failed;
4693 res = obj2ast_int(tmp, &level, arena);
4694 if (res != 0) goto failed;
4695 Py_XDECREF(tmp);
4696 tmp = NULL;
4697 } else {
4698 level = 0;
4699 }
4700 *out = ImportFrom(module, names, level, lineno, col_offset,
4701 arena);
4702 if (*out == NULL) goto failed;
4703 return 0;
4704 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004705 isinstance = PyObject_IsInstance(obj, (PyObject*)Global_type);
4706 if (isinstance == -1) {
4707 return 1;
4708 }
4709 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004710 asdl_seq* names;
4711
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004712 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004713 int res;
4714 Py_ssize_t len;
4715 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004716 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004717 if (tmp == NULL) goto failed;
4718 if (!PyList_Check(tmp)) {
4719 PyErr_Format(PyExc_TypeError, "Global field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4720 goto failed;
4721 }
4722 len = PyList_GET_SIZE(tmp);
4723 names = asdl_seq_new(len, arena);
4724 if (names == NULL) goto failed;
4725 for (i = 0; i < len; i++) {
4726 identifier value;
4727 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4728 if (res != 0) goto failed;
4729 asdl_seq_SET(names, i, value);
4730 }
4731 Py_XDECREF(tmp);
4732 tmp = NULL;
4733 } else {
4734 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Global");
4735 return 1;
4736 }
4737 *out = Global(names, lineno, col_offset, arena);
4738 if (*out == NULL) goto failed;
4739 return 0;
4740 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004741 isinstance = PyObject_IsInstance(obj, (PyObject*)Nonlocal_type);
4742 if (isinstance == -1) {
4743 return 1;
4744 }
4745 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004746 asdl_seq* names;
4747
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004748 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004749 int res;
4750 Py_ssize_t len;
4751 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004752 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004753 if (tmp == NULL) goto failed;
4754 if (!PyList_Check(tmp)) {
4755 PyErr_Format(PyExc_TypeError, "Nonlocal field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4756 goto failed;
4757 }
4758 len = PyList_GET_SIZE(tmp);
4759 names = asdl_seq_new(len, arena);
4760 if (names == NULL) goto failed;
4761 for (i = 0; i < len; i++) {
4762 identifier value;
4763 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4764 if (res != 0) goto failed;
4765 asdl_seq_SET(names, i, value);
4766 }
4767 Py_XDECREF(tmp);
4768 tmp = NULL;
4769 } else {
4770 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Nonlocal");
4771 return 1;
4772 }
4773 *out = Nonlocal(names, lineno, col_offset, arena);
4774 if (*out == NULL) goto failed;
4775 return 0;
4776 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004777 isinstance = PyObject_IsInstance(obj, (PyObject*)Expr_type);
4778 if (isinstance == -1) {
4779 return 1;
4780 }
4781 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004782 expr_ty value;
4783
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004784 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004785 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004786 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004787 if (tmp == NULL) goto failed;
4788 res = obj2ast_expr(tmp, &value, arena);
4789 if (res != 0) goto failed;
4790 Py_XDECREF(tmp);
4791 tmp = NULL;
4792 } else {
4793 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Expr");
4794 return 1;
4795 }
4796 *out = Expr(value, lineno, col_offset, arena);
4797 if (*out == NULL) goto failed;
4798 return 0;
4799 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004800 isinstance = PyObject_IsInstance(obj, (PyObject*)Pass_type);
4801 if (isinstance == -1) {
4802 return 1;
4803 }
4804 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004805
4806 *out = Pass(lineno, col_offset, arena);
4807 if (*out == NULL) goto failed;
4808 return 0;
4809 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004810 isinstance = PyObject_IsInstance(obj, (PyObject*)Break_type);
4811 if (isinstance == -1) {
4812 return 1;
4813 }
4814 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004815
4816 *out = Break(lineno, col_offset, arena);
4817 if (*out == NULL) goto failed;
4818 return 0;
4819 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004820 isinstance = PyObject_IsInstance(obj, (PyObject*)Continue_type);
4821 if (isinstance == -1) {
4822 return 1;
4823 }
4824 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004825
4826 *out = Continue(lineno, col_offset, arena);
4827 if (*out == NULL) goto failed;
4828 return 0;
4829 }
4830
Benjamin Peterson5b066812010-11-20 01:38:49 +00004831 PyErr_Format(PyExc_TypeError, "expected some sort of stmt, but got %R", obj);
4832 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00004833 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004834 return 1;
4835}
4836
4837int
4838obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena)
4839{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004840 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004841
Benjamin Petersond8f65972010-11-20 04:31:07 +00004842 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004843 int lineno;
4844 int col_offset;
4845
4846 if (obj == Py_None) {
4847 *out = NULL;
4848 return 0;
4849 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004850 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004851 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004852 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004853 if (tmp == NULL) goto failed;
4854 res = obj2ast_int(tmp, &lineno, arena);
4855 if (res != 0) goto failed;
4856 Py_XDECREF(tmp);
4857 tmp = NULL;
4858 } else {
4859 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from expr");
4860 return 1;
4861 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004862 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004863 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004864 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004865 if (tmp == NULL) goto failed;
4866 res = obj2ast_int(tmp, &col_offset, arena);
4867 if (res != 0) goto failed;
4868 Py_XDECREF(tmp);
4869 tmp = NULL;
4870 } else {
4871 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from expr");
4872 return 1;
4873 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004874 isinstance = PyObject_IsInstance(obj, (PyObject*)BoolOp_type);
4875 if (isinstance == -1) {
4876 return 1;
4877 }
4878 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004879 boolop_ty op;
4880 asdl_seq* values;
4881
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004882 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004883 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004884 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004885 if (tmp == NULL) goto failed;
4886 res = obj2ast_boolop(tmp, &op, arena);
4887 if (res != 0) goto failed;
4888 Py_XDECREF(tmp);
4889 tmp = NULL;
4890 } else {
4891 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BoolOp");
4892 return 1;
4893 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004894 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004895 int res;
4896 Py_ssize_t len;
4897 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004898 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004899 if (tmp == NULL) goto failed;
4900 if (!PyList_Check(tmp)) {
4901 PyErr_Format(PyExc_TypeError, "BoolOp field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4902 goto failed;
4903 }
4904 len = PyList_GET_SIZE(tmp);
4905 values = asdl_seq_new(len, arena);
4906 if (values == NULL) goto failed;
4907 for (i = 0; i < len; i++) {
4908 expr_ty value;
4909 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4910 if (res != 0) goto failed;
4911 asdl_seq_SET(values, i, value);
4912 }
4913 Py_XDECREF(tmp);
4914 tmp = NULL;
4915 } else {
4916 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from BoolOp");
4917 return 1;
4918 }
4919 *out = BoolOp(op, values, lineno, col_offset, arena);
4920 if (*out == NULL) goto failed;
4921 return 0;
4922 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004923 isinstance = PyObject_IsInstance(obj, (PyObject*)BinOp_type);
4924 if (isinstance == -1) {
4925 return 1;
4926 }
4927 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004928 expr_ty left;
4929 operator_ty op;
4930 expr_ty right;
4931
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004932 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004933 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004934 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004935 if (tmp == NULL) goto failed;
4936 res = obj2ast_expr(tmp, &left, arena);
4937 if (res != 0) goto failed;
4938 Py_XDECREF(tmp);
4939 tmp = NULL;
4940 } else {
4941 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from BinOp");
4942 return 1;
4943 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004944 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004945 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004946 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004947 if (tmp == NULL) goto failed;
4948 res = obj2ast_operator(tmp, &op, arena);
4949 if (res != 0) goto failed;
4950 Py_XDECREF(tmp);
4951 tmp = NULL;
4952 } else {
4953 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BinOp");
4954 return 1;
4955 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004956 if (_PyObject_HasAttrId(obj, &PyId_right)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004957 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004958 tmp = _PyObject_GetAttrId(obj, &PyId_right);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004959 if (tmp == NULL) goto failed;
4960 res = obj2ast_expr(tmp, &right, arena);
4961 if (res != 0) goto failed;
4962 Py_XDECREF(tmp);
4963 tmp = NULL;
4964 } else {
4965 PyErr_SetString(PyExc_TypeError, "required field \"right\" missing from BinOp");
4966 return 1;
4967 }
4968 *out = BinOp(left, op, right, lineno, col_offset, arena);
4969 if (*out == NULL) goto failed;
4970 return 0;
4971 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004972 isinstance = PyObject_IsInstance(obj, (PyObject*)UnaryOp_type);
4973 if (isinstance == -1) {
4974 return 1;
4975 }
4976 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004977 unaryop_ty op;
4978 expr_ty operand;
4979
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004980 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004981 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004982 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004983 if (tmp == NULL) goto failed;
4984 res = obj2ast_unaryop(tmp, &op, arena);
4985 if (res != 0) goto failed;
4986 Py_XDECREF(tmp);
4987 tmp = NULL;
4988 } else {
4989 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from UnaryOp");
4990 return 1;
4991 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004992 if (_PyObject_HasAttrId(obj, &PyId_operand)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004993 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004994 tmp = _PyObject_GetAttrId(obj, &PyId_operand);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004995 if (tmp == NULL) goto failed;
4996 res = obj2ast_expr(tmp, &operand, arena);
4997 if (res != 0) goto failed;
4998 Py_XDECREF(tmp);
4999 tmp = NULL;
5000 } else {
5001 PyErr_SetString(PyExc_TypeError, "required field \"operand\" missing from UnaryOp");
5002 return 1;
5003 }
5004 *out = UnaryOp(op, operand, lineno, col_offset, arena);
5005 if (*out == NULL) goto failed;
5006 return 0;
5007 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005008 isinstance = PyObject_IsInstance(obj, (PyObject*)Lambda_type);
5009 if (isinstance == -1) {
5010 return 1;
5011 }
5012 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005013 arguments_ty args;
5014 expr_ty body;
5015
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005016 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005017 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005018 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005019 if (tmp == NULL) goto failed;
5020 res = obj2ast_arguments(tmp, &args, arena);
5021 if (res != 0) goto failed;
5022 Py_XDECREF(tmp);
5023 tmp = NULL;
5024 } else {
5025 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Lambda");
5026 return 1;
5027 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005028 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005029 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005030 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005031 if (tmp == NULL) goto failed;
5032 res = obj2ast_expr(tmp, &body, arena);
5033 if (res != 0) goto failed;
5034 Py_XDECREF(tmp);
5035 tmp = NULL;
5036 } else {
5037 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Lambda");
5038 return 1;
5039 }
5040 *out = Lambda(args, body, lineno, col_offset, arena);
5041 if (*out == NULL) goto failed;
5042 return 0;
5043 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005044 isinstance = PyObject_IsInstance(obj, (PyObject*)IfExp_type);
5045 if (isinstance == -1) {
5046 return 1;
5047 }
5048 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005049 expr_ty test;
5050 expr_ty body;
5051 expr_ty orelse;
5052
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005053 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005054 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005055 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005056 if (tmp == NULL) goto failed;
5057 res = obj2ast_expr(tmp, &test, arena);
5058 if (res != 0) goto failed;
5059 Py_XDECREF(tmp);
5060 tmp = NULL;
5061 } else {
5062 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from IfExp");
5063 return 1;
5064 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005065 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005066 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005067 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005068 if (tmp == NULL) goto failed;
5069 res = obj2ast_expr(tmp, &body, arena);
5070 if (res != 0) goto failed;
5071 Py_XDECREF(tmp);
5072 tmp = NULL;
5073 } else {
5074 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from IfExp");
5075 return 1;
5076 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005077 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005078 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005079 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005080 if (tmp == NULL) goto failed;
5081 res = obj2ast_expr(tmp, &orelse, arena);
5082 if (res != 0) goto failed;
5083 Py_XDECREF(tmp);
5084 tmp = NULL;
5085 } else {
5086 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from IfExp");
5087 return 1;
5088 }
5089 *out = IfExp(test, body, orelse, lineno, col_offset, arena);
5090 if (*out == NULL) goto failed;
5091 return 0;
5092 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005093 isinstance = PyObject_IsInstance(obj, (PyObject*)Dict_type);
5094 if (isinstance == -1) {
5095 return 1;
5096 }
5097 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005098 asdl_seq* keys;
5099 asdl_seq* values;
5100
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005101 if (_PyObject_HasAttrId(obj, &PyId_keys)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005102 int res;
5103 Py_ssize_t len;
5104 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005105 tmp = _PyObject_GetAttrId(obj, &PyId_keys);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005106 if (tmp == NULL) goto failed;
5107 if (!PyList_Check(tmp)) {
5108 PyErr_Format(PyExc_TypeError, "Dict field \"keys\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5109 goto failed;
5110 }
5111 len = PyList_GET_SIZE(tmp);
5112 keys = asdl_seq_new(len, arena);
5113 if (keys == NULL) goto failed;
5114 for (i = 0; i < len; i++) {
5115 expr_ty value;
5116 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5117 if (res != 0) goto failed;
5118 asdl_seq_SET(keys, i, value);
5119 }
5120 Py_XDECREF(tmp);
5121 tmp = NULL;
5122 } else {
5123 PyErr_SetString(PyExc_TypeError, "required field \"keys\" missing from Dict");
5124 return 1;
5125 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005126 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005127 int res;
5128 Py_ssize_t len;
5129 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005130 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005131 if (tmp == NULL) goto failed;
5132 if (!PyList_Check(tmp)) {
5133 PyErr_Format(PyExc_TypeError, "Dict field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5134 goto failed;
5135 }
5136 len = PyList_GET_SIZE(tmp);
5137 values = asdl_seq_new(len, arena);
5138 if (values == NULL) goto failed;
5139 for (i = 0; i < len; i++) {
5140 expr_ty value;
5141 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5142 if (res != 0) goto failed;
5143 asdl_seq_SET(values, i, value);
5144 }
5145 Py_XDECREF(tmp);
5146 tmp = NULL;
5147 } else {
5148 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from Dict");
5149 return 1;
5150 }
5151 *out = Dict(keys, values, lineno, col_offset, arena);
5152 if (*out == NULL) goto failed;
5153 return 0;
5154 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005155 isinstance = PyObject_IsInstance(obj, (PyObject*)Set_type);
5156 if (isinstance == -1) {
5157 return 1;
5158 }
5159 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005160 asdl_seq* elts;
5161
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005162 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005163 int res;
5164 Py_ssize_t len;
5165 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005166 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005167 if (tmp == NULL) goto failed;
5168 if (!PyList_Check(tmp)) {
5169 PyErr_Format(PyExc_TypeError, "Set field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5170 goto failed;
5171 }
5172 len = PyList_GET_SIZE(tmp);
5173 elts = asdl_seq_new(len, arena);
5174 if (elts == NULL) goto failed;
5175 for (i = 0; i < len; i++) {
5176 expr_ty value;
5177 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5178 if (res != 0) goto failed;
5179 asdl_seq_SET(elts, i, value);
5180 }
5181 Py_XDECREF(tmp);
5182 tmp = NULL;
5183 } else {
5184 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Set");
5185 return 1;
5186 }
5187 *out = Set(elts, lineno, col_offset, arena);
5188 if (*out == NULL) goto failed;
5189 return 0;
5190 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005191 isinstance = PyObject_IsInstance(obj, (PyObject*)ListComp_type);
5192 if (isinstance == -1) {
5193 return 1;
5194 }
5195 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005196 expr_ty elt;
5197 asdl_seq* generators;
5198
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005199 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005200 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005201 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005202 if (tmp == NULL) goto failed;
5203 res = obj2ast_expr(tmp, &elt, arena);
5204 if (res != 0) goto failed;
5205 Py_XDECREF(tmp);
5206 tmp = NULL;
5207 } else {
5208 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from ListComp");
5209 return 1;
5210 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005211 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005212 int res;
5213 Py_ssize_t len;
5214 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005215 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005216 if (tmp == NULL) goto failed;
5217 if (!PyList_Check(tmp)) {
5218 PyErr_Format(PyExc_TypeError, "ListComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5219 goto failed;
5220 }
5221 len = PyList_GET_SIZE(tmp);
5222 generators = asdl_seq_new(len, arena);
5223 if (generators == NULL) goto failed;
5224 for (i = 0; i < len; i++) {
5225 comprehension_ty value;
5226 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5227 if (res != 0) goto failed;
5228 asdl_seq_SET(generators, i, value);
5229 }
5230 Py_XDECREF(tmp);
5231 tmp = NULL;
5232 } else {
5233 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from ListComp");
5234 return 1;
5235 }
5236 *out = ListComp(elt, generators, lineno, col_offset, arena);
5237 if (*out == NULL) goto failed;
5238 return 0;
5239 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005240 isinstance = PyObject_IsInstance(obj, (PyObject*)SetComp_type);
5241 if (isinstance == -1) {
5242 return 1;
5243 }
5244 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005245 expr_ty elt;
5246 asdl_seq* generators;
5247
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005248 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005249 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005250 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005251 if (tmp == NULL) goto failed;
5252 res = obj2ast_expr(tmp, &elt, arena);
5253 if (res != 0) goto failed;
5254 Py_XDECREF(tmp);
5255 tmp = NULL;
5256 } else {
5257 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from SetComp");
5258 return 1;
5259 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005260 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005261 int res;
5262 Py_ssize_t len;
5263 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005264 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005265 if (tmp == NULL) goto failed;
5266 if (!PyList_Check(tmp)) {
5267 PyErr_Format(PyExc_TypeError, "SetComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5268 goto failed;
5269 }
5270 len = PyList_GET_SIZE(tmp);
5271 generators = asdl_seq_new(len, arena);
5272 if (generators == NULL) goto failed;
5273 for (i = 0; i < len; i++) {
5274 comprehension_ty value;
5275 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5276 if (res != 0) goto failed;
5277 asdl_seq_SET(generators, i, value);
5278 }
5279 Py_XDECREF(tmp);
5280 tmp = NULL;
5281 } else {
5282 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from SetComp");
5283 return 1;
5284 }
5285 *out = SetComp(elt, generators, lineno, col_offset, arena);
5286 if (*out == NULL) goto failed;
5287 return 0;
5288 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005289 isinstance = PyObject_IsInstance(obj, (PyObject*)DictComp_type);
5290 if (isinstance == -1) {
5291 return 1;
5292 }
5293 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005294 expr_ty key;
5295 expr_ty value;
5296 asdl_seq* generators;
5297
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005298 if (_PyObject_HasAttrId(obj, &PyId_key)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005299 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005300 tmp = _PyObject_GetAttrId(obj, &PyId_key);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005301 if (tmp == NULL) goto failed;
5302 res = obj2ast_expr(tmp, &key, arena);
5303 if (res != 0) goto failed;
5304 Py_XDECREF(tmp);
5305 tmp = NULL;
5306 } else {
5307 PyErr_SetString(PyExc_TypeError, "required field \"key\" missing from DictComp");
5308 return 1;
5309 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005310 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005311 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005312 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005313 if (tmp == NULL) goto failed;
5314 res = obj2ast_expr(tmp, &value, arena);
5315 if (res != 0) goto failed;
5316 Py_XDECREF(tmp);
5317 tmp = NULL;
5318 } else {
5319 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from DictComp");
5320 return 1;
5321 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005322 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005323 int res;
5324 Py_ssize_t len;
5325 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005326 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005327 if (tmp == NULL) goto failed;
5328 if (!PyList_Check(tmp)) {
5329 PyErr_Format(PyExc_TypeError, "DictComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5330 goto failed;
5331 }
5332 len = PyList_GET_SIZE(tmp);
5333 generators = asdl_seq_new(len, arena);
5334 if (generators == NULL) goto failed;
5335 for (i = 0; i < len; i++) {
5336 comprehension_ty value;
5337 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5338 if (res != 0) goto failed;
5339 asdl_seq_SET(generators, i, value);
5340 }
5341 Py_XDECREF(tmp);
5342 tmp = NULL;
5343 } else {
5344 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from DictComp");
5345 return 1;
5346 }
5347 *out = DictComp(key, value, generators, lineno, col_offset,
5348 arena);
5349 if (*out == NULL) goto failed;
5350 return 0;
5351 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005352 isinstance = PyObject_IsInstance(obj, (PyObject*)GeneratorExp_type);
5353 if (isinstance == -1) {
5354 return 1;
5355 }
5356 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005357 expr_ty elt;
5358 asdl_seq* generators;
5359
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005360 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005361 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005362 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005363 if (tmp == NULL) goto failed;
5364 res = obj2ast_expr(tmp, &elt, arena);
5365 if (res != 0) goto failed;
5366 Py_XDECREF(tmp);
5367 tmp = NULL;
5368 } else {
5369 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from GeneratorExp");
5370 return 1;
5371 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005372 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005373 int res;
5374 Py_ssize_t len;
5375 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005376 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005377 if (tmp == NULL) goto failed;
5378 if (!PyList_Check(tmp)) {
5379 PyErr_Format(PyExc_TypeError, "GeneratorExp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5380 goto failed;
5381 }
5382 len = PyList_GET_SIZE(tmp);
5383 generators = asdl_seq_new(len, arena);
5384 if (generators == NULL) goto failed;
5385 for (i = 0; i < len; i++) {
5386 comprehension_ty value;
5387 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5388 if (res != 0) goto failed;
5389 asdl_seq_SET(generators, i, value);
5390 }
5391 Py_XDECREF(tmp);
5392 tmp = NULL;
5393 } else {
5394 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from GeneratorExp");
5395 return 1;
5396 }
5397 *out = GeneratorExp(elt, generators, lineno, col_offset, arena);
5398 if (*out == NULL) goto failed;
5399 return 0;
5400 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005401 isinstance = PyObject_IsInstance(obj, (PyObject*)Yield_type);
5402 if (isinstance == -1) {
5403 return 1;
5404 }
5405 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005406 expr_ty value;
5407
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005408 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005409 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005410 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005411 if (tmp == NULL) goto failed;
5412 res = obj2ast_expr(tmp, &value, arena);
5413 if (res != 0) goto failed;
5414 Py_XDECREF(tmp);
5415 tmp = NULL;
5416 } else {
5417 value = NULL;
5418 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05005419 *out = Yield(value, lineno, col_offset, arena);
5420 if (*out == NULL) goto failed;
5421 return 0;
5422 }
5423 isinstance = PyObject_IsInstance(obj, (PyObject*)YieldFrom_type);
5424 if (isinstance == -1) {
5425 return 1;
5426 }
5427 if (isinstance) {
5428 expr_ty value;
5429
5430 if (_PyObject_HasAttrId(obj, &PyId_value)) {
5431 int res;
5432 tmp = _PyObject_GetAttrId(obj, &PyId_value);
5433 if (tmp == NULL) goto failed;
5434 res = obj2ast_expr(tmp, &value, arena);
5435 if (res != 0) goto failed;
5436 Py_XDECREF(tmp);
5437 tmp = NULL;
5438 } else {
Mark Dickinsonded35ae2012-11-25 14:36:26 +00005439 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from YieldFrom");
5440 return 1;
Benjamin Peterson527c6222012-01-14 08:58:23 -05005441 }
5442 *out = YieldFrom(value, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005443 if (*out == NULL) goto failed;
5444 return 0;
5445 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005446 isinstance = PyObject_IsInstance(obj, (PyObject*)Compare_type);
5447 if (isinstance == -1) {
5448 return 1;
5449 }
5450 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005451 expr_ty left;
5452 asdl_int_seq* ops;
5453 asdl_seq* comparators;
5454
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005455 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005456 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005457 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005458 if (tmp == NULL) goto failed;
5459 res = obj2ast_expr(tmp, &left, arena);
5460 if (res != 0) goto failed;
5461 Py_XDECREF(tmp);
5462 tmp = NULL;
5463 } else {
5464 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from Compare");
5465 return 1;
5466 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005467 if (_PyObject_HasAttrId(obj, &PyId_ops)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005468 int res;
5469 Py_ssize_t len;
5470 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005471 tmp = _PyObject_GetAttrId(obj, &PyId_ops);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005472 if (tmp == NULL) goto failed;
5473 if (!PyList_Check(tmp)) {
5474 PyErr_Format(PyExc_TypeError, "Compare field \"ops\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5475 goto failed;
5476 }
5477 len = PyList_GET_SIZE(tmp);
5478 ops = asdl_int_seq_new(len, arena);
5479 if (ops == NULL) goto failed;
5480 for (i = 0; i < len; i++) {
5481 cmpop_ty value;
5482 res = obj2ast_cmpop(PyList_GET_ITEM(tmp, i), &value, arena);
5483 if (res != 0) goto failed;
5484 asdl_seq_SET(ops, i, value);
5485 }
5486 Py_XDECREF(tmp);
5487 tmp = NULL;
5488 } else {
5489 PyErr_SetString(PyExc_TypeError, "required field \"ops\" missing from Compare");
5490 return 1;
5491 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005492 if (_PyObject_HasAttrId(obj, &PyId_comparators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005493 int res;
5494 Py_ssize_t len;
5495 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005496 tmp = _PyObject_GetAttrId(obj, &PyId_comparators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005497 if (tmp == NULL) goto failed;
5498 if (!PyList_Check(tmp)) {
5499 PyErr_Format(PyExc_TypeError, "Compare field \"comparators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5500 goto failed;
5501 }
5502 len = PyList_GET_SIZE(tmp);
5503 comparators = asdl_seq_new(len, arena);
5504 if (comparators == NULL) goto failed;
5505 for (i = 0; i < len; i++) {
5506 expr_ty value;
5507 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5508 if (res != 0) goto failed;
5509 asdl_seq_SET(comparators, i, value);
5510 }
5511 Py_XDECREF(tmp);
5512 tmp = NULL;
5513 } else {
5514 PyErr_SetString(PyExc_TypeError, "required field \"comparators\" missing from Compare");
5515 return 1;
5516 }
5517 *out = Compare(left, ops, comparators, lineno, col_offset,
5518 arena);
5519 if (*out == NULL) goto failed;
5520 return 0;
5521 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005522 isinstance = PyObject_IsInstance(obj, (PyObject*)Call_type);
5523 if (isinstance == -1) {
5524 return 1;
5525 }
5526 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005527 expr_ty func;
5528 asdl_seq* args;
5529 asdl_seq* keywords;
5530 expr_ty starargs;
5531 expr_ty kwargs;
5532
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005533 if (_PyObject_HasAttrId(obj, &PyId_func)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005534 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005535 tmp = _PyObject_GetAttrId(obj, &PyId_func);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005536 if (tmp == NULL) goto failed;
5537 res = obj2ast_expr(tmp, &func, arena);
5538 if (res != 0) goto failed;
5539 Py_XDECREF(tmp);
5540 tmp = NULL;
5541 } else {
5542 PyErr_SetString(PyExc_TypeError, "required field \"func\" missing from Call");
5543 return 1;
5544 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005545 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005546 int res;
5547 Py_ssize_t len;
5548 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005549 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005550 if (tmp == NULL) goto failed;
5551 if (!PyList_Check(tmp)) {
5552 PyErr_Format(PyExc_TypeError, "Call field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5553 goto failed;
5554 }
5555 len = PyList_GET_SIZE(tmp);
5556 args = asdl_seq_new(len, arena);
5557 if (args == NULL) goto failed;
5558 for (i = 0; i < len; i++) {
5559 expr_ty value;
5560 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5561 if (res != 0) goto failed;
5562 asdl_seq_SET(args, i, value);
5563 }
5564 Py_XDECREF(tmp);
5565 tmp = NULL;
5566 } else {
5567 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Call");
5568 return 1;
5569 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005570 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005571 int res;
5572 Py_ssize_t len;
5573 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005574 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005575 if (tmp == NULL) goto failed;
5576 if (!PyList_Check(tmp)) {
5577 PyErr_Format(PyExc_TypeError, "Call field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5578 goto failed;
5579 }
5580 len = PyList_GET_SIZE(tmp);
5581 keywords = asdl_seq_new(len, arena);
5582 if (keywords == NULL) goto failed;
5583 for (i = 0; i < len; i++) {
5584 keyword_ty value;
5585 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
5586 if (res != 0) goto failed;
5587 asdl_seq_SET(keywords, i, value);
5588 }
5589 Py_XDECREF(tmp);
5590 tmp = NULL;
5591 } else {
5592 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from Call");
5593 return 1;
5594 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005595 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005596 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005597 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005598 if (tmp == NULL) goto failed;
5599 res = obj2ast_expr(tmp, &starargs, arena);
5600 if (res != 0) goto failed;
5601 Py_XDECREF(tmp);
5602 tmp = NULL;
5603 } else {
5604 starargs = NULL;
5605 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005606 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005607 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005608 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005609 if (tmp == NULL) goto failed;
5610 res = obj2ast_expr(tmp, &kwargs, arena);
5611 if (res != 0) goto failed;
5612 Py_XDECREF(tmp);
5613 tmp = NULL;
5614 } else {
5615 kwargs = NULL;
5616 }
5617 *out = Call(func, args, keywords, starargs, kwargs, lineno,
5618 col_offset, arena);
5619 if (*out == NULL) goto failed;
5620 return 0;
5621 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005622 isinstance = PyObject_IsInstance(obj, (PyObject*)Num_type);
5623 if (isinstance == -1) {
5624 return 1;
5625 }
5626 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005627 object n;
5628
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005629 if (_PyObject_HasAttrId(obj, &PyId_n)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005630 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005631 tmp = _PyObject_GetAttrId(obj, &PyId_n);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005632 if (tmp == NULL) goto failed;
5633 res = obj2ast_object(tmp, &n, arena);
5634 if (res != 0) goto failed;
5635 Py_XDECREF(tmp);
5636 tmp = NULL;
5637 } else {
5638 PyErr_SetString(PyExc_TypeError, "required field \"n\" missing from Num");
5639 return 1;
5640 }
5641 *out = Num(n, lineno, col_offset, arena);
5642 if (*out == NULL) goto failed;
5643 return 0;
5644 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005645 isinstance = PyObject_IsInstance(obj, (PyObject*)Str_type);
5646 if (isinstance == -1) {
5647 return 1;
5648 }
5649 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005650 string s;
5651
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005652 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005653 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005654 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005655 if (tmp == NULL) goto failed;
5656 res = obj2ast_string(tmp, &s, arena);
5657 if (res != 0) goto failed;
5658 Py_XDECREF(tmp);
5659 tmp = NULL;
5660 } else {
5661 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Str");
5662 return 1;
5663 }
5664 *out = Str(s, lineno, col_offset, arena);
5665 if (*out == NULL) goto failed;
5666 return 0;
5667 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005668 isinstance = PyObject_IsInstance(obj, (PyObject*)Bytes_type);
5669 if (isinstance == -1) {
5670 return 1;
5671 }
5672 if (isinstance) {
Benjamin Petersone2498412011-08-09 16:08:39 -05005673 bytes s;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005674
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005675 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005676 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005677 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005678 if (tmp == NULL) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05005679 res = obj2ast_bytes(tmp, &s, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005680 if (res != 0) goto failed;
5681 Py_XDECREF(tmp);
5682 tmp = NULL;
5683 } else {
5684 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Bytes");
5685 return 1;
5686 }
5687 *out = Bytes(s, lineno, col_offset, arena);
5688 if (*out == NULL) goto failed;
5689 return 0;
5690 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005691 isinstance = PyObject_IsInstance(obj, (PyObject*)Ellipsis_type);
5692 if (isinstance == -1) {
5693 return 1;
5694 }
5695 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005696
5697 *out = Ellipsis(lineno, col_offset, arena);
5698 if (*out == NULL) goto failed;
5699 return 0;
5700 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005701 isinstance = PyObject_IsInstance(obj, (PyObject*)Attribute_type);
5702 if (isinstance == -1) {
5703 return 1;
5704 }
5705 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005706 expr_ty value;
5707 identifier attr;
5708 expr_context_ty ctx;
5709
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005710 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005711 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005712 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005713 if (tmp == NULL) goto failed;
5714 res = obj2ast_expr(tmp, &value, arena);
5715 if (res != 0) goto failed;
5716 Py_XDECREF(tmp);
5717 tmp = NULL;
5718 } else {
5719 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Attribute");
5720 return 1;
5721 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005722 if (_PyObject_HasAttrId(obj, &PyId_attr)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005723 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005724 tmp = _PyObject_GetAttrId(obj, &PyId_attr);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005725 if (tmp == NULL) goto failed;
5726 res = obj2ast_identifier(tmp, &attr, arena);
5727 if (res != 0) goto failed;
5728 Py_XDECREF(tmp);
5729 tmp = NULL;
5730 } else {
5731 PyErr_SetString(PyExc_TypeError, "required field \"attr\" missing from Attribute");
5732 return 1;
5733 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005734 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005735 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005736 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005737 if (tmp == NULL) goto failed;
5738 res = obj2ast_expr_context(tmp, &ctx, arena);
5739 if (res != 0) goto failed;
5740 Py_XDECREF(tmp);
5741 tmp = NULL;
5742 } else {
5743 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Attribute");
5744 return 1;
5745 }
5746 *out = Attribute(value, attr, ctx, lineno, col_offset, arena);
5747 if (*out == NULL) goto failed;
5748 return 0;
5749 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005750 isinstance = PyObject_IsInstance(obj, (PyObject*)Subscript_type);
5751 if (isinstance == -1) {
5752 return 1;
5753 }
5754 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005755 expr_ty value;
5756 slice_ty slice;
5757 expr_context_ty ctx;
5758
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005759 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005760 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005761 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005762 if (tmp == NULL) goto failed;
5763 res = obj2ast_expr(tmp, &value, arena);
5764 if (res != 0) goto failed;
5765 Py_XDECREF(tmp);
5766 tmp = NULL;
5767 } else {
5768 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Subscript");
5769 return 1;
5770 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005771 if (_PyObject_HasAttrId(obj, &PyId_slice)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005772 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005773 tmp = _PyObject_GetAttrId(obj, &PyId_slice);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005774 if (tmp == NULL) goto failed;
5775 res = obj2ast_slice(tmp, &slice, arena);
5776 if (res != 0) goto failed;
5777 Py_XDECREF(tmp);
5778 tmp = NULL;
5779 } else {
5780 PyErr_SetString(PyExc_TypeError, "required field \"slice\" missing from Subscript");
5781 return 1;
5782 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005783 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005784 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005785 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005786 if (tmp == NULL) goto failed;
5787 res = obj2ast_expr_context(tmp, &ctx, arena);
5788 if (res != 0) goto failed;
5789 Py_XDECREF(tmp);
5790 tmp = NULL;
5791 } else {
5792 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Subscript");
5793 return 1;
5794 }
5795 *out = Subscript(value, slice, ctx, lineno, col_offset, arena);
5796 if (*out == NULL) goto failed;
5797 return 0;
5798 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005799 isinstance = PyObject_IsInstance(obj, (PyObject*)Starred_type);
5800 if (isinstance == -1) {
5801 return 1;
5802 }
5803 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005804 expr_ty value;
5805 expr_context_ty ctx;
5806
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005807 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005808 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005809 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005810 if (tmp == NULL) goto failed;
5811 res = obj2ast_expr(tmp, &value, arena);
5812 if (res != 0) goto failed;
5813 Py_XDECREF(tmp);
5814 tmp = NULL;
5815 } else {
5816 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Starred");
5817 return 1;
5818 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005819 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005820 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005821 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005822 if (tmp == NULL) goto failed;
5823 res = obj2ast_expr_context(tmp, &ctx, arena);
5824 if (res != 0) goto failed;
5825 Py_XDECREF(tmp);
5826 tmp = NULL;
5827 } else {
5828 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Starred");
5829 return 1;
5830 }
5831 *out = Starred(value, ctx, lineno, col_offset, arena);
5832 if (*out == NULL) goto failed;
5833 return 0;
5834 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005835 isinstance = PyObject_IsInstance(obj, (PyObject*)Name_type);
5836 if (isinstance == -1) {
5837 return 1;
5838 }
5839 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005840 identifier id;
5841 expr_context_ty ctx;
5842
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005843 if (_PyObject_HasAttrId(obj, &PyId_id)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005844 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005845 tmp = _PyObject_GetAttrId(obj, &PyId_id);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005846 if (tmp == NULL) goto failed;
5847 res = obj2ast_identifier(tmp, &id, arena);
5848 if (res != 0) goto failed;
5849 Py_XDECREF(tmp);
5850 tmp = NULL;
5851 } else {
5852 PyErr_SetString(PyExc_TypeError, "required field \"id\" missing from Name");
5853 return 1;
5854 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005855 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005856 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005857 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005858 if (tmp == NULL) goto failed;
5859 res = obj2ast_expr_context(tmp, &ctx, arena);
5860 if (res != 0) goto failed;
5861 Py_XDECREF(tmp);
5862 tmp = NULL;
5863 } else {
5864 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Name");
5865 return 1;
5866 }
5867 *out = Name(id, ctx, lineno, col_offset, arena);
5868 if (*out == NULL) goto failed;
5869 return 0;
5870 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005871 isinstance = PyObject_IsInstance(obj, (PyObject*)List_type);
5872 if (isinstance == -1) {
5873 return 1;
5874 }
5875 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005876 asdl_seq* elts;
5877 expr_context_ty ctx;
5878
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005879 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005880 int res;
5881 Py_ssize_t len;
5882 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005883 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005884 if (tmp == NULL) goto failed;
5885 if (!PyList_Check(tmp)) {
5886 PyErr_Format(PyExc_TypeError, "List field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5887 goto failed;
5888 }
5889 len = PyList_GET_SIZE(tmp);
5890 elts = asdl_seq_new(len, arena);
5891 if (elts == NULL) goto failed;
5892 for (i = 0; i < len; i++) {
5893 expr_ty value;
5894 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5895 if (res != 0) goto failed;
5896 asdl_seq_SET(elts, i, value);
5897 }
5898 Py_XDECREF(tmp);
5899 tmp = NULL;
5900 } else {
5901 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from List");
5902 return 1;
5903 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005904 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005905 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005906 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005907 if (tmp == NULL) goto failed;
5908 res = obj2ast_expr_context(tmp, &ctx, arena);
5909 if (res != 0) goto failed;
5910 Py_XDECREF(tmp);
5911 tmp = NULL;
5912 } else {
5913 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from List");
5914 return 1;
5915 }
5916 *out = List(elts, ctx, lineno, col_offset, arena);
5917 if (*out == NULL) goto failed;
5918 return 0;
5919 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005920 isinstance = PyObject_IsInstance(obj, (PyObject*)Tuple_type);
5921 if (isinstance == -1) {
5922 return 1;
5923 }
5924 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005925 asdl_seq* elts;
5926 expr_context_ty ctx;
5927
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005928 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005929 int res;
5930 Py_ssize_t len;
5931 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005932 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005933 if (tmp == NULL) goto failed;
5934 if (!PyList_Check(tmp)) {
5935 PyErr_Format(PyExc_TypeError, "Tuple field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5936 goto failed;
5937 }
5938 len = PyList_GET_SIZE(tmp);
5939 elts = asdl_seq_new(len, arena);
5940 if (elts == NULL) goto failed;
5941 for (i = 0; i < len; i++) {
5942 expr_ty value;
5943 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5944 if (res != 0) goto failed;
5945 asdl_seq_SET(elts, i, value);
5946 }
5947 Py_XDECREF(tmp);
5948 tmp = NULL;
5949 } else {
5950 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Tuple");
5951 return 1;
5952 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005953 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005954 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005955 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005956 if (tmp == NULL) goto failed;
5957 res = obj2ast_expr_context(tmp, &ctx, arena);
5958 if (res != 0) goto failed;
5959 Py_XDECREF(tmp);
5960 tmp = NULL;
5961 } else {
5962 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Tuple");
5963 return 1;
5964 }
5965 *out = Tuple(elts, ctx, lineno, col_offset, arena);
5966 if (*out == NULL) goto failed;
5967 return 0;
5968 }
5969
Benjamin Peterson5b066812010-11-20 01:38:49 +00005970 PyErr_Format(PyExc_TypeError, "expected some sort of expr, but got %R", obj);
5971 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00005972 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005973 return 1;
5974}
5975
5976int
5977obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena* arena)
5978{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005979 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005980
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005981 isinstance = PyObject_IsInstance(obj, (PyObject *)Load_type);
5982 if (isinstance == -1) {
5983 return 1;
5984 }
5985 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005986 *out = Load;
5987 return 0;
5988 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005989 isinstance = PyObject_IsInstance(obj, (PyObject *)Store_type);
5990 if (isinstance == -1) {
5991 return 1;
5992 }
5993 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005994 *out = Store;
5995 return 0;
5996 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005997 isinstance = PyObject_IsInstance(obj, (PyObject *)Del_type);
5998 if (isinstance == -1) {
5999 return 1;
6000 }
6001 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006002 *out = Del;
6003 return 0;
6004 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006005 isinstance = PyObject_IsInstance(obj, (PyObject *)AugLoad_type);
6006 if (isinstance == -1) {
6007 return 1;
6008 }
6009 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006010 *out = AugLoad;
6011 return 0;
6012 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006013 isinstance = PyObject_IsInstance(obj, (PyObject *)AugStore_type);
6014 if (isinstance == -1) {
6015 return 1;
6016 }
6017 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006018 *out = AugStore;
6019 return 0;
6020 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006021 isinstance = PyObject_IsInstance(obj, (PyObject *)Param_type);
6022 if (isinstance == -1) {
6023 return 1;
6024 }
6025 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006026 *out = Param;
6027 return 0;
6028 }
6029
Benjamin Peterson5b066812010-11-20 01:38:49 +00006030 PyErr_Format(PyExc_TypeError, "expected some sort of expr_context, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006031 return 1;
6032}
6033
6034int
6035obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena)
6036{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006037 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006038
Benjamin Petersond8f65972010-11-20 04:31:07 +00006039 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006040
6041 if (obj == Py_None) {
6042 *out = NULL;
6043 return 0;
6044 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006045 isinstance = PyObject_IsInstance(obj, (PyObject*)Slice_type);
6046 if (isinstance == -1) {
6047 return 1;
6048 }
6049 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006050 expr_ty lower;
6051 expr_ty upper;
6052 expr_ty step;
6053
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006054 if (_PyObject_HasAttrId(obj, &PyId_lower)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006055 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006056 tmp = _PyObject_GetAttrId(obj, &PyId_lower);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006057 if (tmp == NULL) goto failed;
6058 res = obj2ast_expr(tmp, &lower, arena);
6059 if (res != 0) goto failed;
6060 Py_XDECREF(tmp);
6061 tmp = NULL;
6062 } else {
6063 lower = NULL;
6064 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006065 if (_PyObject_HasAttrId(obj, &PyId_upper)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006066 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006067 tmp = _PyObject_GetAttrId(obj, &PyId_upper);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006068 if (tmp == NULL) goto failed;
6069 res = obj2ast_expr(tmp, &upper, arena);
6070 if (res != 0) goto failed;
6071 Py_XDECREF(tmp);
6072 tmp = NULL;
6073 } else {
6074 upper = NULL;
6075 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006076 if (_PyObject_HasAttrId(obj, &PyId_step)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006077 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006078 tmp = _PyObject_GetAttrId(obj, &PyId_step);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006079 if (tmp == NULL) goto failed;
6080 res = obj2ast_expr(tmp, &step, arena);
6081 if (res != 0) goto failed;
6082 Py_XDECREF(tmp);
6083 tmp = NULL;
6084 } else {
6085 step = NULL;
6086 }
6087 *out = Slice(lower, upper, step, arena);
6088 if (*out == NULL) goto failed;
6089 return 0;
6090 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006091 isinstance = PyObject_IsInstance(obj, (PyObject*)ExtSlice_type);
6092 if (isinstance == -1) {
6093 return 1;
6094 }
6095 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006096 asdl_seq* dims;
6097
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006098 if (_PyObject_HasAttrId(obj, &PyId_dims)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006099 int res;
6100 Py_ssize_t len;
6101 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006102 tmp = _PyObject_GetAttrId(obj, &PyId_dims);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006103 if (tmp == NULL) goto failed;
6104 if (!PyList_Check(tmp)) {
6105 PyErr_Format(PyExc_TypeError, "ExtSlice field \"dims\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6106 goto failed;
6107 }
6108 len = PyList_GET_SIZE(tmp);
6109 dims = asdl_seq_new(len, arena);
6110 if (dims == NULL) goto failed;
6111 for (i = 0; i < len; i++) {
6112 slice_ty value;
6113 res = obj2ast_slice(PyList_GET_ITEM(tmp, i), &value, arena);
6114 if (res != 0) goto failed;
6115 asdl_seq_SET(dims, i, value);
6116 }
6117 Py_XDECREF(tmp);
6118 tmp = NULL;
6119 } else {
6120 PyErr_SetString(PyExc_TypeError, "required field \"dims\" missing from ExtSlice");
6121 return 1;
6122 }
6123 *out = ExtSlice(dims, arena);
6124 if (*out == NULL) goto failed;
6125 return 0;
6126 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006127 isinstance = PyObject_IsInstance(obj, (PyObject*)Index_type);
6128 if (isinstance == -1) {
6129 return 1;
6130 }
6131 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006132 expr_ty value;
6133
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006134 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006135 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006136 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006137 if (tmp == NULL) goto failed;
6138 res = obj2ast_expr(tmp, &value, arena);
6139 if (res != 0) goto failed;
6140 Py_XDECREF(tmp);
6141 tmp = NULL;
6142 } else {
6143 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Index");
6144 return 1;
6145 }
6146 *out = Index(value, arena);
6147 if (*out == NULL) goto failed;
6148 return 0;
6149 }
6150
Benjamin Peterson5b066812010-11-20 01:38:49 +00006151 PyErr_Format(PyExc_TypeError, "expected some sort of slice, but got %R", obj);
6152 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006153 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006154 return 1;
6155}
6156
6157int
6158obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena)
6159{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006160 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006161
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006162 isinstance = PyObject_IsInstance(obj, (PyObject *)And_type);
6163 if (isinstance == -1) {
6164 return 1;
6165 }
6166 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006167 *out = And;
6168 return 0;
6169 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006170 isinstance = PyObject_IsInstance(obj, (PyObject *)Or_type);
6171 if (isinstance == -1) {
6172 return 1;
6173 }
6174 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006175 *out = Or;
6176 return 0;
6177 }
6178
Benjamin Peterson5b066812010-11-20 01:38:49 +00006179 PyErr_Format(PyExc_TypeError, "expected some sort of boolop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006180 return 1;
6181}
6182
6183int
6184obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena)
6185{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006186 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006187
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006188 isinstance = PyObject_IsInstance(obj, (PyObject *)Add_type);
6189 if (isinstance == -1) {
6190 return 1;
6191 }
6192 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006193 *out = Add;
6194 return 0;
6195 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006196 isinstance = PyObject_IsInstance(obj, (PyObject *)Sub_type);
6197 if (isinstance == -1) {
6198 return 1;
6199 }
6200 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006201 *out = Sub;
6202 return 0;
6203 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006204 isinstance = PyObject_IsInstance(obj, (PyObject *)Mult_type);
6205 if (isinstance == -1) {
6206 return 1;
6207 }
6208 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006209 *out = Mult;
6210 return 0;
6211 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006212 isinstance = PyObject_IsInstance(obj, (PyObject *)Div_type);
6213 if (isinstance == -1) {
6214 return 1;
6215 }
6216 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006217 *out = Div;
6218 return 0;
6219 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006220 isinstance = PyObject_IsInstance(obj, (PyObject *)Mod_type);
6221 if (isinstance == -1) {
6222 return 1;
6223 }
6224 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006225 *out = Mod;
6226 return 0;
6227 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006228 isinstance = PyObject_IsInstance(obj, (PyObject *)Pow_type);
6229 if (isinstance == -1) {
6230 return 1;
6231 }
6232 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006233 *out = Pow;
6234 return 0;
6235 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006236 isinstance = PyObject_IsInstance(obj, (PyObject *)LShift_type);
6237 if (isinstance == -1) {
6238 return 1;
6239 }
6240 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006241 *out = LShift;
6242 return 0;
6243 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006244 isinstance = PyObject_IsInstance(obj, (PyObject *)RShift_type);
6245 if (isinstance == -1) {
6246 return 1;
6247 }
6248 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006249 *out = RShift;
6250 return 0;
6251 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006252 isinstance = PyObject_IsInstance(obj, (PyObject *)BitOr_type);
6253 if (isinstance == -1) {
6254 return 1;
6255 }
6256 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006257 *out = BitOr;
6258 return 0;
6259 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006260 isinstance = PyObject_IsInstance(obj, (PyObject *)BitXor_type);
6261 if (isinstance == -1) {
6262 return 1;
6263 }
6264 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006265 *out = BitXor;
6266 return 0;
6267 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006268 isinstance = PyObject_IsInstance(obj, (PyObject *)BitAnd_type);
6269 if (isinstance == -1) {
6270 return 1;
6271 }
6272 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006273 *out = BitAnd;
6274 return 0;
6275 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006276 isinstance = PyObject_IsInstance(obj, (PyObject *)FloorDiv_type);
6277 if (isinstance == -1) {
6278 return 1;
6279 }
6280 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006281 *out = FloorDiv;
6282 return 0;
6283 }
6284
Benjamin Peterson5b066812010-11-20 01:38:49 +00006285 PyErr_Format(PyExc_TypeError, "expected some sort of operator, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006286 return 1;
6287}
6288
6289int
6290obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena)
6291{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006292 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006293
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006294 isinstance = PyObject_IsInstance(obj, (PyObject *)Invert_type);
6295 if (isinstance == -1) {
6296 return 1;
6297 }
6298 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006299 *out = Invert;
6300 return 0;
6301 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006302 isinstance = PyObject_IsInstance(obj, (PyObject *)Not_type);
6303 if (isinstance == -1) {
6304 return 1;
6305 }
6306 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006307 *out = Not;
6308 return 0;
6309 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006310 isinstance = PyObject_IsInstance(obj, (PyObject *)UAdd_type);
6311 if (isinstance == -1) {
6312 return 1;
6313 }
6314 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006315 *out = UAdd;
6316 return 0;
6317 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006318 isinstance = PyObject_IsInstance(obj, (PyObject *)USub_type);
6319 if (isinstance == -1) {
6320 return 1;
6321 }
6322 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006323 *out = USub;
6324 return 0;
6325 }
6326
Benjamin Peterson5b066812010-11-20 01:38:49 +00006327 PyErr_Format(PyExc_TypeError, "expected some sort of unaryop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006328 return 1;
6329}
6330
6331int
6332obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena)
6333{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006334 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006335
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006336 isinstance = PyObject_IsInstance(obj, (PyObject *)Eq_type);
6337 if (isinstance == -1) {
6338 return 1;
6339 }
6340 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006341 *out = Eq;
6342 return 0;
6343 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006344 isinstance = PyObject_IsInstance(obj, (PyObject *)NotEq_type);
6345 if (isinstance == -1) {
6346 return 1;
6347 }
6348 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006349 *out = NotEq;
6350 return 0;
6351 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006352 isinstance = PyObject_IsInstance(obj, (PyObject *)Lt_type);
6353 if (isinstance == -1) {
6354 return 1;
6355 }
6356 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006357 *out = Lt;
6358 return 0;
6359 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006360 isinstance = PyObject_IsInstance(obj, (PyObject *)LtE_type);
6361 if (isinstance == -1) {
6362 return 1;
6363 }
6364 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006365 *out = LtE;
6366 return 0;
6367 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006368 isinstance = PyObject_IsInstance(obj, (PyObject *)Gt_type);
6369 if (isinstance == -1) {
6370 return 1;
6371 }
6372 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006373 *out = Gt;
6374 return 0;
6375 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006376 isinstance = PyObject_IsInstance(obj, (PyObject *)GtE_type);
6377 if (isinstance == -1) {
6378 return 1;
6379 }
6380 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006381 *out = GtE;
6382 return 0;
6383 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006384 isinstance = PyObject_IsInstance(obj, (PyObject *)Is_type);
6385 if (isinstance == -1) {
6386 return 1;
6387 }
6388 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006389 *out = Is;
6390 return 0;
6391 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006392 isinstance = PyObject_IsInstance(obj, (PyObject *)IsNot_type);
6393 if (isinstance == -1) {
6394 return 1;
6395 }
6396 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006397 *out = IsNot;
6398 return 0;
6399 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006400 isinstance = PyObject_IsInstance(obj, (PyObject *)In_type);
6401 if (isinstance == -1) {
6402 return 1;
6403 }
6404 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006405 *out = In;
6406 return 0;
6407 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006408 isinstance = PyObject_IsInstance(obj, (PyObject *)NotIn_type);
6409 if (isinstance == -1) {
6410 return 1;
6411 }
6412 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006413 *out = NotIn;
6414 return 0;
6415 }
6416
Benjamin Peterson5b066812010-11-20 01:38:49 +00006417 PyErr_Format(PyExc_TypeError, "expected some sort of cmpop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006418 return 1;
6419}
6420
6421int
6422obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena* arena)
6423{
6424 PyObject* tmp = NULL;
6425 expr_ty target;
6426 expr_ty iter;
6427 asdl_seq* ifs;
6428
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006429 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006430 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006431 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006432 if (tmp == NULL) goto failed;
6433 res = obj2ast_expr(tmp, &target, arena);
6434 if (res != 0) goto failed;
6435 Py_XDECREF(tmp);
6436 tmp = NULL;
6437 } else {
6438 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from comprehension");
6439 return 1;
6440 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006441 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006442 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006443 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006444 if (tmp == NULL) goto failed;
6445 res = obj2ast_expr(tmp, &iter, arena);
6446 if (res != 0) goto failed;
6447 Py_XDECREF(tmp);
6448 tmp = NULL;
6449 } else {
6450 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from comprehension");
6451 return 1;
6452 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006453 if (_PyObject_HasAttrId(obj, &PyId_ifs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006454 int res;
6455 Py_ssize_t len;
6456 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006457 tmp = _PyObject_GetAttrId(obj, &PyId_ifs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006458 if (tmp == NULL) goto failed;
6459 if (!PyList_Check(tmp)) {
6460 PyErr_Format(PyExc_TypeError, "comprehension field \"ifs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6461 goto failed;
6462 }
6463 len = PyList_GET_SIZE(tmp);
6464 ifs = asdl_seq_new(len, arena);
6465 if (ifs == NULL) goto failed;
6466 for (i = 0; i < len; i++) {
6467 expr_ty value;
6468 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6469 if (res != 0) goto failed;
6470 asdl_seq_SET(ifs, i, value);
6471 }
6472 Py_XDECREF(tmp);
6473 tmp = NULL;
6474 } else {
6475 PyErr_SetString(PyExc_TypeError, "required field \"ifs\" missing from comprehension");
6476 return 1;
6477 }
6478 *out = comprehension(target, iter, ifs, arena);
6479 return 0;
6480failed:
6481 Py_XDECREF(tmp);
6482 return 1;
6483}
6484
6485int
6486obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena* arena)
6487{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006488 int isinstance;
Neal Norwitzad74aa82008-03-31 05:14:30 +00006489
Benjamin Petersond8f65972010-11-20 04:31:07 +00006490 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006491 int lineno;
6492 int col_offset;
6493
Neal Norwitzad74aa82008-03-31 05:14:30 +00006494 if (obj == Py_None) {
6495 *out = NULL;
6496 return 0;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006497 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006498 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006499 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006500 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006501 if (tmp == NULL) goto failed;
6502 res = obj2ast_int(tmp, &lineno, arena);
6503 if (res != 0) goto failed;
6504 Py_XDECREF(tmp);
6505 tmp = NULL;
6506 } else {
6507 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from excepthandler");
6508 return 1;
6509 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006510 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006511 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006512 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006513 if (tmp == NULL) goto failed;
6514 res = obj2ast_int(tmp, &col_offset, arena);
6515 if (res != 0) goto failed;
6516 Py_XDECREF(tmp);
6517 tmp = NULL;
6518 } else {
6519 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from excepthandler");
6520 return 1;
6521 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006522 isinstance = PyObject_IsInstance(obj, (PyObject*)ExceptHandler_type);
6523 if (isinstance == -1) {
6524 return 1;
6525 }
6526 if (isinstance) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006527 expr_ty type;
6528 identifier name;
6529 asdl_seq* body;
6530
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006531 if (_PyObject_HasAttrId(obj, &PyId_type)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006532 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006533 tmp = _PyObject_GetAttrId(obj, &PyId_type);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006534 if (tmp == NULL) goto failed;
6535 res = obj2ast_expr(tmp, &type, arena);
6536 if (res != 0) goto failed;
6537 Py_XDECREF(tmp);
6538 tmp = NULL;
6539 } else {
6540 type = NULL;
6541 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006542 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006543 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006544 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006545 if (tmp == NULL) goto failed;
6546 res = obj2ast_identifier(tmp, &name, arena);
6547 if (res != 0) goto failed;
6548 Py_XDECREF(tmp);
6549 tmp = NULL;
6550 } else {
6551 name = NULL;
6552 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006553 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006554 int res;
6555 Py_ssize_t len;
6556 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006557 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006558 if (tmp == NULL) goto failed;
6559 if (!PyList_Check(tmp)) {
6560 PyErr_Format(PyExc_TypeError, "ExceptHandler field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6561 goto failed;
6562 }
6563 len = PyList_GET_SIZE(tmp);
6564 body = asdl_seq_new(len, arena);
6565 if (body == NULL) goto failed;
6566 for (i = 0; i < len; i++) {
6567 stmt_ty value;
6568 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
6569 if (res != 0) goto failed;
6570 asdl_seq_SET(body, i, value);
6571 }
6572 Py_XDECREF(tmp);
6573 tmp = NULL;
6574 } else {
6575 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ExceptHandler");
6576 return 1;
6577 }
6578 *out = ExceptHandler(type, name, body, lineno, col_offset,
6579 arena);
6580 if (*out == NULL) goto failed;
6581 return 0;
6582 }
6583
Benjamin Peterson5b066812010-11-20 01:38:49 +00006584 PyErr_Format(PyExc_TypeError, "expected some sort of excepthandler, but got %R", obj);
6585 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006586 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006587 return 1;
6588}
6589
6590int
6591obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena)
6592{
6593 PyObject* tmp = NULL;
6594 asdl_seq* args;
6595 identifier vararg;
6596 expr_ty varargannotation;
6597 asdl_seq* kwonlyargs;
6598 identifier kwarg;
6599 expr_ty kwargannotation;
6600 asdl_seq* defaults;
6601 asdl_seq* kw_defaults;
6602
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006603 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006604 int res;
6605 Py_ssize_t len;
6606 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006607 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006608 if (tmp == NULL) goto failed;
6609 if (!PyList_Check(tmp)) {
6610 PyErr_Format(PyExc_TypeError, "arguments field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6611 goto failed;
6612 }
6613 len = PyList_GET_SIZE(tmp);
6614 args = asdl_seq_new(len, arena);
6615 if (args == NULL) goto failed;
6616 for (i = 0; i < len; i++) {
6617 arg_ty value;
6618 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6619 if (res != 0) goto failed;
6620 asdl_seq_SET(args, i, value);
6621 }
6622 Py_XDECREF(tmp);
6623 tmp = NULL;
6624 } else {
6625 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from arguments");
6626 return 1;
6627 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006628 if (_PyObject_HasAttrId(obj, &PyId_vararg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006629 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006630 tmp = _PyObject_GetAttrId(obj, &PyId_vararg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006631 if (tmp == NULL) goto failed;
6632 res = obj2ast_identifier(tmp, &vararg, arena);
6633 if (res != 0) goto failed;
6634 Py_XDECREF(tmp);
6635 tmp = NULL;
6636 } else {
6637 vararg = NULL;
6638 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006639 if (_PyObject_HasAttrId(obj, &PyId_varargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006640 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006641 tmp = _PyObject_GetAttrId(obj, &PyId_varargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006642 if (tmp == NULL) goto failed;
6643 res = obj2ast_expr(tmp, &varargannotation, arena);
6644 if (res != 0) goto failed;
6645 Py_XDECREF(tmp);
6646 tmp = NULL;
6647 } else {
6648 varargannotation = NULL;
6649 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006650 if (_PyObject_HasAttrId(obj, &PyId_kwonlyargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006651 int res;
6652 Py_ssize_t len;
6653 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006654 tmp = _PyObject_GetAttrId(obj, &PyId_kwonlyargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006655 if (tmp == NULL) goto failed;
6656 if (!PyList_Check(tmp)) {
6657 PyErr_Format(PyExc_TypeError, "arguments field \"kwonlyargs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6658 goto failed;
6659 }
6660 len = PyList_GET_SIZE(tmp);
6661 kwonlyargs = asdl_seq_new(len, arena);
6662 if (kwonlyargs == NULL) goto failed;
6663 for (i = 0; i < len; i++) {
6664 arg_ty value;
6665 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6666 if (res != 0) goto failed;
6667 asdl_seq_SET(kwonlyargs, i, value);
6668 }
6669 Py_XDECREF(tmp);
6670 tmp = NULL;
6671 } else {
6672 PyErr_SetString(PyExc_TypeError, "required field \"kwonlyargs\" missing from arguments");
6673 return 1;
6674 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006675 if (_PyObject_HasAttrId(obj, &PyId_kwarg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006676 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006677 tmp = _PyObject_GetAttrId(obj, &PyId_kwarg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006678 if (tmp == NULL) goto failed;
6679 res = obj2ast_identifier(tmp, &kwarg, arena);
6680 if (res != 0) goto failed;
6681 Py_XDECREF(tmp);
6682 tmp = NULL;
6683 } else {
6684 kwarg = NULL;
6685 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006686 if (_PyObject_HasAttrId(obj, &PyId_kwargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006687 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006688 tmp = _PyObject_GetAttrId(obj, &PyId_kwargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006689 if (tmp == NULL) goto failed;
6690 res = obj2ast_expr(tmp, &kwargannotation, arena);
6691 if (res != 0) goto failed;
6692 Py_XDECREF(tmp);
6693 tmp = NULL;
6694 } else {
6695 kwargannotation = NULL;
6696 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006697 if (_PyObject_HasAttrId(obj, &PyId_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006698 int res;
6699 Py_ssize_t len;
6700 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006701 tmp = _PyObject_GetAttrId(obj, &PyId_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006702 if (tmp == NULL) goto failed;
6703 if (!PyList_Check(tmp)) {
6704 PyErr_Format(PyExc_TypeError, "arguments field \"defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6705 goto failed;
6706 }
6707 len = PyList_GET_SIZE(tmp);
6708 defaults = asdl_seq_new(len, arena);
6709 if (defaults == NULL) goto failed;
6710 for (i = 0; i < len; i++) {
6711 expr_ty value;
6712 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6713 if (res != 0) goto failed;
6714 asdl_seq_SET(defaults, i, value);
6715 }
6716 Py_XDECREF(tmp);
6717 tmp = NULL;
6718 } else {
6719 PyErr_SetString(PyExc_TypeError, "required field \"defaults\" missing from arguments");
6720 return 1;
6721 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006722 if (_PyObject_HasAttrId(obj, &PyId_kw_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006723 int res;
6724 Py_ssize_t len;
6725 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006726 tmp = _PyObject_GetAttrId(obj, &PyId_kw_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006727 if (tmp == NULL) goto failed;
6728 if (!PyList_Check(tmp)) {
6729 PyErr_Format(PyExc_TypeError, "arguments field \"kw_defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6730 goto failed;
6731 }
6732 len = PyList_GET_SIZE(tmp);
6733 kw_defaults = asdl_seq_new(len, arena);
6734 if (kw_defaults == NULL) goto failed;
6735 for (i = 0; i < len; i++) {
6736 expr_ty value;
6737 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6738 if (res != 0) goto failed;
6739 asdl_seq_SET(kw_defaults, i, value);
6740 }
6741 Py_XDECREF(tmp);
6742 tmp = NULL;
6743 } else {
6744 PyErr_SetString(PyExc_TypeError, "required field \"kw_defaults\" missing from arguments");
6745 return 1;
6746 }
6747 *out = arguments(args, vararg, varargannotation, kwonlyargs, kwarg,
6748 kwargannotation, defaults, kw_defaults, arena);
6749 return 0;
6750failed:
6751 Py_XDECREF(tmp);
6752 return 1;
6753}
6754
6755int
6756obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena)
6757{
6758 PyObject* tmp = NULL;
6759 identifier arg;
6760 expr_ty annotation;
6761
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006762 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006763 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006764 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006765 if (tmp == NULL) goto failed;
6766 res = obj2ast_identifier(tmp, &arg, arena);
6767 if (res != 0) goto failed;
6768 Py_XDECREF(tmp);
6769 tmp = NULL;
6770 } else {
6771 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from arg");
6772 return 1;
6773 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006774 if (_PyObject_HasAttrId(obj, &PyId_annotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006775 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006776 tmp = _PyObject_GetAttrId(obj, &PyId_annotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006777 if (tmp == NULL) goto failed;
6778 res = obj2ast_expr(tmp, &annotation, arena);
6779 if (res != 0) goto failed;
6780 Py_XDECREF(tmp);
6781 tmp = NULL;
6782 } else {
6783 annotation = NULL;
6784 }
6785 *out = arg(arg, annotation, arena);
6786 return 0;
6787failed:
6788 Py_XDECREF(tmp);
6789 return 1;
6790}
6791
6792int
6793obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena)
6794{
6795 PyObject* tmp = NULL;
6796 identifier arg;
6797 expr_ty value;
6798
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006799 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006800 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006801 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006802 if (tmp == NULL) goto failed;
6803 res = obj2ast_identifier(tmp, &arg, arena);
6804 if (res != 0) goto failed;
6805 Py_XDECREF(tmp);
6806 tmp = NULL;
6807 } else {
6808 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from keyword");
6809 return 1;
6810 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006811 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006812 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006813 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006814 if (tmp == NULL) goto failed;
6815 res = obj2ast_expr(tmp, &value, arena);
6816 if (res != 0) goto failed;
6817 Py_XDECREF(tmp);
6818 tmp = NULL;
6819 } else {
6820 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from keyword");
6821 return 1;
6822 }
6823 *out = keyword(arg, value, arena);
6824 return 0;
6825failed:
6826 Py_XDECREF(tmp);
6827 return 1;
6828}
6829
6830int
6831obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena)
6832{
6833 PyObject* tmp = NULL;
6834 identifier name;
6835 identifier asname;
6836
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006837 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006838 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006839 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006840 if (tmp == NULL) goto failed;
6841 res = obj2ast_identifier(tmp, &name, arena);
6842 if (res != 0) goto failed;
6843 Py_XDECREF(tmp);
6844 tmp = NULL;
6845 } else {
6846 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from alias");
6847 return 1;
6848 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006849 if (_PyObject_HasAttrId(obj, &PyId_asname)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006850 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006851 tmp = _PyObject_GetAttrId(obj, &PyId_asname);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006852 if (tmp == NULL) goto failed;
6853 res = obj2ast_identifier(tmp, &asname, arena);
6854 if (res != 0) goto failed;
6855 Py_XDECREF(tmp);
6856 tmp = NULL;
6857 } else {
6858 asname = NULL;
6859 }
6860 *out = alias(name, asname, arena);
6861 return 0;
6862failed:
6863 Py_XDECREF(tmp);
6864 return 1;
6865}
6866
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006867int
6868obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena)
6869{
6870 PyObject* tmp = NULL;
6871 expr_ty context_expr;
6872 expr_ty optional_vars;
6873
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006874 if (_PyObject_HasAttrId(obj, &PyId_context_expr)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006875 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006876 tmp = _PyObject_GetAttrId(obj, &PyId_context_expr);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006877 if (tmp == NULL) goto failed;
6878 res = obj2ast_expr(tmp, &context_expr, arena);
6879 if (res != 0) goto failed;
6880 Py_XDECREF(tmp);
6881 tmp = NULL;
6882 } else {
6883 PyErr_SetString(PyExc_TypeError, "required field \"context_expr\" missing from withitem");
6884 return 1;
6885 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006886 if (_PyObject_HasAttrId(obj, &PyId_optional_vars)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006887 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006888 tmp = _PyObject_GetAttrId(obj, &PyId_optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006889 if (tmp == NULL) goto failed;
6890 res = obj2ast_expr(tmp, &optional_vars, arena);
6891 if (res != 0) goto failed;
6892 Py_XDECREF(tmp);
6893 tmp = NULL;
6894 } else {
6895 optional_vars = NULL;
6896 }
6897 *out = withitem(context_expr, optional_vars, arena);
6898 return 0;
6899failed:
6900 Py_XDECREF(tmp);
6901 return 1;
6902}
6903
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006904
Martin v. Löwis1a214512008-06-11 05:26:20 +00006905static struct PyModuleDef _astmodule = {
6906 PyModuleDef_HEAD_INIT, "_ast"
6907};
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006908PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00006909PyInit__ast(void)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006910{
6911 PyObject *m, *d;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006912 if (!init_types()) return NULL;
6913 m = PyModule_Create(&_astmodule);
6914 if (!m) return NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006915 d = PyModule_GetDict(m);
Martin v. Löwis1a214512008-06-11 05:26:20 +00006916 if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return
6917 NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006918 if (PyModule_AddIntConstant(m, "PyCF_ONLY_AST", PyCF_ONLY_AST) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006919 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006920 if (PyDict_SetItemString(d, "mod", (PyObject*)mod_type) < 0) return
6921 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006922 if (PyDict_SetItemString(d, "Module", (PyObject*)Module_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006923 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006924 if (PyDict_SetItemString(d, "Interactive", (PyObject*)Interactive_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006925 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006926 if (PyDict_SetItemString(d, "Expression", (PyObject*)Expression_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006927 0) return NULL;
6928 if (PyDict_SetItemString(d, "Suite", (PyObject*)Suite_type) < 0) return
6929 NULL;
6930 if (PyDict_SetItemString(d, "stmt", (PyObject*)stmt_type) < 0) return
6931 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006932 if (PyDict_SetItemString(d, "FunctionDef", (PyObject*)FunctionDef_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006933 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006934 if (PyDict_SetItemString(d, "ClassDef", (PyObject*)ClassDef_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, "Return", (PyObject*)Return_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006937 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006938 if (PyDict_SetItemString(d, "Delete", (PyObject*)Delete_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006939 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006940 if (PyDict_SetItemString(d, "Assign", (PyObject*)Assign_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006941 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006942 if (PyDict_SetItemString(d, "AugAssign", (PyObject*)AugAssign_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006943 0) return NULL;
6944 if (PyDict_SetItemString(d, "For", (PyObject*)For_type) < 0) return
6945 NULL;
6946 if (PyDict_SetItemString(d, "While", (PyObject*)While_type) < 0) return
6947 NULL;
6948 if (PyDict_SetItemString(d, "If", (PyObject*)If_type) < 0) return NULL;
6949 if (PyDict_SetItemString(d, "With", (PyObject*)With_type) < 0) return
6950 NULL;
6951 if (PyDict_SetItemString(d, "Raise", (PyObject*)Raise_type) < 0) return
6952 NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05006953 if (PyDict_SetItemString(d, "Try", (PyObject*)Try_type) < 0) return
6954 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006955 if (PyDict_SetItemString(d, "Assert", (PyObject*)Assert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006956 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006957 if (PyDict_SetItemString(d, "Import", (PyObject*)Import_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006958 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006959 if (PyDict_SetItemString(d, "ImportFrom", (PyObject*)ImportFrom_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006960 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006961 if (PyDict_SetItemString(d, "Global", (PyObject*)Global_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006962 return NULL;
Jeremy Hylton81e95022007-02-27 06:50:52 +00006963 if (PyDict_SetItemString(d, "Nonlocal", (PyObject*)Nonlocal_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006964 return NULL;
6965 if (PyDict_SetItemString(d, "Expr", (PyObject*)Expr_type) < 0) return
6966 NULL;
6967 if (PyDict_SetItemString(d, "Pass", (PyObject*)Pass_type) < 0) return
6968 NULL;
6969 if (PyDict_SetItemString(d, "Break", (PyObject*)Break_type) < 0) return
6970 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006971 if (PyDict_SetItemString(d, "Continue", (PyObject*)Continue_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006972 return NULL;
6973 if (PyDict_SetItemString(d, "expr", (PyObject*)expr_type) < 0) return
6974 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006975 if (PyDict_SetItemString(d, "BoolOp", (PyObject*)BoolOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006976 return NULL;
6977 if (PyDict_SetItemString(d, "BinOp", (PyObject*)BinOp_type) < 0) return
6978 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006979 if (PyDict_SetItemString(d, "UnaryOp", (PyObject*)UnaryOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006980 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006981 if (PyDict_SetItemString(d, "Lambda", (PyObject*)Lambda_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006982 return NULL;
6983 if (PyDict_SetItemString(d, "IfExp", (PyObject*)IfExp_type) < 0) return
6984 NULL;
6985 if (PyDict_SetItemString(d, "Dict", (PyObject*)Dict_type) < 0) return
6986 NULL;
6987 if (PyDict_SetItemString(d, "Set", (PyObject*)Set_type) < 0) return
6988 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006989 if (PyDict_SetItemString(d, "ListComp", (PyObject*)ListComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006990 return NULL;
Nick Coghlan650f0d02007-04-15 12:05:43 +00006991 if (PyDict_SetItemString(d, "SetComp", (PyObject*)SetComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006992 return NULL;
Guido van Rossum992d4a32007-07-11 13:09:30 +00006993 if (PyDict_SetItemString(d, "DictComp", (PyObject*)DictComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006994 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006995 if (PyDict_SetItemString(d, "GeneratorExp",
Martin v. Löwis1a214512008-06-11 05:26:20 +00006996 (PyObject*)GeneratorExp_type) < 0) return NULL;
6997 if (PyDict_SetItemString(d, "Yield", (PyObject*)Yield_type) < 0) return
6998 NULL;
Benjamin Peterson527c6222012-01-14 08:58:23 -05006999 if (PyDict_SetItemString(d, "YieldFrom", (PyObject*)YieldFrom_type) <
7000 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007001 if (PyDict_SetItemString(d, "Compare", (PyObject*)Compare_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007002 return NULL;
7003 if (PyDict_SetItemString(d, "Call", (PyObject*)Call_type) < 0) return
7004 NULL;
7005 if (PyDict_SetItemString(d, "Num", (PyObject*)Num_type) < 0) return
7006 NULL;
7007 if (PyDict_SetItemString(d, "Str", (PyObject*)Str_type) < 0) return
7008 NULL;
7009 if (PyDict_SetItemString(d, "Bytes", (PyObject*)Bytes_type) < 0) return
7010 NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00007011 if (PyDict_SetItemString(d, "Ellipsis", (PyObject*)Ellipsis_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007012 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007013 if (PyDict_SetItemString(d, "Attribute", (PyObject*)Attribute_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007014 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007015 if (PyDict_SetItemString(d, "Subscript", (PyObject*)Subscript_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007016 0) return NULL;
Guido van Rossum0368b722007-05-11 16:50:42 +00007017 if (PyDict_SetItemString(d, "Starred", (PyObject*)Starred_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007018 return NULL;
7019 if (PyDict_SetItemString(d, "Name", (PyObject*)Name_type) < 0) return
7020 NULL;
7021 if (PyDict_SetItemString(d, "List", (PyObject*)List_type) < 0) return
7022 NULL;
7023 if (PyDict_SetItemString(d, "Tuple", (PyObject*)Tuple_type) < 0) return
7024 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007025 if (PyDict_SetItemString(d, "expr_context",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007026 (PyObject*)expr_context_type) < 0) return NULL;
7027 if (PyDict_SetItemString(d, "Load", (PyObject*)Load_type) < 0) return
7028 NULL;
7029 if (PyDict_SetItemString(d, "Store", (PyObject*)Store_type) < 0) return
7030 NULL;
7031 if (PyDict_SetItemString(d, "Del", (PyObject*)Del_type) < 0) return
7032 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007033 if (PyDict_SetItemString(d, "AugLoad", (PyObject*)AugLoad_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007034 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007035 if (PyDict_SetItemString(d, "AugStore", (PyObject*)AugStore_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007036 return NULL;
7037 if (PyDict_SetItemString(d, "Param", (PyObject*)Param_type) < 0) return
7038 NULL;
7039 if (PyDict_SetItemString(d, "slice", (PyObject*)slice_type) < 0) return
7040 NULL;
7041 if (PyDict_SetItemString(d, "Slice", (PyObject*)Slice_type) < 0) return
7042 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007043 if (PyDict_SetItemString(d, "ExtSlice", (PyObject*)ExtSlice_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007044 return NULL;
7045 if (PyDict_SetItemString(d, "Index", (PyObject*)Index_type) < 0) return
7046 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007047 if (PyDict_SetItemString(d, "boolop", (PyObject*)boolop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007048 return NULL;
7049 if (PyDict_SetItemString(d, "And", (PyObject*)And_type) < 0) return
7050 NULL;
7051 if (PyDict_SetItemString(d, "Or", (PyObject*)Or_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007052 if (PyDict_SetItemString(d, "operator", (PyObject*)operator_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007053 return NULL;
7054 if (PyDict_SetItemString(d, "Add", (PyObject*)Add_type) < 0) return
7055 NULL;
7056 if (PyDict_SetItemString(d, "Sub", (PyObject*)Sub_type) < 0) return
7057 NULL;
7058 if (PyDict_SetItemString(d, "Mult", (PyObject*)Mult_type) < 0) return
7059 NULL;
7060 if (PyDict_SetItemString(d, "Div", (PyObject*)Div_type) < 0) return
7061 NULL;
7062 if (PyDict_SetItemString(d, "Mod", (PyObject*)Mod_type) < 0) return
7063 NULL;
7064 if (PyDict_SetItemString(d, "Pow", (PyObject*)Pow_type) < 0) return
7065 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007066 if (PyDict_SetItemString(d, "LShift", (PyObject*)LShift_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, "RShift", (PyObject*)RShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007069 return NULL;
7070 if (PyDict_SetItemString(d, "BitOr", (PyObject*)BitOr_type) < 0) return
7071 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007072 if (PyDict_SetItemString(d, "BitXor", (PyObject*)BitXor_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, "BitAnd", (PyObject*)BitAnd_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007075 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007076 if (PyDict_SetItemString(d, "FloorDiv", (PyObject*)FloorDiv_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007077 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007078 if (PyDict_SetItemString(d, "unaryop", (PyObject*)unaryop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007079 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007080 if (PyDict_SetItemString(d, "Invert", (PyObject*)Invert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007081 return NULL;
7082 if (PyDict_SetItemString(d, "Not", (PyObject*)Not_type) < 0) return
7083 NULL;
7084 if (PyDict_SetItemString(d, "UAdd", (PyObject*)UAdd_type) < 0) return
7085 NULL;
7086 if (PyDict_SetItemString(d, "USub", (PyObject*)USub_type) < 0) return
7087 NULL;
7088 if (PyDict_SetItemString(d, "cmpop", (PyObject*)cmpop_type) < 0) return
7089 NULL;
7090 if (PyDict_SetItemString(d, "Eq", (PyObject*)Eq_type) < 0) return NULL;
7091 if (PyDict_SetItemString(d, "NotEq", (PyObject*)NotEq_type) < 0) return
7092 NULL;
7093 if (PyDict_SetItemString(d, "Lt", (PyObject*)Lt_type) < 0) return NULL;
7094 if (PyDict_SetItemString(d, "LtE", (PyObject*)LtE_type) < 0) return
7095 NULL;
7096 if (PyDict_SetItemString(d, "Gt", (PyObject*)Gt_type) < 0) return NULL;
7097 if (PyDict_SetItemString(d, "GtE", (PyObject*)GtE_type) < 0) return
7098 NULL;
7099 if (PyDict_SetItemString(d, "Is", (PyObject*)Is_type) < 0) return NULL;
7100 if (PyDict_SetItemString(d, "IsNot", (PyObject*)IsNot_type) < 0) return
7101 NULL;
7102 if (PyDict_SetItemString(d, "In", (PyObject*)In_type) < 0) return NULL;
7103 if (PyDict_SetItemString(d, "NotIn", (PyObject*)NotIn_type) < 0) return
7104 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007105 if (PyDict_SetItemString(d, "comprehension",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007106 (PyObject*)comprehension_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007107 if (PyDict_SetItemString(d, "excepthandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007108 (PyObject*)excepthandler_type) < 0) return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00007109 if (PyDict_SetItemString(d, "ExceptHandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007110 (PyObject*)ExceptHandler_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007111 if (PyDict_SetItemString(d, "arguments", (PyObject*)arguments_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007112 0) return NULL;
7113 if (PyDict_SetItemString(d, "arg", (PyObject*)arg_type) < 0) return
7114 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007115 if (PyDict_SetItemString(d, "keyword", (PyObject*)keyword_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007116 return NULL;
7117 if (PyDict_SetItemString(d, "alias", (PyObject*)alias_type) < 0) return
7118 NULL;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05007119 if (PyDict_SetItemString(d, "withitem", (PyObject*)withitem_type) < 0)
7120 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00007121 return m;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00007122}
7123
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007124
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007125PyObject* PyAST_mod2obj(mod_ty t)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007126{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007127 init_types();
7128 return ast2obj_mod(t);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007129}
Martin v. Löwis5b222132007-06-10 09:51:05 +00007130
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007131/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
7132mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007133{
7134 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007135 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
7136 (PyObject*)Interactive_type};
7137 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007138 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007139 assert(0 <= mode && mode <= 2);
7140
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007141 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007142
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007143 isinstance = PyObject_IsInstance(ast, req_type[mode]);
7144 if (isinstance == -1)
7145 return NULL;
7146 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007147 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
7148 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007149 return NULL;
7150 }
7151 if (obj2ast_mod(ast, &res, arena) != 0)
7152 return NULL;
7153 else
7154 return res;
7155}
7156
7157int PyAST_Check(PyObject* obj)
7158{
7159 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00007160 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007161}
7162
Martin v. Löwis5b222132007-06-10 09:51:05 +00007163