python - 如何在 Python 3 中为 xmlrpc.client 保留 cookie？

Question

默认的 Python xmlrpc.client.Transport(可与 xmlrpc.client.ServerProxy 一起使用)不保留 cookie，基于 cookie 的登录有时需要这些 cookie。

例如，以下代理在与 TapaTalk API(login method 使用 cookie 进行身份验证)一起使用时，将在尝试修改帖子时给出权限错误。

proxy = xmlrpc.client.ServerProxy(URL, xmlrpc.client.Transport())

有some solutions适用于网络上的 Python 2，但它们与 Python 3 不兼容。

如何使用保留 cookie 的 Transport？

Answer 1

GermainZ 的现有答案仅适用于 HTTP。折腾了很久，终于有了 HTTPS 适配。请注意至关重要的 context 选项。

class CookiesTransport(xmlrpc.client.SafeTransport):
    """A SafeTransport (HTTPS) subclass that retains cookies over its lifetime."""

    # Note context option - it's required for success
    def __init__(self, context=None):
        super().__init__(context=context)
        self._cookies = []

    def send_headers(self, connection, headers):
        if self._cookies:
            connection.putheader("Cookie", "; ".join(self._cookies))
        super().send_headers(connection, headers)

    def parse_response(self, response):
        # This check is required if in some responses we receive no cookies at all
        if response.msg.get_all("Set-Cookie"):
            for header in response.msg.get_all("Set-Cookie"):
                cookie = header.split(";", 1)[0]
                self._cookies.append(cookie)
        return super().parse_response(response)

原因是 ServerProxy 不考虑与传输相关的 context 选项，如果指定了传输，我们需要直接在传输构造函数中使用它。

用法:

import xmlrpc.client
import ssl


transport = CookiesTransport(context=ssl._create_unverified_context())
# Note the closing slash in address as well, very important
server = xmlrpc.client.ServerProxy("https://<api_link>/", transport=transport)

# do stuff with server
server.myApiFunc({'param1': 'x', 'param2': 'y'})