summaryrefslogtreecommitdiff
path: root/django
diff options
context:
space:
mode:
authorGrzegorz Nosek <root@localdomain.pl>2013-02-23 20:59:09 +0100
committerJacob Kaplan-Moss <jacob@jacobian.org>2013-02-23 14:35:27 -0600
commit2cd0edaa477b327024e4007c8eaf46646dcd0f21 (patch)
treec67d5460147f9d841752a223e01953f9e54a156a /django
parent4c05fdb467d1de68348401fc47869e8f35dcc278 (diff)
Fixed 19895 -- Made second iteration over invalid queryset raise an exception too
When iteration over a queryset raised an exception, the result cache remained initialized with an empty list, so subsequent iterations returned an empty list instead of raising an exception
Diffstat (limited to 'django')
-rw-r--r--django/db/models/query.py14
1 files changed, 13 insertions, 1 deletions
diff --git a/django/db/models/query.py b/django/db/models/query.py
index 0f3a79a25d..ec35f8aba3 100644
--- a/django/db/models/query.py
+++ b/django/db/models/query.py
@@ -104,7 +104,7 @@ class QuerySet(object):
len(self)
if self._result_cache is None:
- self._iter = self.iterator()
+ self._iter = self._safe_iterator(self.iterator())
self._result_cache = []
if self._iter:
return self._result_iter()
@@ -341,6 +341,18 @@ class QuerySet(object):
yield obj
+ def _safe_iterator(self, iterator):
+ # ensure result cache is cleared when iterating over a queryset
+ # raises an exception
+ try:
+ for item in iterator:
+ yield item
+ except StopIteration:
+ raise
+ except Exception:
+ self._result_cache = None
+ raise
+
def aggregate(self, *args, **kwargs):
"""
Returns a dictionary containing the calculations (aggregation)