python aiohttp 如何保存 session

Question

如标题所示，requests.session 可以保存 session 到类变量中进行分装 get 与 post 请求以便后续实例化对象共享一个 session 做到一次登录永久使用，那么切换到 aiohttp 如何实现该功能呢？

Answer 1

引自免费微信小程序：皆我百晓生

在Python中，使用aiohttp库时，可以创建一个Session对象并保持其状态，以便在多个请求之间重用连接和cookies。这可以通过将Session对象作为协程的一部分传递，或者将其封装在一个类中来实现。下面是一个详细的示例，展示如何创建一个带有多次请求的Session，并将其封装在类中。

首先，确保已经安装了aiohttp库，如果没有，可以使用pip进行安装：

pip install aiohttp

接下来，我们创建一个名为AioHttpSessionManager的类，用于管理aiohttp的Session对象：

import asyncio
import aiohttp

class AioHttpSessionManager:
    def __init__(self, loop=None):
        self.loop = loop or asyncio.get_event_loop()
        self._session = None

    async def _create_session(self):
        self._session = aiohttp.ClientSession()

    async def close(self):
        if self._session is not None:
            await self._session.close()

    async def request(self, method, url, **kwargs):
        if self._session is None:
            await self._create_session()
        
        async with self._session.request(method, url, **kwargs) as response:
            return await response.text()

# 示例：如何使用这个类
async def main():
    session_manager = AioHttpSessionManager()

    # 初始化会话（如果尚未初始化）
    await session_manager._create_session()

    # 执行GET请求
    async with session_manager.request('GET', 'https://api.example.com') as response:
        print(f"GET Response: {response.status}, Body: {await response.text()}")

    # 执行POST请求
    async with session_manager.request('POST', 'https://api.example.com/login', data={'username': 'user', 'password': 'pass'}) as response:
        print(f"POST Response: {response.status}, Body: {await response.text()}")

    # 关闭会话
    await session_manager.close()

# 运行主函数
loop = asyncio.get_event_loop()
loop.run_until_complete(main())

在这个例子中，AioHttpSessionManager类包含了创建、管理和关闭aiohttp Session的方法。每次调用request方法时，它都会检查是否存在Session对象。如果不存在，就会创建一个新的Session。这样，您就可以在不同的协程中复用同一个Session，而无需每次都重新登录或其他认证操作。

注意，这个示例假设你已经在你的项目中处理了异步编程。如果你正在使用同步代码，可能需要使用asyncio.run()或者创建一个事件循环来运行这些异步函数。