1. 程式人生 > >介面自動化之requests學習(七)--獲取token,並傳入下一個請求中

介面自動化之requests學習(七)--獲取token,並傳入下一個請求中

#!usr/bin/env python
#-*- coding:utf-8 -*-
"""
@author:Administrator
@file: requests_post.py
@time: 2018/10/13
"""
"""
獲取token並返回給下一個函式的方法
1.呼叫登入介面,拿到獲取的token
2.將拿到的token放在header中,呼叫確認登入介面
3.確認登入介面呼叫成功,介面返回資訊正常
"""
import requests
import login

headers = {"Content-Type": "application/json;charset=UTF-8"}

def login():
    """獲取token"""
    data = {"loginName":"***",
            "loginPassword":"***"
            }

    # cookies = {"graphi_vcode_flag":"d21673e7-a903-41ad-8ca4-86fe7f9e9d46"}

    url = "http://***/WebApi/PassCheck/LoginPassCheck"

    r = requests.post(url=url, json=data, headers=headers)
    #將獲取到的token返回
    return (r.json()["data"]["token"])


def confirm_login():
    """呼叫獲取登入資訊介面,將登入成功後,返回的token放在該請求的header中"""
    #將login()方法中返回的token放入header中
    headers["Token"] = login()
    r = requests.post(
        url="http://***/WebApi/Designer/QueryDesignerBasicInfor",
        headers=headers
    )
    print(r.headers)
    print(r.json())

confirm_login()