python接口自动化测试二十：函数写接口测试

# coding:utf-8
import requests
import re
from bs4 import BeautifulSoup

# s = requests.session()  # 全局的s

def get_token(s):
    '''
    fuction: 获取token
    args: s 参数 -》s = requests.session()
    :return  anti_token  ->{'X-Anit-Forge-Token': 'xx', 'X-Anit-Forge-Code': '38515842'}
   '''
    # 局部的s没定义，从外部传入s
    url = 'https://passport.lagou.com/login/login.html'
    h1 = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:44.0) Gecko/20100101 Firefox/44.0"
        }
    r1 = s.get(url, headers=h1, verify=False)
    # print(r1.text)

    soup = BeautifulSoup(r1.content, "html.parser", from_encoding='utf-8')
    tokenCode = {}
    try:
        t = soup.find_all('script')[1].get_text()
        print(t)
        tokenCode['X_Anti_Forge_Token'] = re.findall(r"Token = '(.+?)'", t)[0]
        tokenCode['X_Anti_Forge_Code'] = re.findall(r"Code = '(.+?)'", t)[0]
        return tokenCode
    except:
        print("获取token和code失败")
        tokenCode['X_Anti_Forge_Token'] = ""
        tokenCode['X_Anti_Forge_Code'] = ""
        return tokenCode


def login_lgw(s, anti_token, user, psw):

    url2 = 'https://passport.lagou.com/login/login.json'
    h2 = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:44.0) Gecko/20100101 Firefox/44.0",
        "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8",
        "X-Requested-With": "XMLHttpRequest",
        "X-Anit-Forge-Token": anti_token['X_Anti_Forge_Token'] ,
        "X-Anit-Forge-Code": anti_token['X_Anti_Forge_Code'],
        "Referer": "https://passport.lagou.com/login/login.html"
        }
    body ={
            "isValidate": "true",
            "username": user,
            "password": psw,
            "request_form_verifyCode": "",
            "submit": ""
        }
    print(s.headers)  # s的头部

    # 更新s的头部
    s.headers.update(h2)
    print(s.headers)
    r2 = s.post(url2, data=body, verify=False)
    print(r2.text)
    return r2.json()

if __name__ == "__main__":
    # 自测的内容
    s = requests.session()
    token = get_token(s)
    print(token)
    user = "1232"
    psw = "322222"
    login_lgw(s, token, user, psw)