2 #include "Python-ast.h"
10 #define UNDEFINED_FUTURE_FEATURE "future feature %.100s is not defined"
13 future_check_features(PyFutureFeatures
*ff
, stmt_ty s
, const char *filename
)
18 assert(s
->kind
== ImportFrom_kind
);
20 names
= s
->v
.ImportFrom
.names
;
21 for (i
= 0; i
< asdl_seq_LEN(names
); i
++) {
22 alias_ty name
= (alias_ty
)asdl_seq_GET(names
, i
);
23 const char *feature
= PyString_AsString(name
->name
);
26 if (strcmp(feature
, FUTURE_NESTED_SCOPES
) == 0) {
28 } else if (strcmp(feature
, FUTURE_GENERATORS
) == 0) {
30 } else if (strcmp(feature
, FUTURE_DIVISION
) == 0) {
31 ff
->ff_features
|= CO_FUTURE_DIVISION
;
32 } else if (strcmp(feature
, FUTURE_ABSOLUTE_IMPORT
) == 0) {
33 ff
->ff_features
|= CO_FUTURE_ABSOLUTE_IMPORT
;
34 } else if (strcmp(feature
, FUTURE_WITH_STATEMENT
) == 0) {
35 ff
->ff_features
|= CO_FUTURE_WITH_STATEMENT
;
36 } else if (strcmp(feature
, FUTURE_PRINT_FUNCTION
) == 0) {
37 ff
->ff_features
|= CO_FUTURE_PRINT_FUNCTION
;
38 } else if (strcmp(feature
, FUTURE_UNICODE_LITERALS
) == 0) {
39 ff
->ff_features
|= CO_FUTURE_UNICODE_LITERALS
;
40 } else if (strcmp(feature
, "braces") == 0) {
41 PyErr_SetString(PyExc_SyntaxError
,
43 PyErr_SyntaxLocation(filename
, s
->lineno
);
46 PyErr_Format(PyExc_SyntaxError
,
47 UNDEFINED_FUTURE_FEATURE
, feature
);
48 PyErr_SyntaxLocation(filename
, s
->lineno
);
56 future_parse(PyFutureFeatures
*ff
, mod_ty mod
, const char *filename
)
58 int i
, found_docstring
= 0, done
= 0, prev_line
= 0;
60 static PyObject
*future
;
62 future
= PyString_InternFromString("__future__");
67 if (!(mod
->kind
== Module_kind
|| mod
->kind
== Interactive_kind
))
70 /* A subsequent pass will detect future imports that don't
71 appear at the beginning of the file. There's one case,
72 however, that is easier to handle here: A series of imports
73 joined by semi-colons, where the first import is a future
74 statement but some subsequent import has the future form
75 but is preceded by a regular import.
79 for (i
= 0; i
< asdl_seq_LEN(mod
->v
.Module
.body
); i
++) {
80 stmt_ty s
= (stmt_ty
)asdl_seq_GET(mod
->v
.Module
.body
, i
);
82 if (done
&& s
->lineno
> prev_line
)
84 prev_line
= s
->lineno
;
86 /* The tests below will return from this function unless it is
87 still possible to find a future statement. The only things
88 that can precede a future statement are another future
89 statement and a doc string.
92 if (s
->kind
== ImportFrom_kind
) {
93 if (s
->v
.ImportFrom
.module
== future
) {
95 PyErr_SetString(PyExc_SyntaxError
,
97 PyErr_SyntaxLocation(filename
,
101 if (!future_check_features(ff
, s
, filename
))
103 ff
->ff_lineno
= s
->lineno
;
108 else if (s
->kind
== Expr_kind
&& !found_docstring
) {
109 expr_ty e
= s
->v
.Expr
.value
;
110 if (e
->kind
!= Str_kind
)
123 PyFuture_FromAST(mod_ty mod
, const char *filename
)
125 PyFutureFeatures
*ff
;
127 ff
= (PyFutureFeatures
*)PyObject_Malloc(sizeof(PyFutureFeatures
));
135 if (!future_parse(ff
, mod
, filename
)) {