asyncio异步使用requests

以下内容为原创,转载请注明出处!

python requests是同步请求,由于我们在编写代码的时候用的是async异步,异步中应该避免使用同步代码,所以需要修改requests,让他支持异步请求。如下(以下代码主要是在Django中使用):

import asyncio

import functools

import requests

async def _get_requests(self, url, key):

    loop = asyncio.get_event_loop()

    # 需要用参数使用functools,不需要的话可以这样写:

    # r = await loop.run_in_executor(None, requests.get, url)

    r = await loop.run_in_executor(None, functools.partial(requests.get, url=url, auth=(settings.HARBOR_USERNAME, settings.HARBOR_PSW)))

    return r, key

async def _get_docker_index_info(self, context):

    tasks = []

    tasks.append(asyncio.ensure_future(self._get_requests(settings.HARBOR_URL + 'repositories?project_id=3', 'repositories')))

    tasks.append(asyncio.ensure_future(self._get_requests(settings.HARBOR_URL + 'projects/3/members', 'members')))

    tasks.append(asyncio.ensure_future(self._get_requests(settings.HARBOR_URL + 'projects/3/logs', 'logs')))

    for task in asyncio.as_completed(tasks):

        r, key = await task

        if r.status_code == 200:

            context[key] = r.json()

    return context

def get_context_data(self, **kwargs):

    context = super(DockerIndexView, self).get_context_data(**kwargs)

    loop = asyncio.new_event_loop()

    task = loop.create_task(self._get_docker_index_info(context))

    loop.run_until_complete(task)

    context = task.result()

    return context


以上是 asyncio异步使用requests 的全部内容, 来源链接: www.h5w3.com/116096.html

回到顶部