update __doc__ for function object

xuantengh · xuantengh · commit 49d32543a367 · 2024-10-29T14:18:05.000+08:00
diff --git a/Objects/funcobject.c b/Objects/funcobject.c
@@ -162,7 +162,7 @@ PyFunction_NewWithQualName(PyObject *code, PyObject *globals, PyObject *qualname
     PyObject *consts = code_obj->co_consts;
     assert(PyTuple_Check(consts));
     PyObject *doc;
-    if (PyTuple_Size(consts) >= 1) {
+    if (PyTuple_Size(consts) >= 1 && code_obj->co_flags & CO_HAS_DOCSTRING) {
         doc = PyTuple_GetItem(consts, 0);
         if (!PyUnicode_Check(doc)) {
             doc = Py_None;
diff --git a/Python/codegen.c b/Python/codegen.c
@@ -772,7 +772,6 @@ _PyCodegen_Body(compiler *c, location loc, asdl_stmt_seq *stmts, bool is_interac
             assert(st->kind == Expr_kind);
             location loc = LOC(st->v.Expr.value);
             ADDOP_LOAD_CONST(c, loc, cleandoc);
-
             Py_DECREF(cleandoc);
             RETURN_IF_ERROR(codegen_nameop(c, NO_LOCATION, &_Py_ID(__doc__), Store));
         }

Original file line number	Diff line number	Diff line change
`@@ -772,7 +772,6 @@ _PyCodegen_Body(compiler c, location loc, asdl_stmt_seq stmts, bool is_interac`
`772`	`772`	`assert(st->kind == Expr_kind);`
`773`	`773`	`location loc = LOC(st->v.Expr.value);`
`774`	`774`	`ADDOP_LOAD_CONST(c, loc, cleandoc);`
`775`		`-`
`776`	`775`	`Py_DECREF(cleandoc);`
`777`	`776`	`RETURN_IF_ERROR(codegen_nameop(c, NO_LOCATION, &_Py_ID(__doc__), Store));`
`778`	`777`	`}`