Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 1 | #include "Python.h" |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 2 | #include "Python-ast.h" |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 3 | #include "token.h" |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 4 | #include "code.h" |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 5 | #include "symtable.h" |
Serhiy Storchaka | 143ce5c | 2018-05-30 10:56:16 +0300 | [diff] [blame] | 6 | #include "ast.h" |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 7 | |
| 8 | #define UNDEFINED_FUTURE_FEATURE "future feature %.100s is not defined" |
Benjamin Peterson | c0747cf | 2008-11-03 20:31:38 +0000 | [diff] [blame] | 9 | #define ERR_LATE_FUTURE \ |
| 10 | "from __future__ imports must occur at the beginning of the file" |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 11 | |
| 12 | static int |
Victor Stinner | 14e461d | 2013-08-26 22:28:21 +0200 | [diff] [blame] | 13 | future_check_features(PyFutureFeatures *ff, stmt_ty s, PyObject *filename) |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 14 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 15 | int i; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 16 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 17 | assert(s->kind == ImportFrom_kind); |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 18 | |
Pablo Galindo | a5634c4 | 2020-09-16 19:42:00 +0100 | [diff] [blame] | 19 | asdl_alias_seq *names = s->v.ImportFrom.names; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 20 | for (i = 0; i < asdl_seq_LEN(names); i++) { |
| 21 | alias_ty name = (alias_ty)asdl_seq_GET(names, i); |
Serhiy Storchaka | 0651583 | 2016-11-20 09:13:07 +0200 | [diff] [blame] | 22 | const char *feature = PyUnicode_AsUTF8(name->name); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 23 | if (!feature) |
| 24 | return 0; |
| 25 | if (strcmp(feature, FUTURE_NESTED_SCOPES) == 0) { |
| 26 | continue; |
| 27 | } else if (strcmp(feature, FUTURE_GENERATORS) == 0) { |
| 28 | continue; |
| 29 | } else if (strcmp(feature, FUTURE_DIVISION) == 0) { |
| 30 | continue; |
| 31 | } else if (strcmp(feature, FUTURE_ABSOLUTE_IMPORT) == 0) { |
| 32 | continue; |
| 33 | } else if (strcmp(feature, FUTURE_WITH_STATEMENT) == 0) { |
| 34 | continue; |
| 35 | } else if (strcmp(feature, FUTURE_PRINT_FUNCTION) == 0) { |
| 36 | continue; |
| 37 | } else if (strcmp(feature, FUTURE_UNICODE_LITERALS) == 0) { |
| 38 | continue; |
| 39 | } else if (strcmp(feature, FUTURE_BARRY_AS_BDFL) == 0) { |
| 40 | ff->ff_features |= CO_FUTURE_BARRY_AS_BDFL; |
Yury Selivanov | 8170e8c | 2015-05-09 11:44:30 -0400 | [diff] [blame] | 41 | } else if (strcmp(feature, FUTURE_GENERATOR_STOP) == 0) { |
Yury Selivanov | 43c47fe | 2018-01-26 15:24:24 -0500 | [diff] [blame] | 42 | continue; |
Guido van Rossum | 95e4d58 | 2018-01-26 08:20:18 -0800 | [diff] [blame] | 43 | } else if (strcmp(feature, FUTURE_ANNOTATIONS) == 0) { |
Batuhan Taskaya | 044a104 | 2020-10-06 23:03:02 +0300 | [diff] [blame] | 44 | continue; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 45 | } else if (strcmp(feature, "braces") == 0) { |
| 46 | PyErr_SetString(PyExc_SyntaxError, |
| 47 | "not a chance"); |
Ammar Askar | 025eb98 | 2018-09-24 17:12:49 -0400 | [diff] [blame] | 48 | PyErr_SyntaxLocationObject(filename, s->lineno, s->col_offset + 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 49 | return 0; |
| 50 | } else { |
| 51 | PyErr_Format(PyExc_SyntaxError, |
| 52 | UNDEFINED_FUTURE_FEATURE, feature); |
Ammar Askar | 025eb98 | 2018-09-24 17:12:49 -0400 | [diff] [blame] | 53 | PyErr_SyntaxLocationObject(filename, s->lineno, s->col_offset + 1); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 54 | return 0; |
| 55 | } |
| 56 | } |
| 57 | return 1; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 58 | } |
| 59 | |
Neal Norwitz | 6576bd8 | 2005-11-13 18:41:28 +0000 | [diff] [blame] | 60 | static int |
Victor Stinner | 14e461d | 2013-08-26 22:28:21 +0200 | [diff] [blame] | 61 | future_parse(PyFutureFeatures *ff, mod_ty mod, PyObject *filename) |
Jeremy Hylton | 39e2f3f | 2001-02-28 01:58:08 +0000 | [diff] [blame] | 62 | { |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 63 | int i, done = 0, prev_line = 0; |
Jeremy Hylton | 39e2f3f | 2001-02-28 01:58:08 +0000 | [diff] [blame] | 64 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 65 | if (!(mod->kind == Module_kind || mod->kind == Interactive_kind)) |
| 66 | return 1; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 67 | |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 68 | if (asdl_seq_LEN(mod->v.Module.body) == 0) |
| 69 | return 1; |
| 70 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 71 | /* A subsequent pass will detect future imports that don't |
| 72 | appear at the beginning of the file. There's one case, |
| 73 | however, that is easier to handle here: A series of imports |
| 74 | joined by semi-colons, where the first import is a future |
| 75 | statement but some subsequent import has the future form |
| 76 | but is preceded by a regular import. |
| 77 | */ |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 78 | |
Serhiy Storchaka | 73cbe7a | 2018-05-29 12:04:55 +0300 | [diff] [blame] | 79 | i = 0; |
Serhiy Storchaka | 143ce5c | 2018-05-30 10:56:16 +0300 | [diff] [blame] | 80 | if (_PyAST_GetDocString(mod->v.Module.body) != NULL) |
Serhiy Storchaka | 73cbe7a | 2018-05-29 12:04:55 +0300 | [diff] [blame] | 81 | i++; |
| 82 | |
| 83 | for (; i < asdl_seq_LEN(mod->v.Module.body); i++) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 84 | stmt_ty s = (stmt_ty)asdl_seq_GET(mod->v.Module.body, i); |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 85 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 86 | if (done && s->lineno > prev_line) |
| 87 | return 1; |
| 88 | prev_line = s->lineno; |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 89 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 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. |
| 94 | */ |
| 95 | |
| 96 | if (s->kind == ImportFrom_kind) { |
Kristján Valur Jónsson | c5d47d5 | 2012-03-23 12:50:53 +0000 | [diff] [blame] | 97 | identifier modname = s->v.ImportFrom.module; |
Benjamin Peterson | ab79c71 | 2012-03-22 08:56:15 -0400 | [diff] [blame] | 98 | if (modname && |
Serhiy Storchaka | f4934ea | 2016-11-16 10:17:58 +0200 | [diff] [blame] | 99 | _PyUnicode_EqualToASCIIString(modname, "__future__")) { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 100 | if (done) { |
| 101 | PyErr_SetString(PyExc_SyntaxError, |
| 102 | ERR_LATE_FUTURE); |
Victor Stinner | 14e461d | 2013-08-26 22:28:21 +0200 | [diff] [blame] | 103 | PyErr_SyntaxLocationObject(filename, s->lineno, s->col_offset); |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 104 | return 0; |
| 105 | } |
| 106 | if (!future_check_features(ff, s, filename)) |
| 107 | return 0; |
| 108 | ff->ff_lineno = s->lineno; |
| 109 | } |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 110 | else { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 111 | done = 1; |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 112 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 113 | } |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 114 | else { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 115 | done = 1; |
Benjamin Peterson | 2d6acd2 | 2013-03-16 09:15:47 -0700 | [diff] [blame] | 116 | } |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 117 | } |
| 118 | return 1; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 119 | } |
| 120 | |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 121 | |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 122 | PyFutureFeatures * |
Victor Stinner | 14e461d | 2013-08-26 22:28:21 +0200 | [diff] [blame] | 123 | PyFuture_FromASTObject(mod_ty mod, PyObject *filename) |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 124 | { |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 125 | PyFutureFeatures *ff; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 126 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 127 | ff = (PyFutureFeatures *)PyObject_Malloc(sizeof(PyFutureFeatures)); |
| 128 | if (ff == NULL) { |
| 129 | PyErr_NoMemory(); |
| 130 | return NULL; |
| 131 | } |
| 132 | ff->ff_features = 0; |
| 133 | ff->ff_lineno = -1; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 134 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 135 | if (!future_parse(ff, mod, filename)) { |
| 136 | PyObject_Free(ff); |
| 137 | return NULL; |
| 138 | } |
| 139 | return ff; |
Jeremy Hylton | 4db62b1 | 2001-02-27 19:07:02 +0000 | [diff] [blame] | 140 | } |
Victor Stinner | 14e461d | 2013-08-26 22:28:21 +0200 | [diff] [blame] | 141 | |
| 142 | |
| 143 | PyFutureFeatures * |
| 144 | PyFuture_FromAST(mod_ty mod, const char *filename_str) |
| 145 | { |
| 146 | PyFutureFeatures *ff; |
| 147 | PyObject *filename; |
| 148 | |
| 149 | filename = PyUnicode_DecodeFSDefault(filename_str); |
| 150 | if (filename == NULL) |
| 151 | return NULL; |
| 152 | ff = PyFuture_FromASTObject(mod, filename); |
| 153 | Py_DECREF(filename); |
| 154 | return ff; |
| 155 | } |