From 7fd236352a678b47b7fc890466d813430d67b8cc Mon Sep 17 00:00:00 2001 From: Lev Vereshchagin Date: Thu, 13 Jun 2024 11:42:01 +0300 Subject: [PATCH] Handle `socket.gaierror` in `Connection.connect()` --- stompman/connection.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stompman/connection.py b/stompman/connection.py index 58be58fb..0282ac94 100644 --- a/stompman/connection.py +++ b/stompman/connection.py @@ -1,4 +1,5 @@ import asyncio +import socket from collections.abc import AsyncGenerator, Iterator from dataclasses import dataclass, field from typing import Protocol, Self, TypedDict, TypeVar, cast @@ -96,7 +97,7 @@ async def connect(self) -> bool: asyncio.open_connection(self.connection_parameters.host, self.connection_parameters.port), timeout=self.connect_timeout, ) - except (TimeoutError, ConnectionError): + except (TimeoutError, ConnectionError, socket.gaierror): return False return True