Python-接口的头部token会话保持，不用每次接口都写入headers

Posted 2021-05-20 小哈别闹

在接口测试的时候，经常会需要登录之后获取到token授权之后才能进行接下来的测试工作

一般token信息都是放在header中的，如果每个接口都编辑一次header信息，就会很麻烦

在python中，requests库可以对头部信息进行保持，在接下来的测试中就不需要再次编辑就能保持token

# coding=utf-8
import requests

def get_token(s,user="test",pwd="123456"):
    url = "登录的接口url"
    params = {
        "username": user,
        "password": pwd
    }

    #登陆
    res = s.post(url, json=params)
    token = res.json().get("token")

    #更新session会话的头部
    headers ={
            "Authorization": "Token %s" %token
        }
    s.headers.update(headers)
    return res

if __name__ == \'__main__\':
    s = requests.Session()
    res = get_token(s)
    print(res.json().get(\'token\'))
    print(s.headers)