python+requests+unittest 搭建接口自动化测试框架(四)

自动获取cookie的方法  autoLogin.py

import requests
import json
import os
cookie_file = os.path.join(os.getcwd()+'/cookies.txt')
session = requests.session()
headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.110 Safari/537.36'}

def cookies_read():

    cookies_txt = open(cookie_file, 'r')
    cookies_dict = json.loads(cookies_txt.read())
    #cookies = requests.utils.cookiejar_from_dict(cookies_dict)
    return cookies_dict
    # 以上4行代码，是cookies读取。

def sign_in():
    url = ' https://wordpress-edu-3autumn.localprod.oc.forchange.cn/wp-login.php'
    data = {'log': input('请输入你的账号'),
            'pwd': input('请输入你的密码'),
            'wp-submit': '登录',
            'redirect_to': 'https://wordpress-edu-3autumn.localprod.oc.forchange.cn',
            'testcookie': '1'}
    session.post(url, headers=headers, data=data)
    cookies_str = json.dumps(session.cookies)
    f = open(cookie_file, 'w')
    f.write(cookies_str)
    f.close()
    # 以上5行代码，是cookies存储。

try:
    session.cookies = cookies_read()
except FileNotFoundError:
    sign_in()