Support Module and other objects in ast.parse

Author: GastonAQS-LuckyRabbit

stdlib/@tests/test_cases/check_ast.py

@@ -0,0 +1,21 @@
+import ast
+from typing_extensions import assert_type
+
+# Test with Module input
+mod1: ast.Module = ast.Module([], [])
+assert_type(ast.parse(mod1), ast.Module)
+assert_type(ast.parse(mod1, mode="exec"), ast.Module)
+mod2: ast.Module = ast.Module(body=[ast.Expr(value=ast.Constant(value=42))], type_ignores=[])
+assert_type(ast.parse(mod2), ast.Module)
+
+# Test with Expression input
+expr1: ast.Expression = ast.Expression(body=ast.Constant(value=42))
+assert_type(ast.parse(expr1, mode="eval"), ast.Expression)
+
+# Test with Interactive input
+inter1: ast.Interactive = ast.Interactive(body=[])
+assert_type(ast.parse(inter1, mode="single"), ast.Interactive)
+
+# Test with FunctionType input
+func1: ast.FunctionType = ast.FunctionType(argtypes=[], returns=ast.Constant(value=None))
+assert_type(ast.parse(func1, mode="func_type"), ast.FunctionType)

stdlib/ast.pyi

@@ -1746,7 +1746,7 @@ _T = _TypeVar("_T", bound=AST)
 if sys.version_info >= (3, 13):
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any] = "<unknown>",
         mode: Literal["exec"] = "exec",
         *,
@@ -1756,7 +1756,7 @@ if sys.version_info >= (3, 13):
     ) -> Module: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["eval"],
         *,
@@ -1766,7 +1766,7 @@ if sys.version_info >= (3, 13):
     ) -> Expression: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["func_type"],
         *,
@@ -1776,7 +1776,7 @@ if sys.version_info >= (3, 13):
     ) -> FunctionType: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["single"],
         *,
@@ -1786,7 +1786,7 @@ if sys.version_info >= (3, 13):
     ) -> Interactive: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["eval"],
         type_comments: bool = False,
@@ -1795,7 +1795,7 @@ if sys.version_info >= (3, 13):
     ) -> Expression: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["func_type"],
         type_comments: bool = False,
@@ -1804,7 +1804,7 @@ if sys.version_info >= (3, 13):
     ) -> FunctionType: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["single"],
         type_comments: bool = False,
@@ -1813,7 +1813,7 @@ if sys.version_info >= (3, 13):
     ) -> Interactive: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any] = "<unknown>",
         mode: str = "exec",
         *,
@@ -1825,7 +1825,7 @@ if sys.version_info >= (3, 13):
 else:
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any] = "<unknown>",
         mode: Literal["exec"] = "exec",
         *,
@@ -1834,7 +1834,7 @@ else:
     ) -> Module: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["eval"],
         *,
@@ -1843,7 +1843,7 @@ else:
     ) -> Expression: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["func_type"],
         *,
@@ -1852,7 +1852,7 @@ else:
     ) -> FunctionType: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any],
         mode: Literal["single"],
         *,
@@ -1861,31 +1861,31 @@ else:
     ) -> Interactive: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["eval"],
         type_comments: bool = False,
         feature_version: None | int | tuple[int, int] = None,
     ) -> Expression: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["func_type"],
         type_comments: bool = False,
         feature_version: None | int | tuple[int, int] = None,
     ) -> FunctionType: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         *,
         mode: Literal["single"],
         type_comments: bool = False,
         feature_version: None | int | tuple[int, int] = None,
     ) -> Interactive: ...
     @overload
     def parse(
-        source: str | ReadableBuffer,
+        source: str | ReadableBuffer | mod,
         filename: str | ReadableBuffer | os.PathLike[Any] = "<unknown>",
         mode: str = "exec",
         *,