diff options
| author | Grzegorz Nosek <root@localdomain.pl> | 2013-02-23 20:59:09 +0100 |
|---|---|---|
| committer | Jacob Kaplan-Moss <jacob@jacobian.org> | 2013-02-23 14:35:27 -0600 |
| commit | 2cd0edaa477b327024e4007c8eaf46646dcd0f21 (patch) | |
| tree | c67d5460147f9d841752a223e01953f9e54a156a /django/db | |
| parent | 4c05fdb467d1de68348401fc47869e8f35dcc278 (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/db')
| -rw-r--r-- | django/db/models/query.py | 14 |
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) |
