improve error handling on timeouts

This commit is contained in:
Jannes Höke 2015-11-11 14:05:57 +01:00
parent 206802cf4d
commit 0b4c23f50e

View file

@ -20,6 +20,7 @@
"""This module contains methods to make POST and GET requests""" """This module contains methods to make POST and GET requests"""
import json import json
import socket
from ssl import SSLError from ssl import SSLError
try: try:
@ -111,11 +112,11 @@ def post(url,
message = _parse(error.read()) message = _parse(error.read())
raise TelegramError(message) raise TelegramError(message)
except SSLError as error: except (SSLError, socket.timeout) as error:
if "operation timed out" in error.message: if "operation timed out" in str(error):
raise TelegramError("Timed out") raise TelegramError("Timed out")
raise TelegramError(error.message) raise TelegramError(str(error))
return _parse(result) return _parse(result)