blob: b940417823497c64405f99c188bb54baf63f8f7b [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};
Benjamin Peterson442f2092012-12-06 17:41:04 -0500274static PyTypeObject *NameConstant_type;
275static char *NameConstant_fields[]={
276 "value",
277};
Georg Brandl52318d62006-09-06 07:06:08 +0000278static PyTypeObject *Ellipsis_type;
Neal Norwitz53d960c2006-02-28 22:47:29 +0000279static PyTypeObject *Attribute_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200280_Py_IDENTIFIER(attr);
281_Py_IDENTIFIER(ctx);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000282static char *Attribute_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000283 "value",
284 "attr",
285 "ctx",
286};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000287static PyTypeObject *Subscript_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200288_Py_IDENTIFIER(slice);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000289static char *Subscript_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000290 "value",
291 "slice",
292 "ctx",
293};
Guido van Rossum0368b722007-05-11 16:50:42 +0000294static PyTypeObject *Starred_type;
295static char *Starred_fields[]={
296 "value",
297 "ctx",
298};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000299static PyTypeObject *Name_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200300_Py_IDENTIFIER(id);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000301static char *Name_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000302 "id",
303 "ctx",
304};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000305static PyTypeObject *List_type;
306static char *List_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000307 "elts",
308 "ctx",
309};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000310static PyTypeObject *Tuple_type;
311static char *Tuple_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000312 "elts",
313 "ctx",
314};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000315static PyTypeObject *expr_context_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000316static PyObject *Load_singleton, *Store_singleton, *Del_singleton,
317*AugLoad_singleton, *AugStore_singleton, *Param_singleton;
318static PyObject* ast2obj_expr_context(expr_context_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000319static PyTypeObject *Load_type;
320static PyTypeObject *Store_type;
321static PyTypeObject *Del_type;
322static PyTypeObject *AugLoad_type;
323static PyTypeObject *AugStore_type;
324static PyTypeObject *Param_type;
325static PyTypeObject *slice_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000326static PyObject* ast2obj_slice(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000327static PyTypeObject *Slice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200328_Py_IDENTIFIER(lower);
329_Py_IDENTIFIER(upper);
330_Py_IDENTIFIER(step);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000331static char *Slice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000332 "lower",
333 "upper",
334 "step",
335};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000336static PyTypeObject *ExtSlice_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200337_Py_IDENTIFIER(dims);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000338static char *ExtSlice_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000339 "dims",
340};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000341static PyTypeObject *Index_type;
342static char *Index_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000343 "value",
344};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000345static PyTypeObject *boolop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000346static PyObject *And_singleton, *Or_singleton;
347static PyObject* ast2obj_boolop(boolop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000348static PyTypeObject *And_type;
349static PyTypeObject *Or_type;
350static PyTypeObject *operator_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000351static PyObject *Add_singleton, *Sub_singleton, *Mult_singleton,
352*Div_singleton, *Mod_singleton, *Pow_singleton, *LShift_singleton,
353*RShift_singleton, *BitOr_singleton, *BitXor_singleton, *BitAnd_singleton,
354*FloorDiv_singleton;
355static PyObject* ast2obj_operator(operator_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000356static PyTypeObject *Add_type;
357static PyTypeObject *Sub_type;
358static PyTypeObject *Mult_type;
359static PyTypeObject *Div_type;
360static PyTypeObject *Mod_type;
361static PyTypeObject *Pow_type;
362static PyTypeObject *LShift_type;
363static PyTypeObject *RShift_type;
364static PyTypeObject *BitOr_type;
365static PyTypeObject *BitXor_type;
366static PyTypeObject *BitAnd_type;
367static PyTypeObject *FloorDiv_type;
368static PyTypeObject *unaryop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000369static PyObject *Invert_singleton, *Not_singleton, *UAdd_singleton,
370*USub_singleton;
371static PyObject* ast2obj_unaryop(unaryop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000372static PyTypeObject *Invert_type;
373static PyTypeObject *Not_type;
374static PyTypeObject *UAdd_type;
375static PyTypeObject *USub_type;
376static PyTypeObject *cmpop_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000377static PyObject *Eq_singleton, *NotEq_singleton, *Lt_singleton, *LtE_singleton,
378*Gt_singleton, *GtE_singleton, *Is_singleton, *IsNot_singleton, *In_singleton,
379*NotIn_singleton;
380static PyObject* ast2obj_cmpop(cmpop_ty);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000381static PyTypeObject *Eq_type;
382static PyTypeObject *NotEq_type;
383static PyTypeObject *Lt_type;
384static PyTypeObject *LtE_type;
385static PyTypeObject *Gt_type;
386static PyTypeObject *GtE_type;
387static PyTypeObject *Is_type;
388static PyTypeObject *IsNot_type;
389static PyTypeObject *In_type;
390static PyTypeObject *NotIn_type;
391static PyTypeObject *comprehension_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000392static PyObject* ast2obj_comprehension(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200393_Py_IDENTIFIER(ifs);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000394static char *comprehension_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000395 "target",
396 "iter",
397 "ifs",
398};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000399static PyTypeObject *excepthandler_type;
Neal Norwitzad74aa82008-03-31 05:14:30 +0000400static char *excepthandler_attributes[] = {
401 "lineno",
402 "col_offset",
403};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000404static PyObject* ast2obj_excepthandler(void*);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000405static PyTypeObject *ExceptHandler_type;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200406_Py_IDENTIFIER(type);
Neal Norwitzad74aa82008-03-31 05:14:30 +0000407static char *ExceptHandler_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000408 "type",
409 "name",
410 "body",
411};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000412static PyTypeObject *arguments_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000413static PyObject* ast2obj_arguments(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200414_Py_IDENTIFIER(vararg);
415_Py_IDENTIFIER(varargannotation);
416_Py_IDENTIFIER(kwonlyargs);
417_Py_IDENTIFIER(kwarg);
418_Py_IDENTIFIER(kwargannotation);
419_Py_IDENTIFIER(defaults);
420_Py_IDENTIFIER(kw_defaults);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000421static char *arguments_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000422 "args",
423 "vararg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000424 "varargannotation",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000425 "kwonlyargs",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000426 "kwarg",
Neal Norwitzc1505362006-12-28 06:47:50 +0000427 "kwargannotation",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000428 "defaults",
Guido van Rossum4f72a782006-10-27 23:31:49 +0000429 "kw_defaults",
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000430};
Neal Norwitzc1505362006-12-28 06:47:50 +0000431static PyTypeObject *arg_type;
432static PyObject* ast2obj_arg(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200433_Py_IDENTIFIER(arg);
434_Py_IDENTIFIER(annotation);
Guido van Rossum1bc535d2007-05-15 18:46:22 +0000435static char *arg_fields[]={
Neal Norwitzc1505362006-12-28 06:47:50 +0000436 "arg",
437 "annotation",
438};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000439static PyTypeObject *keyword_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000440static PyObject* ast2obj_keyword(void*);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000441static char *keyword_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000442 "arg",
443 "value",
444};
Neal Norwitz53d960c2006-02-28 22:47:29 +0000445static PyTypeObject *alias_type;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000446static PyObject* ast2obj_alias(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200447_Py_IDENTIFIER(asname);
Neal Norwitz53d960c2006-02-28 22:47:29 +0000448static char *alias_fields[]={
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000449 "name",
450 "asname",
451};
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500452static PyTypeObject *withitem_type;
453static PyObject* ast2obj_withitem(void*);
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200454_Py_IDENTIFIER(context_expr);
455_Py_IDENTIFIER(optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500456static char *withitem_fields[]={
457 "context_expr",
458 "optional_vars",
459};
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000460
461
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700462typedef struct {
Victor Stinner45e50de2012-03-13 01:17:31 +0100463 PyObject_HEAD
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700464 PyObject *dict;
465} AST_object;
466
Benjamin Peterson1767e022012-03-14 21:50:29 -0500467static void
468ast_dealloc(AST_object *self)
469{
470 Py_CLEAR(self->dict);
Antoine Pitrou50750742012-07-08 12:43:32 +0200471 Py_TYPE(self)->tp_free(self);
Benjamin Peterson1767e022012-03-14 21:50:29 -0500472}
473
Neal Norwitz207c9f32008-03-31 04:42:11 +0000474static int
Benjamin Peterson81071762012-07-08 11:03:46 -0700475ast_traverse(AST_object *self, visitproc visit, void *arg)
476{
477 Py_VISIT(self->dict);
478 return 0;
479}
480
481static void
482ast_clear(AST_object *self)
483{
484 Py_CLEAR(self->dict);
485}
486
487static int
Neal Norwitz207c9f32008-03-31 04:42:11 +0000488ast_type_init(PyObject *self, PyObject *args, PyObject *kw)
489{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200490 _Py_IDENTIFIER(_fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000491 Py_ssize_t i, numfields = 0;
492 int res = -1;
493 PyObject *key, *value, *fields;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200494 fields = _PyObject_GetAttrId((PyObject*)Py_TYPE(self), &PyId__fields);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000495 if (!fields)
496 PyErr_Clear();
497 if (fields) {
498 numfields = PySequence_Size(fields);
499 if (numfields == -1)
500 goto cleanup;
501 }
502 res = 0; /* if no error occurs, this stays 0 to the end */
503 if (PyTuple_GET_SIZE(args) > 0) {
504 if (numfields != PyTuple_GET_SIZE(args)) {
505 PyErr_Format(PyExc_TypeError, "%.400s constructor takes %s"
Amaury Forgeot d'Arc245c70b2008-09-10 22:24:24 +0000506 "%zd positional argument%s",
Neal Norwitz207c9f32008-03-31 04:42:11 +0000507 Py_TYPE(self)->tp_name,
508 numfields == 0 ? "" : "either 0 or ",
509 numfields, numfields == 1 ? "" : "s");
510 res = -1;
511 goto cleanup;
512 }
513 for (i = 0; i < PyTuple_GET_SIZE(args); i++) {
514 /* cannot be reached when fields is NULL */
515 PyObject *name = PySequence_GetItem(fields, i);
516 if (!name) {
517 res = -1;
518 goto cleanup;
519 }
520 res = PyObject_SetAttr(self, name, PyTuple_GET_ITEM(args, i));
521 Py_DECREF(name);
522 if (res < 0)
523 goto cleanup;
524 }
525 }
526 if (kw) {
527 i = 0; /* needed by PyDict_Next */
528 while (PyDict_Next(kw, &i, &key, &value)) {
529 res = PyObject_SetAttr(self, key, value);
530 if (res < 0)
531 goto cleanup;
532 }
533 }
534 cleanup:
535 Py_XDECREF(fields);
536 return res;
537}
538
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000539/* Pickling support */
540static PyObject *
541ast_type_reduce(PyObject *self, PyObject *unused)
542{
543 PyObject *res;
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200544 _Py_IDENTIFIER(__dict__);
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +0200545 PyObject *dict = _PyObject_GetAttrId(self, &PyId___dict__);
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000546 if (dict == NULL) {
547 if (PyErr_ExceptionMatches(PyExc_AttributeError))
548 PyErr_Clear();
549 else
550 return NULL;
551 }
552 if (dict) {
553 res = Py_BuildValue("O()O", Py_TYPE(self), dict);
554 Py_DECREF(dict);
555 return res;
556 }
557 return Py_BuildValue("O()", Py_TYPE(self));
558}
559
560static PyMethodDef ast_type_methods[] = {
561 {"__reduce__", ast_type_reduce, METH_NOARGS, NULL},
562 {NULL}
563};
564
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700565static PyGetSetDef ast_type_getsets[] = {
566 {"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict},
567 {NULL}
568};
569
Neal Norwitz207c9f32008-03-31 04:42:11 +0000570static PyTypeObject AST_type = {
571 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000572 "_ast.AST",
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700573 sizeof(AST_object),
Neal Norwitz207c9f32008-03-31 04:42:11 +0000574 0,
Benjamin Peterson1767e022012-03-14 21:50:29 -0500575 (destructor)ast_dealloc, /* tp_dealloc */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000576 0, /* tp_print */
577 0, /* tp_getattr */
578 0, /* tp_setattr */
Mark Dickinsone94c6792009-02-02 20:36:42 +0000579 0, /* tp_reserved */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000580 0, /* tp_repr */
581 0, /* tp_as_number */
582 0, /* tp_as_sequence */
583 0, /* tp_as_mapping */
584 0, /* tp_hash */
585 0, /* tp_call */
586 0, /* tp_str */
587 PyObject_GenericGetAttr, /* tp_getattro */
588 PyObject_GenericSetAttr, /* tp_setattro */
589 0, /* tp_as_buffer */
Benjamin Peterson81071762012-07-08 11:03:46 -0700590 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, /* tp_flags */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000591 0, /* tp_doc */
Benjamin Peterson81071762012-07-08 11:03:46 -0700592 (traverseproc)ast_traverse, /* tp_traverse */
593 (inquiry)ast_clear, /* tp_clear */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000594 0, /* tp_richcompare */
595 0, /* tp_weaklistoffset */
596 0, /* tp_iter */
597 0, /* tp_iternext */
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000598 ast_type_methods, /* tp_methods */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000599 0, /* tp_members */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700600 ast_type_getsets, /* tp_getset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000601 0, /* tp_base */
602 0, /* tp_dict */
603 0, /* tp_descr_get */
604 0, /* tp_descr_set */
Benjamin Peterson7e0dbfb2012-03-12 09:46:44 -0700605 offsetof(AST_object, dict),/* tp_dictoffset */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000606 (initproc)ast_type_init, /* tp_init */
607 PyType_GenericAlloc, /* tp_alloc */
608 PyType_GenericNew, /* tp_new */
Benjamin Peterson81071762012-07-08 11:03:46 -0700609 PyObject_GC_Del, /* tp_free */
Neal Norwitz207c9f32008-03-31 04:42:11 +0000610};
611
612
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000613static PyTypeObject* make_type(char *type, PyTypeObject* base, char**fields, int num_fields)
614{
615 PyObject *fnames, *result;
616 int i;
Neal Norwitzee9b10a2008-03-31 05:29:39 +0000617 fnames = PyTuple_New(num_fields);
618 if (!fnames) return NULL;
619 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000620 PyObject *field = PyUnicode_FromString(fields[i]);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000621 if (!field) {
622 Py_DECREF(fnames);
623 return NULL;
624 }
625 PyTuple_SET_ITEM(fnames, i, field);
626 }
Victor Stinner7eeb5b52010-06-07 19:57:46 +0000627 result = PyObject_CallFunction((PyObject*)&PyType_Type, "s(O){sOss}",
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000628 type, base, "_fields", fnames, "__module__", "_ast");
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000629 Py_DECREF(fnames);
630 return (PyTypeObject*)result;
631}
632
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000633static int add_attributes(PyTypeObject* type, char**attrs, int num_fields)
634{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000635 int i, result;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200636 _Py_IDENTIFIER(_attributes);
Neal Norwitz207c9f32008-03-31 04:42:11 +0000637 PyObject *s, *l = PyTuple_New(num_fields);
Benjamin Peterson3e5cd1d2010-06-27 21:45:24 +0000638 if (!l)
639 return 0;
640 for (i = 0; i < num_fields; i++) {
Neal Norwitze4dc3242007-08-25 01:33:49 +0000641 s = PyUnicode_FromString(attrs[i]);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000642 if (!s) {
643 Py_DECREF(l);
644 return 0;
645 }
Neal Norwitz207c9f32008-03-31 04:42:11 +0000646 PyTuple_SET_ITEM(l, i, s);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000647 }
Martin v. Löwis1c67dd92011-10-14 15:16:45 +0200648 result = _PyObject_SetAttrId((PyObject*)type, &PyId__attributes, l) >= 0;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000649 Py_DECREF(l);
650 return result;
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000651}
652
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000653/* Conversion AST -> Python */
654
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000655static PyObject* ast2obj_list(asdl_seq *seq, PyObject* (*func)(void*))
656{
Benjamin Peterson77fa9372012-05-15 10:10:27 -0700657 Py_ssize_t i, n = asdl_seq_LEN(seq);
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000658 PyObject *result = PyList_New(n);
659 PyObject *value;
660 if (!result)
661 return NULL;
662 for (i = 0; i < n; i++) {
663 value = func(asdl_seq_GET(seq, i));
664 if (!value) {
665 Py_DECREF(result);
666 return NULL;
667 }
668 PyList_SET_ITEM(result, i, value);
669 }
670 return result;
671}
672
673static PyObject* ast2obj_object(void *o)
674{
675 if (!o)
676 o = Py_None;
677 Py_INCREF((PyObject*)o);
678 return (PyObject*)o;
679}
Benjamin Peterson442f2092012-12-06 17:41:04 -0500680#define ast2obj_singleton ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000681#define ast2obj_identifier ast2obj_object
682#define ast2obj_string ast2obj_object
Benjamin Petersone2498412011-08-09 16:08:39 -0500683#define ast2obj_bytes ast2obj_object
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000684
Thomas Woutersa44f3a32007-02-26 18:20:15 +0000685static PyObject* ast2obj_int(long b)
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000686{
Christian Heimes217cfd12007-12-02 14:31:20 +0000687 return PyLong_FromLong(b);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000688}
689
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000690/* Conversion Python -> AST */
691
Benjamin Peterson442f2092012-12-06 17:41:04 -0500692static int obj2ast_singleton(PyObject *obj, PyObject** out, PyArena* arena)
693{
694 if (obj != Py_None && obj != Py_True && obj != Py_False) {
695 PyErr_SetString(PyExc_ValueError,
696 "AST singleton must be True, False, or None");
697 return 1;
698 }
699 *out = obj;
700 return 0;
701}
702
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000703static int obj2ast_object(PyObject* obj, PyObject** out, PyArena* arena)
704{
705 if (obj == Py_None)
706 obj = NULL;
707 if (obj)
708 PyArena_AddPyObject(arena, obj);
709 Py_XINCREF(obj);
710 *out = obj;
711 return 0;
712}
713
Benjamin Peterson180e6352011-07-22 11:09:07 -0500714static int obj2ast_identifier(PyObject* obj, PyObject** out, PyArena* arena)
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500715{
Benjamin Peterson180e6352011-07-22 11:09:07 -0500716 if (!PyUnicode_CheckExact(obj) && obj != Py_None) {
717 PyErr_SetString(PyExc_TypeError, "AST identifier must be of type str");
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500718 return 1;
719 }
720 return obj2ast_object(obj, out, arena);
721}
722
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500723static int obj2ast_string(PyObject* obj, PyObject** out, PyArena* arena)
724{
Benjamin Peterson0224d4e2011-08-31 22:13:03 -0400725 if (!PyUnicode_CheckExact(obj) && !PyBytes_CheckExact(obj)) {
Benjamin Peterson180e6352011-07-22 11:09:07 -0500726 PyErr_SetString(PyExc_TypeError, "AST string must be of type str");
727 return 1;
728 }
729 return obj2ast_object(obj, out, arena);
Benjamin Peterson2193d2b2011-07-22 10:50:23 -0500730}
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000731
Benjamin Petersone2498412011-08-09 16:08:39 -0500732static int obj2ast_bytes(PyObject* obj, PyObject** out, PyArena* arena)
733{
734 if (!PyBytes_CheckExact(obj)) {
735 PyErr_SetString(PyExc_TypeError, "AST bytes must be of type bytes");
736 return 1;
737 }
738 return obj2ast_object(obj, out, arena);
739}
740
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000741static int obj2ast_int(PyObject* obj, int* out, PyArena* arena)
742{
743 int i;
744 if (!PyLong_Check(obj)) {
Amaury Forgeot d'Arc5e8f8102011-11-22 21:52:30 +0100745 PyErr_Format(PyExc_ValueError, "invalid integer value: %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000746 return 1;
747 }
748
749 i = (int)PyLong_AsLong(obj);
750 if (i == -1 && PyErr_Occurred())
751 return 1;
752 *out = i;
753 return 0;
754}
755
Benjamin Peterson1a6e0d02008-10-25 15:49:17 +0000756static int add_ast_fields(void)
Benjamin Petersonce825f12008-10-24 23:11:02 +0000757{
758 PyObject *empty_tuple, *d;
759 if (PyType_Ready(&AST_type) < 0)
760 return -1;
761 d = AST_type.tp_dict;
762 empty_tuple = PyTuple_New(0);
763 if (!empty_tuple ||
764 PyDict_SetItemString(d, "_fields", empty_tuple) < 0 ||
765 PyDict_SetItemString(d, "_attributes", empty_tuple) < 0) {
766 Py_XDECREF(empty_tuple);
767 return -1;
768 }
769 Py_DECREF(empty_tuple);
770 return 0;
771}
772
Martin v. Löwis618dc5e2008-03-30 20:03:44 +0000773
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000774static int init_types(void)
775{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000776 static int initialized;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000777 if (initialized) return 1;
Benjamin Petersonce825f12008-10-24 23:11:02 +0000778 if (add_ast_fields() < 0) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000779 mod_type = make_type("mod", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000780 if (!mod_type) return 0;
781 if (!add_attributes(mod_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000782 Module_type = make_type("Module", mod_type, Module_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000783 if (!Module_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000784 Interactive_type = make_type("Interactive", mod_type,
785 Interactive_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000786 if (!Interactive_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000787 Expression_type = make_type("Expression", mod_type, Expression_fields,
788 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000789 if (!Expression_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000790 Suite_type = make_type("Suite", mod_type, Suite_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000791 if (!Suite_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000792 stmt_type = make_type("stmt", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000793 if (!stmt_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000794 if (!add_attributes(stmt_type, stmt_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000795 FunctionDef_type = make_type("FunctionDef", stmt_type,
Neal Norwitzc1505362006-12-28 06:47:50 +0000796 FunctionDef_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000797 if (!FunctionDef_type) return 0;
Guido van Rossumd59da4b2007-05-22 18:11:13 +0000798 ClassDef_type = make_type("ClassDef", stmt_type, ClassDef_fields, 7);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000799 if (!ClassDef_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000800 Return_type = make_type("Return", stmt_type, Return_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000801 if (!Return_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000802 Delete_type = make_type("Delete", stmt_type, Delete_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000803 if (!Delete_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000804 Assign_type = make_type("Assign", stmt_type, Assign_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000805 if (!Assign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000806 AugAssign_type = make_type("AugAssign", stmt_type, AugAssign_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000807 if (!AugAssign_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000808 For_type = make_type("For", stmt_type, For_fields, 4);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000809 if (!For_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000810 While_type = make_type("While", stmt_type, While_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000811 if (!While_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000812 If_type = make_type("If", stmt_type, If_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000813 if (!If_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -0500814 With_type = make_type("With", stmt_type, With_fields, 2);
Guido van Rossumc2e20742006-02-27 22:32:47 +0000815 if (!With_type) return 0;
Collin Winter828f04a2007-08-31 00:04:24 +0000816 Raise_type = make_type("Raise", stmt_type, Raise_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000817 if (!Raise_type) return 0;
Benjamin Peterson43af12b2011-05-29 11:43:10 -0500818 Try_type = make_type("Try", stmt_type, Try_fields, 4);
819 if (!Try_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000820 Assert_type = make_type("Assert", stmt_type, Assert_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000821 if (!Assert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000822 Import_type = make_type("Import", stmt_type, Import_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000823 if (!Import_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000824 ImportFrom_type = make_type("ImportFrom", stmt_type, ImportFrom_fields,
Thomas Woutersf7f438b2006-02-28 16:09:29 +0000825 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000826 if (!ImportFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000827 Global_type = make_type("Global", stmt_type, Global_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000828 if (!Global_type) return 0;
Jeremy Hylton81e95022007-02-27 06:50:52 +0000829 Nonlocal_type = make_type("Nonlocal", stmt_type, Nonlocal_fields, 1);
830 if (!Nonlocal_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000831 Expr_type = make_type("Expr", stmt_type, Expr_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000832 if (!Expr_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000833 Pass_type = make_type("Pass", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000834 if (!Pass_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000835 Break_type = make_type("Break", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000836 if (!Break_type) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000837 Continue_type = make_type("Continue", stmt_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000838 if (!Continue_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000839 expr_type = make_type("expr", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000840 if (!expr_type) return 0;
Martin v. Löwis49c5da12006-03-01 22:49:05 +0000841 if (!add_attributes(expr_type, expr_attributes, 2)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000842 BoolOp_type = make_type("BoolOp", expr_type, BoolOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000843 if (!BoolOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000844 BinOp_type = make_type("BinOp", expr_type, BinOp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000845 if (!BinOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000846 UnaryOp_type = make_type("UnaryOp", expr_type, UnaryOp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000847 if (!UnaryOp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000848 Lambda_type = make_type("Lambda", expr_type, Lambda_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000849 if (!Lambda_type) return 0;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +0000850 IfExp_type = make_type("IfExp", expr_type, IfExp_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000851 if (!IfExp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000852 Dict_type = make_type("Dict", expr_type, Dict_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000853 if (!Dict_type) return 0;
Guido van Rossum86e58e22006-08-28 15:27:34 +0000854 Set_type = make_type("Set", expr_type, Set_fields, 1);
855 if (!Set_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000856 ListComp_type = make_type("ListComp", expr_type, ListComp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000857 if (!ListComp_type) return 0;
Nick Coghlan650f0d02007-04-15 12:05:43 +0000858 SetComp_type = make_type("SetComp", expr_type, SetComp_fields, 2);
859 if (!SetComp_type) return 0;
Guido van Rossum992d4a32007-07-11 13:09:30 +0000860 DictComp_type = make_type("DictComp", expr_type, DictComp_fields, 3);
861 if (!DictComp_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000862 GeneratorExp_type = make_type("GeneratorExp", expr_type,
863 GeneratorExp_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000864 if (!GeneratorExp_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500865 Yield_type = make_type("Yield", expr_type, Yield_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000866 if (!Yield_type) return 0;
Benjamin Peterson527c6222012-01-14 08:58:23 -0500867 YieldFrom_type = make_type("YieldFrom", expr_type, YieldFrom_fields, 1);
868 if (!YieldFrom_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000869 Compare_type = make_type("Compare", expr_type, Compare_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000870 if (!Compare_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000871 Call_type = make_type("Call", expr_type, Call_fields, 5);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000872 if (!Call_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000873 Num_type = make_type("Num", expr_type, Num_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000874 if (!Num_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000875 Str_type = make_type("Str", expr_type, Str_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000876 if (!Str_type) return 0;
Thomas Wouters00e41de2007-02-23 19:56:57 +0000877 Bytes_type = make_type("Bytes", expr_type, Bytes_fields, 1);
878 if (!Bytes_type) return 0;
Benjamin Peterson442f2092012-12-06 17:41:04 -0500879 NameConstant_type = make_type("NameConstant", expr_type,
880 NameConstant_fields, 1);
881 if (!NameConstant_type) return 0;
Georg Brandl52318d62006-09-06 07:06:08 +0000882 Ellipsis_type = make_type("Ellipsis", expr_type, NULL, 0);
883 if (!Ellipsis_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000884 Attribute_type = make_type("Attribute", expr_type, Attribute_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000885 if (!Attribute_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000886 Subscript_type = make_type("Subscript", expr_type, Subscript_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000887 if (!Subscript_type) return 0;
Guido van Rossum0368b722007-05-11 16:50:42 +0000888 Starred_type = make_type("Starred", expr_type, Starred_fields, 2);
889 if (!Starred_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000890 Name_type = make_type("Name", expr_type, Name_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000891 if (!Name_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000892 List_type = make_type("List", expr_type, List_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000893 if (!List_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000894 Tuple_type = make_type("Tuple", expr_type, Tuple_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000895 if (!Tuple_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000896 expr_context_type = make_type("expr_context", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000897 if (!expr_context_type) return 0;
898 if (!add_attributes(expr_context_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000899 Load_type = make_type("Load", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000900 if (!Load_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000901 Load_singleton = PyType_GenericNew(Load_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000902 if (!Load_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000903 Store_type = make_type("Store", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000904 if (!Store_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000905 Store_singleton = PyType_GenericNew(Store_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000906 if (!Store_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000907 Del_type = make_type("Del", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000908 if (!Del_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000909 Del_singleton = PyType_GenericNew(Del_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000910 if (!Del_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000911 AugLoad_type = make_type("AugLoad", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000912 if (!AugLoad_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000913 AugLoad_singleton = PyType_GenericNew(AugLoad_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000914 if (!AugLoad_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000915 AugStore_type = make_type("AugStore", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000916 if (!AugStore_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000917 AugStore_singleton = PyType_GenericNew(AugStore_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000918 if (!AugStore_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000919 Param_type = make_type("Param", expr_context_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000920 if (!Param_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000921 Param_singleton = PyType_GenericNew(Param_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000922 if (!Param_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000923 slice_type = make_type("slice", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000924 if (!slice_type) return 0;
925 if (!add_attributes(slice_type, NULL, 0)) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000926 Slice_type = make_type("Slice", slice_type, Slice_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000927 if (!Slice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000928 ExtSlice_type = make_type("ExtSlice", slice_type, ExtSlice_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000929 if (!ExtSlice_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000930 Index_type = make_type("Index", slice_type, Index_fields, 1);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000931 if (!Index_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000932 boolop_type = make_type("boolop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000933 if (!boolop_type) return 0;
934 if (!add_attributes(boolop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000935 And_type = make_type("And", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000936 if (!And_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000937 And_singleton = PyType_GenericNew(And_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000938 if (!And_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000939 Or_type = make_type("Or", boolop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000940 if (!Or_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000941 Or_singleton = PyType_GenericNew(Or_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000942 if (!Or_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000943 operator_type = make_type("operator", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000944 if (!operator_type) return 0;
945 if (!add_attributes(operator_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000946 Add_type = make_type("Add", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000947 if (!Add_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000948 Add_singleton = PyType_GenericNew(Add_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000949 if (!Add_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000950 Sub_type = make_type("Sub", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000951 if (!Sub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000952 Sub_singleton = PyType_GenericNew(Sub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000953 if (!Sub_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000954 Mult_type = make_type("Mult", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000955 if (!Mult_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000956 Mult_singleton = PyType_GenericNew(Mult_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000957 if (!Mult_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000958 Div_type = make_type("Div", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000959 if (!Div_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000960 Div_singleton = PyType_GenericNew(Div_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000961 if (!Div_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000962 Mod_type = make_type("Mod", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000963 if (!Mod_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000964 Mod_singleton = PyType_GenericNew(Mod_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000965 if (!Mod_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000966 Pow_type = make_type("Pow", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000967 if (!Pow_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000968 Pow_singleton = PyType_GenericNew(Pow_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000969 if (!Pow_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000970 LShift_type = make_type("LShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000971 if (!LShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000972 LShift_singleton = PyType_GenericNew(LShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000973 if (!LShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000974 RShift_type = make_type("RShift", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000975 if (!RShift_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000976 RShift_singleton = PyType_GenericNew(RShift_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000977 if (!RShift_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000978 BitOr_type = make_type("BitOr", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000979 if (!BitOr_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000980 BitOr_singleton = PyType_GenericNew(BitOr_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000981 if (!BitOr_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000982 BitXor_type = make_type("BitXor", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000983 if (!BitXor_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000984 BitXor_singleton = PyType_GenericNew(BitXor_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000985 if (!BitXor_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000986 BitAnd_type = make_type("BitAnd", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000987 if (!BitAnd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000988 BitAnd_singleton = PyType_GenericNew(BitAnd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000989 if (!BitAnd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000990 FloorDiv_type = make_type("FloorDiv", operator_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000991 if (!FloorDiv_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000992 FloorDiv_singleton = PyType_GenericNew(FloorDiv_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000993 if (!FloorDiv_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +0000994 unaryop_type = make_type("unaryop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000995 if (!unaryop_type) return 0;
996 if (!add_attributes(unaryop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +0000997 Invert_type = make_type("Invert", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +0000998 if (!Invert_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +0000999 Invert_singleton = PyType_GenericNew(Invert_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001000 if (!Invert_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001001 Not_type = make_type("Not", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001002 if (!Not_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001003 Not_singleton = PyType_GenericNew(Not_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001004 if (!Not_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001005 UAdd_type = make_type("UAdd", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001006 if (!UAdd_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001007 UAdd_singleton = PyType_GenericNew(UAdd_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001008 if (!UAdd_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001009 USub_type = make_type("USub", unaryop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001010 if (!USub_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001011 USub_singleton = PyType_GenericNew(USub_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001012 if (!USub_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001013 cmpop_type = make_type("cmpop", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001014 if (!cmpop_type) return 0;
1015 if (!add_attributes(cmpop_type, NULL, 0)) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001016 Eq_type = make_type("Eq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001017 if (!Eq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001018 Eq_singleton = PyType_GenericNew(Eq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001019 if (!Eq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001020 NotEq_type = make_type("NotEq", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001021 if (!NotEq_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001022 NotEq_singleton = PyType_GenericNew(NotEq_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001023 if (!NotEq_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001024 Lt_type = make_type("Lt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001025 if (!Lt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001026 Lt_singleton = PyType_GenericNew(Lt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001027 if (!Lt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001028 LtE_type = make_type("LtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001029 if (!LtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001030 LtE_singleton = PyType_GenericNew(LtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001031 if (!LtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001032 Gt_type = make_type("Gt", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001033 if (!Gt_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001034 Gt_singleton = PyType_GenericNew(Gt_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001035 if (!Gt_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001036 GtE_type = make_type("GtE", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001037 if (!GtE_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001038 GtE_singleton = PyType_GenericNew(GtE_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001039 if (!GtE_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001040 Is_type = make_type("Is", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001041 if (!Is_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001042 Is_singleton = PyType_GenericNew(Is_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001043 if (!Is_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001044 IsNot_type = make_type("IsNot", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001045 if (!IsNot_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001046 IsNot_singleton = PyType_GenericNew(IsNot_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001047 if (!IsNot_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001048 In_type = make_type("In", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001049 if (!In_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001050 In_singleton = PyType_GenericNew(In_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001051 if (!In_singleton) return 0;
Martin v. Löwis8d0701d2006-02-26 23:40:20 +00001052 NotIn_type = make_type("NotIn", cmpop_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001053 if (!NotIn_type) return 0;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001054 NotIn_singleton = PyType_GenericNew(NotIn_type, NULL, NULL);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001055 if (!NotIn_singleton) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001056 comprehension_type = make_type("comprehension", &AST_type,
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001057 comprehension_fields, 3);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001058 if (!comprehension_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001059 excepthandler_type = make_type("excepthandler", &AST_type, NULL, 0);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001060 if (!excepthandler_type) return 0;
Neal Norwitzad74aa82008-03-31 05:14:30 +00001061 if (!add_attributes(excepthandler_type, excepthandler_attributes, 2))
1062 return 0;
1063 ExceptHandler_type = make_type("ExceptHandler", excepthandler_type,
1064 ExceptHandler_fields, 3);
1065 if (!ExceptHandler_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001066 arguments_type = make_type("arguments", &AST_type, arguments_fields, 8);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001067 if (!arguments_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001068 arg_type = make_type("arg", &AST_type, arg_fields, 2);
Neal Norwitzc1505362006-12-28 06:47:50 +00001069 if (!arg_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001070 keyword_type = make_type("keyword", &AST_type, keyword_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001071 if (!keyword_type) return 0;
Neal Norwitz207c9f32008-03-31 04:42:11 +00001072 alias_type = make_type("alias", &AST_type, alias_fields, 2);
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001073 if (!alias_type) return 0;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001074 withitem_type = make_type("withitem", &AST_type, withitem_fields, 2);
1075 if (!withitem_type) return 0;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00001076 initialized = 1;
1077 return 1;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00001078}
Neal Norwitz7b5a6042005-11-13 19:14:20 +00001079
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001080static int obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena);
1081static int obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena);
1082static int obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena);
1083static int obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena*
1084 arena);
1085static int obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena);
1086static int obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena);
1087static int obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena);
1088static int obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena);
1089static int obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena);
1090static int obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena*
1091 arena);
1092static int obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena*
1093 arena);
1094static int obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena);
1095static int obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena);
1096static int obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena);
1097static int obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001098static int obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00001099
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001100mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001101Module(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001102{
1103 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001104 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001105 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001106 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001107 p->kind = Module_kind;
1108 p->v.Module.body = body;
1109 return p;
1110}
1111
1112mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001113Interactive(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001114{
1115 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001116 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001117 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001118 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001119 p->kind = Interactive_kind;
1120 p->v.Interactive.body = body;
1121 return p;
1122}
1123
1124mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001125Expression(expr_ty body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001126{
1127 mod_ty p;
1128 if (!body) {
1129 PyErr_SetString(PyExc_ValueError,
1130 "field body is required for Expression");
1131 return NULL;
1132 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001133 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001134 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001135 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001136 p->kind = Expression_kind;
1137 p->v.Expression.body = body;
1138 return p;
1139}
1140
1141mod_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001142Suite(asdl_seq * body, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001143{
1144 mod_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001145 p = (mod_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001146 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001147 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001148 p->kind = Suite_kind;
1149 p->v.Suite.body = body;
1150 return p;
1151}
1152
1153stmt_ty
1154FunctionDef(identifier name, arguments_ty args, asdl_seq * body, asdl_seq *
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001155 decorator_list, expr_ty returns, int lineno, int col_offset,
1156 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001157{
1158 stmt_ty p;
1159 if (!name) {
1160 PyErr_SetString(PyExc_ValueError,
1161 "field name is required for FunctionDef");
1162 return NULL;
1163 }
1164 if (!args) {
1165 PyErr_SetString(PyExc_ValueError,
1166 "field args is required for FunctionDef");
1167 return NULL;
1168 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001169 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001170 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001171 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001172 p->kind = FunctionDef_kind;
1173 p->v.FunctionDef.name = name;
1174 p->v.FunctionDef.args = args;
1175 p->v.FunctionDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001176 p->v.FunctionDef.decorator_list = decorator_list;
Neal Norwitzc1505362006-12-28 06:47:50 +00001177 p->v.FunctionDef.returns = returns;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001178 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001179 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001180 return p;
1181}
1182
1183stmt_ty
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001184ClassDef(identifier name, asdl_seq * bases, asdl_seq * keywords, expr_ty
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001185 starargs, expr_ty kwargs, asdl_seq * body, asdl_seq * decorator_list,
1186 int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001187{
1188 stmt_ty p;
1189 if (!name) {
1190 PyErr_SetString(PyExc_ValueError,
1191 "field name is required for ClassDef");
1192 return NULL;
1193 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001194 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001195 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001196 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001197 p->kind = ClassDef_kind;
1198 p->v.ClassDef.name = name;
1199 p->v.ClassDef.bases = bases;
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001200 p->v.ClassDef.keywords = keywords;
1201 p->v.ClassDef.starargs = starargs;
1202 p->v.ClassDef.kwargs = kwargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001203 p->v.ClassDef.body = body;
Guido van Rossumd59da4b2007-05-22 18:11:13 +00001204 p->v.ClassDef.decorator_list = decorator_list;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001205 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001206 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001207 return p;
1208}
1209
1210stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001211Return(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001212{
1213 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001214 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001215 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001216 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001217 p->kind = Return_kind;
1218 p->v.Return.value = value;
1219 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001220 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001221 return p;
1222}
1223
1224stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001225Delete(asdl_seq * targets, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001226{
1227 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001228 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001229 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001230 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001231 p->kind = Delete_kind;
1232 p->v.Delete.targets = targets;
1233 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001234 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001235 return p;
1236}
1237
1238stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001239Assign(asdl_seq * targets, expr_ty value, int lineno, int col_offset, PyArena
1240 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001241{
1242 stmt_ty p;
1243 if (!value) {
1244 PyErr_SetString(PyExc_ValueError,
1245 "field value is required for Assign");
1246 return NULL;
1247 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001248 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001249 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001250 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001251 p->kind = Assign_kind;
1252 p->v.Assign.targets = targets;
1253 p->v.Assign.value = value;
1254 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001255 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001256 return p;
1257}
1258
1259stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001260AugAssign(expr_ty target, operator_ty op, expr_ty value, int lineno, int
1261 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001262{
1263 stmt_ty p;
1264 if (!target) {
1265 PyErr_SetString(PyExc_ValueError,
1266 "field target is required for AugAssign");
1267 return NULL;
1268 }
1269 if (!op) {
1270 PyErr_SetString(PyExc_ValueError,
1271 "field op is required for AugAssign");
1272 return NULL;
1273 }
1274 if (!value) {
1275 PyErr_SetString(PyExc_ValueError,
1276 "field value is required for AugAssign");
1277 return NULL;
1278 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001279 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001280 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001281 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001282 p->kind = AugAssign_kind;
1283 p->v.AugAssign.target = target;
1284 p->v.AugAssign.op = op;
1285 p->v.AugAssign.value = value;
1286 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001287 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001288 return p;
1289}
1290
1291stmt_ty
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001292For(expr_ty target, expr_ty iter, asdl_seq * body, asdl_seq * orelse, int
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001293 lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001294{
1295 stmt_ty p;
1296 if (!target) {
1297 PyErr_SetString(PyExc_ValueError,
1298 "field target is required for For");
1299 return NULL;
1300 }
1301 if (!iter) {
1302 PyErr_SetString(PyExc_ValueError,
1303 "field iter is required for For");
1304 return NULL;
1305 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001306 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001307 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001308 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001309 p->kind = For_kind;
1310 p->v.For.target = target;
1311 p->v.For.iter = iter;
1312 p->v.For.body = body;
1313 p->v.For.orelse = orelse;
1314 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001315 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001316 return p;
1317}
1318
1319stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001320While(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1321 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001322{
1323 stmt_ty p;
1324 if (!test) {
1325 PyErr_SetString(PyExc_ValueError,
1326 "field test is required for While");
1327 return NULL;
1328 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001329 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001330 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001331 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001332 p->kind = While_kind;
1333 p->v.While.test = test;
1334 p->v.While.body = body;
1335 p->v.While.orelse = orelse;
1336 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001337 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001338 return p;
1339}
1340
1341stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001342If(expr_ty test, asdl_seq * body, asdl_seq * orelse, int lineno, int
1343 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001344{
1345 stmt_ty p;
1346 if (!test) {
1347 PyErr_SetString(PyExc_ValueError,
1348 "field test is required for If");
1349 return NULL;
1350 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001351 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001352 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001353 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001354 p->kind = If_kind;
1355 p->v.If.test = test;
1356 p->v.If.body = body;
1357 p->v.If.orelse = orelse;
1358 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001359 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001360 return p;
1361}
1362
1363stmt_ty
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001364With(asdl_seq * items, asdl_seq * body, int lineno, int col_offset, PyArena
1365 *arena)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001366{
1367 stmt_ty p;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001368 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001369 if (!p)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001370 return NULL;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001371 p->kind = With_kind;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05001372 p->v.With.items = items;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001373 p->v.With.body = body;
1374 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001375 p->col_offset = col_offset;
Guido van Rossumc2e20742006-02-27 22:32:47 +00001376 return p;
1377}
1378
1379stmt_ty
Collin Winter828f04a2007-08-31 00:04:24 +00001380Raise(expr_ty exc, expr_ty cause, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001381{
1382 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001383 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001384 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001385 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001386 p->kind = Raise_kind;
Collin Winter828f04a2007-08-31 00:04:24 +00001387 p->v.Raise.exc = exc;
1388 p->v.Raise.cause = cause;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001389 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001390 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001391 return p;
1392}
1393
1394stmt_ty
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001395Try(asdl_seq * body, asdl_seq * handlers, asdl_seq * orelse, asdl_seq *
1396 finalbody, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001397{
1398 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001399 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001400 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001401 return NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05001402 p->kind = Try_kind;
1403 p->v.Try.body = body;
1404 p->v.Try.handlers = handlers;
1405 p->v.Try.orelse = orelse;
1406 p->v.Try.finalbody = finalbody;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001407 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001408 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001409 return p;
1410}
1411
1412stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001413Assert(expr_ty test, expr_ty msg, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001414{
1415 stmt_ty p;
1416 if (!test) {
1417 PyErr_SetString(PyExc_ValueError,
1418 "field test is required for Assert");
1419 return NULL;
1420 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001421 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001422 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001423 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001424 p->kind = Assert_kind;
1425 p->v.Assert.test = test;
1426 p->v.Assert.msg = msg;
1427 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001428 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001429 return p;
1430}
1431
1432stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001433Import(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001434{
1435 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001436 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001437 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001438 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001439 p->kind = Import_kind;
1440 p->v.Import.names = names;
1441 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001442 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001443 return p;
1444}
1445
1446stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001447ImportFrom(identifier module, asdl_seq * names, int level, int lineno, int
1448 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001449{
1450 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001451 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001452 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001453 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001454 p->kind = ImportFrom_kind;
1455 p->v.ImportFrom.module = module;
1456 p->v.ImportFrom.names = names;
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001457 p->v.ImportFrom.level = level;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001458 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001459 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001460 return p;
1461}
1462
1463stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001464Global(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001465{
1466 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001467 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001468 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001469 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001470 p->kind = Global_kind;
1471 p->v.Global.names = names;
1472 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001473 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001474 return p;
1475}
1476
1477stmt_ty
Jeremy Hylton81e95022007-02-27 06:50:52 +00001478Nonlocal(asdl_seq * names, int lineno, int col_offset, PyArena *arena)
1479{
1480 stmt_ty p;
1481 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
1482 if (!p)
1483 return NULL;
1484 p->kind = Nonlocal_kind;
1485 p->v.Nonlocal.names = names;
1486 p->lineno = lineno;
1487 p->col_offset = col_offset;
1488 return p;
1489}
1490
1491stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001492Expr(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001493{
1494 stmt_ty p;
1495 if (!value) {
1496 PyErr_SetString(PyExc_ValueError,
1497 "field value is required for Expr");
1498 return NULL;
1499 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001500 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001501 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001502 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001503 p->kind = Expr_kind;
1504 p->v.Expr.value = value;
1505 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001506 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001507 return p;
1508}
1509
1510stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001511Pass(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001512{
1513 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001514 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001515 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001516 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001517 p->kind = Pass_kind;
1518 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001519 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001520 return p;
1521}
1522
1523stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001524Break(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001525{
1526 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001527 p = (stmt_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001528 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001529 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001530 p->kind = Break_kind;
1531 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001532 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001533 return p;
1534}
1535
1536stmt_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001537Continue(int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001538{
1539 stmt_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001540 p = (stmt_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 = Continue_kind;
1544 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001545 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001546 return p;
1547}
1548
1549expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001550BoolOp(boolop_ty op, asdl_seq * values, int lineno, int col_offset, PyArena
1551 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001552{
1553 expr_ty p;
1554 if (!op) {
1555 PyErr_SetString(PyExc_ValueError,
1556 "field op is required for BoolOp");
1557 return NULL;
1558 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001559 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001560 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001561 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001562 p->kind = BoolOp_kind;
1563 p->v.BoolOp.op = op;
1564 p->v.BoolOp.values = values;
1565 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001566 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001567 return p;
1568}
1569
1570expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001571BinOp(expr_ty left, operator_ty op, expr_ty right, int lineno, int col_offset,
1572 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001573{
1574 expr_ty p;
1575 if (!left) {
1576 PyErr_SetString(PyExc_ValueError,
1577 "field left is required for BinOp");
1578 return NULL;
1579 }
1580 if (!op) {
1581 PyErr_SetString(PyExc_ValueError,
1582 "field op is required for BinOp");
1583 return NULL;
1584 }
1585 if (!right) {
1586 PyErr_SetString(PyExc_ValueError,
1587 "field right is required for BinOp");
1588 return NULL;
1589 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001590 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001591 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001592 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001593 p->kind = BinOp_kind;
1594 p->v.BinOp.left = left;
1595 p->v.BinOp.op = op;
1596 p->v.BinOp.right = right;
1597 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001598 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001599 return p;
1600}
1601
1602expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001603UnaryOp(unaryop_ty op, expr_ty operand, int lineno, int col_offset, PyArena
1604 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001605{
1606 expr_ty p;
1607 if (!op) {
1608 PyErr_SetString(PyExc_ValueError,
1609 "field op is required for UnaryOp");
1610 return NULL;
1611 }
1612 if (!operand) {
1613 PyErr_SetString(PyExc_ValueError,
1614 "field operand is required for UnaryOp");
1615 return NULL;
1616 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001617 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001618 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001619 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001620 p->kind = UnaryOp_kind;
1621 p->v.UnaryOp.op = op;
1622 p->v.UnaryOp.operand = operand;
1623 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001624 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001625 return p;
1626}
1627
1628expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001629Lambda(arguments_ty args, expr_ty body, int lineno, int col_offset, PyArena
1630 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001631{
1632 expr_ty p;
1633 if (!args) {
1634 PyErr_SetString(PyExc_ValueError,
1635 "field args is required for Lambda");
1636 return NULL;
1637 }
1638 if (!body) {
1639 PyErr_SetString(PyExc_ValueError,
1640 "field body is required for Lambda");
1641 return NULL;
1642 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001643 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001644 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001645 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001646 p->kind = Lambda_kind;
1647 p->v.Lambda.args = args;
1648 p->v.Lambda.body = body;
1649 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001650 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001651 return p;
1652}
1653
1654expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001655IfExp(expr_ty test, expr_ty body, expr_ty orelse, int lineno, int col_offset,
1656 PyArena *arena)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001657{
1658 expr_ty p;
1659 if (!test) {
1660 PyErr_SetString(PyExc_ValueError,
1661 "field test is required for IfExp");
1662 return NULL;
1663 }
1664 if (!body) {
1665 PyErr_SetString(PyExc_ValueError,
1666 "field body is required for IfExp");
1667 return NULL;
1668 }
1669 if (!orelse) {
1670 PyErr_SetString(PyExc_ValueError,
1671 "field orelse is required for IfExp");
1672 return NULL;
1673 }
1674 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001675 if (!p)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001676 return NULL;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001677 p->kind = IfExp_kind;
1678 p->v.IfExp.test = test;
1679 p->v.IfExp.body = body;
1680 p->v.IfExp.orelse = orelse;
1681 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001682 p->col_offset = col_offset;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00001683 return p;
1684}
1685
1686expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001687Dict(asdl_seq * keys, asdl_seq * values, int lineno, int col_offset, PyArena
1688 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001689{
1690 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001691 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001692 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001693 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001694 p->kind = Dict_kind;
1695 p->v.Dict.keys = keys;
1696 p->v.Dict.values = values;
1697 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001698 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001699 return p;
1700}
1701
1702expr_ty
Guido van Rossum86e58e22006-08-28 15:27:34 +00001703Set(asdl_seq * elts, int lineno, int col_offset, PyArena *arena)
1704{
1705 expr_ty p;
1706 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001707 if (!p)
Guido van Rossum86e58e22006-08-28 15:27:34 +00001708 return NULL;
Guido van Rossum86e58e22006-08-28 15:27:34 +00001709 p->kind = Set_kind;
1710 p->v.Set.elts = elts;
1711 p->lineno = lineno;
1712 p->col_offset = col_offset;
1713 return p;
1714}
1715
1716expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001717ListComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1718 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001719{
1720 expr_ty p;
1721 if (!elt) {
1722 PyErr_SetString(PyExc_ValueError,
1723 "field elt is required for ListComp");
1724 return NULL;
1725 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001726 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001727 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001728 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001729 p->kind = ListComp_kind;
1730 p->v.ListComp.elt = elt;
1731 p->v.ListComp.generators = generators;
1732 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001733 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001734 return p;
1735}
1736
1737expr_ty
Nick Coghlan650f0d02007-04-15 12:05:43 +00001738SetComp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset, PyArena
1739 *arena)
1740{
1741 expr_ty p;
1742 if (!elt) {
1743 PyErr_SetString(PyExc_ValueError,
1744 "field elt is required for SetComp");
1745 return NULL;
1746 }
1747 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1748 if (!p)
1749 return NULL;
1750 p->kind = SetComp_kind;
1751 p->v.SetComp.elt = elt;
1752 p->v.SetComp.generators = generators;
1753 p->lineno = lineno;
1754 p->col_offset = col_offset;
1755 return p;
1756}
1757
1758expr_ty
Guido van Rossum992d4a32007-07-11 13:09:30 +00001759DictComp(expr_ty key, expr_ty value, asdl_seq * generators, int lineno, int
1760 col_offset, PyArena *arena)
1761{
1762 expr_ty p;
1763 if (!key) {
1764 PyErr_SetString(PyExc_ValueError,
1765 "field key is required for DictComp");
1766 return NULL;
1767 }
1768 if (!value) {
1769 PyErr_SetString(PyExc_ValueError,
1770 "field value is required for DictComp");
1771 return NULL;
1772 }
1773 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1774 if (!p)
1775 return NULL;
1776 p->kind = DictComp_kind;
1777 p->v.DictComp.key = key;
1778 p->v.DictComp.value = value;
1779 p->v.DictComp.generators = generators;
1780 p->lineno = lineno;
1781 p->col_offset = col_offset;
1782 return p;
1783}
1784
1785expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001786GeneratorExp(expr_ty elt, asdl_seq * generators, int lineno, int col_offset,
1787 PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001788{
1789 expr_ty p;
1790 if (!elt) {
1791 PyErr_SetString(PyExc_ValueError,
1792 "field elt is required for GeneratorExp");
1793 return NULL;
1794 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001795 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001796 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001797 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001798 p->kind = GeneratorExp_kind;
1799 p->v.GeneratorExp.elt = elt;
1800 p->v.GeneratorExp.generators = generators;
1801 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001802 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001803 return p;
1804}
1805
1806expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001807Yield(expr_ty value, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001808{
1809 expr_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001810 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001811 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001812 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001813 p->kind = Yield_kind;
1814 p->v.Yield.value = value;
1815 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001816 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001817 return p;
1818}
1819
1820expr_ty
Benjamin Peterson527c6222012-01-14 08:58:23 -05001821YieldFrom(expr_ty value, int lineno, int col_offset, PyArena *arena)
1822{
1823 expr_ty p;
Mark Dickinsonded35ae2012-11-25 14:36:26 +00001824 if (!value) {
1825 PyErr_SetString(PyExc_ValueError,
1826 "field value is required for YieldFrom");
1827 return NULL;
1828 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05001829 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1830 if (!p)
1831 return NULL;
1832 p->kind = YieldFrom_kind;
1833 p->v.YieldFrom.value = value;
1834 p->lineno = lineno;
1835 p->col_offset = col_offset;
1836 return p;
1837}
1838
1839expr_ty
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001840Compare(expr_ty left, asdl_int_seq * ops, asdl_seq * comparators, int lineno,
1841 int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001842{
1843 expr_ty p;
1844 if (!left) {
1845 PyErr_SetString(PyExc_ValueError,
1846 "field left is required for Compare");
1847 return NULL;
1848 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001849 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001850 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001851 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001852 p->kind = Compare_kind;
1853 p->v.Compare.left = left;
1854 p->v.Compare.ops = ops;
1855 p->v.Compare.comparators = comparators;
1856 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001857 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001858 return p;
1859}
1860
1861expr_ty
1862Call(expr_ty func, asdl_seq * args, asdl_seq * keywords, expr_ty starargs,
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001863 expr_ty kwargs, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001864{
1865 expr_ty p;
1866 if (!func) {
1867 PyErr_SetString(PyExc_ValueError,
1868 "field func is required for Call");
1869 return NULL;
1870 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001871 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001872 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001873 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001874 p->kind = Call_kind;
1875 p->v.Call.func = func;
1876 p->v.Call.args = args;
1877 p->v.Call.keywords = keywords;
1878 p->v.Call.starargs = starargs;
1879 p->v.Call.kwargs = kwargs;
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 +00001886Num(object n, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001887{
1888 expr_ty p;
1889 if (!n) {
1890 PyErr_SetString(PyExc_ValueError,
1891 "field n is required for Num");
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 = Num_kind;
1898 p->v.Num.n = n;
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
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001905Str(string s, int lineno, int col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001906{
1907 expr_ty p;
1908 if (!s) {
1909 PyErr_SetString(PyExc_ValueError,
1910 "field s is required for Str");
1911 return NULL;
1912 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001913 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001914 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001915 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001916 p->kind = Str_kind;
1917 p->v.Str.s = s;
1918 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001919 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001920 return p;
1921}
1922
1923expr_ty
Benjamin Petersone2498412011-08-09 16:08:39 -05001924Bytes(bytes s, int lineno, int col_offset, PyArena *arena)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001925{
1926 expr_ty p;
1927 if (!s) {
1928 PyErr_SetString(PyExc_ValueError,
1929 "field s is required for Bytes");
1930 return NULL;
1931 }
1932 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001933 if (!p)
Thomas Wouters00e41de2007-02-23 19:56:57 +00001934 return NULL;
Thomas Wouters00e41de2007-02-23 19:56:57 +00001935 p->kind = Bytes_kind;
1936 p->v.Bytes.s = s;
1937 p->lineno = lineno;
1938 p->col_offset = col_offset;
1939 return p;
1940}
1941
1942expr_ty
Benjamin Peterson442f2092012-12-06 17:41:04 -05001943NameConstant(singleton value, int lineno, int col_offset, PyArena *arena)
1944{
1945 expr_ty p;
1946 if (!value) {
1947 PyErr_SetString(PyExc_ValueError,
1948 "field value is required for NameConstant");
1949 return NULL;
1950 }
1951 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
1952 if (!p)
1953 return NULL;
1954 p->kind = NameConstant_kind;
1955 p->v.NameConstant.value = value;
1956 p->lineno = lineno;
1957 p->col_offset = col_offset;
1958 return p;
1959}
1960
1961expr_ty
Georg Brandl52318d62006-09-06 07:06:08 +00001962Ellipsis(int lineno, int col_offset, PyArena *arena)
1963{
1964 expr_ty p;
1965 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001966 if (!p)
Georg Brandl52318d62006-09-06 07:06:08 +00001967 return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00001968 p->kind = Ellipsis_kind;
1969 p->lineno = lineno;
1970 p->col_offset = col_offset;
1971 return p;
1972}
1973
1974expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00001975Attribute(expr_ty value, identifier attr, expr_context_ty ctx, int lineno, int
1976 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001977{
1978 expr_ty p;
1979 if (!value) {
1980 PyErr_SetString(PyExc_ValueError,
1981 "field value is required for Attribute");
1982 return NULL;
1983 }
1984 if (!attr) {
1985 PyErr_SetString(PyExc_ValueError,
1986 "field attr is required for Attribute");
1987 return NULL;
1988 }
1989 if (!ctx) {
1990 PyErr_SetString(PyExc_ValueError,
1991 "field ctx is required for Attribute");
1992 return NULL;
1993 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00001994 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00001995 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001996 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00001997 p->kind = Attribute_kind;
1998 p->v.Attribute.value = value;
1999 p->v.Attribute.attr = attr;
2000 p->v.Attribute.ctx = ctx;
2001 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002002 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002003 return p;
2004}
2005
2006expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002007Subscript(expr_ty value, slice_ty slice, expr_context_ty ctx, int lineno, int
2008 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002009{
2010 expr_ty p;
2011 if (!value) {
2012 PyErr_SetString(PyExc_ValueError,
2013 "field value is required for Subscript");
2014 return NULL;
2015 }
2016 if (!slice) {
2017 PyErr_SetString(PyExc_ValueError,
2018 "field slice is required for Subscript");
2019 return NULL;
2020 }
2021 if (!ctx) {
2022 PyErr_SetString(PyExc_ValueError,
2023 "field ctx is required for Subscript");
2024 return NULL;
2025 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002026 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002027 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002028 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002029 p->kind = Subscript_kind;
2030 p->v.Subscript.value = value;
2031 p->v.Subscript.slice = slice;
2032 p->v.Subscript.ctx = ctx;
2033 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002034 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002035 return p;
2036}
2037
2038expr_ty
Guido van Rossum0368b722007-05-11 16:50:42 +00002039Starred(expr_ty value, expr_context_ty ctx, int lineno, int col_offset, PyArena
2040 *arena)
2041{
2042 expr_ty p;
2043 if (!value) {
2044 PyErr_SetString(PyExc_ValueError,
2045 "field value is required for Starred");
2046 return NULL;
2047 }
2048 if (!ctx) {
2049 PyErr_SetString(PyExc_ValueError,
2050 "field ctx is required for Starred");
2051 return NULL;
2052 }
2053 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
2054 if (!p)
2055 return NULL;
2056 p->kind = Starred_kind;
2057 p->v.Starred.value = value;
2058 p->v.Starred.ctx = ctx;
2059 p->lineno = lineno;
2060 p->col_offset = col_offset;
2061 return p;
2062}
2063
2064expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002065Name(identifier id, expr_context_ty ctx, int lineno, int col_offset, PyArena
2066 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002067{
2068 expr_ty p;
2069 if (!id) {
2070 PyErr_SetString(PyExc_ValueError,
2071 "field id is required for Name");
2072 return NULL;
2073 }
2074 if (!ctx) {
2075 PyErr_SetString(PyExc_ValueError,
2076 "field ctx is required for Name");
2077 return NULL;
2078 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002079 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002080 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002081 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002082 p->kind = Name_kind;
2083 p->v.Name.id = id;
2084 p->v.Name.ctx = ctx;
2085 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002086 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002087 return p;
2088}
2089
2090expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002091List(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2092 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002093{
2094 expr_ty p;
2095 if (!ctx) {
2096 PyErr_SetString(PyExc_ValueError,
2097 "field ctx is required for List");
2098 return NULL;
2099 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002100 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002101 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002102 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002103 p->kind = List_kind;
2104 p->v.List.elts = elts;
2105 p->v.List.ctx = ctx;
2106 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002107 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002108 return p;
2109}
2110
2111expr_ty
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002112Tuple(asdl_seq * elts, expr_context_ty ctx, int lineno, int col_offset, PyArena
2113 *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002114{
2115 expr_ty p;
2116 if (!ctx) {
2117 PyErr_SetString(PyExc_ValueError,
2118 "field ctx is required for Tuple");
2119 return NULL;
2120 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002121 p = (expr_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002122 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002123 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002124 p->kind = Tuple_kind;
2125 p->v.Tuple.elts = elts;
2126 p->v.Tuple.ctx = ctx;
2127 p->lineno = lineno;
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002128 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002129 return p;
2130}
2131
2132slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002133Slice(expr_ty lower, expr_ty upper, expr_ty step, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002134{
2135 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002136 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002137 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002138 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002139 p->kind = Slice_kind;
2140 p->v.Slice.lower = lower;
2141 p->v.Slice.upper = upper;
2142 p->v.Slice.step = step;
2143 return p;
2144}
2145
2146slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002147ExtSlice(asdl_seq * dims, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002148{
2149 slice_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002150 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002151 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002152 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002153 p->kind = ExtSlice_kind;
2154 p->v.ExtSlice.dims = dims;
2155 return p;
2156}
2157
2158slice_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002159Index(expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002160{
2161 slice_ty p;
2162 if (!value) {
2163 PyErr_SetString(PyExc_ValueError,
2164 "field value is required for Index");
2165 return NULL;
2166 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002167 p = (slice_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002168 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002169 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002170 p->kind = Index_kind;
2171 p->v.Index.value = value;
2172 return p;
2173}
2174
2175comprehension_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002176comprehension(expr_ty target, expr_ty iter, asdl_seq * ifs, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002177{
2178 comprehension_ty p;
2179 if (!target) {
2180 PyErr_SetString(PyExc_ValueError,
2181 "field target is required for comprehension");
2182 return NULL;
2183 }
2184 if (!iter) {
2185 PyErr_SetString(PyExc_ValueError,
2186 "field iter is required for comprehension");
2187 return NULL;
2188 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002189 p = (comprehension_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002190 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002191 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002192 p->target = target;
2193 p->iter = iter;
2194 p->ifs = ifs;
2195 return p;
2196}
2197
2198excepthandler_ty
Neal Norwitzad74aa82008-03-31 05:14:30 +00002199ExceptHandler(expr_ty type, identifier name, asdl_seq * body, int lineno, int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002200 col_offset, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002201{
2202 excepthandler_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002203 p = (excepthandler_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002204 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002205 return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00002206 p->kind = ExceptHandler_kind;
2207 p->v.ExceptHandler.type = type;
2208 p->v.ExceptHandler.name = name;
2209 p->v.ExceptHandler.body = body;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002210 p->lineno = lineno;
2211 p->col_offset = col_offset;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002212 return p;
2213}
2214
2215arguments_ty
Neal Norwitzc1505362006-12-28 06:47:50 +00002216arguments(asdl_seq * args, identifier vararg, expr_ty varargannotation,
2217 asdl_seq * kwonlyargs, identifier kwarg, expr_ty kwargannotation,
2218 asdl_seq * defaults, asdl_seq * kw_defaults, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002219{
2220 arguments_ty p;
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002221 p = (arguments_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002222 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002223 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002224 p->args = args;
2225 p->vararg = vararg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002226 p->varargannotation = varargannotation;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002227 p->kwonlyargs = kwonlyargs;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002228 p->kwarg = kwarg;
Neal Norwitzc1505362006-12-28 06:47:50 +00002229 p->kwargannotation = kwargannotation;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002230 p->defaults = defaults;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002231 p->kw_defaults = kw_defaults;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002232 return p;
2233}
2234
Neal Norwitzc1505362006-12-28 06:47:50 +00002235arg_ty
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002236arg(identifier arg, expr_ty annotation, PyArena *arena)
Neal Norwitzc1505362006-12-28 06:47:50 +00002237{
2238 arg_ty p;
2239 if (!arg) {
2240 PyErr_SetString(PyExc_ValueError,
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002241 "field arg is required for arg");
Neal Norwitzc1505362006-12-28 06:47:50 +00002242 return NULL;
2243 }
2244 p = (arg_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002245 if (!p)
Neal Norwitzc1505362006-12-28 06:47:50 +00002246 return NULL;
Guido van Rossum1bc535d2007-05-15 18:46:22 +00002247 p->arg = arg;
2248 p->annotation = annotation;
Neal Norwitzc1505362006-12-28 06:47:50 +00002249 return p;
2250}
2251
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002252keyword_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002253keyword(identifier arg, expr_ty value, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002254{
2255 keyword_ty p;
2256 if (!arg) {
2257 PyErr_SetString(PyExc_ValueError,
2258 "field arg is required for keyword");
2259 return NULL;
2260 }
2261 if (!value) {
2262 PyErr_SetString(PyExc_ValueError,
2263 "field value is required for keyword");
2264 return NULL;
2265 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002266 p = (keyword_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002267 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002268 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002269 p->arg = arg;
2270 p->value = value;
2271 return p;
2272}
2273
2274alias_ty
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002275alias(identifier name, identifier asname, PyArena *arena)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002276{
2277 alias_ty p;
2278 if (!name) {
2279 PyErr_SetString(PyExc_ValueError,
2280 "field name is required for alias");
2281 return NULL;
2282 }
Neal Norwitzadb69fc2005-12-17 20:54:49 +00002283 p = (alias_ty)PyArena_Malloc(arena, sizeof(*p));
Thomas Woutersa44f3a32007-02-26 18:20:15 +00002284 if (!p)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002285 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002286 p->name = name;
2287 p->asname = asname;
2288 return p;
2289}
2290
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002291withitem_ty
2292withitem(expr_ty context_expr, expr_ty optional_vars, PyArena *arena)
2293{
2294 withitem_ty p;
2295 if (!context_expr) {
2296 PyErr_SetString(PyExc_ValueError,
2297 "field context_expr is required for withitem");
2298 return NULL;
2299 }
2300 p = (withitem_ty)PyArena_Malloc(arena, sizeof(*p));
2301 if (!p)
2302 return NULL;
2303 p->context_expr = context_expr;
2304 p->optional_vars = optional_vars;
2305 return p;
2306}
2307
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002308
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002309PyObject*
2310ast2obj_mod(void* _o)
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002311{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002312 mod_ty o = (mod_ty)_o;
2313 PyObject *result = NULL, *value = NULL;
2314 if (!o) {
2315 Py_INCREF(Py_None);
2316 return Py_None;
2317 }
Neal Norwitz7b5a6042005-11-13 19:14:20 +00002318
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002319 switch (o->kind) {
2320 case Module_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002321 result = PyType_GenericNew(Module_type, NULL, NULL);
2322 if (!result) goto failed;
2323 value = ast2obj_list(o->v.Module.body, ast2obj_stmt);
2324 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002325 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002326 goto failed;
2327 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002328 break;
2329 case Interactive_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002330 result = PyType_GenericNew(Interactive_type, NULL, NULL);
2331 if (!result) goto failed;
2332 value = ast2obj_list(o->v.Interactive.body, ast2obj_stmt);
2333 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002334 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002335 goto failed;
2336 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002337 break;
2338 case Expression_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002339 result = PyType_GenericNew(Expression_type, NULL, NULL);
2340 if (!result) goto failed;
2341 value = ast2obj_expr(o->v.Expression.body);
2342 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002343 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002344 goto failed;
2345 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002346 break;
2347 case Suite_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002348 result = PyType_GenericNew(Suite_type, NULL, NULL);
2349 if (!result) goto failed;
2350 value = ast2obj_list(o->v.Suite.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);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002355 break;
2356 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002357 return result;
2358failed:
2359 Py_XDECREF(value);
2360 Py_XDECREF(result);
2361 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002362}
2363
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002364PyObject*
2365ast2obj_stmt(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002366{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002367 stmt_ty o = (stmt_ty)_o;
2368 PyObject *result = NULL, *value = NULL;
2369 if (!o) {
2370 Py_INCREF(Py_None);
2371 return Py_None;
2372 }
2373
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002374 switch (o->kind) {
2375 case FunctionDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002376 result = PyType_GenericNew(FunctionDef_type, NULL, NULL);
2377 if (!result) goto failed;
2378 value = ast2obj_identifier(o->v.FunctionDef.name);
2379 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002380 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002381 goto failed;
2382 Py_DECREF(value);
2383 value = ast2obj_arguments(o->v.FunctionDef.args);
2384 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002385 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002386 goto failed;
2387 Py_DECREF(value);
2388 value = ast2obj_list(o->v.FunctionDef.body, ast2obj_stmt);
2389 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002390 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002391 goto failed;
2392 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002393 value = ast2obj_list(o->v.FunctionDef.decorator_list,
2394 ast2obj_expr);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002395 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002396 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002397 -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002398 goto failed;
2399 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00002400 value = ast2obj_expr(o->v.FunctionDef.returns);
2401 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002402 if (_PyObject_SetAttrId(result, &PyId_returns, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00002403 goto failed;
2404 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002405 break;
2406 case ClassDef_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002407 result = PyType_GenericNew(ClassDef_type, NULL, NULL);
2408 if (!result) goto failed;
2409 value = ast2obj_identifier(o->v.ClassDef.name);
2410 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002411 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002412 goto failed;
2413 Py_DECREF(value);
2414 value = ast2obj_list(o->v.ClassDef.bases, ast2obj_expr);
2415 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002416 if (_PyObject_SetAttrId(result, &PyId_bases, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002417 goto failed;
2418 Py_DECREF(value);
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002419 value = ast2obj_list(o->v.ClassDef.keywords, ast2obj_keyword);
2420 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002421 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002422 goto failed;
2423 Py_DECREF(value);
2424 value = ast2obj_expr(o->v.ClassDef.starargs);
2425 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002426 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002427 goto failed;
2428 Py_DECREF(value);
2429 value = ast2obj_expr(o->v.ClassDef.kwargs);
2430 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002431 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Guido van Rossum52cc1d82007-03-18 15:41:51 +00002432 goto failed;
2433 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002434 value = ast2obj_list(o->v.ClassDef.body, ast2obj_stmt);
2435 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002436 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002437 goto failed;
2438 Py_DECREF(value);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002439 value = ast2obj_list(o->v.ClassDef.decorator_list,
2440 ast2obj_expr);
2441 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002442 if (_PyObject_SetAttrId(result, &PyId_decorator_list, value) ==
Guido van Rossumd59da4b2007-05-22 18:11:13 +00002443 -1)
2444 goto failed;
2445 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002446 break;
2447 case Return_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002448 result = PyType_GenericNew(Return_type, NULL, NULL);
2449 if (!result) goto failed;
2450 value = ast2obj_expr(o->v.Return.value);
2451 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002452 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002453 goto failed;
2454 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002455 break;
2456 case Delete_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002457 result = PyType_GenericNew(Delete_type, NULL, NULL);
2458 if (!result) goto failed;
2459 value = ast2obj_list(o->v.Delete.targets, ast2obj_expr);
2460 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002461 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002462 goto failed;
2463 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002464 break;
2465 case Assign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002466 result = PyType_GenericNew(Assign_type, NULL, NULL);
2467 if (!result) goto failed;
2468 value = ast2obj_list(o->v.Assign.targets, ast2obj_expr);
2469 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002470 if (_PyObject_SetAttrId(result, &PyId_targets, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002471 goto failed;
2472 Py_DECREF(value);
2473 value = ast2obj_expr(o->v.Assign.value);
2474 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002475 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002476 goto failed;
2477 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002478 break;
2479 case AugAssign_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002480 result = PyType_GenericNew(AugAssign_type, NULL, NULL);
2481 if (!result) goto failed;
2482 value = ast2obj_expr(o->v.AugAssign.target);
2483 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002484 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002485 goto failed;
2486 Py_DECREF(value);
2487 value = ast2obj_operator(o->v.AugAssign.op);
2488 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002489 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002490 goto failed;
2491 Py_DECREF(value);
2492 value = ast2obj_expr(o->v.AugAssign.value);
2493 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002494 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002495 goto failed;
2496 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002497 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002498 case For_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002499 result = PyType_GenericNew(For_type, NULL, NULL);
2500 if (!result) goto failed;
2501 value = ast2obj_expr(o->v.For.target);
2502 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002503 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002504 goto failed;
2505 Py_DECREF(value);
2506 value = ast2obj_expr(o->v.For.iter);
2507 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002508 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002509 goto failed;
2510 Py_DECREF(value);
2511 value = ast2obj_list(o->v.For.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.For.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;
2522 case While_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002523 result = PyType_GenericNew(While_type, NULL, NULL);
2524 if (!result) goto failed;
2525 value = ast2obj_expr(o->v.While.test);
2526 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002527 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002528 goto failed;
2529 Py_DECREF(value);
2530 value = ast2obj_list(o->v.While.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)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002533 goto failed;
2534 Py_DECREF(value);
2535 value = ast2obj_list(o->v.While.orelse, ast2obj_stmt);
2536 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002537 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002538 goto failed;
2539 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002540 break;
2541 case If_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002542 result = PyType_GenericNew(If_type, NULL, NULL);
2543 if (!result) goto failed;
2544 value = ast2obj_expr(o->v.If.test);
2545 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002546 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002547 goto failed;
2548 Py_DECREF(value);
2549 value = ast2obj_list(o->v.If.body, ast2obj_stmt);
2550 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002551 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002552 goto failed;
2553 Py_DECREF(value);
2554 value = ast2obj_list(o->v.If.orelse, ast2obj_stmt);
2555 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002556 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002557 goto failed;
2558 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002559 break;
Guido van Rossumc2e20742006-02-27 22:32:47 +00002560 case With_kind:
2561 result = PyType_GenericNew(With_type, NULL, NULL);
2562 if (!result) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05002563 value = ast2obj_list(o->v.With.items, ast2obj_withitem);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002564 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002565 if (_PyObject_SetAttrId(result, &PyId_items, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002566 goto failed;
2567 Py_DECREF(value);
2568 value = ast2obj_list(o->v.With.body, ast2obj_stmt);
2569 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002570 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002571 goto failed;
2572 Py_DECREF(value);
2573 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002574 case Raise_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002575 result = PyType_GenericNew(Raise_type, NULL, NULL);
2576 if (!result) goto failed;
Collin Winter828f04a2007-08-31 00:04:24 +00002577 value = ast2obj_expr(o->v.Raise.exc);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002578 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002579 if (_PyObject_SetAttrId(result, &PyId_exc, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002580 goto failed;
2581 Py_DECREF(value);
Collin Winter828f04a2007-08-31 00:04:24 +00002582 value = ast2obj_expr(o->v.Raise.cause);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002583 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002584 if (_PyObject_SetAttrId(result, &PyId_cause, 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;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002588 case Try_kind:
2589 result = PyType_GenericNew(Try_type, NULL, NULL);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002590 if (!result) goto failed;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002591 value = ast2obj_list(o->v.Try.body, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002592 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002593 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002594 goto failed;
2595 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002596 value = ast2obj_list(o->v.Try.handlers, ast2obj_excepthandler);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002597 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002598 if (_PyObject_SetAttrId(result, &PyId_handlers, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002599 goto failed;
2600 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002601 value = ast2obj_list(o->v.Try.orelse, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002602 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002603 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002604 goto failed;
2605 Py_DECREF(value);
Benjamin Peterson43af12b2011-05-29 11:43:10 -05002606 value = ast2obj_list(o->v.Try.finalbody, ast2obj_stmt);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002607 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002608 if (_PyObject_SetAttrId(result, &PyId_finalbody, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002609 goto failed;
2610 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002611 break;
2612 case Assert_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002613 result = PyType_GenericNew(Assert_type, NULL, NULL);
2614 if (!result) goto failed;
2615 value = ast2obj_expr(o->v.Assert.test);
2616 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002617 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002618 goto failed;
2619 Py_DECREF(value);
2620 value = ast2obj_expr(o->v.Assert.msg);
2621 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002622 if (_PyObject_SetAttrId(result, &PyId_msg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002623 goto failed;
2624 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002625 break;
2626 case Import_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002627 result = PyType_GenericNew(Import_type, NULL, NULL);
2628 if (!result) goto failed;
2629 value = ast2obj_list(o->v.Import.names, ast2obj_alias);
2630 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002631 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002632 goto failed;
2633 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002634 break;
2635 case ImportFrom_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002636 result = PyType_GenericNew(ImportFrom_type, NULL, NULL);
2637 if (!result) goto failed;
2638 value = ast2obj_identifier(o->v.ImportFrom.module);
2639 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002640 if (_PyObject_SetAttrId(result, &PyId_module, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002641 goto failed;
2642 Py_DECREF(value);
2643 value = ast2obj_list(o->v.ImportFrom.names, ast2obj_alias);
2644 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002645 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002646 goto failed;
2647 Py_DECREF(value);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002648 value = ast2obj_int(o->v.ImportFrom.level);
2649 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002650 if (_PyObject_SetAttrId(result, &PyId_level, value) == -1)
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002651 goto failed;
2652 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002653 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002654 case Global_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002655 result = PyType_GenericNew(Global_type, NULL, NULL);
2656 if (!result) goto failed;
2657 value = ast2obj_list(o->v.Global.names, ast2obj_identifier);
2658 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002659 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002660 goto failed;
2661 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002662 break;
Jeremy Hylton81e95022007-02-27 06:50:52 +00002663 case Nonlocal_kind:
2664 result = PyType_GenericNew(Nonlocal_type, NULL, NULL);
2665 if (!result) goto failed;
2666 value = ast2obj_list(o->v.Nonlocal.names, ast2obj_identifier);
2667 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002668 if (_PyObject_SetAttrId(result, &PyId_names, value) == -1)
Jeremy Hylton81e95022007-02-27 06:50:52 +00002669 goto failed;
2670 Py_DECREF(value);
2671 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002672 case Expr_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002673 result = PyType_GenericNew(Expr_type, NULL, NULL);
2674 if (!result) goto failed;
2675 value = ast2obj_expr(o->v.Expr.value);
2676 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002677 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002678 goto failed;
2679 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002680 break;
2681 case Pass_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002682 result = PyType_GenericNew(Pass_type, NULL, NULL);
2683 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002684 break;
2685 case Break_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002686 result = PyType_GenericNew(Break_type, NULL, NULL);
2687 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002688 break;
2689 case Continue_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002690 result = PyType_GenericNew(Continue_type, NULL, NULL);
2691 if (!result) goto failed;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002692 break;
2693 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00002694 value = ast2obj_int(o->lineno);
2695 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002696 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002697 goto failed;
2698 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00002699 value = ast2obj_int(o->col_offset);
2700 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002701 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00002702 goto failed;
2703 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002704 return result;
2705failed:
2706 Py_XDECREF(value);
2707 Py_XDECREF(result);
2708 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002709}
2710
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002711PyObject*
2712ast2obj_expr(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002713{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002714 expr_ty o = (expr_ty)_o;
2715 PyObject *result = NULL, *value = NULL;
2716 if (!o) {
2717 Py_INCREF(Py_None);
2718 return Py_None;
2719 }
2720
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002721 switch (o->kind) {
2722 case BoolOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002723 result = PyType_GenericNew(BoolOp_type, NULL, NULL);
2724 if (!result) goto failed;
2725 value = ast2obj_boolop(o->v.BoolOp.op);
2726 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002727 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002728 goto failed;
2729 Py_DECREF(value);
2730 value = ast2obj_list(o->v.BoolOp.values, ast2obj_expr);
2731 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002732 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002733 goto failed;
2734 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002735 break;
2736 case BinOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002737 result = PyType_GenericNew(BinOp_type, NULL, NULL);
2738 if (!result) goto failed;
2739 value = ast2obj_expr(o->v.BinOp.left);
2740 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002741 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002742 goto failed;
2743 Py_DECREF(value);
2744 value = ast2obj_operator(o->v.BinOp.op);
2745 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002746 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002747 goto failed;
2748 Py_DECREF(value);
2749 value = ast2obj_expr(o->v.BinOp.right);
2750 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002751 if (_PyObject_SetAttrId(result, &PyId_right, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002752 goto failed;
2753 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002754 break;
2755 case UnaryOp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002756 result = PyType_GenericNew(UnaryOp_type, NULL, NULL);
2757 if (!result) goto failed;
2758 value = ast2obj_unaryop(o->v.UnaryOp.op);
2759 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002760 if (_PyObject_SetAttrId(result, &PyId_op, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002761 goto failed;
2762 Py_DECREF(value);
2763 value = ast2obj_expr(o->v.UnaryOp.operand);
2764 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002765 if (_PyObject_SetAttrId(result, &PyId_operand, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002766 goto failed;
2767 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002768 break;
2769 case Lambda_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002770 result = PyType_GenericNew(Lambda_type, NULL, NULL);
2771 if (!result) goto failed;
2772 value = ast2obj_arguments(o->v.Lambda.args);
2773 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002774 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002775 goto failed;
2776 Py_DECREF(value);
2777 value = ast2obj_expr(o->v.Lambda.body);
2778 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002779 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002780 goto failed;
2781 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002782 break;
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002783 case IfExp_kind:
2784 result = PyType_GenericNew(IfExp_type, NULL, NULL);
2785 if (!result) goto failed;
2786 value = ast2obj_expr(o->v.IfExp.test);
2787 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002788 if (_PyObject_SetAttrId(result, &PyId_test, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002789 goto failed;
2790 Py_DECREF(value);
2791 value = ast2obj_expr(o->v.IfExp.body);
2792 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002793 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002794 goto failed;
2795 Py_DECREF(value);
2796 value = ast2obj_expr(o->v.IfExp.orelse);
2797 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002798 if (_PyObject_SetAttrId(result, &PyId_orelse, value) == -1)
Thomas Woutersdca3b9c2006-02-27 00:24:13 +00002799 goto failed;
2800 Py_DECREF(value);
2801 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002802 case Dict_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002803 result = PyType_GenericNew(Dict_type, NULL, NULL);
2804 if (!result) goto failed;
2805 value = ast2obj_list(o->v.Dict.keys, ast2obj_expr);
2806 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002807 if (_PyObject_SetAttrId(result, &PyId_keys, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002808 goto failed;
2809 Py_DECREF(value);
2810 value = ast2obj_list(o->v.Dict.values, ast2obj_expr);
2811 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002812 if (_PyObject_SetAttrId(result, &PyId_values, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002813 goto failed;
2814 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002815 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002816 case Set_kind:
2817 result = PyType_GenericNew(Set_type, NULL, NULL);
2818 if (!result) goto failed;
2819 value = ast2obj_list(o->v.Set.elts, ast2obj_expr);
2820 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002821 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Guido van Rossum86e58e22006-08-28 15:27:34 +00002822 goto failed;
2823 Py_DECREF(value);
2824 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002825 case ListComp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002826 result = PyType_GenericNew(ListComp_type, NULL, NULL);
2827 if (!result) goto failed;
2828 value = ast2obj_expr(o->v.ListComp.elt);
2829 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002830 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002831 goto failed;
2832 Py_DECREF(value);
2833 value = ast2obj_list(o->v.ListComp.generators,
2834 ast2obj_comprehension);
2835 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002836 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002837 goto failed;
2838 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002839 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00002840 case SetComp_kind:
2841 result = PyType_GenericNew(SetComp_type, NULL, NULL);
2842 if (!result) goto failed;
2843 value = ast2obj_expr(o->v.SetComp.elt);
2844 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002845 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002846 goto failed;
2847 Py_DECREF(value);
2848 value = ast2obj_list(o->v.SetComp.generators,
2849 ast2obj_comprehension);
2850 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002851 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Nick Coghlan650f0d02007-04-15 12:05:43 +00002852 goto failed;
2853 Py_DECREF(value);
2854 break;
Guido van Rossum992d4a32007-07-11 13:09:30 +00002855 case DictComp_kind:
2856 result = PyType_GenericNew(DictComp_type, NULL, NULL);
2857 if (!result) goto failed;
2858 value = ast2obj_expr(o->v.DictComp.key);
2859 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002860 if (_PyObject_SetAttrId(result, &PyId_key, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002861 goto failed;
2862 Py_DECREF(value);
2863 value = ast2obj_expr(o->v.DictComp.value);
2864 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002865 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002866 goto failed;
2867 Py_DECREF(value);
2868 value = ast2obj_list(o->v.DictComp.generators,
2869 ast2obj_comprehension);
2870 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002871 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Guido van Rossum992d4a32007-07-11 13:09:30 +00002872 goto failed;
2873 Py_DECREF(value);
2874 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002875 case GeneratorExp_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002876 result = PyType_GenericNew(GeneratorExp_type, NULL, NULL);
2877 if (!result) goto failed;
2878 value = ast2obj_expr(o->v.GeneratorExp.elt);
2879 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002880 if (_PyObject_SetAttrId(result, &PyId_elt, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002881 goto failed;
2882 Py_DECREF(value);
2883 value = ast2obj_list(o->v.GeneratorExp.generators,
2884 ast2obj_comprehension);
2885 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002886 if (_PyObject_SetAttrId(result, &PyId_generators, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002887 goto failed;
2888 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002889 break;
2890 case Yield_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002891 result = PyType_GenericNew(Yield_type, NULL, NULL);
2892 if (!result) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002893 value = ast2obj_expr(o->v.Yield.value);
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002894 if (!value) goto failed;
Benjamin Peterson527c6222012-01-14 08:58:23 -05002895 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Nick Coghlan1f7ce622012-01-13 21:43:40 +10002896 goto failed;
2897 Py_DECREF(value);
Benjamin Peterson527c6222012-01-14 08:58:23 -05002898 break;
2899 case YieldFrom_kind:
2900 result = PyType_GenericNew(YieldFrom_type, NULL, NULL);
2901 if (!result) goto failed;
2902 value = ast2obj_expr(o->v.YieldFrom.value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002903 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002904 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002905 goto failed;
2906 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002907 break;
2908 case Compare_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002909 result = PyType_GenericNew(Compare_type, NULL, NULL);
2910 if (!result) goto failed;
2911 value = ast2obj_expr(o->v.Compare.left);
2912 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002913 if (_PyObject_SetAttrId(result, &PyId_left, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002914 goto failed;
2915 Py_DECREF(value);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002916 {
Benjamin Peterson77fa9372012-05-15 10:10:27 -07002917 Py_ssize_t i, n = asdl_seq_LEN(o->v.Compare.ops);
Martin v. Löwisce1d5d22006-02-26 20:51:25 +00002918 value = PyList_New(n);
2919 if (!value) goto failed;
2920 for(i = 0; i < n; i++)
2921 PyList_SET_ITEM(value, i, ast2obj_cmpop((cmpop_ty)asdl_seq_GET(o->v.Compare.ops, i)));
2922 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002923 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002924 if (_PyObject_SetAttrId(result, &PyId_ops, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002925 goto failed;
2926 Py_DECREF(value);
2927 value = ast2obj_list(o->v.Compare.comparators, ast2obj_expr);
2928 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002929 if (_PyObject_SetAttrId(result, &PyId_comparators, 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 Call_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002934 result = PyType_GenericNew(Call_type, NULL, NULL);
2935 if (!result) goto failed;
2936 value = ast2obj_expr(o->v.Call.func);
2937 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002938 if (_PyObject_SetAttrId(result, &PyId_func, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002939 goto failed;
2940 Py_DECREF(value);
2941 value = ast2obj_list(o->v.Call.args, ast2obj_expr);
2942 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002943 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002944 goto failed;
2945 Py_DECREF(value);
2946 value = ast2obj_list(o->v.Call.keywords, ast2obj_keyword);
2947 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002948 if (_PyObject_SetAttrId(result, &PyId_keywords, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002949 goto failed;
2950 Py_DECREF(value);
2951 value = ast2obj_expr(o->v.Call.starargs);
2952 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002953 if (_PyObject_SetAttrId(result, &PyId_starargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002954 goto failed;
2955 Py_DECREF(value);
2956 value = ast2obj_expr(o->v.Call.kwargs);
2957 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002958 if (_PyObject_SetAttrId(result, &PyId_kwargs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002959 goto failed;
2960 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002961 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002962 case Num_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002963 result = PyType_GenericNew(Num_type, NULL, NULL);
2964 if (!result) goto failed;
2965 value = ast2obj_object(o->v.Num.n);
2966 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002967 if (_PyObject_SetAttrId(result, &PyId_n, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002968 goto failed;
2969 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002970 break;
2971 case Str_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002972 result = PyType_GenericNew(Str_type, NULL, NULL);
2973 if (!result) goto failed;
2974 value = ast2obj_string(o->v.Str.s);
2975 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002976 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00002977 goto failed;
2978 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002979 break;
Thomas Wouters00e41de2007-02-23 19:56:57 +00002980 case Bytes_kind:
2981 result = PyType_GenericNew(Bytes_type, NULL, NULL);
2982 if (!result) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05002983 value = ast2obj_bytes(o->v.Bytes.s);
Thomas Wouters00e41de2007-02-23 19:56:57 +00002984 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02002985 if (_PyObject_SetAttrId(result, &PyId_s, value) == -1)
Thomas Wouters00e41de2007-02-23 19:56:57 +00002986 goto failed;
2987 Py_DECREF(value);
2988 break;
Benjamin Peterson442f2092012-12-06 17:41:04 -05002989 case NameConstant_kind:
2990 result = PyType_GenericNew(NameConstant_type, NULL, NULL);
2991 if (!result) goto failed;
2992 value = ast2obj_singleton(o->v.NameConstant.value);
2993 if (!value) goto failed;
2994 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
2995 goto failed;
2996 Py_DECREF(value);
2997 break;
Georg Brandl52318d62006-09-06 07:06:08 +00002998 case Ellipsis_kind:
2999 result = PyType_GenericNew(Ellipsis_type, NULL, NULL);
3000 if (!result) goto failed;
3001 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003002 case Attribute_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003003 result = PyType_GenericNew(Attribute_type, NULL, NULL);
3004 if (!result) goto failed;
3005 value = ast2obj_expr(o->v.Attribute.value);
3006 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003007 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003008 goto failed;
3009 Py_DECREF(value);
3010 value = ast2obj_identifier(o->v.Attribute.attr);
3011 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003012 if (_PyObject_SetAttrId(result, &PyId_attr, 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.Attribute.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 Subscript_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003022 result = PyType_GenericNew(Subscript_type, NULL, NULL);
3023 if (!result) goto failed;
3024 value = ast2obj_expr(o->v.Subscript.value);
3025 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003026 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003027 goto failed;
3028 Py_DECREF(value);
3029 value = ast2obj_slice(o->v.Subscript.slice);
3030 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003031 if (_PyObject_SetAttrId(result, &PyId_slice, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003032 goto failed;
3033 Py_DECREF(value);
3034 value = ast2obj_expr_context(o->v.Subscript.ctx);
3035 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003036 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003037 goto failed;
3038 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003039 break;
Guido van Rossum0368b722007-05-11 16:50:42 +00003040 case Starred_kind:
3041 result = PyType_GenericNew(Starred_type, NULL, NULL);
3042 if (!result) goto failed;
3043 value = ast2obj_expr(o->v.Starred.value);
3044 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003045 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00003046 goto failed;
3047 Py_DECREF(value);
3048 value = ast2obj_expr_context(o->v.Starred.ctx);
3049 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003050 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Guido van Rossum0368b722007-05-11 16:50:42 +00003051 goto failed;
3052 Py_DECREF(value);
3053 break;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003054 case Name_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003055 result = PyType_GenericNew(Name_type, NULL, NULL);
3056 if (!result) goto failed;
3057 value = ast2obj_identifier(o->v.Name.id);
3058 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003059 if (_PyObject_SetAttrId(result, &PyId_id, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003060 goto failed;
3061 Py_DECREF(value);
3062 value = ast2obj_expr_context(o->v.Name.ctx);
3063 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003064 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003065 goto failed;
3066 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003067 break;
3068 case List_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003069 result = PyType_GenericNew(List_type, NULL, NULL);
3070 if (!result) goto failed;
3071 value = ast2obj_list(o->v.List.elts, ast2obj_expr);
3072 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003073 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003074 goto failed;
3075 Py_DECREF(value);
3076 value = ast2obj_expr_context(o->v.List.ctx);
3077 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003078 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003079 goto failed;
3080 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003081 break;
3082 case Tuple_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003083 result = PyType_GenericNew(Tuple_type, NULL, NULL);
3084 if (!result) goto failed;
3085 value = ast2obj_list(o->v.Tuple.elts, ast2obj_expr);
3086 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003087 if (_PyObject_SetAttrId(result, &PyId_elts, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003088 goto failed;
3089 Py_DECREF(value);
3090 value = ast2obj_expr_context(o->v.Tuple.ctx);
3091 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003092 if (_PyObject_SetAttrId(result, &PyId_ctx, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003093 goto failed;
3094 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003095 break;
3096 }
Martin v. Löwis577b5b92006-02-27 15:23:19 +00003097 value = ast2obj_int(o->lineno);
3098 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003099 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003100 goto failed;
3101 Py_DECREF(value);
Martin v. Löwis49c5da12006-03-01 22:49:05 +00003102 value = ast2obj_int(o->col_offset);
3103 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003104 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Martin v. Löwis03e5bc02006-03-02 00:31:27 +00003105 goto failed;
3106 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003107 return result;
3108failed:
3109 Py_XDECREF(value);
3110 Py_XDECREF(result);
3111 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003112}
3113
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003114PyObject* ast2obj_expr_context(expr_context_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003115{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003116 switch(o) {
3117 case Load:
3118 Py_INCREF(Load_singleton);
3119 return Load_singleton;
3120 case Store:
3121 Py_INCREF(Store_singleton);
3122 return Store_singleton;
3123 case Del:
3124 Py_INCREF(Del_singleton);
3125 return Del_singleton;
3126 case AugLoad:
3127 Py_INCREF(AugLoad_singleton);
3128 return AugLoad_singleton;
3129 case AugStore:
3130 Py_INCREF(AugStore_singleton);
3131 return AugStore_singleton;
3132 case Param:
3133 Py_INCREF(Param_singleton);
3134 return Param_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003135 default:
3136 /* should never happen, but just in case ... */
3137 PyErr_Format(PyExc_SystemError, "unknown expr_context found");
3138 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003139 }
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003140}
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003141PyObject*
3142ast2obj_slice(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003143{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003144 slice_ty o = (slice_ty)_o;
3145 PyObject *result = NULL, *value = NULL;
3146 if (!o) {
3147 Py_INCREF(Py_None);
3148 return Py_None;
3149 }
3150
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003151 switch (o->kind) {
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003152 case Slice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003153 result = PyType_GenericNew(Slice_type, NULL, NULL);
3154 if (!result) goto failed;
3155 value = ast2obj_expr(o->v.Slice.lower);
3156 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003157 if (_PyObject_SetAttrId(result, &PyId_lower, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003158 goto failed;
3159 Py_DECREF(value);
3160 value = ast2obj_expr(o->v.Slice.upper);
3161 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003162 if (_PyObject_SetAttrId(result, &PyId_upper, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003163 goto failed;
3164 Py_DECREF(value);
3165 value = ast2obj_expr(o->v.Slice.step);
3166 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003167 if (_PyObject_SetAttrId(result, &PyId_step, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003168 goto failed;
3169 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003170 break;
3171 case ExtSlice_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003172 result = PyType_GenericNew(ExtSlice_type, NULL, NULL);
3173 if (!result) goto failed;
3174 value = ast2obj_list(o->v.ExtSlice.dims, ast2obj_slice);
3175 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003176 if (_PyObject_SetAttrId(result, &PyId_dims, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003177 goto failed;
3178 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003179 break;
3180 case Index_kind:
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003181 result = PyType_GenericNew(Index_type, NULL, NULL);
3182 if (!result) goto failed;
3183 value = ast2obj_expr(o->v.Index.value);
3184 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003185 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003186 goto failed;
3187 Py_DECREF(value);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003188 break;
3189 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003190 return result;
3191failed:
3192 Py_XDECREF(value);
3193 Py_XDECREF(result);
3194 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003195}
3196
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003197PyObject* ast2obj_boolop(boolop_ty o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003198{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003199 switch(o) {
3200 case And:
3201 Py_INCREF(And_singleton);
3202 return And_singleton;
3203 case Or:
3204 Py_INCREF(Or_singleton);
3205 return Or_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003206 default:
3207 /* should never happen, but just in case ... */
3208 PyErr_Format(PyExc_SystemError, "unknown boolop found");
3209 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003210 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003211}
3212PyObject* ast2obj_operator(operator_ty o)
3213{
3214 switch(o) {
3215 case Add:
3216 Py_INCREF(Add_singleton);
3217 return Add_singleton;
3218 case Sub:
3219 Py_INCREF(Sub_singleton);
3220 return Sub_singleton;
3221 case Mult:
3222 Py_INCREF(Mult_singleton);
3223 return Mult_singleton;
3224 case Div:
3225 Py_INCREF(Div_singleton);
3226 return Div_singleton;
3227 case Mod:
3228 Py_INCREF(Mod_singleton);
3229 return Mod_singleton;
3230 case Pow:
3231 Py_INCREF(Pow_singleton);
3232 return Pow_singleton;
3233 case LShift:
3234 Py_INCREF(LShift_singleton);
3235 return LShift_singleton;
3236 case RShift:
3237 Py_INCREF(RShift_singleton);
3238 return RShift_singleton;
3239 case BitOr:
3240 Py_INCREF(BitOr_singleton);
3241 return BitOr_singleton;
3242 case BitXor:
3243 Py_INCREF(BitXor_singleton);
3244 return BitXor_singleton;
3245 case BitAnd:
3246 Py_INCREF(BitAnd_singleton);
3247 return BitAnd_singleton;
3248 case FloorDiv:
3249 Py_INCREF(FloorDiv_singleton);
3250 return FloorDiv_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003251 default:
3252 /* should never happen, but just in case ... */
3253 PyErr_Format(PyExc_SystemError, "unknown operator found");
3254 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003255 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003256}
3257PyObject* ast2obj_unaryop(unaryop_ty o)
3258{
3259 switch(o) {
3260 case Invert:
3261 Py_INCREF(Invert_singleton);
3262 return Invert_singleton;
3263 case Not:
3264 Py_INCREF(Not_singleton);
3265 return Not_singleton;
3266 case UAdd:
3267 Py_INCREF(UAdd_singleton);
3268 return UAdd_singleton;
3269 case USub:
3270 Py_INCREF(USub_singleton);
3271 return USub_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003272 default:
3273 /* should never happen, but just in case ... */
3274 PyErr_Format(PyExc_SystemError, "unknown unaryop found");
3275 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003276 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003277}
3278PyObject* ast2obj_cmpop(cmpop_ty o)
3279{
3280 switch(o) {
3281 case Eq:
3282 Py_INCREF(Eq_singleton);
3283 return Eq_singleton;
3284 case NotEq:
3285 Py_INCREF(NotEq_singleton);
3286 return NotEq_singleton;
3287 case Lt:
3288 Py_INCREF(Lt_singleton);
3289 return Lt_singleton;
3290 case LtE:
3291 Py_INCREF(LtE_singleton);
3292 return LtE_singleton;
3293 case Gt:
3294 Py_INCREF(Gt_singleton);
3295 return Gt_singleton;
3296 case GtE:
3297 Py_INCREF(GtE_singleton);
3298 return GtE_singleton;
3299 case Is:
3300 Py_INCREF(Is_singleton);
3301 return Is_singleton;
3302 case IsNot:
3303 Py_INCREF(IsNot_singleton);
3304 return IsNot_singleton;
3305 case In:
3306 Py_INCREF(In_singleton);
3307 return In_singleton;
3308 case NotIn:
3309 Py_INCREF(NotIn_singleton);
3310 return NotIn_singleton;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003311 default:
3312 /* should never happen, but just in case ... */
3313 PyErr_Format(PyExc_SystemError, "unknown cmpop found");
3314 return NULL;
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003315 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003316}
3317PyObject*
3318ast2obj_comprehension(void* _o)
3319{
3320 comprehension_ty o = (comprehension_ty)_o;
3321 PyObject *result = NULL, *value = NULL;
3322 if (!o) {
3323 Py_INCREF(Py_None);
3324 return Py_None;
3325 }
3326
3327 result = PyType_GenericNew(comprehension_type, NULL, NULL);
3328 if (!result) return NULL;
3329 value = ast2obj_expr(o->target);
3330 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003331 if (_PyObject_SetAttrId(result, &PyId_target, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003332 goto failed;
3333 Py_DECREF(value);
3334 value = ast2obj_expr(o->iter);
3335 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003336 if (_PyObject_SetAttrId(result, &PyId_iter, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003337 goto failed;
3338 Py_DECREF(value);
3339 value = ast2obj_list(o->ifs, ast2obj_expr);
3340 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003341 if (_PyObject_SetAttrId(result, &PyId_ifs, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003342 goto failed;
3343 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003344 return result;
3345failed:
3346 Py_XDECREF(value);
3347 Py_XDECREF(result);
3348 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003349}
3350
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003351PyObject*
3352ast2obj_excepthandler(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003353{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003354 excepthandler_ty o = (excepthandler_ty)_o;
3355 PyObject *result = NULL, *value = NULL;
3356 if (!o) {
3357 Py_INCREF(Py_None);
3358 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003359 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003360
Neal Norwitzad74aa82008-03-31 05:14:30 +00003361 switch (o->kind) {
3362 case ExceptHandler_kind:
3363 result = PyType_GenericNew(ExceptHandler_type, NULL, NULL);
3364 if (!result) goto failed;
3365 value = ast2obj_expr(o->v.ExceptHandler.type);
3366 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003367 if (_PyObject_SetAttrId(result, &PyId_type, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003368 goto failed;
3369 Py_DECREF(value);
3370 value = ast2obj_identifier(o->v.ExceptHandler.name);
3371 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003372 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003373 goto failed;
3374 Py_DECREF(value);
3375 value = ast2obj_list(o->v.ExceptHandler.body, ast2obj_stmt);
3376 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003377 if (_PyObject_SetAttrId(result, &PyId_body, value) == -1)
Neal Norwitzad74aa82008-03-31 05:14:30 +00003378 goto failed;
3379 Py_DECREF(value);
3380 break;
3381 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003382 value = ast2obj_int(o->lineno);
3383 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003384 if (_PyObject_SetAttrId(result, &PyId_lineno, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003385 goto failed;
3386 Py_DECREF(value);
3387 value = ast2obj_int(o->col_offset);
3388 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003389 if (_PyObject_SetAttrId(result, &PyId_col_offset, value) < 0)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003390 goto failed;
3391 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003392 return result;
3393failed:
3394 Py_XDECREF(value);
3395 Py_XDECREF(result);
3396 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003397}
3398
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003399PyObject*
3400ast2obj_arguments(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003401{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003402 arguments_ty o = (arguments_ty)_o;
3403 PyObject *result = NULL, *value = NULL;
3404 if (!o) {
3405 Py_INCREF(Py_None);
3406 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003407 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003408
3409 result = PyType_GenericNew(arguments_type, NULL, NULL);
3410 if (!result) return NULL;
Neal Norwitzc1505362006-12-28 06:47:50 +00003411 value = ast2obj_list(o->args, ast2obj_arg);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003412 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003413 if (_PyObject_SetAttrId(result, &PyId_args, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003414 goto failed;
3415 Py_DECREF(value);
3416 value = ast2obj_identifier(o->vararg);
3417 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003418 if (_PyObject_SetAttrId(result, &PyId_vararg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003419 goto failed;
3420 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003421 value = ast2obj_expr(o->varargannotation);
3422 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003423 if (_PyObject_SetAttrId(result, &PyId_varargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003424 goto failed;
3425 Py_DECREF(value);
3426 value = ast2obj_list(o->kwonlyargs, ast2obj_arg);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003427 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003428 if (_PyObject_SetAttrId(result, &PyId_kwonlyargs, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003429 goto failed;
3430 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003431 value = ast2obj_identifier(o->kwarg);
3432 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003433 if (_PyObject_SetAttrId(result, &PyId_kwarg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003434 goto failed;
3435 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003436 value = ast2obj_expr(o->kwargannotation);
3437 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003438 if (_PyObject_SetAttrId(result, &PyId_kwargannotation, value) == -1)
Neal Norwitzc1505362006-12-28 06:47:50 +00003439 goto failed;
3440 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003441 value = ast2obj_list(o->defaults, ast2obj_expr);
3442 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003443 if (_PyObject_SetAttrId(result, &PyId_defaults, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003444 goto failed;
3445 Py_DECREF(value);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003446 value = ast2obj_list(o->kw_defaults, ast2obj_expr);
3447 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003448 if (_PyObject_SetAttrId(result, &PyId_kw_defaults, value) == -1)
Guido van Rossum4f72a782006-10-27 23:31:49 +00003449 goto failed;
3450 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003451 return result;
3452failed:
3453 Py_XDECREF(value);
3454 Py_XDECREF(result);
3455 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003456}
3457
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003458PyObject*
Neal Norwitzc1505362006-12-28 06:47:50 +00003459ast2obj_arg(void* _o)
3460{
3461 arg_ty o = (arg_ty)_o;
3462 PyObject *result = NULL, *value = NULL;
3463 if (!o) {
3464 Py_INCREF(Py_None);
3465 return Py_None;
3466 }
3467
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003468 result = PyType_GenericNew(arg_type, NULL, NULL);
3469 if (!result) return NULL;
3470 value = ast2obj_identifier(o->arg);
3471 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003472 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003473 goto failed;
3474 Py_DECREF(value);
3475 value = ast2obj_expr(o->annotation);
3476 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003477 if (_PyObject_SetAttrId(result, &PyId_annotation, value) == -1)
Guido van Rossum1bc535d2007-05-15 18:46:22 +00003478 goto failed;
3479 Py_DECREF(value);
Neal Norwitzc1505362006-12-28 06:47:50 +00003480 return result;
3481failed:
3482 Py_XDECREF(value);
3483 Py_XDECREF(result);
3484 return NULL;
3485}
3486
3487PyObject*
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003488ast2obj_keyword(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003489{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003490 keyword_ty o = (keyword_ty)_o;
3491 PyObject *result = NULL, *value = NULL;
3492 if (!o) {
3493 Py_INCREF(Py_None);
3494 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003495 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003496
3497 result = PyType_GenericNew(keyword_type, NULL, NULL);
3498 if (!result) return NULL;
3499 value = ast2obj_identifier(o->arg);
3500 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003501 if (_PyObject_SetAttrId(result, &PyId_arg, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003502 goto failed;
3503 Py_DECREF(value);
3504 value = ast2obj_expr(o->value);
3505 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003506 if (_PyObject_SetAttrId(result, &PyId_value, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003507 goto failed;
3508 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003509 return result;
3510failed:
3511 Py_XDECREF(value);
3512 Py_XDECREF(result);
3513 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003514}
3515
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003516PyObject*
3517ast2obj_alias(void* _o)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003518{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003519 alias_ty o = (alias_ty)_o;
3520 PyObject *result = NULL, *value = NULL;
3521 if (!o) {
3522 Py_INCREF(Py_None);
3523 return Py_None;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003524 }
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003525
3526 result = PyType_GenericNew(alias_type, NULL, NULL);
3527 if (!result) return NULL;
3528 value = ast2obj_identifier(o->name);
3529 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003530 if (_PyObject_SetAttrId(result, &PyId_name, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003531 goto failed;
3532 Py_DECREF(value);
3533 value = ast2obj_identifier(o->asname);
3534 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003535 if (_PyObject_SetAttrId(result, &PyId_asname, value) == -1)
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003536 goto failed;
3537 Py_DECREF(value);
Martin v. Löwisbd260da2006-02-26 19:42:26 +00003538 return result;
3539failed:
3540 Py_XDECREF(value);
3541 Py_XDECREF(result);
3542 return NULL;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003543}
3544
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003545PyObject*
3546ast2obj_withitem(void* _o)
3547{
3548 withitem_ty o = (withitem_ty)_o;
3549 PyObject *result = NULL, *value = NULL;
3550 if (!o) {
3551 Py_INCREF(Py_None);
3552 return Py_None;
3553 }
3554
3555 result = PyType_GenericNew(withitem_type, NULL, NULL);
3556 if (!result) return NULL;
3557 value = ast2obj_expr(o->context_expr);
3558 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003559 if (_PyObject_SetAttrId(result, &PyId_context_expr, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003560 goto failed;
3561 Py_DECREF(value);
3562 value = ast2obj_expr(o->optional_vars);
3563 if (!value) goto failed;
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02003564 if (_PyObject_SetAttrId(result, &PyId_optional_vars, value) == -1)
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05003565 goto failed;
3566 Py_DECREF(value);
3567 return result;
3568failed:
3569 Py_XDECREF(value);
3570 Py_XDECREF(result);
3571 return NULL;
3572}
3573
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003574
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003575int
3576obj2ast_mod(PyObject* obj, mod_ty* out, PyArena* arena)
3577{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003578 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003579
Benjamin Petersond8f65972010-11-20 04:31:07 +00003580 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003581
3582 if (obj == Py_None) {
3583 *out = NULL;
3584 return 0;
3585 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003586 isinstance = PyObject_IsInstance(obj, (PyObject*)Module_type);
3587 if (isinstance == -1) {
3588 return 1;
3589 }
3590 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003591 asdl_seq* body;
3592
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003593 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003594 int res;
3595 Py_ssize_t len;
3596 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003597 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003598 if (tmp == NULL) goto failed;
3599 if (!PyList_Check(tmp)) {
3600 PyErr_Format(PyExc_TypeError, "Module field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3601 goto failed;
3602 }
3603 len = PyList_GET_SIZE(tmp);
3604 body = asdl_seq_new(len, arena);
3605 if (body == NULL) goto failed;
3606 for (i = 0; i < len; i++) {
3607 stmt_ty value;
3608 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3609 if (res != 0) goto failed;
3610 asdl_seq_SET(body, i, value);
3611 }
3612 Py_XDECREF(tmp);
3613 tmp = NULL;
3614 } else {
3615 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Module");
3616 return 1;
3617 }
3618 *out = Module(body, arena);
3619 if (*out == NULL) goto failed;
3620 return 0;
3621 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003622 isinstance = PyObject_IsInstance(obj, (PyObject*)Interactive_type);
3623 if (isinstance == -1) {
3624 return 1;
3625 }
3626 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003627 asdl_seq* body;
3628
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003629 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003630 int res;
3631 Py_ssize_t len;
3632 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003633 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003634 if (tmp == NULL) goto failed;
3635 if (!PyList_Check(tmp)) {
3636 PyErr_Format(PyExc_TypeError, "Interactive field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3637 goto failed;
3638 }
3639 len = PyList_GET_SIZE(tmp);
3640 body = asdl_seq_new(len, arena);
3641 if (body == NULL) goto failed;
3642 for (i = 0; i < len; i++) {
3643 stmt_ty value;
3644 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3645 if (res != 0) goto failed;
3646 asdl_seq_SET(body, i, value);
3647 }
3648 Py_XDECREF(tmp);
3649 tmp = NULL;
3650 } else {
3651 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Interactive");
3652 return 1;
3653 }
3654 *out = Interactive(body, arena);
3655 if (*out == NULL) goto failed;
3656 return 0;
3657 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003658 isinstance = PyObject_IsInstance(obj, (PyObject*)Expression_type);
3659 if (isinstance == -1) {
3660 return 1;
3661 }
3662 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003663 expr_ty body;
3664
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003665 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003666 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003667 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003668 if (tmp == NULL) goto failed;
3669 res = obj2ast_expr(tmp, &body, arena);
3670 if (res != 0) goto failed;
3671 Py_XDECREF(tmp);
3672 tmp = NULL;
3673 } else {
3674 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Expression");
3675 return 1;
3676 }
3677 *out = Expression(body, arena);
3678 if (*out == NULL) goto failed;
3679 return 0;
3680 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003681 isinstance = PyObject_IsInstance(obj, (PyObject*)Suite_type);
3682 if (isinstance == -1) {
3683 return 1;
3684 }
3685 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003686 asdl_seq* body;
3687
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003688 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003689 int res;
3690 Py_ssize_t len;
3691 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003692 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003693 if (tmp == NULL) goto failed;
3694 if (!PyList_Check(tmp)) {
3695 PyErr_Format(PyExc_TypeError, "Suite field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3696 goto failed;
3697 }
3698 len = PyList_GET_SIZE(tmp);
3699 body = asdl_seq_new(len, arena);
3700 if (body == NULL) goto failed;
3701 for (i = 0; i < len; i++) {
3702 stmt_ty value;
3703 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3704 if (res != 0) goto failed;
3705 asdl_seq_SET(body, i, value);
3706 }
3707 Py_XDECREF(tmp);
3708 tmp = NULL;
3709 } else {
3710 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Suite");
3711 return 1;
3712 }
3713 *out = Suite(body, arena);
3714 if (*out == NULL) goto failed;
3715 return 0;
3716 }
3717
Benjamin Peterson5b066812010-11-20 01:38:49 +00003718 PyErr_Format(PyExc_TypeError, "expected some sort of mod, but got %R", obj);
3719 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00003720 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003721 return 1;
3722}
3723
3724int
3725obj2ast_stmt(PyObject* obj, stmt_ty* out, PyArena* arena)
3726{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003727 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003728
Benjamin Petersond8f65972010-11-20 04:31:07 +00003729 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003730 int lineno;
3731 int col_offset;
3732
3733 if (obj == Py_None) {
3734 *out = NULL;
3735 return 0;
3736 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003737 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
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_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003740 if (tmp == NULL) goto failed;
3741 res = obj2ast_int(tmp, &lineno, arena);
3742 if (res != 0) goto failed;
3743 Py_XDECREF(tmp);
3744 tmp = NULL;
3745 } else {
3746 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from stmt");
3747 return 1;
3748 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003749 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003750 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003751 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003752 if (tmp == NULL) goto failed;
3753 res = obj2ast_int(tmp, &col_offset, arena);
3754 if (res != 0) goto failed;
3755 Py_XDECREF(tmp);
3756 tmp = NULL;
3757 } else {
3758 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from stmt");
3759 return 1;
3760 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003761 isinstance = PyObject_IsInstance(obj, (PyObject*)FunctionDef_type);
3762 if (isinstance == -1) {
3763 return 1;
3764 }
3765 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003766 identifier name;
3767 arguments_ty args;
3768 asdl_seq* body;
3769 asdl_seq* decorator_list;
3770 expr_ty returns;
3771
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003772 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003773 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003774 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003775 if (tmp == NULL) goto failed;
3776 res = obj2ast_identifier(tmp, &name, arena);
3777 if (res != 0) goto failed;
3778 Py_XDECREF(tmp);
3779 tmp = NULL;
3780 } else {
3781 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from FunctionDef");
3782 return 1;
3783 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003784 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003785 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003786 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003787 if (tmp == NULL) goto failed;
3788 res = obj2ast_arguments(tmp, &args, arena);
3789 if (res != 0) goto failed;
3790 Py_XDECREF(tmp);
3791 tmp = NULL;
3792 } else {
3793 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from FunctionDef");
3794 return 1;
3795 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003796 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003797 int res;
3798 Py_ssize_t len;
3799 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003800 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003801 if (tmp == NULL) goto failed;
3802 if (!PyList_Check(tmp)) {
3803 PyErr_Format(PyExc_TypeError, "FunctionDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3804 goto failed;
3805 }
3806 len = PyList_GET_SIZE(tmp);
3807 body = asdl_seq_new(len, arena);
3808 if (body == NULL) goto failed;
3809 for (i = 0; i < len; i++) {
3810 stmt_ty value;
3811 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3812 if (res != 0) goto failed;
3813 asdl_seq_SET(body, i, value);
3814 }
3815 Py_XDECREF(tmp);
3816 tmp = NULL;
3817 } else {
3818 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from FunctionDef");
3819 return 1;
3820 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003821 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003822 int res;
3823 Py_ssize_t len;
3824 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003825 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003826 if (tmp == NULL) goto failed;
3827 if (!PyList_Check(tmp)) {
3828 PyErr_Format(PyExc_TypeError, "FunctionDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3829 goto failed;
3830 }
3831 len = PyList_GET_SIZE(tmp);
3832 decorator_list = asdl_seq_new(len, arena);
3833 if (decorator_list == NULL) goto failed;
3834 for (i = 0; i < len; i++) {
3835 expr_ty value;
3836 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3837 if (res != 0) goto failed;
3838 asdl_seq_SET(decorator_list, i, value);
3839 }
3840 Py_XDECREF(tmp);
3841 tmp = NULL;
3842 } else {
3843 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from FunctionDef");
3844 return 1;
3845 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003846 if (_PyObject_HasAttrId(obj, &PyId_returns)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003847 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003848 tmp = _PyObject_GetAttrId(obj, &PyId_returns);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003849 if (tmp == NULL) goto failed;
3850 res = obj2ast_expr(tmp, &returns, arena);
3851 if (res != 0) goto failed;
3852 Py_XDECREF(tmp);
3853 tmp = NULL;
3854 } else {
3855 returns = NULL;
3856 }
3857 *out = FunctionDef(name, args, body, decorator_list, returns,
3858 lineno, col_offset, arena);
3859 if (*out == NULL) goto failed;
3860 return 0;
3861 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00003862 isinstance = PyObject_IsInstance(obj, (PyObject*)ClassDef_type);
3863 if (isinstance == -1) {
3864 return 1;
3865 }
3866 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003867 identifier name;
3868 asdl_seq* bases;
3869 asdl_seq* keywords;
3870 expr_ty starargs;
3871 expr_ty kwargs;
3872 asdl_seq* body;
3873 asdl_seq* decorator_list;
3874
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003875 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003876 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003877 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003878 if (tmp == NULL) goto failed;
3879 res = obj2ast_identifier(tmp, &name, arena);
3880 if (res != 0) goto failed;
3881 Py_XDECREF(tmp);
3882 tmp = NULL;
3883 } else {
3884 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from ClassDef");
3885 return 1;
3886 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003887 if (_PyObject_HasAttrId(obj, &PyId_bases)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003888 int res;
3889 Py_ssize_t len;
3890 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003891 tmp = _PyObject_GetAttrId(obj, &PyId_bases);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003892 if (tmp == NULL) goto failed;
3893 if (!PyList_Check(tmp)) {
3894 PyErr_Format(PyExc_TypeError, "ClassDef field \"bases\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3895 goto failed;
3896 }
3897 len = PyList_GET_SIZE(tmp);
3898 bases = asdl_seq_new(len, arena);
3899 if (bases == NULL) goto failed;
3900 for (i = 0; i < len; i++) {
3901 expr_ty value;
3902 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
3903 if (res != 0) goto failed;
3904 asdl_seq_SET(bases, i, value);
3905 }
3906 Py_XDECREF(tmp);
3907 tmp = NULL;
3908 } else {
3909 PyErr_SetString(PyExc_TypeError, "required field \"bases\" missing from ClassDef");
3910 return 1;
3911 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003912 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
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_keywords);
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 \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3920 goto failed;
3921 }
3922 len = PyList_GET_SIZE(tmp);
3923 keywords = asdl_seq_new(len, arena);
3924 if (keywords == NULL) goto failed;
3925 for (i = 0; i < len; i++) {
3926 keyword_ty value;
3927 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
3928 if (res != 0) goto failed;
3929 asdl_seq_SET(keywords, i, value);
3930 }
3931 Py_XDECREF(tmp);
3932 tmp = NULL;
3933 } else {
3934 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from ClassDef");
3935 return 1;
3936 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003937 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003938 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003939 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003940 if (tmp == NULL) goto failed;
3941 res = obj2ast_expr(tmp, &starargs, arena);
3942 if (res != 0) goto failed;
3943 Py_XDECREF(tmp);
3944 tmp = NULL;
3945 } else {
3946 starargs = NULL;
3947 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003948 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003949 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003950 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003951 if (tmp == NULL) goto failed;
3952 res = obj2ast_expr(tmp, &kwargs, arena);
3953 if (res != 0) goto failed;
3954 Py_XDECREF(tmp);
3955 tmp = NULL;
3956 } else {
3957 kwargs = NULL;
3958 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003959 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003960 int res;
3961 Py_ssize_t len;
3962 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003963 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003964 if (tmp == NULL) goto failed;
3965 if (!PyList_Check(tmp)) {
3966 PyErr_Format(PyExc_TypeError, "ClassDef field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3967 goto failed;
3968 }
3969 len = PyList_GET_SIZE(tmp);
3970 body = asdl_seq_new(len, arena);
3971 if (body == NULL) goto failed;
3972 for (i = 0; i < len; i++) {
3973 stmt_ty value;
3974 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
3975 if (res != 0) goto failed;
3976 asdl_seq_SET(body, i, value);
3977 }
3978 Py_XDECREF(tmp);
3979 tmp = NULL;
3980 } else {
3981 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ClassDef");
3982 return 1;
3983 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003984 if (_PyObject_HasAttrId(obj, &PyId_decorator_list)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003985 int res;
3986 Py_ssize_t len;
3987 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02003988 tmp = _PyObject_GetAttrId(obj, &PyId_decorator_list);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00003989 if (tmp == NULL) goto failed;
3990 if (!PyList_Check(tmp)) {
3991 PyErr_Format(PyExc_TypeError, "ClassDef field \"decorator_list\" must be a list, not a %.200s", tmp->ob_type->tp_name);
3992 goto failed;
3993 }
3994 len = PyList_GET_SIZE(tmp);
3995 decorator_list = asdl_seq_new(len, arena);
3996 if (decorator_list == NULL) goto failed;
3997 for (i = 0; i < len; i++) {
3998 expr_ty value;
3999 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4000 if (res != 0) goto failed;
4001 asdl_seq_SET(decorator_list, i, value);
4002 }
4003 Py_XDECREF(tmp);
4004 tmp = NULL;
4005 } else {
4006 PyErr_SetString(PyExc_TypeError, "required field \"decorator_list\" missing from ClassDef");
4007 return 1;
4008 }
4009 *out = ClassDef(name, bases, keywords, starargs, kwargs, body,
4010 decorator_list, lineno, col_offset, arena);
4011 if (*out == NULL) goto failed;
4012 return 0;
4013 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004014 isinstance = PyObject_IsInstance(obj, (PyObject*)Return_type);
4015 if (isinstance == -1) {
4016 return 1;
4017 }
4018 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004019 expr_ty value;
4020
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004021 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004022 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004023 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004024 if (tmp == NULL) goto failed;
4025 res = obj2ast_expr(tmp, &value, arena);
4026 if (res != 0) goto failed;
4027 Py_XDECREF(tmp);
4028 tmp = NULL;
4029 } else {
4030 value = NULL;
4031 }
4032 *out = Return(value, lineno, col_offset, arena);
4033 if (*out == NULL) goto failed;
4034 return 0;
4035 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004036 isinstance = PyObject_IsInstance(obj, (PyObject*)Delete_type);
4037 if (isinstance == -1) {
4038 return 1;
4039 }
4040 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004041 asdl_seq* targets;
4042
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004043 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004044 int res;
4045 Py_ssize_t len;
4046 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004047 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004048 if (tmp == NULL) goto failed;
4049 if (!PyList_Check(tmp)) {
4050 PyErr_Format(PyExc_TypeError, "Delete field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4051 goto failed;
4052 }
4053 len = PyList_GET_SIZE(tmp);
4054 targets = asdl_seq_new(len, arena);
4055 if (targets == NULL) goto failed;
4056 for (i = 0; i < len; i++) {
4057 expr_ty value;
4058 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4059 if (res != 0) goto failed;
4060 asdl_seq_SET(targets, i, value);
4061 }
4062 Py_XDECREF(tmp);
4063 tmp = NULL;
4064 } else {
4065 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Delete");
4066 return 1;
4067 }
4068 *out = Delete(targets, lineno, col_offset, arena);
4069 if (*out == NULL) goto failed;
4070 return 0;
4071 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004072 isinstance = PyObject_IsInstance(obj, (PyObject*)Assign_type);
4073 if (isinstance == -1) {
4074 return 1;
4075 }
4076 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004077 asdl_seq* targets;
4078 expr_ty value;
4079
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004080 if (_PyObject_HasAttrId(obj, &PyId_targets)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004081 int res;
4082 Py_ssize_t len;
4083 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004084 tmp = _PyObject_GetAttrId(obj, &PyId_targets);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004085 if (tmp == NULL) goto failed;
4086 if (!PyList_Check(tmp)) {
4087 PyErr_Format(PyExc_TypeError, "Assign field \"targets\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4088 goto failed;
4089 }
4090 len = PyList_GET_SIZE(tmp);
4091 targets = asdl_seq_new(len, arena);
4092 if (targets == NULL) goto failed;
4093 for (i = 0; i < len; i++) {
4094 expr_ty value;
4095 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4096 if (res != 0) goto failed;
4097 asdl_seq_SET(targets, i, value);
4098 }
4099 Py_XDECREF(tmp);
4100 tmp = NULL;
4101 } else {
4102 PyErr_SetString(PyExc_TypeError, "required field \"targets\" missing from Assign");
4103 return 1;
4104 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004105 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004106 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004107 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004108 if (tmp == NULL) goto failed;
4109 res = obj2ast_expr(tmp, &value, arena);
4110 if (res != 0) goto failed;
4111 Py_XDECREF(tmp);
4112 tmp = NULL;
4113 } else {
4114 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Assign");
4115 return 1;
4116 }
4117 *out = Assign(targets, value, lineno, col_offset, arena);
4118 if (*out == NULL) goto failed;
4119 return 0;
4120 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004121 isinstance = PyObject_IsInstance(obj, (PyObject*)AugAssign_type);
4122 if (isinstance == -1) {
4123 return 1;
4124 }
4125 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004126 expr_ty target;
4127 operator_ty op;
4128 expr_ty value;
4129
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004130 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004131 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004132 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004133 if (tmp == NULL) goto failed;
4134 res = obj2ast_expr(tmp, &target, arena);
4135 if (res != 0) goto failed;
4136 Py_XDECREF(tmp);
4137 tmp = NULL;
4138 } else {
4139 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from AugAssign");
4140 return 1;
4141 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004142 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004143 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004144 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004145 if (tmp == NULL) goto failed;
4146 res = obj2ast_operator(tmp, &op, arena);
4147 if (res != 0) goto failed;
4148 Py_XDECREF(tmp);
4149 tmp = NULL;
4150 } else {
4151 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from AugAssign");
4152 return 1;
4153 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004154 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004155 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004156 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004157 if (tmp == NULL) goto failed;
4158 res = obj2ast_expr(tmp, &value, arena);
4159 if (res != 0) goto failed;
4160 Py_XDECREF(tmp);
4161 tmp = NULL;
4162 } else {
4163 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from AugAssign");
4164 return 1;
4165 }
4166 *out = AugAssign(target, op, value, lineno, col_offset, arena);
4167 if (*out == NULL) goto failed;
4168 return 0;
4169 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004170 isinstance = PyObject_IsInstance(obj, (PyObject*)For_type);
4171 if (isinstance == -1) {
4172 return 1;
4173 }
4174 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004175 expr_ty target;
4176 expr_ty iter;
4177 asdl_seq* body;
4178 asdl_seq* orelse;
4179
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004180 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004181 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004182 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004183 if (tmp == NULL) goto failed;
4184 res = obj2ast_expr(tmp, &target, arena);
4185 if (res != 0) goto failed;
4186 Py_XDECREF(tmp);
4187 tmp = NULL;
4188 } else {
4189 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from For");
4190 return 1;
4191 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004192 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004193 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004194 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004195 if (tmp == NULL) goto failed;
4196 res = obj2ast_expr(tmp, &iter, arena);
4197 if (res != 0) goto failed;
4198 Py_XDECREF(tmp);
4199 tmp = NULL;
4200 } else {
4201 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from For");
4202 return 1;
4203 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004204 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004205 int res;
4206 Py_ssize_t len;
4207 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004208 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004209 if (tmp == NULL) goto failed;
4210 if (!PyList_Check(tmp)) {
4211 PyErr_Format(PyExc_TypeError, "For field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4212 goto failed;
4213 }
4214 len = PyList_GET_SIZE(tmp);
4215 body = asdl_seq_new(len, arena);
4216 if (body == NULL) goto failed;
4217 for (i = 0; i < len; i++) {
4218 stmt_ty value;
4219 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4220 if (res != 0) goto failed;
4221 asdl_seq_SET(body, i, value);
4222 }
4223 Py_XDECREF(tmp);
4224 tmp = NULL;
4225 } else {
4226 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from For");
4227 return 1;
4228 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004229 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004230 int res;
4231 Py_ssize_t len;
4232 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004233 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004234 if (tmp == NULL) goto failed;
4235 if (!PyList_Check(tmp)) {
4236 PyErr_Format(PyExc_TypeError, "For field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4237 goto failed;
4238 }
4239 len = PyList_GET_SIZE(tmp);
4240 orelse = asdl_seq_new(len, arena);
4241 if (orelse == NULL) goto failed;
4242 for (i = 0; i < len; i++) {
4243 stmt_ty value;
4244 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4245 if (res != 0) goto failed;
4246 asdl_seq_SET(orelse, i, value);
4247 }
4248 Py_XDECREF(tmp);
4249 tmp = NULL;
4250 } else {
4251 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from For");
4252 return 1;
4253 }
4254 *out = For(target, iter, body, orelse, lineno, col_offset,
4255 arena);
4256 if (*out == NULL) goto failed;
4257 return 0;
4258 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004259 isinstance = PyObject_IsInstance(obj, (PyObject*)While_type);
4260 if (isinstance == -1) {
4261 return 1;
4262 }
4263 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004264 expr_ty test;
4265 asdl_seq* body;
4266 asdl_seq* orelse;
4267
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004268 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004269 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004270 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004271 if (tmp == NULL) goto failed;
4272 res = obj2ast_expr(tmp, &test, arena);
4273 if (res != 0) goto failed;
4274 Py_XDECREF(tmp);
4275 tmp = NULL;
4276 } else {
4277 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from While");
4278 return 1;
4279 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004280 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004281 int res;
4282 Py_ssize_t len;
4283 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004284 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004285 if (tmp == NULL) goto failed;
4286 if (!PyList_Check(tmp)) {
4287 PyErr_Format(PyExc_TypeError, "While field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4288 goto failed;
4289 }
4290 len = PyList_GET_SIZE(tmp);
4291 body = asdl_seq_new(len, arena);
4292 if (body == NULL) goto failed;
4293 for (i = 0; i < len; i++) {
4294 stmt_ty value;
4295 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4296 if (res != 0) goto failed;
4297 asdl_seq_SET(body, i, value);
4298 }
4299 Py_XDECREF(tmp);
4300 tmp = NULL;
4301 } else {
4302 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from While");
4303 return 1;
4304 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004305 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004306 int res;
4307 Py_ssize_t len;
4308 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004309 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004310 if (tmp == NULL) goto failed;
4311 if (!PyList_Check(tmp)) {
4312 PyErr_Format(PyExc_TypeError, "While field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4313 goto failed;
4314 }
4315 len = PyList_GET_SIZE(tmp);
4316 orelse = asdl_seq_new(len, arena);
4317 if (orelse == NULL) goto failed;
4318 for (i = 0; i < len; i++) {
4319 stmt_ty value;
4320 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4321 if (res != 0) goto failed;
4322 asdl_seq_SET(orelse, i, value);
4323 }
4324 Py_XDECREF(tmp);
4325 tmp = NULL;
4326 } else {
4327 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from While");
4328 return 1;
4329 }
4330 *out = While(test, body, orelse, lineno, col_offset, arena);
4331 if (*out == NULL) goto failed;
4332 return 0;
4333 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004334 isinstance = PyObject_IsInstance(obj, (PyObject*)If_type);
4335 if (isinstance == -1) {
4336 return 1;
4337 }
4338 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004339 expr_ty test;
4340 asdl_seq* body;
4341 asdl_seq* orelse;
4342
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004343 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004344 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004345 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004346 if (tmp == NULL) goto failed;
4347 res = obj2ast_expr(tmp, &test, arena);
4348 if (res != 0) goto failed;
4349 Py_XDECREF(tmp);
4350 tmp = NULL;
4351 } else {
4352 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from If");
4353 return 1;
4354 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004355 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004356 int res;
4357 Py_ssize_t len;
4358 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004359 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004360 if (tmp == NULL) goto failed;
4361 if (!PyList_Check(tmp)) {
4362 PyErr_Format(PyExc_TypeError, "If field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4363 goto failed;
4364 }
4365 len = PyList_GET_SIZE(tmp);
4366 body = asdl_seq_new(len, arena);
4367 if (body == NULL) goto failed;
4368 for (i = 0; i < len; i++) {
4369 stmt_ty value;
4370 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4371 if (res != 0) goto failed;
4372 asdl_seq_SET(body, i, value);
4373 }
4374 Py_XDECREF(tmp);
4375 tmp = NULL;
4376 } else {
4377 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from If");
4378 return 1;
4379 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004380 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004381 int res;
4382 Py_ssize_t len;
4383 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004384 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004385 if (tmp == NULL) goto failed;
4386 if (!PyList_Check(tmp)) {
4387 PyErr_Format(PyExc_TypeError, "If field \"orelse\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4388 goto failed;
4389 }
4390 len = PyList_GET_SIZE(tmp);
4391 orelse = asdl_seq_new(len, arena);
4392 if (orelse == NULL) goto failed;
4393 for (i = 0; i < len; i++) {
4394 stmt_ty value;
4395 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4396 if (res != 0) goto failed;
4397 asdl_seq_SET(orelse, i, value);
4398 }
4399 Py_XDECREF(tmp);
4400 tmp = NULL;
4401 } else {
4402 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from If");
4403 return 1;
4404 }
4405 *out = If(test, body, orelse, lineno, col_offset, arena);
4406 if (*out == NULL) goto failed;
4407 return 0;
4408 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004409 isinstance = PyObject_IsInstance(obj, (PyObject*)With_type);
4410 if (isinstance == -1) {
4411 return 1;
4412 }
4413 if (isinstance) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004414 asdl_seq* items;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004415 asdl_seq* body;
4416
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004417 if (_PyObject_HasAttrId(obj, &PyId_items)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004418 int res;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004419 Py_ssize_t len;
4420 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004421 tmp = _PyObject_GetAttrId(obj, &PyId_items);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004422 if (tmp == NULL) goto failed;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004423 if (!PyList_Check(tmp)) {
4424 PyErr_Format(PyExc_TypeError, "With field \"items\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4425 goto failed;
4426 }
4427 len = PyList_GET_SIZE(tmp);
4428 items = asdl_seq_new(len, arena);
4429 if (items == NULL) goto failed;
4430 for (i = 0; i < len; i++) {
4431 withitem_ty value;
4432 res = obj2ast_withitem(PyList_GET_ITEM(tmp, i), &value, arena);
4433 if (res != 0) goto failed;
4434 asdl_seq_SET(items, i, value);
4435 }
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004436 Py_XDECREF(tmp);
4437 tmp = NULL;
4438 } else {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004439 PyErr_SetString(PyExc_TypeError, "required field \"items\" missing from With");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004440 return 1;
4441 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004442 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004443 int res;
4444 Py_ssize_t len;
4445 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004446 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004447 if (tmp == NULL) goto failed;
4448 if (!PyList_Check(tmp)) {
4449 PyErr_Format(PyExc_TypeError, "With field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4450 goto failed;
4451 }
4452 len = PyList_GET_SIZE(tmp);
4453 body = asdl_seq_new(len, arena);
4454 if (body == NULL) goto failed;
4455 for (i = 0; i < len; i++) {
4456 stmt_ty value;
4457 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4458 if (res != 0) goto failed;
4459 asdl_seq_SET(body, i, value);
4460 }
4461 Py_XDECREF(tmp);
4462 tmp = NULL;
4463 } else {
4464 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from With");
4465 return 1;
4466 }
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05004467 *out = With(items, body, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004468 if (*out == NULL) goto failed;
4469 return 0;
4470 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004471 isinstance = PyObject_IsInstance(obj, (PyObject*)Raise_type);
4472 if (isinstance == -1) {
4473 return 1;
4474 }
4475 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004476 expr_ty exc;
4477 expr_ty cause;
4478
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004479 if (_PyObject_HasAttrId(obj, &PyId_exc)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004480 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004481 tmp = _PyObject_GetAttrId(obj, &PyId_exc);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004482 if (tmp == NULL) goto failed;
4483 res = obj2ast_expr(tmp, &exc, arena);
4484 if (res != 0) goto failed;
4485 Py_XDECREF(tmp);
4486 tmp = NULL;
4487 } else {
4488 exc = NULL;
4489 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004490 if (_PyObject_HasAttrId(obj, &PyId_cause)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004491 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004492 tmp = _PyObject_GetAttrId(obj, &PyId_cause);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004493 if (tmp == NULL) goto failed;
4494 res = obj2ast_expr(tmp, &cause, arena);
4495 if (res != 0) goto failed;
4496 Py_XDECREF(tmp);
4497 tmp = NULL;
4498 } else {
4499 cause = NULL;
4500 }
4501 *out = Raise(exc, cause, lineno, col_offset, arena);
4502 if (*out == NULL) goto failed;
4503 return 0;
4504 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004505 isinstance = PyObject_IsInstance(obj, (PyObject*)Try_type);
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004506 if (isinstance == -1) {
4507 return 1;
4508 }
4509 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004510 asdl_seq* body;
4511 asdl_seq* handlers;
4512 asdl_seq* orelse;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004513 asdl_seq* finalbody;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004514
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004515 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004516 int res;
4517 Py_ssize_t len;
4518 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004519 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004520 if (tmp == NULL) goto failed;
4521 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004522 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 +00004523 goto failed;
4524 }
4525 len = PyList_GET_SIZE(tmp);
4526 body = asdl_seq_new(len, arena);
4527 if (body == NULL) goto failed;
4528 for (i = 0; i < len; i++) {
4529 stmt_ty value;
4530 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4531 if (res != 0) goto failed;
4532 asdl_seq_SET(body, i, value);
4533 }
4534 Py_XDECREF(tmp);
4535 tmp = NULL;
4536 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004537 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004538 return 1;
4539 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004540 if (_PyObject_HasAttrId(obj, &PyId_handlers)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004541 int res;
4542 Py_ssize_t len;
4543 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004544 tmp = _PyObject_GetAttrId(obj, &PyId_handlers);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004545 if (tmp == NULL) goto failed;
4546 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004547 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 +00004548 goto failed;
4549 }
4550 len = PyList_GET_SIZE(tmp);
4551 handlers = asdl_seq_new(len, arena);
4552 if (handlers == NULL) goto failed;
4553 for (i = 0; i < len; i++) {
4554 excepthandler_ty value;
4555 res = obj2ast_excepthandler(PyList_GET_ITEM(tmp, i), &value, arena);
4556 if (res != 0) goto failed;
4557 asdl_seq_SET(handlers, i, value);
4558 }
4559 Py_XDECREF(tmp);
4560 tmp = NULL;
4561 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004562 PyErr_SetString(PyExc_TypeError, "required field \"handlers\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004563 return 1;
4564 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004565 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004566 int res;
4567 Py_ssize_t len;
4568 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004569 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004570 if (tmp == NULL) goto failed;
4571 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004572 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 +00004573 goto failed;
4574 }
4575 len = PyList_GET_SIZE(tmp);
4576 orelse = asdl_seq_new(len, arena);
4577 if (orelse == NULL) goto failed;
4578 for (i = 0; i < len; i++) {
4579 stmt_ty value;
4580 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4581 if (res != 0) goto failed;
4582 asdl_seq_SET(orelse, i, value);
4583 }
4584 Py_XDECREF(tmp);
4585 tmp = NULL;
4586 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004587 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004588 return 1;
4589 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004590 if (_PyObject_HasAttrId(obj, &PyId_finalbody)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004591 int res;
4592 Py_ssize_t len;
4593 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004594 tmp = _PyObject_GetAttrId(obj, &PyId_finalbody);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004595 if (tmp == NULL) goto failed;
4596 if (!PyList_Check(tmp)) {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004597 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 +00004598 goto failed;
4599 }
4600 len = PyList_GET_SIZE(tmp);
4601 finalbody = asdl_seq_new(len, arena);
4602 if (finalbody == NULL) goto failed;
4603 for (i = 0; i < len; i++) {
4604 stmt_ty value;
4605 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
4606 if (res != 0) goto failed;
4607 asdl_seq_SET(finalbody, i, value);
4608 }
4609 Py_XDECREF(tmp);
4610 tmp = NULL;
4611 } else {
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004612 PyErr_SetString(PyExc_TypeError, "required field \"finalbody\" missing from Try");
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004613 return 1;
4614 }
Benjamin Peterson43af12b2011-05-29 11:43:10 -05004615 *out = Try(body, handlers, orelse, finalbody, lineno,
4616 col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004617 if (*out == NULL) goto failed;
4618 return 0;
4619 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004620 isinstance = PyObject_IsInstance(obj, (PyObject*)Assert_type);
4621 if (isinstance == -1) {
4622 return 1;
4623 }
4624 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004625 expr_ty test;
4626 expr_ty msg;
4627
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004628 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004629 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004630 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004631 if (tmp == NULL) goto failed;
4632 res = obj2ast_expr(tmp, &test, arena);
4633 if (res != 0) goto failed;
4634 Py_XDECREF(tmp);
4635 tmp = NULL;
4636 } else {
4637 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from Assert");
4638 return 1;
4639 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004640 if (_PyObject_HasAttrId(obj, &PyId_msg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004641 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004642 tmp = _PyObject_GetAttrId(obj, &PyId_msg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004643 if (tmp == NULL) goto failed;
4644 res = obj2ast_expr(tmp, &msg, arena);
4645 if (res != 0) goto failed;
4646 Py_XDECREF(tmp);
4647 tmp = NULL;
4648 } else {
4649 msg = NULL;
4650 }
4651 *out = Assert(test, msg, lineno, col_offset, arena);
4652 if (*out == NULL) goto failed;
4653 return 0;
4654 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004655 isinstance = PyObject_IsInstance(obj, (PyObject*)Import_type);
4656 if (isinstance == -1) {
4657 return 1;
4658 }
4659 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004660 asdl_seq* names;
4661
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004662 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004663 int res;
4664 Py_ssize_t len;
4665 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004666 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004667 if (tmp == NULL) goto failed;
4668 if (!PyList_Check(tmp)) {
4669 PyErr_Format(PyExc_TypeError, "Import field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4670 goto failed;
4671 }
4672 len = PyList_GET_SIZE(tmp);
4673 names = asdl_seq_new(len, arena);
4674 if (names == NULL) goto failed;
4675 for (i = 0; i < len; i++) {
4676 alias_ty value;
4677 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4678 if (res != 0) goto failed;
4679 asdl_seq_SET(names, i, value);
4680 }
4681 Py_XDECREF(tmp);
4682 tmp = NULL;
4683 } else {
4684 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Import");
4685 return 1;
4686 }
4687 *out = Import(names, lineno, col_offset, arena);
4688 if (*out == NULL) goto failed;
4689 return 0;
4690 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004691 isinstance = PyObject_IsInstance(obj, (PyObject*)ImportFrom_type);
4692 if (isinstance == -1) {
4693 return 1;
4694 }
4695 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004696 identifier module;
4697 asdl_seq* names;
4698 int level;
4699
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004700 if (_PyObject_HasAttrId(obj, &PyId_module)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004701 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004702 tmp = _PyObject_GetAttrId(obj, &PyId_module);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004703 if (tmp == NULL) goto failed;
4704 res = obj2ast_identifier(tmp, &module, arena);
4705 if (res != 0) goto failed;
4706 Py_XDECREF(tmp);
4707 tmp = NULL;
4708 } else {
Benjamin Peterson78565b22009-06-28 19:19:51 +00004709 module = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004710 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004711 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004712 int res;
4713 Py_ssize_t len;
4714 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004715 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004716 if (tmp == NULL) goto failed;
4717 if (!PyList_Check(tmp)) {
4718 PyErr_Format(PyExc_TypeError, "ImportFrom field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4719 goto failed;
4720 }
4721 len = PyList_GET_SIZE(tmp);
4722 names = asdl_seq_new(len, arena);
4723 if (names == NULL) goto failed;
4724 for (i = 0; i < len; i++) {
4725 alias_ty value;
4726 res = obj2ast_alias(PyList_GET_ITEM(tmp, i), &value, arena);
4727 if (res != 0) goto failed;
4728 asdl_seq_SET(names, i, value);
4729 }
4730 Py_XDECREF(tmp);
4731 tmp = NULL;
4732 } else {
4733 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from ImportFrom");
4734 return 1;
4735 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004736 if (_PyObject_HasAttrId(obj, &PyId_level)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004737 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004738 tmp = _PyObject_GetAttrId(obj, &PyId_level);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004739 if (tmp == NULL) goto failed;
4740 res = obj2ast_int(tmp, &level, arena);
4741 if (res != 0) goto failed;
4742 Py_XDECREF(tmp);
4743 tmp = NULL;
4744 } else {
4745 level = 0;
4746 }
4747 *out = ImportFrom(module, names, level, lineno, col_offset,
4748 arena);
4749 if (*out == NULL) goto failed;
4750 return 0;
4751 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004752 isinstance = PyObject_IsInstance(obj, (PyObject*)Global_type);
4753 if (isinstance == -1) {
4754 return 1;
4755 }
4756 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004757 asdl_seq* names;
4758
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004759 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004760 int res;
4761 Py_ssize_t len;
4762 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004763 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004764 if (tmp == NULL) goto failed;
4765 if (!PyList_Check(tmp)) {
4766 PyErr_Format(PyExc_TypeError, "Global field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4767 goto failed;
4768 }
4769 len = PyList_GET_SIZE(tmp);
4770 names = asdl_seq_new(len, arena);
4771 if (names == NULL) goto failed;
4772 for (i = 0; i < len; i++) {
4773 identifier value;
4774 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4775 if (res != 0) goto failed;
4776 asdl_seq_SET(names, i, value);
4777 }
4778 Py_XDECREF(tmp);
4779 tmp = NULL;
4780 } else {
4781 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Global");
4782 return 1;
4783 }
4784 *out = Global(names, lineno, col_offset, arena);
4785 if (*out == NULL) goto failed;
4786 return 0;
4787 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004788 isinstance = PyObject_IsInstance(obj, (PyObject*)Nonlocal_type);
4789 if (isinstance == -1) {
4790 return 1;
4791 }
4792 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004793 asdl_seq* names;
4794
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004795 if (_PyObject_HasAttrId(obj, &PyId_names)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004796 int res;
4797 Py_ssize_t len;
4798 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004799 tmp = _PyObject_GetAttrId(obj, &PyId_names);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004800 if (tmp == NULL) goto failed;
4801 if (!PyList_Check(tmp)) {
4802 PyErr_Format(PyExc_TypeError, "Nonlocal field \"names\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4803 goto failed;
4804 }
4805 len = PyList_GET_SIZE(tmp);
4806 names = asdl_seq_new(len, arena);
4807 if (names == NULL) goto failed;
4808 for (i = 0; i < len; i++) {
4809 identifier value;
4810 res = obj2ast_identifier(PyList_GET_ITEM(tmp, i), &value, arena);
4811 if (res != 0) goto failed;
4812 asdl_seq_SET(names, i, value);
4813 }
4814 Py_XDECREF(tmp);
4815 tmp = NULL;
4816 } else {
4817 PyErr_SetString(PyExc_TypeError, "required field \"names\" missing from Nonlocal");
4818 return 1;
4819 }
4820 *out = Nonlocal(names, lineno, col_offset, arena);
4821 if (*out == NULL) goto failed;
4822 return 0;
4823 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004824 isinstance = PyObject_IsInstance(obj, (PyObject*)Expr_type);
4825 if (isinstance == -1) {
4826 return 1;
4827 }
4828 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004829 expr_ty value;
4830
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004831 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004832 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004833 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004834 if (tmp == NULL) goto failed;
4835 res = obj2ast_expr(tmp, &value, arena);
4836 if (res != 0) goto failed;
4837 Py_XDECREF(tmp);
4838 tmp = NULL;
4839 } else {
4840 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Expr");
4841 return 1;
4842 }
4843 *out = Expr(value, lineno, col_offset, arena);
4844 if (*out == NULL) goto failed;
4845 return 0;
4846 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004847 isinstance = PyObject_IsInstance(obj, (PyObject*)Pass_type);
4848 if (isinstance == -1) {
4849 return 1;
4850 }
4851 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004852
4853 *out = Pass(lineno, col_offset, arena);
4854 if (*out == NULL) goto failed;
4855 return 0;
4856 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004857 isinstance = PyObject_IsInstance(obj, (PyObject*)Break_type);
4858 if (isinstance == -1) {
4859 return 1;
4860 }
4861 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004862
4863 *out = Break(lineno, col_offset, arena);
4864 if (*out == NULL) goto failed;
4865 return 0;
4866 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004867 isinstance = PyObject_IsInstance(obj, (PyObject*)Continue_type);
4868 if (isinstance == -1) {
4869 return 1;
4870 }
4871 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004872
4873 *out = Continue(lineno, col_offset, arena);
4874 if (*out == NULL) goto failed;
4875 return 0;
4876 }
4877
Benjamin Peterson5b066812010-11-20 01:38:49 +00004878 PyErr_Format(PyExc_TypeError, "expected some sort of stmt, but got %R", obj);
4879 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00004880 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004881 return 1;
4882}
4883
4884int
4885obj2ast_expr(PyObject* obj, expr_ty* out, PyArena* arena)
4886{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004887 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004888
Benjamin Petersond8f65972010-11-20 04:31:07 +00004889 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004890 int lineno;
4891 int col_offset;
4892
4893 if (obj == Py_None) {
4894 *out = NULL;
4895 return 0;
4896 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004897 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004898 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004899 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004900 if (tmp == NULL) goto failed;
4901 res = obj2ast_int(tmp, &lineno, arena);
4902 if (res != 0) goto failed;
4903 Py_XDECREF(tmp);
4904 tmp = NULL;
4905 } else {
4906 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from expr");
4907 return 1;
4908 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004909 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004910 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004911 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004912 if (tmp == NULL) goto failed;
4913 res = obj2ast_int(tmp, &col_offset, arena);
4914 if (res != 0) goto failed;
4915 Py_XDECREF(tmp);
4916 tmp = NULL;
4917 } else {
4918 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from expr");
4919 return 1;
4920 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004921 isinstance = PyObject_IsInstance(obj, (PyObject*)BoolOp_type);
4922 if (isinstance == -1) {
4923 return 1;
4924 }
4925 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004926 boolop_ty op;
4927 asdl_seq* values;
4928
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004929 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004930 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004931 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004932 if (tmp == NULL) goto failed;
4933 res = obj2ast_boolop(tmp, &op, arena);
4934 if (res != 0) goto failed;
4935 Py_XDECREF(tmp);
4936 tmp = NULL;
4937 } else {
4938 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BoolOp");
4939 return 1;
4940 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004941 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004942 int res;
4943 Py_ssize_t len;
4944 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004945 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004946 if (tmp == NULL) goto failed;
4947 if (!PyList_Check(tmp)) {
4948 PyErr_Format(PyExc_TypeError, "BoolOp field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
4949 goto failed;
4950 }
4951 len = PyList_GET_SIZE(tmp);
4952 values = asdl_seq_new(len, arena);
4953 if (values == NULL) goto failed;
4954 for (i = 0; i < len; i++) {
4955 expr_ty value;
4956 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
4957 if (res != 0) goto failed;
4958 asdl_seq_SET(values, i, value);
4959 }
4960 Py_XDECREF(tmp);
4961 tmp = NULL;
4962 } else {
4963 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from BoolOp");
4964 return 1;
4965 }
4966 *out = BoolOp(op, values, lineno, col_offset, arena);
4967 if (*out == NULL) goto failed;
4968 return 0;
4969 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00004970 isinstance = PyObject_IsInstance(obj, (PyObject*)BinOp_type);
4971 if (isinstance == -1) {
4972 return 1;
4973 }
4974 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004975 expr_ty left;
4976 operator_ty op;
4977 expr_ty right;
4978
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004979 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004980 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004981 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004982 if (tmp == NULL) goto failed;
4983 res = obj2ast_expr(tmp, &left, arena);
4984 if (res != 0) goto failed;
4985 Py_XDECREF(tmp);
4986 tmp = NULL;
4987 } else {
4988 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from BinOp");
4989 return 1;
4990 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004991 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004992 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02004993 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00004994 if (tmp == NULL) goto failed;
4995 res = obj2ast_operator(tmp, &op, arena);
4996 if (res != 0) goto failed;
4997 Py_XDECREF(tmp);
4998 tmp = NULL;
4999 } else {
5000 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from BinOp");
5001 return 1;
5002 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005003 if (_PyObject_HasAttrId(obj, &PyId_right)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005004 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005005 tmp = _PyObject_GetAttrId(obj, &PyId_right);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005006 if (tmp == NULL) goto failed;
5007 res = obj2ast_expr(tmp, &right, arena);
5008 if (res != 0) goto failed;
5009 Py_XDECREF(tmp);
5010 tmp = NULL;
5011 } else {
5012 PyErr_SetString(PyExc_TypeError, "required field \"right\" missing from BinOp");
5013 return 1;
5014 }
5015 *out = BinOp(left, op, right, lineno, col_offset, arena);
5016 if (*out == NULL) goto failed;
5017 return 0;
5018 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005019 isinstance = PyObject_IsInstance(obj, (PyObject*)UnaryOp_type);
5020 if (isinstance == -1) {
5021 return 1;
5022 }
5023 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005024 unaryop_ty op;
5025 expr_ty operand;
5026
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005027 if (_PyObject_HasAttrId(obj, &PyId_op)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005028 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005029 tmp = _PyObject_GetAttrId(obj, &PyId_op);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005030 if (tmp == NULL) goto failed;
5031 res = obj2ast_unaryop(tmp, &op, arena);
5032 if (res != 0) goto failed;
5033 Py_XDECREF(tmp);
5034 tmp = NULL;
5035 } else {
5036 PyErr_SetString(PyExc_TypeError, "required field \"op\" missing from UnaryOp");
5037 return 1;
5038 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005039 if (_PyObject_HasAttrId(obj, &PyId_operand)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005040 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005041 tmp = _PyObject_GetAttrId(obj, &PyId_operand);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005042 if (tmp == NULL) goto failed;
5043 res = obj2ast_expr(tmp, &operand, arena);
5044 if (res != 0) goto failed;
5045 Py_XDECREF(tmp);
5046 tmp = NULL;
5047 } else {
5048 PyErr_SetString(PyExc_TypeError, "required field \"operand\" missing from UnaryOp");
5049 return 1;
5050 }
5051 *out = UnaryOp(op, operand, lineno, col_offset, arena);
5052 if (*out == NULL) goto failed;
5053 return 0;
5054 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005055 isinstance = PyObject_IsInstance(obj, (PyObject*)Lambda_type);
5056 if (isinstance == -1) {
5057 return 1;
5058 }
5059 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005060 arguments_ty args;
5061 expr_ty body;
5062
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005063 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005064 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005065 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005066 if (tmp == NULL) goto failed;
5067 res = obj2ast_arguments(tmp, &args, arena);
5068 if (res != 0) goto failed;
5069 Py_XDECREF(tmp);
5070 tmp = NULL;
5071 } else {
5072 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Lambda");
5073 return 1;
5074 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005075 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005076 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005077 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005078 if (tmp == NULL) goto failed;
5079 res = obj2ast_expr(tmp, &body, arena);
5080 if (res != 0) goto failed;
5081 Py_XDECREF(tmp);
5082 tmp = NULL;
5083 } else {
5084 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from Lambda");
5085 return 1;
5086 }
5087 *out = Lambda(args, body, lineno, col_offset, arena);
5088 if (*out == NULL) goto failed;
5089 return 0;
5090 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005091 isinstance = PyObject_IsInstance(obj, (PyObject*)IfExp_type);
5092 if (isinstance == -1) {
5093 return 1;
5094 }
5095 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005096 expr_ty test;
5097 expr_ty body;
5098 expr_ty orelse;
5099
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005100 if (_PyObject_HasAttrId(obj, &PyId_test)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005101 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005102 tmp = _PyObject_GetAttrId(obj, &PyId_test);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005103 if (tmp == NULL) goto failed;
5104 res = obj2ast_expr(tmp, &test, arena);
5105 if (res != 0) goto failed;
5106 Py_XDECREF(tmp);
5107 tmp = NULL;
5108 } else {
5109 PyErr_SetString(PyExc_TypeError, "required field \"test\" missing from IfExp");
5110 return 1;
5111 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005112 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005113 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005114 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005115 if (tmp == NULL) goto failed;
5116 res = obj2ast_expr(tmp, &body, arena);
5117 if (res != 0) goto failed;
5118 Py_XDECREF(tmp);
5119 tmp = NULL;
5120 } else {
5121 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from IfExp");
5122 return 1;
5123 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005124 if (_PyObject_HasAttrId(obj, &PyId_orelse)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005125 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005126 tmp = _PyObject_GetAttrId(obj, &PyId_orelse);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005127 if (tmp == NULL) goto failed;
5128 res = obj2ast_expr(tmp, &orelse, arena);
5129 if (res != 0) goto failed;
5130 Py_XDECREF(tmp);
5131 tmp = NULL;
5132 } else {
5133 PyErr_SetString(PyExc_TypeError, "required field \"orelse\" missing from IfExp");
5134 return 1;
5135 }
5136 *out = IfExp(test, body, orelse, lineno, col_offset, arena);
5137 if (*out == NULL) goto failed;
5138 return 0;
5139 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005140 isinstance = PyObject_IsInstance(obj, (PyObject*)Dict_type);
5141 if (isinstance == -1) {
5142 return 1;
5143 }
5144 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005145 asdl_seq* keys;
5146 asdl_seq* values;
5147
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005148 if (_PyObject_HasAttrId(obj, &PyId_keys)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005149 int res;
5150 Py_ssize_t len;
5151 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005152 tmp = _PyObject_GetAttrId(obj, &PyId_keys);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005153 if (tmp == NULL) goto failed;
5154 if (!PyList_Check(tmp)) {
5155 PyErr_Format(PyExc_TypeError, "Dict field \"keys\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5156 goto failed;
5157 }
5158 len = PyList_GET_SIZE(tmp);
5159 keys = asdl_seq_new(len, arena);
5160 if (keys == NULL) goto failed;
5161 for (i = 0; i < len; i++) {
5162 expr_ty value;
5163 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5164 if (res != 0) goto failed;
5165 asdl_seq_SET(keys, i, value);
5166 }
5167 Py_XDECREF(tmp);
5168 tmp = NULL;
5169 } else {
5170 PyErr_SetString(PyExc_TypeError, "required field \"keys\" missing from Dict");
5171 return 1;
5172 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005173 if (_PyObject_HasAttrId(obj, &PyId_values)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005174 int res;
5175 Py_ssize_t len;
5176 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005177 tmp = _PyObject_GetAttrId(obj, &PyId_values);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005178 if (tmp == NULL) goto failed;
5179 if (!PyList_Check(tmp)) {
5180 PyErr_Format(PyExc_TypeError, "Dict field \"values\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5181 goto failed;
5182 }
5183 len = PyList_GET_SIZE(tmp);
5184 values = asdl_seq_new(len, arena);
5185 if (values == NULL) goto failed;
5186 for (i = 0; i < len; i++) {
5187 expr_ty value;
5188 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5189 if (res != 0) goto failed;
5190 asdl_seq_SET(values, i, value);
5191 }
5192 Py_XDECREF(tmp);
5193 tmp = NULL;
5194 } else {
5195 PyErr_SetString(PyExc_TypeError, "required field \"values\" missing from Dict");
5196 return 1;
5197 }
5198 *out = Dict(keys, values, lineno, col_offset, arena);
5199 if (*out == NULL) goto failed;
5200 return 0;
5201 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005202 isinstance = PyObject_IsInstance(obj, (PyObject*)Set_type);
5203 if (isinstance == -1) {
5204 return 1;
5205 }
5206 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005207 asdl_seq* elts;
5208
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005209 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005210 int res;
5211 Py_ssize_t len;
5212 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005213 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005214 if (tmp == NULL) goto failed;
5215 if (!PyList_Check(tmp)) {
5216 PyErr_Format(PyExc_TypeError, "Set field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5217 goto failed;
5218 }
5219 len = PyList_GET_SIZE(tmp);
5220 elts = asdl_seq_new(len, arena);
5221 if (elts == NULL) goto failed;
5222 for (i = 0; i < len; i++) {
5223 expr_ty value;
5224 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5225 if (res != 0) goto failed;
5226 asdl_seq_SET(elts, i, value);
5227 }
5228 Py_XDECREF(tmp);
5229 tmp = NULL;
5230 } else {
5231 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Set");
5232 return 1;
5233 }
5234 *out = Set(elts, lineno, col_offset, arena);
5235 if (*out == NULL) goto failed;
5236 return 0;
5237 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005238 isinstance = PyObject_IsInstance(obj, (PyObject*)ListComp_type);
5239 if (isinstance == -1) {
5240 return 1;
5241 }
5242 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005243 expr_ty elt;
5244 asdl_seq* generators;
5245
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005246 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005247 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005248 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005249 if (tmp == NULL) goto failed;
5250 res = obj2ast_expr(tmp, &elt, arena);
5251 if (res != 0) goto failed;
5252 Py_XDECREF(tmp);
5253 tmp = NULL;
5254 } else {
5255 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from ListComp");
5256 return 1;
5257 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005258 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005259 int res;
5260 Py_ssize_t len;
5261 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005262 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005263 if (tmp == NULL) goto failed;
5264 if (!PyList_Check(tmp)) {
5265 PyErr_Format(PyExc_TypeError, "ListComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5266 goto failed;
5267 }
5268 len = PyList_GET_SIZE(tmp);
5269 generators = asdl_seq_new(len, arena);
5270 if (generators == NULL) goto failed;
5271 for (i = 0; i < len; i++) {
5272 comprehension_ty value;
5273 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5274 if (res != 0) goto failed;
5275 asdl_seq_SET(generators, i, value);
5276 }
5277 Py_XDECREF(tmp);
5278 tmp = NULL;
5279 } else {
5280 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from ListComp");
5281 return 1;
5282 }
5283 *out = ListComp(elt, generators, lineno, col_offset, arena);
5284 if (*out == NULL) goto failed;
5285 return 0;
5286 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005287 isinstance = PyObject_IsInstance(obj, (PyObject*)SetComp_type);
5288 if (isinstance == -1) {
5289 return 1;
5290 }
5291 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005292 expr_ty elt;
5293 asdl_seq* generators;
5294
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005295 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005296 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005297 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005298 if (tmp == NULL) goto failed;
5299 res = obj2ast_expr(tmp, &elt, arena);
5300 if (res != 0) goto failed;
5301 Py_XDECREF(tmp);
5302 tmp = NULL;
5303 } else {
5304 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from SetComp");
5305 return 1;
5306 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005307 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005308 int res;
5309 Py_ssize_t len;
5310 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005311 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005312 if (tmp == NULL) goto failed;
5313 if (!PyList_Check(tmp)) {
5314 PyErr_Format(PyExc_TypeError, "SetComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5315 goto failed;
5316 }
5317 len = PyList_GET_SIZE(tmp);
5318 generators = asdl_seq_new(len, arena);
5319 if (generators == NULL) goto failed;
5320 for (i = 0; i < len; i++) {
5321 comprehension_ty value;
5322 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5323 if (res != 0) goto failed;
5324 asdl_seq_SET(generators, i, value);
5325 }
5326 Py_XDECREF(tmp);
5327 tmp = NULL;
5328 } else {
5329 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from SetComp");
5330 return 1;
5331 }
5332 *out = SetComp(elt, generators, lineno, col_offset, arena);
5333 if (*out == NULL) goto failed;
5334 return 0;
5335 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005336 isinstance = PyObject_IsInstance(obj, (PyObject*)DictComp_type);
5337 if (isinstance == -1) {
5338 return 1;
5339 }
5340 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005341 expr_ty key;
5342 expr_ty value;
5343 asdl_seq* generators;
5344
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005345 if (_PyObject_HasAttrId(obj, &PyId_key)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005346 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005347 tmp = _PyObject_GetAttrId(obj, &PyId_key);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005348 if (tmp == NULL) goto failed;
5349 res = obj2ast_expr(tmp, &key, arena);
5350 if (res != 0) goto failed;
5351 Py_XDECREF(tmp);
5352 tmp = NULL;
5353 } else {
5354 PyErr_SetString(PyExc_TypeError, "required field \"key\" missing from DictComp");
5355 return 1;
5356 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005357 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005358 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005359 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005360 if (tmp == NULL) goto failed;
5361 res = obj2ast_expr(tmp, &value, arena);
5362 if (res != 0) goto failed;
5363 Py_XDECREF(tmp);
5364 tmp = NULL;
5365 } else {
5366 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from DictComp");
5367 return 1;
5368 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005369 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005370 int res;
5371 Py_ssize_t len;
5372 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005373 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005374 if (tmp == NULL) goto failed;
5375 if (!PyList_Check(tmp)) {
5376 PyErr_Format(PyExc_TypeError, "DictComp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5377 goto failed;
5378 }
5379 len = PyList_GET_SIZE(tmp);
5380 generators = asdl_seq_new(len, arena);
5381 if (generators == NULL) goto failed;
5382 for (i = 0; i < len; i++) {
5383 comprehension_ty value;
5384 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5385 if (res != 0) goto failed;
5386 asdl_seq_SET(generators, i, value);
5387 }
5388 Py_XDECREF(tmp);
5389 tmp = NULL;
5390 } else {
5391 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from DictComp");
5392 return 1;
5393 }
5394 *out = DictComp(key, value, generators, lineno, col_offset,
5395 arena);
5396 if (*out == NULL) goto failed;
5397 return 0;
5398 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005399 isinstance = PyObject_IsInstance(obj, (PyObject*)GeneratorExp_type);
5400 if (isinstance == -1) {
5401 return 1;
5402 }
5403 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005404 expr_ty elt;
5405 asdl_seq* generators;
5406
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005407 if (_PyObject_HasAttrId(obj, &PyId_elt)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005408 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005409 tmp = _PyObject_GetAttrId(obj, &PyId_elt);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005410 if (tmp == NULL) goto failed;
5411 res = obj2ast_expr(tmp, &elt, arena);
5412 if (res != 0) goto failed;
5413 Py_XDECREF(tmp);
5414 tmp = NULL;
5415 } else {
5416 PyErr_SetString(PyExc_TypeError, "required field \"elt\" missing from GeneratorExp");
5417 return 1;
5418 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005419 if (_PyObject_HasAttrId(obj, &PyId_generators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005420 int res;
5421 Py_ssize_t len;
5422 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005423 tmp = _PyObject_GetAttrId(obj, &PyId_generators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005424 if (tmp == NULL) goto failed;
5425 if (!PyList_Check(tmp)) {
5426 PyErr_Format(PyExc_TypeError, "GeneratorExp field \"generators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5427 goto failed;
5428 }
5429 len = PyList_GET_SIZE(tmp);
5430 generators = asdl_seq_new(len, arena);
5431 if (generators == NULL) goto failed;
5432 for (i = 0; i < len; i++) {
5433 comprehension_ty value;
5434 res = obj2ast_comprehension(PyList_GET_ITEM(tmp, i), &value, arena);
5435 if (res != 0) goto failed;
5436 asdl_seq_SET(generators, i, value);
5437 }
5438 Py_XDECREF(tmp);
5439 tmp = NULL;
5440 } else {
5441 PyErr_SetString(PyExc_TypeError, "required field \"generators\" missing from GeneratorExp");
5442 return 1;
5443 }
5444 *out = GeneratorExp(elt, generators, lineno, col_offset, arena);
5445 if (*out == NULL) goto failed;
5446 return 0;
5447 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005448 isinstance = PyObject_IsInstance(obj, (PyObject*)Yield_type);
5449 if (isinstance == -1) {
5450 return 1;
5451 }
5452 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005453 expr_ty value;
5454
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005455 if (_PyObject_HasAttrId(obj, &PyId_value)) {
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_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005458 if (tmp == NULL) goto failed;
5459 res = obj2ast_expr(tmp, &value, arena);
5460 if (res != 0) goto failed;
5461 Py_XDECREF(tmp);
5462 tmp = NULL;
5463 } else {
5464 value = NULL;
5465 }
Benjamin Peterson527c6222012-01-14 08:58:23 -05005466 *out = Yield(value, lineno, col_offset, arena);
5467 if (*out == NULL) goto failed;
5468 return 0;
5469 }
5470 isinstance = PyObject_IsInstance(obj, (PyObject*)YieldFrom_type);
5471 if (isinstance == -1) {
5472 return 1;
5473 }
5474 if (isinstance) {
5475 expr_ty value;
5476
5477 if (_PyObject_HasAttrId(obj, &PyId_value)) {
5478 int res;
5479 tmp = _PyObject_GetAttrId(obj, &PyId_value);
5480 if (tmp == NULL) goto failed;
5481 res = obj2ast_expr(tmp, &value, arena);
5482 if (res != 0) goto failed;
5483 Py_XDECREF(tmp);
5484 tmp = NULL;
5485 } else {
Mark Dickinsonded35ae2012-11-25 14:36:26 +00005486 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from YieldFrom");
5487 return 1;
Benjamin Peterson527c6222012-01-14 08:58:23 -05005488 }
5489 *out = YieldFrom(value, lineno, col_offset, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005490 if (*out == NULL) goto failed;
5491 return 0;
5492 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005493 isinstance = PyObject_IsInstance(obj, (PyObject*)Compare_type);
5494 if (isinstance == -1) {
5495 return 1;
5496 }
5497 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005498 expr_ty left;
5499 asdl_int_seq* ops;
5500 asdl_seq* comparators;
5501
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005502 if (_PyObject_HasAttrId(obj, &PyId_left)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005503 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005504 tmp = _PyObject_GetAttrId(obj, &PyId_left);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005505 if (tmp == NULL) goto failed;
5506 res = obj2ast_expr(tmp, &left, arena);
5507 if (res != 0) goto failed;
5508 Py_XDECREF(tmp);
5509 tmp = NULL;
5510 } else {
5511 PyErr_SetString(PyExc_TypeError, "required field \"left\" missing from Compare");
5512 return 1;
5513 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005514 if (_PyObject_HasAttrId(obj, &PyId_ops)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005515 int res;
5516 Py_ssize_t len;
5517 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005518 tmp = _PyObject_GetAttrId(obj, &PyId_ops);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005519 if (tmp == NULL) goto failed;
5520 if (!PyList_Check(tmp)) {
5521 PyErr_Format(PyExc_TypeError, "Compare field \"ops\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5522 goto failed;
5523 }
5524 len = PyList_GET_SIZE(tmp);
5525 ops = asdl_int_seq_new(len, arena);
5526 if (ops == NULL) goto failed;
5527 for (i = 0; i < len; i++) {
5528 cmpop_ty value;
5529 res = obj2ast_cmpop(PyList_GET_ITEM(tmp, i), &value, arena);
5530 if (res != 0) goto failed;
5531 asdl_seq_SET(ops, i, value);
5532 }
5533 Py_XDECREF(tmp);
5534 tmp = NULL;
5535 } else {
5536 PyErr_SetString(PyExc_TypeError, "required field \"ops\" missing from Compare");
5537 return 1;
5538 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005539 if (_PyObject_HasAttrId(obj, &PyId_comparators)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005540 int res;
5541 Py_ssize_t len;
5542 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005543 tmp = _PyObject_GetAttrId(obj, &PyId_comparators);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005544 if (tmp == NULL) goto failed;
5545 if (!PyList_Check(tmp)) {
5546 PyErr_Format(PyExc_TypeError, "Compare field \"comparators\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5547 goto failed;
5548 }
5549 len = PyList_GET_SIZE(tmp);
5550 comparators = asdl_seq_new(len, arena);
5551 if (comparators == NULL) goto failed;
5552 for (i = 0; i < len; i++) {
5553 expr_ty value;
5554 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5555 if (res != 0) goto failed;
5556 asdl_seq_SET(comparators, i, value);
5557 }
5558 Py_XDECREF(tmp);
5559 tmp = NULL;
5560 } else {
5561 PyErr_SetString(PyExc_TypeError, "required field \"comparators\" missing from Compare");
5562 return 1;
5563 }
5564 *out = Compare(left, ops, comparators, lineno, col_offset,
5565 arena);
5566 if (*out == NULL) goto failed;
5567 return 0;
5568 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005569 isinstance = PyObject_IsInstance(obj, (PyObject*)Call_type);
5570 if (isinstance == -1) {
5571 return 1;
5572 }
5573 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005574 expr_ty func;
5575 asdl_seq* args;
5576 asdl_seq* keywords;
5577 expr_ty starargs;
5578 expr_ty kwargs;
5579
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005580 if (_PyObject_HasAttrId(obj, &PyId_func)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005581 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005582 tmp = _PyObject_GetAttrId(obj, &PyId_func);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005583 if (tmp == NULL) goto failed;
5584 res = obj2ast_expr(tmp, &func, arena);
5585 if (res != 0) goto failed;
5586 Py_XDECREF(tmp);
5587 tmp = NULL;
5588 } else {
5589 PyErr_SetString(PyExc_TypeError, "required field \"func\" missing from Call");
5590 return 1;
5591 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005592 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005593 int res;
5594 Py_ssize_t len;
5595 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005596 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005597 if (tmp == NULL) goto failed;
5598 if (!PyList_Check(tmp)) {
5599 PyErr_Format(PyExc_TypeError, "Call field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5600 goto failed;
5601 }
5602 len = PyList_GET_SIZE(tmp);
5603 args = asdl_seq_new(len, arena);
5604 if (args == NULL) goto failed;
5605 for (i = 0; i < len; i++) {
5606 expr_ty value;
5607 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5608 if (res != 0) goto failed;
5609 asdl_seq_SET(args, i, value);
5610 }
5611 Py_XDECREF(tmp);
5612 tmp = NULL;
5613 } else {
5614 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from Call");
5615 return 1;
5616 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005617 if (_PyObject_HasAttrId(obj, &PyId_keywords)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005618 int res;
5619 Py_ssize_t len;
5620 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005621 tmp = _PyObject_GetAttrId(obj, &PyId_keywords);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005622 if (tmp == NULL) goto failed;
5623 if (!PyList_Check(tmp)) {
5624 PyErr_Format(PyExc_TypeError, "Call field \"keywords\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5625 goto failed;
5626 }
5627 len = PyList_GET_SIZE(tmp);
5628 keywords = asdl_seq_new(len, arena);
5629 if (keywords == NULL) goto failed;
5630 for (i = 0; i < len; i++) {
5631 keyword_ty value;
5632 res = obj2ast_keyword(PyList_GET_ITEM(tmp, i), &value, arena);
5633 if (res != 0) goto failed;
5634 asdl_seq_SET(keywords, i, value);
5635 }
5636 Py_XDECREF(tmp);
5637 tmp = NULL;
5638 } else {
5639 PyErr_SetString(PyExc_TypeError, "required field \"keywords\" missing from Call");
5640 return 1;
5641 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005642 if (_PyObject_HasAttrId(obj, &PyId_starargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005643 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005644 tmp = _PyObject_GetAttrId(obj, &PyId_starargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005645 if (tmp == NULL) goto failed;
5646 res = obj2ast_expr(tmp, &starargs, arena);
5647 if (res != 0) goto failed;
5648 Py_XDECREF(tmp);
5649 tmp = NULL;
5650 } else {
5651 starargs = NULL;
5652 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005653 if (_PyObject_HasAttrId(obj, &PyId_kwargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005654 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005655 tmp = _PyObject_GetAttrId(obj, &PyId_kwargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005656 if (tmp == NULL) goto failed;
5657 res = obj2ast_expr(tmp, &kwargs, arena);
5658 if (res != 0) goto failed;
5659 Py_XDECREF(tmp);
5660 tmp = NULL;
5661 } else {
5662 kwargs = NULL;
5663 }
5664 *out = Call(func, args, keywords, starargs, kwargs, lineno,
5665 col_offset, arena);
5666 if (*out == NULL) goto failed;
5667 return 0;
5668 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005669 isinstance = PyObject_IsInstance(obj, (PyObject*)Num_type);
5670 if (isinstance == -1) {
5671 return 1;
5672 }
5673 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005674 object n;
5675
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005676 if (_PyObject_HasAttrId(obj, &PyId_n)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005677 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005678 tmp = _PyObject_GetAttrId(obj, &PyId_n);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005679 if (tmp == NULL) goto failed;
5680 res = obj2ast_object(tmp, &n, arena);
5681 if (res != 0) goto failed;
5682 Py_XDECREF(tmp);
5683 tmp = NULL;
5684 } else {
5685 PyErr_SetString(PyExc_TypeError, "required field \"n\" missing from Num");
5686 return 1;
5687 }
5688 *out = Num(n, lineno, col_offset, arena);
5689 if (*out == NULL) goto failed;
5690 return 0;
5691 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005692 isinstance = PyObject_IsInstance(obj, (PyObject*)Str_type);
5693 if (isinstance == -1) {
5694 return 1;
5695 }
5696 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005697 string s;
5698
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005699 if (_PyObject_HasAttrId(obj, &PyId_s)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005700 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005701 tmp = _PyObject_GetAttrId(obj, &PyId_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005702 if (tmp == NULL) goto failed;
5703 res = obj2ast_string(tmp, &s, arena);
5704 if (res != 0) goto failed;
5705 Py_XDECREF(tmp);
5706 tmp = NULL;
5707 } else {
5708 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Str");
5709 return 1;
5710 }
5711 *out = Str(s, lineno, col_offset, arena);
5712 if (*out == NULL) goto failed;
5713 return 0;
5714 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005715 isinstance = PyObject_IsInstance(obj, (PyObject*)Bytes_type);
5716 if (isinstance == -1) {
5717 return 1;
5718 }
5719 if (isinstance) {
Benjamin Petersone2498412011-08-09 16:08:39 -05005720 bytes s;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005721
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005722 if (_PyObject_HasAttrId(obj, &PyId_s)) {
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_s);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005725 if (tmp == NULL) goto failed;
Benjamin Petersone2498412011-08-09 16:08:39 -05005726 res = obj2ast_bytes(tmp, &s, arena);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005727 if (res != 0) goto failed;
5728 Py_XDECREF(tmp);
5729 tmp = NULL;
5730 } else {
5731 PyErr_SetString(PyExc_TypeError, "required field \"s\" missing from Bytes");
5732 return 1;
5733 }
5734 *out = Bytes(s, lineno, col_offset, arena);
5735 if (*out == NULL) goto failed;
5736 return 0;
5737 }
Benjamin Peterson442f2092012-12-06 17:41:04 -05005738 isinstance = PyObject_IsInstance(obj, (PyObject*)NameConstant_type);
5739 if (isinstance == -1) {
5740 return 1;
5741 }
5742 if (isinstance) {
5743 singleton value;
5744
5745 if (_PyObject_HasAttrId(obj, &PyId_value)) {
5746 int res;
5747 tmp = _PyObject_GetAttrId(obj, &PyId_value);
5748 if (tmp == NULL) goto failed;
5749 res = obj2ast_singleton(tmp, &value, arena);
5750 if (res != 0) goto failed;
5751 Py_XDECREF(tmp);
5752 tmp = NULL;
5753 } else {
5754 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from NameConstant");
5755 return 1;
5756 }
5757 *out = NameConstant(value, lineno, col_offset, arena);
5758 if (*out == NULL) goto failed;
5759 return 0;
5760 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005761 isinstance = PyObject_IsInstance(obj, (PyObject*)Ellipsis_type);
5762 if (isinstance == -1) {
5763 return 1;
5764 }
5765 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005766
5767 *out = Ellipsis(lineno, col_offset, arena);
5768 if (*out == NULL) goto failed;
5769 return 0;
5770 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005771 isinstance = PyObject_IsInstance(obj, (PyObject*)Attribute_type);
5772 if (isinstance == -1) {
5773 return 1;
5774 }
5775 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005776 expr_ty value;
5777 identifier attr;
5778 expr_context_ty ctx;
5779
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005780 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005781 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005782 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005783 if (tmp == NULL) goto failed;
5784 res = obj2ast_expr(tmp, &value, arena);
5785 if (res != 0) goto failed;
5786 Py_XDECREF(tmp);
5787 tmp = NULL;
5788 } else {
5789 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Attribute");
5790 return 1;
5791 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005792 if (_PyObject_HasAttrId(obj, &PyId_attr)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005793 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005794 tmp = _PyObject_GetAttrId(obj, &PyId_attr);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005795 if (tmp == NULL) goto failed;
5796 res = obj2ast_identifier(tmp, &attr, arena);
5797 if (res != 0) goto failed;
5798 Py_XDECREF(tmp);
5799 tmp = NULL;
5800 } else {
5801 PyErr_SetString(PyExc_TypeError, "required field \"attr\" missing from Attribute");
5802 return 1;
5803 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005804 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005805 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005806 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005807 if (tmp == NULL) goto failed;
5808 res = obj2ast_expr_context(tmp, &ctx, arena);
5809 if (res != 0) goto failed;
5810 Py_XDECREF(tmp);
5811 tmp = NULL;
5812 } else {
5813 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Attribute");
5814 return 1;
5815 }
5816 *out = Attribute(value, attr, ctx, lineno, col_offset, arena);
5817 if (*out == NULL) goto failed;
5818 return 0;
5819 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005820 isinstance = PyObject_IsInstance(obj, (PyObject*)Subscript_type);
5821 if (isinstance == -1) {
5822 return 1;
5823 }
5824 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005825 expr_ty value;
5826 slice_ty slice;
5827 expr_context_ty ctx;
5828
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005829 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005830 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005831 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005832 if (tmp == NULL) goto failed;
5833 res = obj2ast_expr(tmp, &value, arena);
5834 if (res != 0) goto failed;
5835 Py_XDECREF(tmp);
5836 tmp = NULL;
5837 } else {
5838 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Subscript");
5839 return 1;
5840 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005841 if (_PyObject_HasAttrId(obj, &PyId_slice)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005842 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005843 tmp = _PyObject_GetAttrId(obj, &PyId_slice);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005844 if (tmp == NULL) goto failed;
5845 res = obj2ast_slice(tmp, &slice, arena);
5846 if (res != 0) goto failed;
5847 Py_XDECREF(tmp);
5848 tmp = NULL;
5849 } else {
5850 PyErr_SetString(PyExc_TypeError, "required field \"slice\" missing from Subscript");
5851 return 1;
5852 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005853 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005854 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005855 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005856 if (tmp == NULL) goto failed;
5857 res = obj2ast_expr_context(tmp, &ctx, arena);
5858 if (res != 0) goto failed;
5859 Py_XDECREF(tmp);
5860 tmp = NULL;
5861 } else {
5862 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Subscript");
5863 return 1;
5864 }
5865 *out = Subscript(value, slice, ctx, lineno, col_offset, arena);
5866 if (*out == NULL) goto failed;
5867 return 0;
5868 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005869 isinstance = PyObject_IsInstance(obj, (PyObject*)Starred_type);
5870 if (isinstance == -1) {
5871 return 1;
5872 }
5873 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005874 expr_ty value;
5875 expr_context_ty ctx;
5876
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005877 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005878 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005879 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005880 if (tmp == NULL) goto failed;
5881 res = obj2ast_expr(tmp, &value, arena);
5882 if (res != 0) goto failed;
5883 Py_XDECREF(tmp);
5884 tmp = NULL;
5885 } else {
5886 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Starred");
5887 return 1;
5888 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005889 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005890 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005891 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005892 if (tmp == NULL) goto failed;
5893 res = obj2ast_expr_context(tmp, &ctx, arena);
5894 if (res != 0) goto failed;
5895 Py_XDECREF(tmp);
5896 tmp = NULL;
5897 } else {
5898 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Starred");
5899 return 1;
5900 }
5901 *out = Starred(value, ctx, lineno, col_offset, arena);
5902 if (*out == NULL) goto failed;
5903 return 0;
5904 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005905 isinstance = PyObject_IsInstance(obj, (PyObject*)Name_type);
5906 if (isinstance == -1) {
5907 return 1;
5908 }
5909 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005910 identifier id;
5911 expr_context_ty ctx;
5912
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005913 if (_PyObject_HasAttrId(obj, &PyId_id)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005914 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005915 tmp = _PyObject_GetAttrId(obj, &PyId_id);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005916 if (tmp == NULL) goto failed;
5917 res = obj2ast_identifier(tmp, &id, arena);
5918 if (res != 0) goto failed;
5919 Py_XDECREF(tmp);
5920 tmp = NULL;
5921 } else {
5922 PyErr_SetString(PyExc_TypeError, "required field \"id\" missing from Name");
5923 return 1;
5924 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005925 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005926 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005927 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005928 if (tmp == NULL) goto failed;
5929 res = obj2ast_expr_context(tmp, &ctx, arena);
5930 if (res != 0) goto failed;
5931 Py_XDECREF(tmp);
5932 tmp = NULL;
5933 } else {
5934 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Name");
5935 return 1;
5936 }
5937 *out = Name(id, ctx, lineno, col_offset, arena);
5938 if (*out == NULL) goto failed;
5939 return 0;
5940 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005941 isinstance = PyObject_IsInstance(obj, (PyObject*)List_type);
5942 if (isinstance == -1) {
5943 return 1;
5944 }
5945 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005946 asdl_seq* elts;
5947 expr_context_ty ctx;
5948
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005949 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005950 int res;
5951 Py_ssize_t len;
5952 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005953 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005954 if (tmp == NULL) goto failed;
5955 if (!PyList_Check(tmp)) {
5956 PyErr_Format(PyExc_TypeError, "List field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
5957 goto failed;
5958 }
5959 len = PyList_GET_SIZE(tmp);
5960 elts = asdl_seq_new(len, arena);
5961 if (elts == NULL) goto failed;
5962 for (i = 0; i < len; i++) {
5963 expr_ty value;
5964 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
5965 if (res != 0) goto failed;
5966 asdl_seq_SET(elts, i, value);
5967 }
5968 Py_XDECREF(tmp);
5969 tmp = NULL;
5970 } else {
5971 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from List");
5972 return 1;
5973 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005974 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005975 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005976 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005977 if (tmp == NULL) goto failed;
5978 res = obj2ast_expr_context(tmp, &ctx, arena);
5979 if (res != 0) goto failed;
5980 Py_XDECREF(tmp);
5981 tmp = NULL;
5982 } else {
5983 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from List");
5984 return 1;
5985 }
5986 *out = List(elts, ctx, lineno, col_offset, arena);
5987 if (*out == NULL) goto failed;
5988 return 0;
5989 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00005990 isinstance = PyObject_IsInstance(obj, (PyObject*)Tuple_type);
5991 if (isinstance == -1) {
5992 return 1;
5993 }
5994 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005995 asdl_seq* elts;
5996 expr_context_ty ctx;
5997
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02005998 if (_PyObject_HasAttrId(obj, &PyId_elts)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00005999 int res;
6000 Py_ssize_t len;
6001 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006002 tmp = _PyObject_GetAttrId(obj, &PyId_elts);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006003 if (tmp == NULL) goto failed;
6004 if (!PyList_Check(tmp)) {
6005 PyErr_Format(PyExc_TypeError, "Tuple field \"elts\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6006 goto failed;
6007 }
6008 len = PyList_GET_SIZE(tmp);
6009 elts = asdl_seq_new(len, arena);
6010 if (elts == NULL) goto failed;
6011 for (i = 0; i < len; i++) {
6012 expr_ty value;
6013 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6014 if (res != 0) goto failed;
6015 asdl_seq_SET(elts, i, value);
6016 }
6017 Py_XDECREF(tmp);
6018 tmp = NULL;
6019 } else {
6020 PyErr_SetString(PyExc_TypeError, "required field \"elts\" missing from Tuple");
6021 return 1;
6022 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006023 if (_PyObject_HasAttrId(obj, &PyId_ctx)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006024 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006025 tmp = _PyObject_GetAttrId(obj, &PyId_ctx);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006026 if (tmp == NULL) goto failed;
6027 res = obj2ast_expr_context(tmp, &ctx, arena);
6028 if (res != 0) goto failed;
6029 Py_XDECREF(tmp);
6030 tmp = NULL;
6031 } else {
6032 PyErr_SetString(PyExc_TypeError, "required field \"ctx\" missing from Tuple");
6033 return 1;
6034 }
6035 *out = Tuple(elts, ctx, lineno, col_offset, arena);
6036 if (*out == NULL) goto failed;
6037 return 0;
6038 }
6039
Benjamin Peterson5b066812010-11-20 01:38:49 +00006040 PyErr_Format(PyExc_TypeError, "expected some sort of expr, but got %R", obj);
6041 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006042 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006043 return 1;
6044}
6045
6046int
6047obj2ast_expr_context(PyObject* obj, expr_context_ty* out, PyArena* arena)
6048{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006049 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006050
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006051 isinstance = PyObject_IsInstance(obj, (PyObject *)Load_type);
6052 if (isinstance == -1) {
6053 return 1;
6054 }
6055 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006056 *out = Load;
6057 return 0;
6058 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006059 isinstance = PyObject_IsInstance(obj, (PyObject *)Store_type);
6060 if (isinstance == -1) {
6061 return 1;
6062 }
6063 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006064 *out = Store;
6065 return 0;
6066 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006067 isinstance = PyObject_IsInstance(obj, (PyObject *)Del_type);
6068 if (isinstance == -1) {
6069 return 1;
6070 }
6071 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006072 *out = Del;
6073 return 0;
6074 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006075 isinstance = PyObject_IsInstance(obj, (PyObject *)AugLoad_type);
6076 if (isinstance == -1) {
6077 return 1;
6078 }
6079 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006080 *out = AugLoad;
6081 return 0;
6082 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006083 isinstance = PyObject_IsInstance(obj, (PyObject *)AugStore_type);
6084 if (isinstance == -1) {
6085 return 1;
6086 }
6087 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006088 *out = AugStore;
6089 return 0;
6090 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006091 isinstance = PyObject_IsInstance(obj, (PyObject *)Param_type);
6092 if (isinstance == -1) {
6093 return 1;
6094 }
6095 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006096 *out = Param;
6097 return 0;
6098 }
6099
Benjamin Peterson5b066812010-11-20 01:38:49 +00006100 PyErr_Format(PyExc_TypeError, "expected some sort of expr_context, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006101 return 1;
6102}
6103
6104int
6105obj2ast_slice(PyObject* obj, slice_ty* out, PyArena* arena)
6106{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006107 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006108
Benjamin Petersond8f65972010-11-20 04:31:07 +00006109 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006110
6111 if (obj == Py_None) {
6112 *out = NULL;
6113 return 0;
6114 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006115 isinstance = PyObject_IsInstance(obj, (PyObject*)Slice_type);
6116 if (isinstance == -1) {
6117 return 1;
6118 }
6119 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006120 expr_ty lower;
6121 expr_ty upper;
6122 expr_ty step;
6123
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006124 if (_PyObject_HasAttrId(obj, &PyId_lower)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006125 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006126 tmp = _PyObject_GetAttrId(obj, &PyId_lower);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006127 if (tmp == NULL) goto failed;
6128 res = obj2ast_expr(tmp, &lower, arena);
6129 if (res != 0) goto failed;
6130 Py_XDECREF(tmp);
6131 tmp = NULL;
6132 } else {
6133 lower = NULL;
6134 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006135 if (_PyObject_HasAttrId(obj, &PyId_upper)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006136 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006137 tmp = _PyObject_GetAttrId(obj, &PyId_upper);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006138 if (tmp == NULL) goto failed;
6139 res = obj2ast_expr(tmp, &upper, arena);
6140 if (res != 0) goto failed;
6141 Py_XDECREF(tmp);
6142 tmp = NULL;
6143 } else {
6144 upper = NULL;
6145 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006146 if (_PyObject_HasAttrId(obj, &PyId_step)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006147 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006148 tmp = _PyObject_GetAttrId(obj, &PyId_step);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006149 if (tmp == NULL) goto failed;
6150 res = obj2ast_expr(tmp, &step, arena);
6151 if (res != 0) goto failed;
6152 Py_XDECREF(tmp);
6153 tmp = NULL;
6154 } else {
6155 step = NULL;
6156 }
6157 *out = Slice(lower, upper, step, arena);
6158 if (*out == NULL) goto failed;
6159 return 0;
6160 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006161 isinstance = PyObject_IsInstance(obj, (PyObject*)ExtSlice_type);
6162 if (isinstance == -1) {
6163 return 1;
6164 }
6165 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006166 asdl_seq* dims;
6167
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006168 if (_PyObject_HasAttrId(obj, &PyId_dims)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006169 int res;
6170 Py_ssize_t len;
6171 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006172 tmp = _PyObject_GetAttrId(obj, &PyId_dims);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006173 if (tmp == NULL) goto failed;
6174 if (!PyList_Check(tmp)) {
6175 PyErr_Format(PyExc_TypeError, "ExtSlice field \"dims\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6176 goto failed;
6177 }
6178 len = PyList_GET_SIZE(tmp);
6179 dims = asdl_seq_new(len, arena);
6180 if (dims == NULL) goto failed;
6181 for (i = 0; i < len; i++) {
6182 slice_ty value;
6183 res = obj2ast_slice(PyList_GET_ITEM(tmp, i), &value, arena);
6184 if (res != 0) goto failed;
6185 asdl_seq_SET(dims, i, value);
6186 }
6187 Py_XDECREF(tmp);
6188 tmp = NULL;
6189 } else {
6190 PyErr_SetString(PyExc_TypeError, "required field \"dims\" missing from ExtSlice");
6191 return 1;
6192 }
6193 *out = ExtSlice(dims, arena);
6194 if (*out == NULL) goto failed;
6195 return 0;
6196 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006197 isinstance = PyObject_IsInstance(obj, (PyObject*)Index_type);
6198 if (isinstance == -1) {
6199 return 1;
6200 }
6201 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006202 expr_ty value;
6203
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006204 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006205 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006206 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006207 if (tmp == NULL) goto failed;
6208 res = obj2ast_expr(tmp, &value, arena);
6209 if (res != 0) goto failed;
6210 Py_XDECREF(tmp);
6211 tmp = NULL;
6212 } else {
6213 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from Index");
6214 return 1;
6215 }
6216 *out = Index(value, arena);
6217 if (*out == NULL) goto failed;
6218 return 0;
6219 }
6220
Benjamin Peterson5b066812010-11-20 01:38:49 +00006221 PyErr_Format(PyExc_TypeError, "expected some sort of slice, but got %R", obj);
6222 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006223 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006224 return 1;
6225}
6226
6227int
6228obj2ast_boolop(PyObject* obj, boolop_ty* out, PyArena* arena)
6229{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006230 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006231
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006232 isinstance = PyObject_IsInstance(obj, (PyObject *)And_type);
6233 if (isinstance == -1) {
6234 return 1;
6235 }
6236 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006237 *out = And;
6238 return 0;
6239 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006240 isinstance = PyObject_IsInstance(obj, (PyObject *)Or_type);
6241 if (isinstance == -1) {
6242 return 1;
6243 }
6244 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006245 *out = Or;
6246 return 0;
6247 }
6248
Benjamin Peterson5b066812010-11-20 01:38:49 +00006249 PyErr_Format(PyExc_TypeError, "expected some sort of boolop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006250 return 1;
6251}
6252
6253int
6254obj2ast_operator(PyObject* obj, operator_ty* out, PyArena* arena)
6255{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006256 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006257
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006258 isinstance = PyObject_IsInstance(obj, (PyObject *)Add_type);
6259 if (isinstance == -1) {
6260 return 1;
6261 }
6262 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006263 *out = Add;
6264 return 0;
6265 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006266 isinstance = PyObject_IsInstance(obj, (PyObject *)Sub_type);
6267 if (isinstance == -1) {
6268 return 1;
6269 }
6270 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006271 *out = Sub;
6272 return 0;
6273 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006274 isinstance = PyObject_IsInstance(obj, (PyObject *)Mult_type);
6275 if (isinstance == -1) {
6276 return 1;
6277 }
6278 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006279 *out = Mult;
6280 return 0;
6281 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006282 isinstance = PyObject_IsInstance(obj, (PyObject *)Div_type);
6283 if (isinstance == -1) {
6284 return 1;
6285 }
6286 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006287 *out = Div;
6288 return 0;
6289 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006290 isinstance = PyObject_IsInstance(obj, (PyObject *)Mod_type);
6291 if (isinstance == -1) {
6292 return 1;
6293 }
6294 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006295 *out = Mod;
6296 return 0;
6297 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006298 isinstance = PyObject_IsInstance(obj, (PyObject *)Pow_type);
6299 if (isinstance == -1) {
6300 return 1;
6301 }
6302 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006303 *out = Pow;
6304 return 0;
6305 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006306 isinstance = PyObject_IsInstance(obj, (PyObject *)LShift_type);
6307 if (isinstance == -1) {
6308 return 1;
6309 }
6310 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006311 *out = LShift;
6312 return 0;
6313 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006314 isinstance = PyObject_IsInstance(obj, (PyObject *)RShift_type);
6315 if (isinstance == -1) {
6316 return 1;
6317 }
6318 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006319 *out = RShift;
6320 return 0;
6321 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006322 isinstance = PyObject_IsInstance(obj, (PyObject *)BitOr_type);
6323 if (isinstance == -1) {
6324 return 1;
6325 }
6326 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006327 *out = BitOr;
6328 return 0;
6329 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006330 isinstance = PyObject_IsInstance(obj, (PyObject *)BitXor_type);
6331 if (isinstance == -1) {
6332 return 1;
6333 }
6334 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006335 *out = BitXor;
6336 return 0;
6337 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006338 isinstance = PyObject_IsInstance(obj, (PyObject *)BitAnd_type);
6339 if (isinstance == -1) {
6340 return 1;
6341 }
6342 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006343 *out = BitAnd;
6344 return 0;
6345 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006346 isinstance = PyObject_IsInstance(obj, (PyObject *)FloorDiv_type);
6347 if (isinstance == -1) {
6348 return 1;
6349 }
6350 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006351 *out = FloorDiv;
6352 return 0;
6353 }
6354
Benjamin Peterson5b066812010-11-20 01:38:49 +00006355 PyErr_Format(PyExc_TypeError, "expected some sort of operator, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006356 return 1;
6357}
6358
6359int
6360obj2ast_unaryop(PyObject* obj, unaryop_ty* out, PyArena* arena)
6361{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006362 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006363
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006364 isinstance = PyObject_IsInstance(obj, (PyObject *)Invert_type);
6365 if (isinstance == -1) {
6366 return 1;
6367 }
6368 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006369 *out = Invert;
6370 return 0;
6371 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006372 isinstance = PyObject_IsInstance(obj, (PyObject *)Not_type);
6373 if (isinstance == -1) {
6374 return 1;
6375 }
6376 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006377 *out = Not;
6378 return 0;
6379 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006380 isinstance = PyObject_IsInstance(obj, (PyObject *)UAdd_type);
6381 if (isinstance == -1) {
6382 return 1;
6383 }
6384 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006385 *out = UAdd;
6386 return 0;
6387 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006388 isinstance = PyObject_IsInstance(obj, (PyObject *)USub_type);
6389 if (isinstance == -1) {
6390 return 1;
6391 }
6392 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006393 *out = USub;
6394 return 0;
6395 }
6396
Benjamin Peterson5b066812010-11-20 01:38:49 +00006397 PyErr_Format(PyExc_TypeError, "expected some sort of unaryop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006398 return 1;
6399}
6400
6401int
6402obj2ast_cmpop(PyObject* obj, cmpop_ty* out, PyArena* arena)
6403{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006404 int isinstance;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006405
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006406 isinstance = PyObject_IsInstance(obj, (PyObject *)Eq_type);
6407 if (isinstance == -1) {
6408 return 1;
6409 }
6410 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006411 *out = Eq;
6412 return 0;
6413 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006414 isinstance = PyObject_IsInstance(obj, (PyObject *)NotEq_type);
6415 if (isinstance == -1) {
6416 return 1;
6417 }
6418 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006419 *out = NotEq;
6420 return 0;
6421 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006422 isinstance = PyObject_IsInstance(obj, (PyObject *)Lt_type);
6423 if (isinstance == -1) {
6424 return 1;
6425 }
6426 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006427 *out = Lt;
6428 return 0;
6429 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006430 isinstance = PyObject_IsInstance(obj, (PyObject *)LtE_type);
6431 if (isinstance == -1) {
6432 return 1;
6433 }
6434 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006435 *out = LtE;
6436 return 0;
6437 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006438 isinstance = PyObject_IsInstance(obj, (PyObject *)Gt_type);
6439 if (isinstance == -1) {
6440 return 1;
6441 }
6442 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006443 *out = Gt;
6444 return 0;
6445 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006446 isinstance = PyObject_IsInstance(obj, (PyObject *)GtE_type);
6447 if (isinstance == -1) {
6448 return 1;
6449 }
6450 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006451 *out = GtE;
6452 return 0;
6453 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006454 isinstance = PyObject_IsInstance(obj, (PyObject *)Is_type);
6455 if (isinstance == -1) {
6456 return 1;
6457 }
6458 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006459 *out = Is;
6460 return 0;
6461 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006462 isinstance = PyObject_IsInstance(obj, (PyObject *)IsNot_type);
6463 if (isinstance == -1) {
6464 return 1;
6465 }
6466 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006467 *out = IsNot;
6468 return 0;
6469 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006470 isinstance = PyObject_IsInstance(obj, (PyObject *)In_type);
6471 if (isinstance == -1) {
6472 return 1;
6473 }
6474 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006475 *out = In;
6476 return 0;
6477 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006478 isinstance = PyObject_IsInstance(obj, (PyObject *)NotIn_type);
6479 if (isinstance == -1) {
6480 return 1;
6481 }
6482 if (isinstance) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006483 *out = NotIn;
6484 return 0;
6485 }
6486
Benjamin Peterson5b066812010-11-20 01:38:49 +00006487 PyErr_Format(PyExc_TypeError, "expected some sort of cmpop, but got %R", obj);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006488 return 1;
6489}
6490
6491int
6492obj2ast_comprehension(PyObject* obj, comprehension_ty* out, PyArena* arena)
6493{
6494 PyObject* tmp = NULL;
6495 expr_ty target;
6496 expr_ty iter;
6497 asdl_seq* ifs;
6498
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006499 if (_PyObject_HasAttrId(obj, &PyId_target)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006500 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006501 tmp = _PyObject_GetAttrId(obj, &PyId_target);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006502 if (tmp == NULL) goto failed;
6503 res = obj2ast_expr(tmp, &target, arena);
6504 if (res != 0) goto failed;
6505 Py_XDECREF(tmp);
6506 tmp = NULL;
6507 } else {
6508 PyErr_SetString(PyExc_TypeError, "required field \"target\" missing from comprehension");
6509 return 1;
6510 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006511 if (_PyObject_HasAttrId(obj, &PyId_iter)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006512 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006513 tmp = _PyObject_GetAttrId(obj, &PyId_iter);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006514 if (tmp == NULL) goto failed;
6515 res = obj2ast_expr(tmp, &iter, arena);
6516 if (res != 0) goto failed;
6517 Py_XDECREF(tmp);
6518 tmp = NULL;
6519 } else {
6520 PyErr_SetString(PyExc_TypeError, "required field \"iter\" missing from comprehension");
6521 return 1;
6522 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006523 if (_PyObject_HasAttrId(obj, &PyId_ifs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006524 int res;
6525 Py_ssize_t len;
6526 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006527 tmp = _PyObject_GetAttrId(obj, &PyId_ifs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006528 if (tmp == NULL) goto failed;
6529 if (!PyList_Check(tmp)) {
6530 PyErr_Format(PyExc_TypeError, "comprehension field \"ifs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6531 goto failed;
6532 }
6533 len = PyList_GET_SIZE(tmp);
6534 ifs = asdl_seq_new(len, arena);
6535 if (ifs == NULL) goto failed;
6536 for (i = 0; i < len; i++) {
6537 expr_ty value;
6538 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6539 if (res != 0) goto failed;
6540 asdl_seq_SET(ifs, i, value);
6541 }
6542 Py_XDECREF(tmp);
6543 tmp = NULL;
6544 } else {
6545 PyErr_SetString(PyExc_TypeError, "required field \"ifs\" missing from comprehension");
6546 return 1;
6547 }
6548 *out = comprehension(target, iter, ifs, arena);
6549 return 0;
6550failed:
6551 Py_XDECREF(tmp);
6552 return 1;
6553}
6554
6555int
6556obj2ast_excepthandler(PyObject* obj, excepthandler_ty* out, PyArena* arena)
6557{
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006558 int isinstance;
Neal Norwitzad74aa82008-03-31 05:14:30 +00006559
Benjamin Petersond8f65972010-11-20 04:31:07 +00006560 PyObject *tmp = NULL;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006561 int lineno;
6562 int col_offset;
6563
Neal Norwitzad74aa82008-03-31 05:14:30 +00006564 if (obj == Py_None) {
6565 *out = NULL;
6566 return 0;
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006567 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006568 if (_PyObject_HasAttrId(obj, &PyId_lineno)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006569 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006570 tmp = _PyObject_GetAttrId(obj, &PyId_lineno);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006571 if (tmp == NULL) goto failed;
6572 res = obj2ast_int(tmp, &lineno, arena);
6573 if (res != 0) goto failed;
6574 Py_XDECREF(tmp);
6575 tmp = NULL;
6576 } else {
6577 PyErr_SetString(PyExc_TypeError, "required field \"lineno\" missing from excepthandler");
6578 return 1;
6579 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006580 if (_PyObject_HasAttrId(obj, &PyId_col_offset)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006581 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006582 tmp = _PyObject_GetAttrId(obj, &PyId_col_offset);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006583 if (tmp == NULL) goto failed;
6584 res = obj2ast_int(tmp, &col_offset, arena);
6585 if (res != 0) goto failed;
6586 Py_XDECREF(tmp);
6587 tmp = NULL;
6588 } else {
6589 PyErr_SetString(PyExc_TypeError, "required field \"col_offset\" missing from excepthandler");
6590 return 1;
6591 }
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00006592 isinstance = PyObject_IsInstance(obj, (PyObject*)ExceptHandler_type);
6593 if (isinstance == -1) {
6594 return 1;
6595 }
6596 if (isinstance) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006597 expr_ty type;
6598 identifier name;
6599 asdl_seq* body;
6600
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006601 if (_PyObject_HasAttrId(obj, &PyId_type)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006602 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006603 tmp = _PyObject_GetAttrId(obj, &PyId_type);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006604 if (tmp == NULL) goto failed;
6605 res = obj2ast_expr(tmp, &type, arena);
6606 if (res != 0) goto failed;
6607 Py_XDECREF(tmp);
6608 tmp = NULL;
6609 } else {
6610 type = NULL;
6611 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006612 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006613 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006614 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006615 if (tmp == NULL) goto failed;
6616 res = obj2ast_identifier(tmp, &name, arena);
6617 if (res != 0) goto failed;
6618 Py_XDECREF(tmp);
6619 tmp = NULL;
6620 } else {
6621 name = NULL;
6622 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006623 if (_PyObject_HasAttrId(obj, &PyId_body)) {
Neal Norwitzad74aa82008-03-31 05:14:30 +00006624 int res;
6625 Py_ssize_t len;
6626 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006627 tmp = _PyObject_GetAttrId(obj, &PyId_body);
Neal Norwitzad74aa82008-03-31 05:14:30 +00006628 if (tmp == NULL) goto failed;
6629 if (!PyList_Check(tmp)) {
6630 PyErr_Format(PyExc_TypeError, "ExceptHandler field \"body\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6631 goto failed;
6632 }
6633 len = PyList_GET_SIZE(tmp);
6634 body = asdl_seq_new(len, arena);
6635 if (body == NULL) goto failed;
6636 for (i = 0; i < len; i++) {
6637 stmt_ty value;
6638 res = obj2ast_stmt(PyList_GET_ITEM(tmp, i), &value, arena);
6639 if (res != 0) goto failed;
6640 asdl_seq_SET(body, i, value);
6641 }
6642 Py_XDECREF(tmp);
6643 tmp = NULL;
6644 } else {
6645 PyErr_SetString(PyExc_TypeError, "required field \"body\" missing from ExceptHandler");
6646 return 1;
6647 }
6648 *out = ExceptHandler(type, name, body, lineno, col_offset,
6649 arena);
6650 if (*out == NULL) goto failed;
6651 return 0;
6652 }
6653
Benjamin Peterson5b066812010-11-20 01:38:49 +00006654 PyErr_Format(PyExc_TypeError, "expected some sort of excepthandler, but got %R", obj);
6655 failed:
Benjamin Peterson0a4dae52010-11-21 15:12:34 +00006656 Py_XDECREF(tmp);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006657 return 1;
6658}
6659
6660int
6661obj2ast_arguments(PyObject* obj, arguments_ty* out, PyArena* arena)
6662{
6663 PyObject* tmp = NULL;
6664 asdl_seq* args;
6665 identifier vararg;
6666 expr_ty varargannotation;
6667 asdl_seq* kwonlyargs;
6668 identifier kwarg;
6669 expr_ty kwargannotation;
6670 asdl_seq* defaults;
6671 asdl_seq* kw_defaults;
6672
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006673 if (_PyObject_HasAttrId(obj, &PyId_args)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006674 int res;
6675 Py_ssize_t len;
6676 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006677 tmp = _PyObject_GetAttrId(obj, &PyId_args);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006678 if (tmp == NULL) goto failed;
6679 if (!PyList_Check(tmp)) {
6680 PyErr_Format(PyExc_TypeError, "arguments field \"args\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6681 goto failed;
6682 }
6683 len = PyList_GET_SIZE(tmp);
6684 args = asdl_seq_new(len, arena);
6685 if (args == NULL) goto failed;
6686 for (i = 0; i < len; i++) {
6687 arg_ty value;
6688 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6689 if (res != 0) goto failed;
6690 asdl_seq_SET(args, i, value);
6691 }
6692 Py_XDECREF(tmp);
6693 tmp = NULL;
6694 } else {
6695 PyErr_SetString(PyExc_TypeError, "required field \"args\" missing from arguments");
6696 return 1;
6697 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006698 if (_PyObject_HasAttrId(obj, &PyId_vararg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006699 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006700 tmp = _PyObject_GetAttrId(obj, &PyId_vararg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006701 if (tmp == NULL) goto failed;
6702 res = obj2ast_identifier(tmp, &vararg, arena);
6703 if (res != 0) goto failed;
6704 Py_XDECREF(tmp);
6705 tmp = NULL;
6706 } else {
6707 vararg = NULL;
6708 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006709 if (_PyObject_HasAttrId(obj, &PyId_varargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006710 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006711 tmp = _PyObject_GetAttrId(obj, &PyId_varargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006712 if (tmp == NULL) goto failed;
6713 res = obj2ast_expr(tmp, &varargannotation, arena);
6714 if (res != 0) goto failed;
6715 Py_XDECREF(tmp);
6716 tmp = NULL;
6717 } else {
6718 varargannotation = NULL;
6719 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006720 if (_PyObject_HasAttrId(obj, &PyId_kwonlyargs)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006721 int res;
6722 Py_ssize_t len;
6723 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006724 tmp = _PyObject_GetAttrId(obj, &PyId_kwonlyargs);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006725 if (tmp == NULL) goto failed;
6726 if (!PyList_Check(tmp)) {
6727 PyErr_Format(PyExc_TypeError, "arguments field \"kwonlyargs\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6728 goto failed;
6729 }
6730 len = PyList_GET_SIZE(tmp);
6731 kwonlyargs = asdl_seq_new(len, arena);
6732 if (kwonlyargs == NULL) goto failed;
6733 for (i = 0; i < len; i++) {
6734 arg_ty value;
6735 res = obj2ast_arg(PyList_GET_ITEM(tmp, i), &value, arena);
6736 if (res != 0) goto failed;
6737 asdl_seq_SET(kwonlyargs, i, value);
6738 }
6739 Py_XDECREF(tmp);
6740 tmp = NULL;
6741 } else {
6742 PyErr_SetString(PyExc_TypeError, "required field \"kwonlyargs\" missing from arguments");
6743 return 1;
6744 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006745 if (_PyObject_HasAttrId(obj, &PyId_kwarg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006746 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006747 tmp = _PyObject_GetAttrId(obj, &PyId_kwarg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006748 if (tmp == NULL) goto failed;
6749 res = obj2ast_identifier(tmp, &kwarg, arena);
6750 if (res != 0) goto failed;
6751 Py_XDECREF(tmp);
6752 tmp = NULL;
6753 } else {
6754 kwarg = NULL;
6755 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006756 if (_PyObject_HasAttrId(obj, &PyId_kwargannotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006757 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006758 tmp = _PyObject_GetAttrId(obj, &PyId_kwargannotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006759 if (tmp == NULL) goto failed;
6760 res = obj2ast_expr(tmp, &kwargannotation, arena);
6761 if (res != 0) goto failed;
6762 Py_XDECREF(tmp);
6763 tmp = NULL;
6764 } else {
6765 kwargannotation = NULL;
6766 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006767 if (_PyObject_HasAttrId(obj, &PyId_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006768 int res;
6769 Py_ssize_t len;
6770 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006771 tmp = _PyObject_GetAttrId(obj, &PyId_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006772 if (tmp == NULL) goto failed;
6773 if (!PyList_Check(tmp)) {
6774 PyErr_Format(PyExc_TypeError, "arguments field \"defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6775 goto failed;
6776 }
6777 len = PyList_GET_SIZE(tmp);
6778 defaults = asdl_seq_new(len, arena);
6779 if (defaults == NULL) goto failed;
6780 for (i = 0; i < len; i++) {
6781 expr_ty value;
6782 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6783 if (res != 0) goto failed;
6784 asdl_seq_SET(defaults, i, value);
6785 }
6786 Py_XDECREF(tmp);
6787 tmp = NULL;
6788 } else {
6789 PyErr_SetString(PyExc_TypeError, "required field \"defaults\" missing from arguments");
6790 return 1;
6791 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006792 if (_PyObject_HasAttrId(obj, &PyId_kw_defaults)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006793 int res;
6794 Py_ssize_t len;
6795 Py_ssize_t i;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006796 tmp = _PyObject_GetAttrId(obj, &PyId_kw_defaults);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006797 if (tmp == NULL) goto failed;
6798 if (!PyList_Check(tmp)) {
6799 PyErr_Format(PyExc_TypeError, "arguments field \"kw_defaults\" must be a list, not a %.200s", tmp->ob_type->tp_name);
6800 goto failed;
6801 }
6802 len = PyList_GET_SIZE(tmp);
6803 kw_defaults = asdl_seq_new(len, arena);
6804 if (kw_defaults == NULL) goto failed;
6805 for (i = 0; i < len; i++) {
6806 expr_ty value;
6807 res = obj2ast_expr(PyList_GET_ITEM(tmp, i), &value, arena);
6808 if (res != 0) goto failed;
6809 asdl_seq_SET(kw_defaults, i, value);
6810 }
6811 Py_XDECREF(tmp);
6812 tmp = NULL;
6813 } else {
6814 PyErr_SetString(PyExc_TypeError, "required field \"kw_defaults\" missing from arguments");
6815 return 1;
6816 }
6817 *out = arguments(args, vararg, varargannotation, kwonlyargs, kwarg,
6818 kwargannotation, defaults, kw_defaults, arena);
6819 return 0;
6820failed:
6821 Py_XDECREF(tmp);
6822 return 1;
6823}
6824
6825int
6826obj2ast_arg(PyObject* obj, arg_ty* out, PyArena* arena)
6827{
6828 PyObject* tmp = NULL;
6829 identifier arg;
6830 expr_ty annotation;
6831
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006832 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006833 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006834 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006835 if (tmp == NULL) goto failed;
6836 res = obj2ast_identifier(tmp, &arg, arena);
6837 if (res != 0) goto failed;
6838 Py_XDECREF(tmp);
6839 tmp = NULL;
6840 } else {
6841 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from arg");
6842 return 1;
6843 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006844 if (_PyObject_HasAttrId(obj, &PyId_annotation)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006845 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006846 tmp = _PyObject_GetAttrId(obj, &PyId_annotation);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006847 if (tmp == NULL) goto failed;
6848 res = obj2ast_expr(tmp, &annotation, arena);
6849 if (res != 0) goto failed;
6850 Py_XDECREF(tmp);
6851 tmp = NULL;
6852 } else {
6853 annotation = NULL;
6854 }
6855 *out = arg(arg, annotation, arena);
6856 return 0;
6857failed:
6858 Py_XDECREF(tmp);
6859 return 1;
6860}
6861
6862int
6863obj2ast_keyword(PyObject* obj, keyword_ty* out, PyArena* arena)
6864{
6865 PyObject* tmp = NULL;
6866 identifier arg;
6867 expr_ty value;
6868
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006869 if (_PyObject_HasAttrId(obj, &PyId_arg)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006870 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006871 tmp = _PyObject_GetAttrId(obj, &PyId_arg);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006872 if (tmp == NULL) goto failed;
6873 res = obj2ast_identifier(tmp, &arg, arena);
6874 if (res != 0) goto failed;
6875 Py_XDECREF(tmp);
6876 tmp = NULL;
6877 } else {
6878 PyErr_SetString(PyExc_TypeError, "required field \"arg\" missing from keyword");
6879 return 1;
6880 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006881 if (_PyObject_HasAttrId(obj, &PyId_value)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006882 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006883 tmp = _PyObject_GetAttrId(obj, &PyId_value);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006884 if (tmp == NULL) goto failed;
6885 res = obj2ast_expr(tmp, &value, arena);
6886 if (res != 0) goto failed;
6887 Py_XDECREF(tmp);
6888 tmp = NULL;
6889 } else {
6890 PyErr_SetString(PyExc_TypeError, "required field \"value\" missing from keyword");
6891 return 1;
6892 }
6893 *out = keyword(arg, value, arena);
6894 return 0;
6895failed:
6896 Py_XDECREF(tmp);
6897 return 1;
6898}
6899
6900int
6901obj2ast_alias(PyObject* obj, alias_ty* out, PyArena* arena)
6902{
6903 PyObject* tmp = NULL;
6904 identifier name;
6905 identifier asname;
6906
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006907 if (_PyObject_HasAttrId(obj, &PyId_name)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006908 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006909 tmp = _PyObject_GetAttrId(obj, &PyId_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006910 if (tmp == NULL) goto failed;
6911 res = obj2ast_identifier(tmp, &name, arena);
6912 if (res != 0) goto failed;
6913 Py_XDECREF(tmp);
6914 tmp = NULL;
6915 } else {
6916 PyErr_SetString(PyExc_TypeError, "required field \"name\" missing from alias");
6917 return 1;
6918 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006919 if (_PyObject_HasAttrId(obj, &PyId_asname)) {
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006920 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006921 tmp = _PyObject_GetAttrId(obj, &PyId_asname);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006922 if (tmp == NULL) goto failed;
6923 res = obj2ast_identifier(tmp, &asname, arena);
6924 if (res != 0) goto failed;
6925 Py_XDECREF(tmp);
6926 tmp = NULL;
6927 } else {
6928 asname = NULL;
6929 }
6930 *out = alias(name, asname, arena);
6931 return 0;
6932failed:
6933 Py_XDECREF(tmp);
6934 return 1;
6935}
6936
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006937int
6938obj2ast_withitem(PyObject* obj, withitem_ty* out, PyArena* arena)
6939{
6940 PyObject* tmp = NULL;
6941 expr_ty context_expr;
6942 expr_ty optional_vars;
6943
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006944 if (_PyObject_HasAttrId(obj, &PyId_context_expr)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006945 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006946 tmp = _PyObject_GetAttrId(obj, &PyId_context_expr);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006947 if (tmp == NULL) goto failed;
6948 res = obj2ast_expr(tmp, &context_expr, arena);
6949 if (res != 0) goto failed;
6950 Py_XDECREF(tmp);
6951 tmp = NULL;
6952 } else {
6953 PyErr_SetString(PyExc_TypeError, "required field \"context_expr\" missing from withitem");
6954 return 1;
6955 }
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006956 if (_PyObject_HasAttrId(obj, &PyId_optional_vars)) {
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006957 int res;
Martin v. Löwis1ee1b6f2011-10-10 18:11:30 +02006958 tmp = _PyObject_GetAttrId(obj, &PyId_optional_vars);
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05006959 if (tmp == NULL) goto failed;
6960 res = obj2ast_expr(tmp, &optional_vars, arena);
6961 if (res != 0) goto failed;
6962 Py_XDECREF(tmp);
6963 tmp = NULL;
6964 } else {
6965 optional_vars = NULL;
6966 }
6967 *out = withitem(context_expr, optional_vars, arena);
6968 return 0;
6969failed:
6970 Py_XDECREF(tmp);
6971 return 1;
6972}
6973
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00006974
Martin v. Löwis1a214512008-06-11 05:26:20 +00006975static struct PyModuleDef _astmodule = {
6976 PyModuleDef_HEAD_INIT, "_ast"
6977};
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006978PyMODINIT_FUNC
Martin v. Löwis1a214512008-06-11 05:26:20 +00006979PyInit__ast(void)
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006980{
6981 PyObject *m, *d;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006982 if (!init_types()) return NULL;
6983 m = PyModule_Create(&_astmodule);
6984 if (!m) return NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006985 d = PyModule_GetDict(m);
Martin v. Löwis1a214512008-06-11 05:26:20 +00006986 if (PyDict_SetItemString(d, "AST", (PyObject*)&AST_type) < 0) return
6987 NULL;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00006988 if (PyModule_AddIntConstant(m, "PyCF_ONLY_AST", PyCF_ONLY_AST) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006989 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00006990 if (PyDict_SetItemString(d, "mod", (PyObject*)mod_type) < 0) return
6991 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006992 if (PyDict_SetItemString(d, "Module", (PyObject*)Module_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006993 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006994 if (PyDict_SetItemString(d, "Interactive", (PyObject*)Interactive_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00006995 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00006996 if (PyDict_SetItemString(d, "Expression", (PyObject*)Expression_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00006997 0) return NULL;
6998 if (PyDict_SetItemString(d, "Suite", (PyObject*)Suite_type) < 0) return
6999 NULL;
7000 if (PyDict_SetItemString(d, "stmt", (PyObject*)stmt_type) < 0) return
7001 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007002 if (PyDict_SetItemString(d, "FunctionDef", (PyObject*)FunctionDef_type)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007003 < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007004 if (PyDict_SetItemString(d, "ClassDef", (PyObject*)ClassDef_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007005 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007006 if (PyDict_SetItemString(d, "Return", (PyObject*)Return_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007007 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007008 if (PyDict_SetItemString(d, "Delete", (PyObject*)Delete_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007009 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007010 if (PyDict_SetItemString(d, "Assign", (PyObject*)Assign_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007011 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007012 if (PyDict_SetItemString(d, "AugAssign", (PyObject*)AugAssign_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007013 0) return NULL;
7014 if (PyDict_SetItemString(d, "For", (PyObject*)For_type) < 0) return
7015 NULL;
7016 if (PyDict_SetItemString(d, "While", (PyObject*)While_type) < 0) return
7017 NULL;
7018 if (PyDict_SetItemString(d, "If", (PyObject*)If_type) < 0) return NULL;
7019 if (PyDict_SetItemString(d, "With", (PyObject*)With_type) < 0) return
7020 NULL;
7021 if (PyDict_SetItemString(d, "Raise", (PyObject*)Raise_type) < 0) return
7022 NULL;
Benjamin Peterson43af12b2011-05-29 11:43:10 -05007023 if (PyDict_SetItemString(d, "Try", (PyObject*)Try_type) < 0) return
7024 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007025 if (PyDict_SetItemString(d, "Assert", (PyObject*)Assert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007026 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007027 if (PyDict_SetItemString(d, "Import", (PyObject*)Import_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007028 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007029 if (PyDict_SetItemString(d, "ImportFrom", (PyObject*)ImportFrom_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007030 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007031 if (PyDict_SetItemString(d, "Global", (PyObject*)Global_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007032 return NULL;
Jeremy Hylton81e95022007-02-27 06:50:52 +00007033 if (PyDict_SetItemString(d, "Nonlocal", (PyObject*)Nonlocal_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007034 return NULL;
7035 if (PyDict_SetItemString(d, "Expr", (PyObject*)Expr_type) < 0) return
7036 NULL;
7037 if (PyDict_SetItemString(d, "Pass", (PyObject*)Pass_type) < 0) return
7038 NULL;
7039 if (PyDict_SetItemString(d, "Break", (PyObject*)Break_type) < 0) return
7040 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007041 if (PyDict_SetItemString(d, "Continue", (PyObject*)Continue_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007042 return NULL;
7043 if (PyDict_SetItemString(d, "expr", (PyObject*)expr_type) < 0) return
7044 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007045 if (PyDict_SetItemString(d, "BoolOp", (PyObject*)BoolOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007046 return NULL;
7047 if (PyDict_SetItemString(d, "BinOp", (PyObject*)BinOp_type) < 0) return
7048 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007049 if (PyDict_SetItemString(d, "UnaryOp", (PyObject*)UnaryOp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007050 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007051 if (PyDict_SetItemString(d, "Lambda", (PyObject*)Lambda_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007052 return NULL;
7053 if (PyDict_SetItemString(d, "IfExp", (PyObject*)IfExp_type) < 0) return
7054 NULL;
7055 if (PyDict_SetItemString(d, "Dict", (PyObject*)Dict_type) < 0) return
7056 NULL;
7057 if (PyDict_SetItemString(d, "Set", (PyObject*)Set_type) < 0) return
7058 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007059 if (PyDict_SetItemString(d, "ListComp", (PyObject*)ListComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007060 return NULL;
Nick Coghlan650f0d02007-04-15 12:05:43 +00007061 if (PyDict_SetItemString(d, "SetComp", (PyObject*)SetComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007062 return NULL;
Guido van Rossum992d4a32007-07-11 13:09:30 +00007063 if (PyDict_SetItemString(d, "DictComp", (PyObject*)DictComp_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007064 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007065 if (PyDict_SetItemString(d, "GeneratorExp",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007066 (PyObject*)GeneratorExp_type) < 0) return NULL;
7067 if (PyDict_SetItemString(d, "Yield", (PyObject*)Yield_type) < 0) return
7068 NULL;
Benjamin Peterson527c6222012-01-14 08:58:23 -05007069 if (PyDict_SetItemString(d, "YieldFrom", (PyObject*)YieldFrom_type) <
7070 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007071 if (PyDict_SetItemString(d, "Compare", (PyObject*)Compare_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007072 return NULL;
7073 if (PyDict_SetItemString(d, "Call", (PyObject*)Call_type) < 0) return
7074 NULL;
7075 if (PyDict_SetItemString(d, "Num", (PyObject*)Num_type) < 0) return
7076 NULL;
7077 if (PyDict_SetItemString(d, "Str", (PyObject*)Str_type) < 0) return
7078 NULL;
7079 if (PyDict_SetItemString(d, "Bytes", (PyObject*)Bytes_type) < 0) return
7080 NULL;
Benjamin Peterson442f2092012-12-06 17:41:04 -05007081 if (PyDict_SetItemString(d, "NameConstant",
7082 (PyObject*)NameConstant_type) < 0) return NULL;
Georg Brandl52318d62006-09-06 07:06:08 +00007083 if (PyDict_SetItemString(d, "Ellipsis", (PyObject*)Ellipsis_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007084 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007085 if (PyDict_SetItemString(d, "Attribute", (PyObject*)Attribute_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007086 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007087 if (PyDict_SetItemString(d, "Subscript", (PyObject*)Subscript_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007088 0) return NULL;
Guido van Rossum0368b722007-05-11 16:50:42 +00007089 if (PyDict_SetItemString(d, "Starred", (PyObject*)Starred_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007090 return NULL;
7091 if (PyDict_SetItemString(d, "Name", (PyObject*)Name_type) < 0) return
7092 NULL;
7093 if (PyDict_SetItemString(d, "List", (PyObject*)List_type) < 0) return
7094 NULL;
7095 if (PyDict_SetItemString(d, "Tuple", (PyObject*)Tuple_type) < 0) return
7096 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007097 if (PyDict_SetItemString(d, "expr_context",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007098 (PyObject*)expr_context_type) < 0) return NULL;
7099 if (PyDict_SetItemString(d, "Load", (PyObject*)Load_type) < 0) return
7100 NULL;
7101 if (PyDict_SetItemString(d, "Store", (PyObject*)Store_type) < 0) return
7102 NULL;
7103 if (PyDict_SetItemString(d, "Del", (PyObject*)Del_type) < 0) return
7104 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007105 if (PyDict_SetItemString(d, "AugLoad", (PyObject*)AugLoad_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007106 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007107 if (PyDict_SetItemString(d, "AugStore", (PyObject*)AugStore_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007108 return NULL;
7109 if (PyDict_SetItemString(d, "Param", (PyObject*)Param_type) < 0) return
7110 NULL;
7111 if (PyDict_SetItemString(d, "slice", (PyObject*)slice_type) < 0) return
7112 NULL;
7113 if (PyDict_SetItemString(d, "Slice", (PyObject*)Slice_type) < 0) return
7114 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007115 if (PyDict_SetItemString(d, "ExtSlice", (PyObject*)ExtSlice_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007116 return NULL;
7117 if (PyDict_SetItemString(d, "Index", (PyObject*)Index_type) < 0) return
7118 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007119 if (PyDict_SetItemString(d, "boolop", (PyObject*)boolop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007120 return NULL;
7121 if (PyDict_SetItemString(d, "And", (PyObject*)And_type) < 0) return
7122 NULL;
7123 if (PyDict_SetItemString(d, "Or", (PyObject*)Or_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007124 if (PyDict_SetItemString(d, "operator", (PyObject*)operator_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007125 return NULL;
7126 if (PyDict_SetItemString(d, "Add", (PyObject*)Add_type) < 0) return
7127 NULL;
7128 if (PyDict_SetItemString(d, "Sub", (PyObject*)Sub_type) < 0) return
7129 NULL;
7130 if (PyDict_SetItemString(d, "Mult", (PyObject*)Mult_type) < 0) return
7131 NULL;
7132 if (PyDict_SetItemString(d, "Div", (PyObject*)Div_type) < 0) return
7133 NULL;
7134 if (PyDict_SetItemString(d, "Mod", (PyObject*)Mod_type) < 0) return
7135 NULL;
7136 if (PyDict_SetItemString(d, "Pow", (PyObject*)Pow_type) < 0) return
7137 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007138 if (PyDict_SetItemString(d, "LShift", (PyObject*)LShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007139 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007140 if (PyDict_SetItemString(d, "RShift", (PyObject*)RShift_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007141 return NULL;
7142 if (PyDict_SetItemString(d, "BitOr", (PyObject*)BitOr_type) < 0) return
7143 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007144 if (PyDict_SetItemString(d, "BitXor", (PyObject*)BitXor_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007145 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007146 if (PyDict_SetItemString(d, "BitAnd", (PyObject*)BitAnd_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007147 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007148 if (PyDict_SetItemString(d, "FloorDiv", (PyObject*)FloorDiv_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007149 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007150 if (PyDict_SetItemString(d, "unaryop", (PyObject*)unaryop_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007151 return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007152 if (PyDict_SetItemString(d, "Invert", (PyObject*)Invert_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007153 return NULL;
7154 if (PyDict_SetItemString(d, "Not", (PyObject*)Not_type) < 0) return
7155 NULL;
7156 if (PyDict_SetItemString(d, "UAdd", (PyObject*)UAdd_type) < 0) return
7157 NULL;
7158 if (PyDict_SetItemString(d, "USub", (PyObject*)USub_type) < 0) return
7159 NULL;
7160 if (PyDict_SetItemString(d, "cmpop", (PyObject*)cmpop_type) < 0) return
7161 NULL;
7162 if (PyDict_SetItemString(d, "Eq", (PyObject*)Eq_type) < 0) return NULL;
7163 if (PyDict_SetItemString(d, "NotEq", (PyObject*)NotEq_type) < 0) return
7164 NULL;
7165 if (PyDict_SetItemString(d, "Lt", (PyObject*)Lt_type) < 0) return NULL;
7166 if (PyDict_SetItemString(d, "LtE", (PyObject*)LtE_type) < 0) return
7167 NULL;
7168 if (PyDict_SetItemString(d, "Gt", (PyObject*)Gt_type) < 0) return NULL;
7169 if (PyDict_SetItemString(d, "GtE", (PyObject*)GtE_type) < 0) return
7170 NULL;
7171 if (PyDict_SetItemString(d, "Is", (PyObject*)Is_type) < 0) return NULL;
7172 if (PyDict_SetItemString(d, "IsNot", (PyObject*)IsNot_type) < 0) return
7173 NULL;
7174 if (PyDict_SetItemString(d, "In", (PyObject*)In_type) < 0) return NULL;
7175 if (PyDict_SetItemString(d, "NotIn", (PyObject*)NotIn_type) < 0) return
7176 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007177 if (PyDict_SetItemString(d, "comprehension",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007178 (PyObject*)comprehension_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007179 if (PyDict_SetItemString(d, "excepthandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007180 (PyObject*)excepthandler_type) < 0) return NULL;
Neal Norwitzad74aa82008-03-31 05:14:30 +00007181 if (PyDict_SetItemString(d, "ExceptHandler",
Martin v. Löwis1a214512008-06-11 05:26:20 +00007182 (PyObject*)ExceptHandler_type) < 0) return NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007183 if (PyDict_SetItemString(d, "arguments", (PyObject*)arguments_type) <
Martin v. Löwis1a214512008-06-11 05:26:20 +00007184 0) return NULL;
7185 if (PyDict_SetItemString(d, "arg", (PyObject*)arg_type) < 0) return
7186 NULL;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00007187 if (PyDict_SetItemString(d, "keyword", (PyObject*)keyword_type) < 0)
Martin v. Löwis1a214512008-06-11 05:26:20 +00007188 return NULL;
7189 if (PyDict_SetItemString(d, "alias", (PyObject*)alias_type) < 0) return
7190 NULL;
Benjamin Petersonbf1bbc12011-05-27 13:58:08 -05007191 if (PyDict_SetItemString(d, "withitem", (PyObject*)withitem_type) < 0)
7192 return NULL;
Martin v. Löwis1a214512008-06-11 05:26:20 +00007193 return m;
Martin v. Löwis577b5b92006-02-27 15:23:19 +00007194}
7195
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007196
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007197PyObject* PyAST_mod2obj(mod_ty t)
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007198{
Martin v. Löwisbd260da2006-02-26 19:42:26 +00007199 init_types();
7200 return ast2obj_mod(t);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00007201}
Martin v. Löwis5b222132007-06-10 09:51:05 +00007202
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007203/* mode is 0 for "exec", 1 for "eval" and 2 for "single" input */
7204mod_ty PyAST_obj2mod(PyObject* ast, PyArena* arena, int mode)
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007205{
7206 mod_ty res;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007207 PyObject *req_type[] = {(PyObject*)Module_type, (PyObject*)Expression_type,
7208 (PyObject*)Interactive_type};
7209 char *req_name[] = {"Module", "Expression", "Interactive"};
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007210 int isinstance;
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007211 assert(0 <= mode && mode <= 2);
7212
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007213 init_types();
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007214
Benjamin Peterson0496c9e2009-12-13 01:24:58 +00007215 isinstance = PyObject_IsInstance(ast, req_type[mode]);
7216 if (isinstance == -1)
7217 return NULL;
7218 if (!isinstance) {
Neal Norwitzdb4115f2008-03-31 04:20:05 +00007219 PyErr_Format(PyExc_TypeError, "expected %s node, got %.400s",
7220 req_name[mode], Py_TYPE(ast)->tp_name);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007221 return NULL;
7222 }
7223 if (obj2ast_mod(ast, &res, arena) != 0)
7224 return NULL;
7225 else
7226 return res;
7227}
7228
7229int PyAST_Check(PyObject* obj)
7230{
7231 init_types();
Neal Norwitz207c9f32008-03-31 04:42:11 +00007232 return PyObject_IsInstance(obj, (PyObject*)&AST_type);
Martin v. Löwis618dc5e2008-03-30 20:03:44 +00007233}
7234
Martin v. Löwis5b222132007-06-10 09:51:05 +00007235