Skip to content

[Core] Update Transpot __Init__ #25344

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Kashif Khan (kashifkhan) merged 9 commits into from
Aug 3, 2022
Merged

[Core] Update Transpot __Init__ #25344

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
c684cda
remove 3.7 if check
kashifkhan Jul 19, 2022
f7dd711
interim changes
kashifkhan Jul 19, 2022
2c2278c
fix return
kashifkhan Jul 21, 2022
1f3787f
fix
kashifkhan Jul 21, 2022
a1cb2b6
fixes
kashifkhan Jul 22, 2022
6797365
handle requests not being installed for trio
kashifkhan Jul 22, 2022
618d9fd
pylint fixes
kashifkhan Aug 1, 2022
cec8209
pylint fixes
kashifkhan Aug 1, 2022
ed8c227
disable some pylint
kashifkhan Aug 1, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
195 changes: 71 additions & 124 deletions sdk/core/azure-core/azure/core/pipeline/transport/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,137 +24,84 @@
#
# --------------------------------------------------------------------------

import sys
from ._base import HttpTransport, HttpRequest, HttpResponse
from ._base_async import AsyncHttpTransport, AsyncHttpResponse

# pylint: disable=undefined-all-variable

__all__ = [
'HttpTransport',
'HttpRequest',
'HttpResponse',
'RequestsTransport',
'RequestsTransportResponse',
'AsyncHttpTransport',
'AsyncHttpResponse',
'AsyncioRequestsTransport',
'AsyncioRequestsTransportResponse'
'TrioRequestsTransport',
'TrioRequestsTransportResponse',
'AioHttpTransport',
'AioHttpTransportResponse',
]

# pylint: disable=unused-import, redefined-outer-name
try:
from ._requests_basic import RequestsTransport, RequestsTransportResponse
__all__.extend([
'RequestsTransport',
'RequestsTransportResponse',
])
try:
from ._base_async import AsyncHttpTransport, AsyncHttpResponse
from ._requests_asyncio import AsyncioRequestsTransport, AsyncioRequestsTransportResponse

__all__.extend([
'AsyncHttpTransport',
'AsyncHttpResponse',
'AsyncioRequestsTransport',
'AsyncioRequestsTransportResponse'
])

if sys.version_info >= (3, 7):
__all__.extend([
'TrioRequestsTransport',
'TrioRequestsTransportResponse',
'AioHttpTransport',
'AioHttpTransportResponse',
])

def __dir__():
return __all__

def __getattr__(name):
if name == 'AioHttpTransport':
try:
from ._aiohttp import AioHttpTransport
return AioHttpTransport
except ImportError:
raise ImportError("aiohttp package is not installed")
if name == 'AioHttpTransportResponse':
try:
from ._aiohttp import AioHttpTransportResponse
return AioHttpTransportResponse
except ImportError:
raise ImportError("aiohttp package is not installed")
if name == 'TrioRequestsTransport':
try:
from ._requests_trio import TrioRequestsTransport
return TrioRequestsTransport
except ImportError:
raise ImportError("trio package is not installed")
if name == 'TrioRequestsTransportResponse':
try:
from ._requests_trio import TrioRequestsTransportResponse
return TrioRequestsTransportResponse
except ImportError:
raise ImportError("trio package is not installed")
if name == '__bases__':
raise AttributeError("module 'azure.core.pipeline.transport' has no attribute '__bases__'")
return name

else:
try:
from ._requests_trio import TrioRequestsTransport, TrioRequestsTransportResponse

__all__.extend([
'TrioRequestsTransport',
'TrioRequestsTransportResponse'
])
except ImportError:
pass # Trio not installed

try:
from ._aiohttp import AioHttpTransport, AioHttpTransportResponse

__all__.extend([
'AioHttpTransport',
'AioHttpTransportResponse',
])
except ImportError:
pass # Aiohttp not installed
except (ImportError, SyntaxError):
# requests library is installed but asynchronous pipelines not supported.
pass
except (ImportError, SyntaxError):
# requests library is not installed
try:
from ._base_async import AsyncHttpTransport, AsyncHttpResponse
__all__.extend([
'AsyncHttpTransport',
'AsyncHttpResponse',
])

if sys.version_info >= (3, 7):
__all__.extend([
'AioHttpTransport',
'AioHttpTransportResponse',
])

def __dir__():
return __all__
# pylint: disable=unused-import, redefined-outer-name, no-member, too-many-statements, too-many-branches

def __getattr__(name):
if name == 'AioHttpTransport':
try:
from ._aiohttp import AioHttpTransport
return AioHttpTransport
except ImportError:
raise ImportError("aiohttp package is not installed")
if name == 'AioHttpTransportResponse':
try:
from ._aiohttp import AioHttpTransportResponse
return AioHttpTransportResponse
except ImportError:
raise ImportError("aiohttp package is not installed")
return name
def __dir__():
return __all__

else:
try:
from ._aiohttp import AioHttpTransport, AioHttpTransportResponse
__all__.extend([
'AioHttpTransport',
'AioHttpTransportResponse',
])
except ImportError:
pass # Aiohttp not installed
except (ImportError, SyntaxError):
pass # Asynchronous pipelines not supported.
def __getattr__(name):
transport = None
if name == 'AsyncioRequestsTransport':
try:
from ._requests_asyncio import AsyncioRequestsTransport
transport = AsyncioRequestsTransport
except ImportError:
raise ImportError("requests package is not installed")
if name == 'AsyncioRequestsTransportResponse':
try:
from ._requests_asyncio import AsyncioRequestsTransportResponse
transport = AsyncioRequestsTransportResponse
except ImportError:
raise ImportError("requests package is not installed")
if name == 'RequestsTransport':
try:
from ._requests_basic import RequestsTransport
transport = RequestsTransport
except ImportError:
raise ImportError("requests package is not installed")
if name == 'RequestsTransportResponse':
try:
from ._requests_basic import RequestsTransportResponse
transport = RequestsTransportResponse
except ImportError:
raise ImportError("requests package is not installed")
if name == 'AioHttpTransport':
try:
from ._aiohttp import AioHttpTransport
transport = AioHttpTransport
except ImportError:
raise ImportError("aiohttp package is not installed")
if name == 'AioHttpTransportResponse':
try:
from ._aiohttp import AioHttpTransportResponse
transport = AioHttpTransportResponse
except ImportError:
raise ImportError("aiohttp package is not installed")
if name == 'TrioRequestsTransport':
try:
from ._requests_trio import TrioRequestsTransport
transport = TrioRequestsTransport
except ImportError as ex:
if ex.msg.endswith("'requests'"):
raise ImportError("requests package is not installed")
raise ImportError("trio package is not installed")
Comment thread
lmazuel marked this conversation as resolved.
if name == 'TrioRequestsTransportResponse':
try:
from ._requests_trio import TrioRequestsTransportResponse
transport = TrioRequestsTransportResponse
except ImportError:
raise ImportError("trio package is not installed")
if transport:
return transport
raise AttributeError(f"module 'azure.core.pipeline.transport' has no attribute {name}")