每次请求后更新 requests.session() 标头

Question

我正在使用requests.Session 为 RESTful API 编写 Python 客户端。 API 在每次请求后生成并返回一个新的 CSRFToken。 API 的开发人员告诉我，我需要像这样更新会话标头：

csrf = session.cookies.get('CSRFToken')
session.headers.update({'x-csrf-token': csrf})

如何在 Session 对象上绑定此功能，以便在每次请求后自动调用此功能（也许一个不错的继承解决方案可能是可能的），这样用户就不必在每次请求后调用它？

Answer 1

这是您可以做的一个最小示例：

import requests

class MySession(requests.Session):
    def __init__(self):
        super().__init__()
        self.headers = dict()
    def get(self, url):
        if (c := self.cookies.get('CSRFToken')):
            self.headers['x-csrf-token'] = c
        return super().get(url, headers=self.headers)


with MySession() as session:
    (r := session.get('https://www.google.com')).raise_for_status()