Update _ast.pyi

improvement: shift `type_comment` attribute to classes which contain it
python · JelleZijlstra · May 18, 2024 · May 9, 2024 · May 9, 2024 · May 9, 2024
commit 75a2e8e1af712de4fb3a81c0edfc9a40223120a9
diff --git a/stdlib/_ast.pyi b/stdlib/_ast.pyi
@@ -18,8 +18,6 @@ class AST:
    _attributes: ClassVar[tuple[str, ...]]
    _fields: ClassVar[tuple[str, ...]]
    def __init__(self, *args: Any, **kwargs: Any) -> None: ...
-    # TODO: Not all nodes have all of the following attributes
-    type_comment: str | None

 class mod(AST): ...
 class type_ignore(AST): ...
@@ -67,6 +65,7 @@ class FunctionDef(stmt):
    body: list[stmt]
    decorator_list: list[expr]
    returns: expr | None
+    type_comment: str | None
    if sys.version_info >= (3, 12):
        type_params: list[type_param]

@@ -80,6 +79,7 @@ class AsyncFunctionDef(stmt):
    body: list[stmt]
    decorator_list: list[expr]
    returns: expr | None
+    type_comment: str | None
    if sys.version_info >= (3, 12):
        type_params: list[type_param]

@@ -111,6 +111,7 @@ class Assign(stmt):
        __match_args__ = ("targets", "value", "type_comment")
    targets: list[expr]
    value: expr
+    type_comment: str | None

 class AugAssign(stmt):
    if sys.version_info >= (3, 10):
@@ -134,6 +135,7 @@ class For(stmt):
    iter: expr
    body: list[stmt]
    orelse: list[stmt]
+    type_comment: str | None

 class AsyncFor(stmt):
    if sys.version_info >= (3, 10):
@@ -142,6 +144,7 @@ class AsyncFor(stmt):
    iter: expr
    body: list[stmt]
    orelse: list[stmt]
+    type_comment: str | None

 class While(stmt):
    if sys.version_info >= (3, 10):
@@ -162,12 +165,14 @@ class With(stmt):
        __match_args__ = ("items", "body", "type_comment")
    items: list[withitem]
    body: list[stmt]
+    type_comment: str | None

 class AsyncWith(stmt):
    if sys.version_info >= (3, 10):
        __match_args__ = ("items", "body", "type_comment")
    items: list[withitem]
    body: list[stmt]
+    type_comment: str | None

 class Raise(stmt):
    if sys.version_info >= (3, 10):
@@ -506,6 +511,7 @@ class arg(AST):
        __match_args__ = ("arg", "annotation", "type_comment")
    arg: _Identifier
    annotation: expr | None
+    type_comment: str | None

 class keyword(AST):
    lineno: int