diff --git a/nut2.py b/nut2.py index 99ea8a0..c3f0adc 100644 --- a/nut2.py +++ b/nut2.py @@ -22,7 +22,7 @@ along with this program. If not, see . """ -import telnetlib +import telnetlib3 import logging @@ -75,7 +75,7 @@ def __del__(self): try: self._srv_handler.write(b"LOGOUT\n") self._srv_handler.close() - except (telnetlib.socket.error, AttributeError): + except (telnetlib3.socket.error, AttributeError): # The socket is already disconnected. pass @@ -94,7 +94,7 @@ def _connect(self): logging.debug("Connecting to host") try: - self._srv_handler = telnetlib.Telnet(self._host, self._port, + self._srv_handler = telnetlib3.Telnet(self._host, self._port, timeout=self._timeout) if self._login is not None: @@ -108,7 +108,7 @@ def _connect(self): result = self._srv_handler.read_until(b"\n", self._timeout).decode('utf-8') if not result == "OK\n": raise PyNUTError(result.replace("\n", "")) - except telnetlib.socket.error: + except telnetlib3.socket.error: raise PyNUTError("Socket error.") def description(self, ups):