python+request.session关联cookie：登录并访问用户中心

1、访问用户中心需要登录接口返回的cookie中的csfrtoken和sessionid

2、用户中心需要的csrftoken和sessionid可以从登录login接口的response header里看到

3、pytest中提供的request.session会自动管理cookie，所以写login接口的时候要用

s=request.session()
r=s.post(url,data=body)

login接口请求成功后request.session会更新cookie。

4、第二个用户中心代码直接引用login代码即可，cookie会自动带过来，不用手动传参。

5、以下是参考代码

#login.py
import requests
import re

#登录
def login(s,base_url,username="[email protected]",password="123456"):
    url=base_url + "/users/login/"

    r = s.get(url)

    #获取token
    csrftoken=re.findall("name='csrfmiddlewaretoken' value='(.+?)'",r.text)
    print(csrftoken) #['TknBaMe6IRcrp618WIMh7iMyNITdZjUgCHT5OzI8G9oEUTsuIhSdsvAgULE0oJwd']

    body = {
        "username": username,
        "password": password,
        "csrfmiddlewaretoken": csrftoken[0]
    }
    r1 = s.post(url, data=body)
    #print(s.cookies)#, ]>
    return  r1

if __name__ == '__main__':
    s=requests.session()
    login(s,base_url="http://199.235.92.12:8200")
    #运行完login函数，打印cookie，可以看到cookie中已经有了csrftoken和sessionid
    #print(s.cookies) #, ]>

#usercenter.py
import requests
from online_work.api.login import login

#个人中心
def usercenter(s,base_url):
    url=base_url + "/users/userinfo/"

    r = s.get(url)

    print(r.text)#返回的数据包含个人信息，证明访问个人中心页面成功
    return  r

if __name__ == '__main__':
    s=requests.session()
    login(s,base_url="http://199.235.92.12:8200")
    print(s.cookies)#, ]>
    #运行完login函数，打印cookie，有csrfttoken和sessionid,再运行usercenter函数返回的数据就是个人信息页面，不带cookie返回的是未登录的数据
    usercenter(s,base_url="http://199.235.92.12:8200")