2 #include "Python-ast.h"
10 #define UNDEFINED_FUTURE_FEATURE "future feature %.100s is not defined"
11 #define ERR_LATE_FUTURE \
12 "from __future__ imports must occur at the beginning of the file"
15 future_check_features(PyFutureFeatures
*ff
, stmt_ty s
, const char *filename
)
20 assert(s
->kind
== ImportFrom_kind
);
22 names
= s
->v
.ImportFrom
.names
;
23 for (i
= 0; i
< asdl_seq_LEN(names
); i
++) {
24 alias_ty name
= (alias_ty
)asdl_seq_GET(names
, i
);
25 const char *feature
= _PyUnicode_AsString(name
->name
);
28 if (strcmp(feature
, FUTURE_NESTED_SCOPES
) == 0) {
30 } else if (strcmp(feature
, FUTURE_GENERATORS
) == 0) {
32 } else if (strcmp(feature
, FUTURE_DIVISION
) == 0) {
34 } else if (strcmp(feature
, FUTURE_ABSOLUTE_IMPORT
) == 0) {
36 } else if (strcmp(feature
, FUTURE_WITH_STATEMENT
) == 0) {
38 } else if (strcmp(feature
, FUTURE_PRINT_FUNCTION
) == 0) {
40 } else if (strcmp(feature
, FUTURE_UNICODE_LITERALS
) == 0) {
42 } else if (strcmp(feature
, FUTURE_BARRY_AS_BDFL
) == 0) {
43 ff
->ff_features
|= CO_FUTURE_BARRY_AS_BDFL
;
44 } else if (strcmp(feature
, "braces") == 0) {
45 PyErr_SetString(PyExc_SyntaxError
,
47 PyErr_SyntaxLocation(filename
, s
->lineno
);
50 PyErr_Format(PyExc_SyntaxError
,
51 UNDEFINED_FUTURE_FEATURE
, feature
);
52 PyErr_SyntaxLocation(filename
, s
->lineno
);
60 future_parse(PyFutureFeatures
*ff
, mod_ty mod
, const char *filename
)
62 int i
, found_docstring
= 0, done
= 0, prev_line
= 0;
64 static PyObject
*future
;
66 future
= PyUnicode_InternFromString("__future__");
71 if (!(mod
->kind
== Module_kind
|| mod
->kind
== Interactive_kind
))
74 /* A subsequent pass will detect future imports that don't
75 appear at the beginning of the file. There's one case,
76 however, that is easier to handle here: A series of imports
77 joined by semi-colons, where the first import is a future
78 statement but some subsequent import has the future form
79 but is preceded by a regular import.
83 for (i
= 0; i
< asdl_seq_LEN(mod
->v
.Module
.body
); i
++) {
84 stmt_ty s
= (stmt_ty
)asdl_seq_GET(mod
->v
.Module
.body
, i
);
86 if (done
&& s
->lineno
> prev_line
)
88 prev_line
= s
->lineno
;
90 /* The tests below will return from this function unless it is
91 still possible to find a future statement. The only things
92 that can precede a future statement are another future
93 statement and a doc string.
96 if (s
->kind
== ImportFrom_kind
) {
97 if (s
->v
.ImportFrom
.module
== future
) {
99 PyErr_SetString(PyExc_SyntaxError
,
101 PyErr_SyntaxLocation(filename
,
105 if (!future_check_features(ff
, s
, filename
))
107 ff
->ff_lineno
= s
->lineno
;
112 else if (s
->kind
== Expr_kind
&& !found_docstring
) {
113 expr_ty e
= s
->v
.Expr
.value
;
114 if (e
->kind
!= Str_kind
)
127 PyFuture_FromAST(mod_ty mod
, const char *filename
)
129 PyFutureFeatures
*ff
;
131 ff
= (PyFutureFeatures
*)PyObject_Malloc(sizeof(PyFutureFeatures
));
139 if (!future_parse(ff
, mod
, filename
)) {