mirror of
https://github.com/astral-sh/ruff.git
synced 2025-09-29 21:34:57 +00:00
[ty] Support "legacy" typing.Self
in combination with PEP 695 generic contexts (#20304)
Some checks are pending
CI / Determine changes (push) Waiting to run
CI / cargo fmt (push) Waiting to run
CI / cargo clippy (push) Blocked by required conditions
CI / cargo test (linux) (push) Blocked by required conditions
CI / cargo test (linux, release) (push) Blocked by required conditions
CI / cargo test (windows) (push) Blocked by required conditions
CI / cargo test (wasm) (push) Blocked by required conditions
CI / cargo build (release) (push) Waiting to run
CI / mkdocs (push) Waiting to run
CI / cargo build (msrv) (push) Blocked by required conditions
CI / cargo fuzz build (push) Blocked by required conditions
CI / fuzz parser (push) Blocked by required conditions
CI / test scripts (push) Blocked by required conditions
CI / ecosystem (push) Blocked by required conditions
CI / Fuzz for new ty panics (push) Blocked by required conditions
CI / cargo shear (push) Blocked by required conditions
CI / python package (push) Waiting to run
CI / pre-commit (push) Waiting to run
CI / formatter instabilities and black similarity (push) Blocked by required conditions
CI / test ruff-lsp (push) Blocked by required conditions
CI / check playground (push) Blocked by required conditions
CI / benchmarks-instrumented (push) Blocked by required conditions
CI / benchmarks-walltime (push) Blocked by required conditions
[ty Playground] Release / publish (push) Waiting to run
Some checks are pending
CI / Determine changes (push) Waiting to run
CI / cargo fmt (push) Waiting to run
CI / cargo clippy (push) Blocked by required conditions
CI / cargo test (linux) (push) Blocked by required conditions
CI / cargo test (linux, release) (push) Blocked by required conditions
CI / cargo test (windows) (push) Blocked by required conditions
CI / cargo test (wasm) (push) Blocked by required conditions
CI / cargo build (release) (push) Waiting to run
CI / mkdocs (push) Waiting to run
CI / cargo build (msrv) (push) Blocked by required conditions
CI / cargo fuzz build (push) Blocked by required conditions
CI / fuzz parser (push) Blocked by required conditions
CI / test scripts (push) Blocked by required conditions
CI / ecosystem (push) Blocked by required conditions
CI / Fuzz for new ty panics (push) Blocked by required conditions
CI / cargo shear (push) Blocked by required conditions
CI / python package (push) Waiting to run
CI / pre-commit (push) Waiting to run
CI / formatter instabilities and black similarity (push) Blocked by required conditions
CI / test ruff-lsp (push) Blocked by required conditions
CI / check playground (push) Blocked by required conditions
CI / benchmarks-instrumented (push) Blocked by required conditions
CI / benchmarks-walltime (push) Blocked by required conditions
[ty Playground] Release / publish (push) Waiting to run
## Summary Support cases like the following, where we need the generic context to include both `Self` and `T` (not just `T`): ```py from typing import Self class C: def method[T](self: Self, arg: T): ... C().method(1) ``` closes https://github.com/astral-sh/ty/issues/1131 ## Test Plan Added regression test
This commit is contained in:
parent
ab86ae1760
commit
d55edb3d74
3 changed files with 51 additions and 4 deletions
|
@ -494,3 +494,25 @@ age, name = team.employees[0]
|
||||||
reveal_type(age) # revealed: Age
|
reveal_type(age) # revealed: Age
|
||||||
reveal_type(name) # revealed: Name
|
reveal_type(name) # revealed: Name
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## `self` in PEP 695 generic methods
|
||||||
|
|
||||||
|
When a generic method uses a PEP 695 generic context, an implict or explicit annotation of
|
||||||
|
`self: Self` is still part of the full generic context:
|
||||||
|
|
||||||
|
```py
|
||||||
|
from typing import Self
|
||||||
|
|
||||||
|
class C:
|
||||||
|
def explicit_self[T](self: Self, x: T) -> tuple[Self, T]:
|
||||||
|
return self, x
|
||||||
|
|
||||||
|
def implicit_self[T](self, x: T) -> tuple[Self, T]:
|
||||||
|
return self, x
|
||||||
|
|
||||||
|
def _(x: int):
|
||||||
|
reveal_type(C().explicit_self(x)) # revealed: tuple[C, int]
|
||||||
|
|
||||||
|
# TODO: this should be `tuple[C, int]` as well, once we support implicit `self`
|
||||||
|
reveal_type(C().implicit_self(x)) # revealed: tuple[Unknown, int]
|
||||||
|
```
|
||||||
|
|
|
@ -134,6 +134,18 @@ impl<'db> GenericContext<'db> {
|
||||||
Self::new(db, type_params.into_iter().collect::<FxOrderSet<_>>())
|
Self::new(db, type_params.into_iter().collect::<FxOrderSet<_>>())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Merge this generic context with another, returning a new generic context that
|
||||||
|
/// contains type variables from both contexts.
|
||||||
|
pub(crate) fn merge(self, db: &'db dyn Db, other: Self) -> Self {
|
||||||
|
Self::from_typevar_instances(
|
||||||
|
db,
|
||||||
|
self.variables(db)
|
||||||
|
.iter()
|
||||||
|
.chain(other.variables(db).iter())
|
||||||
|
.copied(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
fn variable_from_type_param(
|
fn variable_from_type_param(
|
||||||
db: &'db dyn Db,
|
db: &'db dyn Db,
|
||||||
index: &'db SemanticIndex<'db>,
|
index: &'db SemanticIndex<'db>,
|
||||||
|
|
|
@ -376,12 +376,25 @@ impl<'db> Signature<'db> {
|
||||||
let legacy_generic_context =
|
let legacy_generic_context =
|
||||||
GenericContext::from_function_params(db, definition, ¶meters, return_ty);
|
GenericContext::from_function_params(db, definition, ¶meters, return_ty);
|
||||||
|
|
||||||
if generic_context.is_some() && legacy_generic_context.is_some() {
|
let full_generic_context = match (legacy_generic_context, generic_context) {
|
||||||
// TODO: Raise a diagnostic!
|
(Some(legacy_ctx), Some(ctx)) => {
|
||||||
}
|
if legacy_ctx
|
||||||
|
.variables(db)
|
||||||
|
.iter()
|
||||||
|
.exactly_one()
|
||||||
|
.is_ok_and(|bound_typevar| bound_typevar.typevar(db).is_self(db))
|
||||||
|
{
|
||||||
|
Some(legacy_ctx.merge(db, ctx))
|
||||||
|
} else {
|
||||||
|
// TODO: Raise a diagnostic — mixing PEP 695 and legacy typevars is not allowed
|
||||||
|
Some(ctx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
(left, right) => left.or(right),
|
||||||
|
};
|
||||||
|
|
||||||
Self {
|
Self {
|
||||||
generic_context: generic_context.or(legacy_generic_context),
|
generic_context: full_generic_context,
|
||||||
inherited_generic_context,
|
inherited_generic_context,
|
||||||
definition: Some(definition),
|
definition: Some(definition),
|
||||||
parameters,
|
parameters,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue