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

本文介绍了如何使用Python的requests库结合session管理cookie进行登录操作,并在登录成功后自动携带cookie访问用户中心。重点在于request.session()会自动处理cookie,无需手动传递csfrtoken和sessionid。

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")

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值