mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-01 06:11:43 +00:00
56f956a238
This PR uses the tokens of the parsed annotation available in the `Checker`, instead of re-lexing (using `SimpleTokenizer`) the annotation. This avoids some limitations of the `SimpleTokenizer`, such as not being able to handle number and string literals. Closes #15816 .
122 lines
1.7 KiB
Python
122 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import (
|
|
Annotated,
|
|
Callable,
|
|
List,
|
|
Literal,
|
|
NamedTuple,
|
|
Tuple,
|
|
TypeVar,
|
|
TypedDict,
|
|
cast,
|
|
)
|
|
|
|
from mypy_extensions import Arg, DefaultArg, DefaultNamedArg, NamedArg, VarArg
|
|
|
|
|
|
def foo(var: "MyClass") -> "MyClass":
|
|
x: "MyClass"
|
|
|
|
|
|
def foo(*, inplace: "bool"):
|
|
pass
|
|
|
|
|
|
def foo(*args: "str", **kwargs: "int"):
|
|
pass
|
|
|
|
|
|
x: Tuple["MyClass"]
|
|
|
|
x: Callable[["MyClass"], None]
|
|
|
|
|
|
class Foo(NamedTuple):
|
|
x: "MyClass"
|
|
|
|
|
|
class D(TypedDict):
|
|
E: TypedDict("E", foo="int", total=False)
|
|
|
|
|
|
class D(TypedDict):
|
|
E: TypedDict("E", {"foo": "int"})
|
|
|
|
|
|
x: Annotated["str", "metadata"]
|
|
|
|
x: Arg("str", "name")
|
|
|
|
x: DefaultArg("str", "name")
|
|
|
|
x: NamedArg("str", "name")
|
|
|
|
x: DefaultNamedArg("str", "name")
|
|
|
|
x: DefaultNamedArg("str", name="name")
|
|
|
|
x: VarArg("str")
|
|
|
|
x: List[List[List["MyClass"]]]
|
|
|
|
x: NamedTuple("X", [("foo", "int"), ("bar", "str")])
|
|
|
|
x: NamedTuple("X", fields=[("foo", "int"), ("bar", "str")])
|
|
|
|
x: NamedTuple(typename="X", fields=[("foo", "int")])
|
|
|
|
X: MyCallable("X")
|
|
|
|
|
|
# OK
|
|
class D(TypedDict):
|
|
E: TypedDict("E")
|
|
|
|
|
|
x: Annotated[()]
|
|
|
|
x: DefaultNamedArg(name="name", quox="str")
|
|
|
|
x: DefaultNamedArg(name="name")
|
|
|
|
x: NamedTuple("X", [("foo",), ("bar",)])
|
|
|
|
x: NamedTuple("X", ["foo", "bar"])
|
|
|
|
x: NamedTuple()
|
|
|
|
x: Literal["foo", "bar"]
|
|
|
|
x = cast(x, "str")
|
|
|
|
|
|
def foo(x, *args, **kwargs):
|
|
...
|
|
|
|
|
|
def foo(*, inplace):
|
|
...
|
|
|
|
|
|
x: Annotated[1:2] = ...
|
|
|
|
x = TypeVar("x", "str", "int")
|
|
|
|
x = cast("str", x)
|
|
|
|
X = List["MyClass"]
|
|
|
|
# Handle end of line comment in string annotation
|
|
# See https://github.com/astral-sh/ruff/issues/15816
|
|
def f() -> "Literal[0]#":
|
|
return 0
|
|
|
|
def g(x: "Literal['abc']#") -> None:
|
|
return
|
|
|
|
def f() -> """Literal[0]
|
|
#
|
|
|
|
""":
|
|
return 0
|