mpd.py: unset iterating flag if an exception is raised

[python-musicpd.git] / mpd.py

diff --git a/mpd.py b/mpd.py
index a850bb03da5c50fa1f6d8baa7fe51b23e117309c..380f5b1bd5426ea36c2f040820067e0a79386eb1 100644 (file)
--- a/mpd.py
+++ b/mpd.py
@@ -275,9 +275,11 @@ class MPDClient(object):
         self._fetch_nothing()
 
     def _iterator_wrapper(self, iterator):
-        for item in iterator:
-            yield item
-        self._iterating = False
+        try:
+            for item in iterator:
+                yield item
+        finally:
+            self._iterating = False
 
     def _wrap_iterator(self, iterator):
         if not self.iterate: