Coverage Report

Created: 2025-07-18 06:09

/src/cpython/Python/future.c
Line
Count
Source (jump to first uncovered line)
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
30
#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
338
{
11
338
    Py_ssize_t i;
12
13
338
    assert(s->kind == ImportFrom_kind);
14
15
338
    asdl_alias_seq *names = s->v.ImportFrom.names;
16
784
    for (i = 0; i < asdl_seq_LEN(names); i++) {
17
476
        alias_ty name = (alias_ty)asdl_seq_GET(names, i);
18
476
        const char *feature = PyUnicode_AsUTF8(name->name);
19
476
        if (!feature)
20
0
            return 0;
21
476
        if (strcmp(feature, FUTURE_NESTED_SCOPES) == 0) {
22
7
            continue;
23
469
        } else if (strcmp(feature, FUTURE_GENERATORS) == 0) {
24
0
            continue;
25
469
        } else if (strcmp(feature, FUTURE_DIVISION) == 0) {
26
1
            continue;
27
468
        } else if (strcmp(feature, FUTURE_ABSOLUTE_IMPORT) == 0) {
28
21
            continue;
29
447
        } else if (strcmp(feature, FUTURE_WITH_STATEMENT) == 0) {
30
35
            continue;
31
412
        } else if (strcmp(feature, FUTURE_PRINT_FUNCTION) == 0) {
32
100
            continue;
33
312
        } else if (strcmp(feature, FUTURE_UNICODE_LITERALS) == 0) {
34
139
            continue;
35
173
        } else if (strcmp(feature, FUTURE_BARRY_AS_BDFL) == 0) {
36
10
            ff->ff_features |= CO_FUTURE_BARRY_AS_BDFL;
37
163
        } else if (strcmp(feature, FUTURE_GENERATOR_STOP) == 0) {
38
6
            continue;
39
157
        } else if (strcmp(feature, FUTURE_ANNOTATIONS) == 0) {
40
127
            ff->ff_features |= CO_FUTURE_ANNOTATIONS;
41
127
        } else if (strcmp(feature, "braces") == 0) {
42
0
            PyErr_SetString(PyExc_SyntaxError,
43
0
                            "not a chance");
44
0
            PyErr_RangedSyntaxLocationObject(filename,
45
0
                                             name->lineno,
46
0
                                             name->col_offset + 1,
47
0
                                             name->end_lineno,
48
0
                                             name->end_col_offset + 1);
49
0
            return 0;
50
30
        } else {
51
30
            PyErr_Format(PyExc_SyntaxError,
52
30
                         UNDEFINED_FUTURE_FEATURE, feature);
53
30
            PyErr_RangedSyntaxLocationObject(filename,
54
30
                                             name->lineno,
55
30
                                             name->col_offset + 1,
56
30
                                             name->end_lineno,
57
30
                                             name->end_col_offset + 1);
58
30
            return 0;
59
30
        }
60
476
    }
61
308
    return 1;
62
338
}
63
64
static int
65
future_parse(_PyFutureFeatures *ff, mod_ty mod, PyObject *filename)
66
6.93k
{
67
6.93k
    if (!(mod->kind == Module_kind || mod->kind == Interactive_kind)) {
68
43
        return 1;
69
43
    }
70
71
6.89k
    Py_ssize_t n = asdl_seq_LEN(mod->v.Module.body);
72
6.89k
    if (n == 0) {
73
746
        return 1;
74
746
    }
75
76
6.14k
    Py_ssize_t i = 0;
77
6.14k
    if (_PyAST_GetDocString(mod->v.Module.body) != NULL) {
78
527
        i++;
79
527
    }
80
81
6.45k
    for (; i < n; i++) {
82
6.13k
        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
6.13k
        if (s->kind == ImportFrom_kind && s->v.ImportFrom.level == 0) {
89
381
            identifier modname = s->v.ImportFrom.module;
90
381
            if (modname &&
91
381
                _PyUnicode_EqualToASCIIString(modname, "__future__")) {
92
338
                if (!future_check_features(ff, s, filename)) {
93
30
                    return 0;
94
30
                }
95
308
                ff->ff_location = SRC_LOCATION_FROM_AST(s);
96
308
            }
97
43
            else {
98
43
                return 1;
99
43
            }
100
381
        }
101
5.74k
        else {
102
5.74k
            return 1;
103
5.74k
        }
104
6.13k
    }
105
322
    return 1;
106
6.14k
}
107
108
109
int
110
_PyFuture_FromAST(mod_ty mod, PyObject *filename, _PyFutureFeatures *ff)
111
6.93k
{
112
6.93k
    ff->ff_features = 0;
113
6.93k
    ff->ff_location = (_Py_SourceLocation){-1, -1, -1, -1};
114
115
6.93k
    if (!future_parse(ff, mod, filename)) {
116
30
        return 0;
117
30
    }
118
6.90k
    return 1;
119
6.93k
}