From 1bf2009f605d4df8d1b53fe7c90490dbcebbe9d0 Mon Sep 17 00:00:00 2001
From: Maciej Mazur <mamciek@gmail.com>
Date: Wed, 15 Jan 2020 12:54:39 +0100
Subject: [PATCH] Catch StopIteration raised by generator

---
 snakebite/client.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/snakebite/client.py b/snakebite/client.py
index d8ef7e3..5fc7ff7 100644
--- a/snakebite/client.py
+++ b/snakebite/client.py
@@ -1551,6 +1551,8 @@ def wrapped(self, *args, **kw):
                     self.__handle_request_error(e)
                 except socket.error as e:
                     self.__handle_socket_error(e)
+                except StopIteration as e:
+                    return e.value
         return wrapped
 
 HAClient._wrap_methods()