Add QuerySet.__aiter__() (#1136)

django-stubs/db/models/query.pyi

@@ -1,6 +1,7 @@
 import datetime
 from typing import (
     Any,
+    AsyncIterator,
     Collection,
     Dict,
     Generic,
@@ -167,6 +168,7 @@ class _QuerySet(Generic[_T, _Row], Collection[_Row], Reversible[_Row], Sized):
     def db(self) -> str: ...
     def resolve_expression(self, *args: Any, **kwargs: Any) -> Any: ...
     def __iter__(self) -> Iterator[_Row]: ...
+    def __aiter__(self) -> AsyncIterator[_Row]: ...
     def __contains__(self, x: object) -> bool: ...
     @overload
     def __getitem__(self, i: int) -> _Row: ...

tests/typecheck/managers/querysets/test_iteration.yml

@@ -0,0 +1,33 @@
+-   case: sync_for
+    main: |
+        from myapp.models import User
+
+        for user in User.objects.all():
+            reveal_type(user)  # N: Revealed type is "myapp.models.User"
+    installed_apps:
+        - myapp
+    files:
+        -   path: myapp/__init__.py
+        -   path: myapp/models.py
+            content: |
+                from django.db import models
+                class User(models.Model):
+                    pass
+
+
+-   case: async_for
+    main: |
+        from myapp.models import User
+
+        async def main():
+            async for user in User.objects.all():
+                reveal_type(user)  # N: Revealed type is "myapp.models.User"
+    installed_apps:
+        - myapp
+    files:
+        -   path: myapp/__init__.py
+        -   path: myapp/models.py
+            content: |
+                from django.db import models
+                class User(models.Model):
+                    pass