Python aiohttp 模块：进阶使用

Travis Tang

2024-07-25

Python aiohttp 模块进阶使用

aiohttp 是一个用于处理网络请求的强大库，支持异步编程，允许用户以非阻塞的方式处理 HTTP 请求。它适配 Python 3.6 及以上版本，深受开发者的喜爱。使用 aiohttp，您可以轻松实现高并发的网络访问，适用于各种 web 应用的开发。

模块介绍

aiohttp 是一个基于 asyncio 库的异步 HTTP 客户端和服务器框架，提供了简单易用的 API，使得编写非阻塞的网络通信变得更加轻松。支持 HTTP 和 WebSocket，能够让开发者以异步的方式处理网络请求。

应用场景

aiohttp 适合于需要处理大量并发请求的场景，如：

爬虫程序：在进行数据采集时，可以同时连接多个网站，提高爬取效率。
微服务架构：在微服务间的通讯需要高并发时，aiohttp 是一个理想选择。
实时应用：例如实时聊天应用中的 WebSocket 通信，aiohttp 提供良好的支持。

安装说明

aiohttp 不是 Python 的默认模块，需要使用 pip 进行安装。执行以下命令进行安装：

1	pip install aiohttp

用法举例

1. 示例一：基本的 GET 请求

import aiohttp  # 导入 aiohttp 库
import asyncio  # 导入 asyncio 库

async def fetch(url):  # 定义异步函数 fetch，接收 URL 参数
    async with aiohttp.ClientSession() as session:  # 创建 aiohttp 会话
        async with session.get(url) as response:  # 发送 GET 请求
            return await response.text()  # 返回响应内容

url = 'https://httpbin.org/get'  # 定义请求的 URL
result = asyncio.run(fetch(url))  # 运行异步函数
print(result)  # 打印返回的内容

在这个示例中，我们使用 aiohttp 库发送了一个 GET 请求，并获取了响应内容。

2. 示例二：处理多个请求

import aiohttp  # 导入 aiohttp 库
import asyncio  # 导入 asyncio 库

async def fetch(url):  # 定义异步函数 fetch
    async with aiohttp.ClientSession() as session:  # 创建会话
        async with session.get(url) as response:  # 发送 GET 请求
            return await response.json()  # 解析为 JSON 格式

async def main():  # 定义主函数
    urls = ['https://httpbin.org/get', 'https://httpbin.org/ip', 'https://api.github.com']  # 定义 URL 列表
    tasks = [fetch(url) for url in urls]  # 创建任务列表
    results = await asyncio.gather(*tasks)  # 并发执行任务
    print(results)  # 打印所有响应结果

asyncio.run(main())  # 运行主函数

这里我们演示了如何并发处理多个请求，利用 asyncio.gather 同时发送多个 GET 请求。

3. 示例三：使用 WebSocket 进行实时通信

import aiohttp  # 导入 aiohttp 库
import asyncio  # 导入 asyncio 库

async def listen_websocket(url):  # 定义监听 WebSocket 的异步函数
    async with aiohttp.ClientSession() as session:  # 创建会话
        async with session.ws_connect(url) as ws:  # 连接 WebSocket
            async for msg in ws:  # 逐条监听消息
                print(f"Received: {msg.data}")  # 打印接收到的消息

url = 'wss://example.websocket.server'  # 定义 WebSocket URL
asyncio.run(listen_websocket(url))  # 运行监听函数

在这个示例中，我们展示了如何使用 aiohttp 库通过 WebSocket 进行实时消息监听。

强烈建议大家关注本人的博客全糖冲击博客，我的博客详细讲解了所有 Python 标准库的使用教程，内容丰富、分类清晰，非常适合学习和查询。关注我的博客，您将能快速掌握 Python 编程的各个方面，提升自己的编程能力，帮助您在职场中更加游刃有余。让我们一起探索 Python 的无限可能吧！