mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-22 17:55:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 15

gh-126072: Set docstring attribute for module and class (#126231)

Browse source
This commit is contained in:
Xuanteng Huang 2024-11-08 23:13:18 +08:00 • committed by GitHub
parent b19d12f447
commit 6ec886531f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 77 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

8
Python/symtable.c
View file

@ -434,6 +434,9 @@ _PySymtable_Build(mod_ty mod, PyObject *filename, _PyFutureFeatures *future)
switch (mod->kind) {
case Module_kind:
seq = mod->v.Module.body;
if (_PyAST_GetDocString(seq)) {
st->st_cur->ste_has_docstring = 1;
}
for (i = 0; i < asdl_seq_LEN(seq); i++)
if (!symtable_visit_stmt(st,
(stmt_ty)asdl_seq_GET(seq, i)))
@ -1909,6 +1912,11 @@ symtable_visit_stmt(struct symtable *st, stmt_ty s)
return 0;
}
}
if (_PyAST_GetDocString(s->v.ClassDef.body)) {
st->st_cur->ste_has_docstring = 1;
}
VISIT_SEQ(st, stmt, s->v.ClassDef.body);
if (!symtable_exit_block(st))
return 0;