catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Wrap botocore patching in tyr/except since we do not require it to be installed.

Browse source
This commit is contained in:
Steve Pulec 2016-03-02 15:20:35 -05:00
commit 75a40e4571
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
moto/__init__.py
View file

@ -33,7 +33,11 @@ from .route53 import mock_route53 # flake8: noqa
from .swf import mock_swf # flake8: noqa
# Need to monkey-patch botocore requests back to underlying urllib3 classes
from botocore.awsrequest import HTTPSConnectionPool, HTTPConnectionPool, HTTPConnection, VerifiedHTTPSConnection
HTTPSConnectionPool.ConnectionCls = VerifiedHTTPSConnection
HTTPConnectionPool.ConnectionCls = HTTPConnection
try:
# Need to monkey-patch botocore requests back to underlying urllib3 classes
from botocore.awsrequest import HTTPSConnectionPool, HTTPConnectionPool, HTTPConnection, VerifiedHTTPSConnection
except ImportError:
pass
else:
HTTPSConnectionPool.ConnectionCls = VerifiedHTTPSConnection
HTTPConnectionPool.ConnectionCls = HTTPConnection