mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-09-29 13:24:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

[ty] Remove false positives when subscripting Generic or Protocol with a ParamSpec or TypeVarTuple (#19749)

Browse source
This commit is contained in:
Alex Waygood 2025-08-04 21:42:46 +01:00 • committed by GitHub
parent 739c94f95a
commit 3a9341f7be
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 37 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

44
crates/ty_python_semantic/src/types/infer.rs
View file

@ -8563,7 +8563,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
fn infer_subscript_expression_types(
&self,
value_node: &ast::Expr,
value_node: &'ast ast::Expr,
value_ty: Type<'db>,
slice_ty: Type<'db>,
expr_context: ExprContext,
@ -8732,7 +8732,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
.map(|context| {
Type::KnownInstance(KnownInstanceType::SubscriptedProtocol(context))
})
.unwrap_or_else(Type::unknown),
.unwrap_or_else(GenericContextError::into_type),
// TODO: emit a diagnostic
TupleSpec::Variable(_) => Type::unknown(),
})
@ -8745,7 +8745,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
LegacyGenericBase::Protocol,
)
.map(|context| Type::KnownInstance(KnownInstanceType::SubscriptedProtocol(context)))
.unwrap_or_else(Type::unknown),
.unwrap_or_else(GenericContextError::into_type),
),
(Type::KnownInstance(KnownInstanceType::SubscriptedProtocol(_)), _) => {
@ -8764,7 +8764,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
.map(|context| {
Type::KnownInstance(KnownInstanceType::SubscriptedGeneric(context))
})
.unwrap_or_else(Type::unknown),
.unwrap_or_else(GenericContextError::into_type),
// TODO: emit a diagnostic
TupleSpec::Variable(_) => Type::unknown(),
})
@ -8777,7 +8777,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
LegacyGenericBase::Generic,
)
.map(|context| Type::KnownInstance(KnownInstanceType::SubscriptedGeneric(context)))
.unwrap_or_else(Type::unknown),
.unwrap_or_else(GenericContextError::into_type),
),
(Type::KnownInstance(KnownInstanceType::SubscriptedGeneric(_)), _) => {
@ -8946,11 +8946,19 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
value_node: &ast::Expr,
typevars: &[Type<'db>],
origin: LegacyGenericBase,
) -> Option<GenericContext<'db>> {
let typevars: Option<FxOrderSet<_>> = typevars
) -> Result<GenericContext<'db>, GenericContextError> {
let typevars: Result<FxOrderSet<_>, GenericContextError> = typevars
.iter()
.map(|typevar| match typevar {
Type::KnownInstance(KnownInstanceType::TypeVar(typevar)) => Some(*typevar),
Type::KnownInstance(KnownInstanceType::TypeVar(typevar)) => Ok(*typevar),
Type::NominalInstance(NominalInstanceType { class, .. })
if matches!(
class.known(self.db()),
Some(KnownClass::TypeVarTuple | KnownClass::ParamSpec)
) =>
{
Err(GenericContextError::NotYetSupported)
}
_ => {
if let Some(builder) =
self.context.report_lint(&INVALID_ARGUMENT_TYPE, value_node)
@ -8960,7 +8968,7 @@ impl<'db, 'ast> TypeInferenceBuilder<'db, 'ast> {
typevar.display(self.db()),
));
}
None
Err(GenericContextError::InvalidArgument)
}
})
.collect();
@ -10860,6 +10868,24 @@ impl<'db> TypeInferenceBuilder<'db, '_> {
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum GenericContextError {
/// It's invalid to subscript `Generic` or `Protocol` with this type
InvalidArgument,
/// It's valid to subscribe `Generic` or `Protocol` with this type,
/// but the type is not yet supported.
NotYetSupported,
}
impl GenericContextError {
const fn into_type<'db>(self) -> Type<'db> {
match self {
GenericContextError::InvalidArgument => Type::unknown(),
GenericContextError::NotYetSupported => todo_type!("ParamSpecs and TypeVarTuples"),
}
}
}
/// The deferred state of a specific expression in an inference region.
#[derive(Default, Debug, Clone, Copy)]
enum DeferredExpressionState {