mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-10-29 00:26:07 +00:00
Code Issues 32 Releases Wiki Activity

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

Browse Source
This commit is contained in:
Mariusz Felisiak
2022-04-27 11:30:43 +02:00
parent 77926176b2
commit 271a8e73ee
2 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
django/db/models/query.py
View File

@@ -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)