1. 程式人生 > 實用技巧 >python+requests+unittest 搭建介面自動化測試框架(四)

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