From bddc07cb9f99da48dbe28f9f987e53b01564e7a7 Mon Sep 17 00:00:00 2001 From: Moses Miller Date: Wed, 27 Dec 2017 16:10:14 -0800 Subject: [PATCH] Fix IndexError instead of StopIteration when MessageIterator is done --- disco/types/channel.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/disco/types/channel.py b/disco/types/channel.py index 2419bec..ef0c17e 100644 --- a/disco/types/channel.py +++ b/disco/types/channel.py @@ -548,6 +548,9 @@ class MessageIterator(object): if not len(self._buffer): self.fill() + if not len(self._buffer): + raise StopIteration + if self.bulk: res = self._buffer self._buffer = []