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="3304496257@qq.com",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)#<RequestsCookieJar[<Cookie csrftoken=kdo41TZFVcyzHpCreD0ZtvDAI2KhqOX0Ivu5AijDY6kS1CzlSy4UNJPAVm0ZFoId for 49.235.92.12/>, <Cookie sessionid=hlv8oylsk8csjgbqsjv4tka2c55fjl8f for 49.235.92.12/>]>
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) #<RequestsCookieJar[<Cookie csrftoken=kdo41TZFVcyzHpCreD0ZtvDAI2KhqOX0Ivu5AijDY6kS1CzlSy4UNJPAVm0ZFoId for 49.235.92.12/>, <Cookie sessionid=hlv8oylsk8csjgbqsjv4tka2c55fjl8f for 49.235.92.12/>]>
#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)#<RequestsCookieJar[<Cookie csrftoken=xoApuRnAR77p4jL3pMG56rafRZsOSWowoNYeN9HxkUCM0W8D7ZMxdlE6urSWuTtv for 49.235.92.12/>, <Cookie sessionid=ckdle629t0g2cyojmm56d5oxksdbbe5x for 49.235.92.12/>]>
#运行完login函数,打印cookie,有csrfttoken和sessionid,再运行usercenter函数返回的数据就是个人信息页面,不带cookie返回的是未登录的数据
usercenter(s,base_url="http://199.235.92.12:8200")
本文介绍了如何使用Python的requests库结合session管理cookie进行登录操作,并在登录成功后自动携带cookie访问用户中心。重点在于request.session()会自动处理cookie,无需手动传递csfrtoken和sessionid。

1321

被折叠的 条评论
为什么被折叠?



