/src/cpython3/Python/future.c
Line | Count | Source |
1 | | #include "Python.h" |
2 | | #include "pycore_ast.h" // _PyAST_GetDocString() |
3 | | #include "pycore_symtable.h" // _PyFutureFeatures |
4 | | #include "pycore_unicodeobject.h" // _PyUnicode_EqualToASCIIString() |
5 | | |
6 | 13 | #define UNDEFINED_FUTURE_FEATURE "future feature %.100s is not defined" |
7 | | |
8 | | static int |
9 | | future_check_features(_PyFutureFeatures *ff, stmt_ty s, PyObject *filename) |
10 | 1.50k | { |
11 | 1.50k | Py_ssize_t i; |
12 | | |
13 | 1.50k | assert(s->kind == ImportFrom_kind); |
14 | | |
15 | 1.50k | asdl_alias_seq *names = s->v.ImportFrom.names; |
16 | 3.00k | for (i = 0; i < asdl_seq_LEN(names); i++) { |
17 | 1.50k | alias_ty name = (alias_ty)asdl_seq_GET(names, i); |
18 | 1.50k | const char *feature = PyUnicode_AsUTF8(name->name); |
19 | 1.50k | if (!feature) |
20 | 0 | return 0; |
21 | 1.50k | if (strcmp(feature, FUTURE_NESTED_SCOPES) == 0) { |
22 | 0 | continue; |
23 | 1.50k | } else if (strcmp(feature, FUTURE_GENERATORS) == 0) { |
24 | 0 | continue; |
25 | 1.50k | } else if (strcmp(feature, FUTURE_DIVISION) == 0) { |
26 | 1 | continue; |
27 | 1.50k | } else if (strcmp(feature, FUTURE_ABSOLUTE_IMPORT) == 0) { |
28 | 0 | continue; |
29 | 1.50k | } else if (strcmp(feature, FUTURE_WITH_STATEMENT) == 0) { |
30 | 1 | continue; |
31 | 1.50k | } else if (strcmp(feature, FUTURE_PRINT_FUNCTION) == 0) { |
32 | 0 | continue; |
33 | 1.50k | } else if (strcmp(feature, FUTURE_UNICODE_LITERALS) == 0) { |
34 | 0 | continue; |
35 | 1.50k | } else if (strcmp(feature, FUTURE_BARRY_AS_BDFL) == 0) { |
36 | 1 | ff->ff_features |= CO_FUTURE_BARRY_AS_BDFL; |
37 | 1.50k | } else if (strcmp(feature, FUTURE_GENERATOR_STOP) == 0) { |
38 | 1 | continue; |
39 | 1.50k | } else if (strcmp(feature, FUTURE_ANNOTATIONS) == 0) { |
40 | 1.49k | ff->ff_features |= CO_FUTURE_ANNOTATIONS; |
41 | 1.49k | } else if (strcmp(feature, "braces") == 0) { |
42 | 1 | PyErr_SetString(PyExc_SyntaxError, |
43 | 1 | "not a chance"); |
44 | 1 | PyErr_RangedSyntaxLocationObject(filename, |
45 | 1 | name->lineno, |
46 | 1 | name->col_offset + 1, |
47 | 1 | name->end_lineno, |
48 | 1 | name->end_col_offset + 1); |
49 | 1 | return 0; |
50 | 13 | } else { |
51 | 13 | PyErr_Format(PyExc_SyntaxError, |
52 | 13 | UNDEFINED_FUTURE_FEATURE, feature); |
53 | 13 | PyErr_RangedSyntaxLocationObject(filename, |
54 | 13 | name->lineno, |
55 | 13 | name->col_offset + 1, |
56 | 13 | name->end_lineno, |
57 | 13 | name->end_col_offset + 1); |
58 | 13 | return 0; |
59 | 13 | } |
60 | 1.50k | } |
61 | 1.49k | return 1; |
62 | 1.50k | } |
63 | | |
64 | | static int |
65 | | future_parse(_PyFutureFeatures *ff, mod_ty mod, PyObject *filename) |
66 | 12.2k | { |
67 | 12.2k | if (!(mod->kind == Module_kind || mod->kind == Interactive_kind)) { |
68 | 2.00k | return 1; |
69 | 2.00k | } |
70 | | |
71 | 10.2k | Py_ssize_t n = asdl_seq_LEN(mod->v.Module.body); |
72 | 10.2k | if (n == 0) { |
73 | 245 | return 1; |
74 | 245 | } |
75 | | |
76 | 9.96k | Py_ssize_t i = 0; |
77 | 9.96k | if (_PyAST_GetDocString(mod->v.Module.body) != NULL) { |
78 | 272 | i++; |
79 | 272 | } |
80 | | |
81 | 11.4k | for (; i < n; i++) { |
82 | 11.2k | stmt_ty s = (stmt_ty)asdl_seq_GET(mod->v.Module.body, i); |
83 | | |
84 | | /* The only things that can precede a future statement |
85 | | * are another future statement and a doc string. |
86 | | */ |
87 | | |
88 | 11.2k | if (s->kind == ImportFrom_kind && s->v.ImportFrom.level == 0) { |
89 | 1.55k | identifier modname = s->v.ImportFrom.module; |
90 | 1.55k | if (modname && |
91 | 1.55k | _PyUnicode_EqualToASCIIString(modname, "__future__")) { |
92 | 1.50k | if (!future_check_features(ff, s, filename)) { |
93 | 14 | return 0; |
94 | 14 | } |
95 | 1.49k | ff->ff_location = SRC_LOCATION_FROM_AST(s); |
96 | 1.49k | } |
97 | 47 | else { |
98 | 47 | return 1; |
99 | 47 | } |
100 | 1.55k | } |
101 | 9.69k | else { |
102 | 9.69k | return 1; |
103 | 9.69k | } |
104 | 11.2k | } |
105 | 213 | return 1; |
106 | 9.96k | } |
107 | | |
108 | | |
109 | | int |
110 | | _PyFuture_FromAST(mod_ty mod, PyObject *filename, _PyFutureFeatures *ff) |
111 | 12.2k | { |
112 | 12.2k | ff->ff_features = 0; |
113 | 12.2k | ff->ff_location = (_Py_SourceLocation){-1, -1, -1, -1}; |
114 | | |
115 | 12.2k | if (!future_parse(ff, mod, filename)) { |
116 | 14 | return 0; |
117 | 14 | } |
118 | 12.1k | return 1; |
119 | 12.2k | } |