python+requests+unittest 搭建介面自動化測試框架(四)
阿新 • • 發佈:2020-11-03
自動獲取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()