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()