python使用requestssession模拟登录

jiezi

5 年前

最近开发一套接口，写个 Python 脚本，使用 requests.session 模拟一下登录.
因为每次需要获取用户信息，登录需要带着 session 信息，所以所有请求需要带着 session。
请求使用 post 方式，请求参数类型为 raw 方式，参数为 json 类型。
登录接口参数和结果如下：

脚本如下:

1. 引入需要的第三方包

#! /usr/bin/env python3
# -*- coding: utf-8 -*-

import requests
# import re
import json
# from bs4 import BeautifulSoup

2. 请求头设置，设置请求参数, 需要 Python 字典转为 json 字符串

url = 'http://beanhome.com/user/login'
header = {
        "Content-Type": 'application/json',
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/76.0.3809.87 Safari/537.36"
    }
# Python 字典数据转为 json，需要使用 json.dumps
data = {
        "email": "common@moviebook.com",
        "passwd": "123456"
    }
# 通过 session 模拟登录，每次请求带着 session
sess = requests.Session()
f = sess.post(url, data=json.dumps(data), headers=header)

3. 打印登录成功后结果

print(json.loads(f.text))
# soup = BeautifulSoup(f.content, "html.parser")
# {'status': 1, 'msg': '操作成功', 'time': 1565317698, 'element': {'id': 1, 'uid': 1, 'name': 'common', 'email': 'common@moviebook.com', 'company': '客户公司', 'type': 1, 'title': '普通用户'}}

4. 打印菜单结果

url = 'http://beanhome.com/user/getMenu'
# 获取菜单
f = sess.post(url,headers=header)
print(json.loads(f.text))

5. 打印配置选项结果

# 获取配置选项
url= 'http://beanhome.com/user/getOptionList'

result = sess.post(url, headers=header)
# print(json.loads(result.text))
for j,k in json.loads(result.text)['element'].items():
    print(j,k)