Refs #33646 -- Made QuerySet.raw() async-compatible.

1 files changed, 16 insertions, 0 deletions

diff --git a/django/db/models/query.py b/django/db/models/query.py
index 80dc83a41c..ce7e295d52 100644
--- a/django/db/models/query.py
+++ b/django/db/models/query.py
@@ -2025,6 +2025,12 @@ class RawQuerySet:
         if self._prefetch_related_lookups and not self._prefetch_done:
             self._prefetch_related_objects()
 
+    async def _async_fetch_all(self):
+        if self._result_cache is None:
+            self._result_cache = [result async for result in RawModelIterable(self)]
+        if self._prefetch_related_lookups and not self._prefetch_done:
+            sync_to_async(self._prefetch_related_objects)()
+
     def __len__(self):
         self._fetch_all()
         return len(self._result_cache)
@@ -2037,6 +2043,16 @@ class RawQuerySet:
         self._fetch_all()
         return iter(self._result_cache)
 
+    def __aiter__(self):
+        # Remember, __aiter__ itself is synchronous, it's the thing it returns
+        # that is async!
+        async def generator():
+            await self._async_fetch_all()
+            for item in self._result_cache:
+                yield item
+
+        return generator()
+
     def iterator(self):
         yield from RawModelIterable(self)