76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
"""
|
|
Adds asyncio support to Zeep. Contains Python 3.5+ only syntax!
|
|
|
|
"""
|
|
import asyncio
|
|
import logging
|
|
|
|
import aiohttp
|
|
from zeep.transports import Transport
|
|
from zeep.utils import get_version
|
|
from zeep.wsdl.utils import etree_to_string
|
|
|
|
__all__ = ['AsyncTransport']
|
|
|
|
|
|
class AsyncTransport(Transport):
|
|
"""Asynchronous Transport class using aiohttp."""
|
|
supports_async = True
|
|
|
|
def __init__(self, loop, cache=None, timeout=300, operation_timeout=None,
|
|
session=None):
|
|
|
|
self.loop = loop if loop else asyncio.get_event_loop()
|
|
self.cache = cache
|
|
self.load_timeout = timeout
|
|
self.operation_timeout = operation_timeout
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
self.session = session or aiohttp.ClientSession(loop=self.loop)
|
|
self.session._default_headers['User-Agent'] = (
|
|
'Zeep/%s (www.python-zeep.org)' % (get_version()))
|
|
|
|
def _load_remote_data(self, url):
|
|
result = None
|
|
|
|
async def _load_remote_data_async():
|
|
nonlocal result
|
|
with aiohttp.Timeout(self.load_timeout):
|
|
response = await self.session.get(url)
|
|
result = await response.read()
|
|
|
|
# Block until we have the data
|
|
self.loop.run_until_complete(_load_remote_data_async())
|
|
return result
|
|
|
|
async def post(self, address, message, headers):
|
|
self.logger.debug("HTTP Post to %s:\n%s", address, message)
|
|
with aiohttp.Timeout(self.operation_timeout):
|
|
response = await self.session.post(
|
|
address, data=message, headers=headers)
|
|
self.logger.debug(
|
|
"HTTP Response from %s (status: %d):\n%s",
|
|
address, response.status, await response.read())
|
|
return response
|
|
|
|
async def post_xml(self, address, envelope, headers):
|
|
message = etree_to_string(envelope)
|
|
response = await self.post(address, message, headers)
|
|
|
|
from pretend import stub
|
|
return stub(
|
|
content=await response.read(),
|
|
status_code=response.status,
|
|
headers=response.headers)
|
|
|
|
async def get(self, address, params, headers):
|
|
with aiohttp.Timeout(self.operation_timeout):
|
|
response = await self.session.get(
|
|
address, params=params, headers=headers)
|
|
|
|
from pretend import stub
|
|
return await stub(
|
|
content=await response.read(),
|
|
status_code=response.status,
|
|
headers=response.headers)
|