Import ConnectionFailure exception

This commit is contained in:
Sylvain Bellemare 2016-12-13 11:44:08 +01:00
parent d3d42747b7
commit 1f996c1746

View File

@ -2,7 +2,7 @@ import time
import logging import logging
from pymongo import MongoClient from pymongo import MongoClient
from pymongo.errors import ConnectionError from pymongo.errors import ConnectionFailure
import bigchaindb import bigchaindb
from bigchaindb.backend.connection import Connection from bigchaindb.backend.connection import Connection
@ -39,7 +39,7 @@ class MongoDBConnection(Connection):
for i in range(self.max_tries): for i in range(self.max_tries):
try: try:
self.conn = MongoClient(self.host, self.port) self.conn = MongoClient(self.host, self.port)
except ConnectionError as exc: except ConnectionFailure as exc:
if i + 1 == self.max_tries: if i + 1 == self.max_tries:
raise raise
else: else: