RawQuerySet has conflicting errors: "is not subscriptable" vs "Missing type parameter"

[hterik]

Bug report

What's wrong

Using RawQuerySet as type annotation is not possible.
Without the generic parameter, mypy fails with is not subscriptable error.
With the generic parameter, python throws TypeError.

from django.db.models.query import RawQuerySet
from models import MyModel

# TypeError: type 'RawQuerySet' is not subscriptable
def foo() -> RawQuerySet[MyModel]:                  
    return MyModel.objects.raw("SELECT id FROM my_model")

# mypy error: Missing type parameters for generic type "RawQuerySet"  [type-arg]
def bar() -> RawQuerySet:                           
    return MyModel.objects.raw("SELECT id FROM my_model")

System information

• OS: Ubuntu 24.04
• python version: 3.12.3
• django version: 5.0.6
• mypy version: 1.10
• django-stubs version: 5.0.2
• django-stubs-ext version: 5.0.2

[flaeppe]

Looks like RawQuerySet is missing from the monkeypatch list of classes to add __class_getitem__ to:

django-stubs/ext/django_stubs_ext/patch.py

Lines 55 to 81 in 590f033

	_need_generic: List[MPGeneric[Any]] = [
	MPGeneric(ModelAdmin),
	MPGeneric(SingleObjectMixin),
	MPGeneric(FormMixin),
	MPGeneric(DeletionMixin),
	MPGeneric(MultipleObjectMixin),
	MPGeneric(BaseModelAdmin),
	MPGeneric(Field),
	MPGeneric(Paginator),
	MPGeneric(BaseFormSet),
	MPGeneric(BaseModelForm),
	MPGeneric(BaseModelFormSet),
	MPGeneric(ModelChoiceField),
	MPGeneric(Feed),
	MPGeneric(Sitemap),
	MPGeneric(SuccessMessageMixin),
	MPGeneric(FileProxyMixin),
	MPGeneric(Lookup),
	MPGeneric(BaseConnectionHandler),
	MPGeneric(ExpressionWrapper),
	MPGeneric(ReverseManyToOneDescriptor),
	# These types do have native `__class_getitem__` method since django 3.1:
	MPGeneric(QuerySet, (3, 1)),
	MPGeneric(BaseManager, (3, 1)),
	# These types do have native `__class_getitem__` method since django 4.1:
	MPGeneric(ForeignKey, (4, 1)),
	]

A PR is welcome for fixing that

[intgr]

As a work-around, you can also use quoted type hints, e.g.

def foo() -> "RawQuerySet[MyModel]":                  
    return MyModel.objects.raw("SELECT id FROM my_model")

or alternatively, add from __future__ import annotations to the beginning of the file, which causes evaluation of all type hints to be deferred.