Python接口自动化测试脚本-实现禅道登录

#!/usr/bin/env python

# -*- coding: UTF-8 -*-

'''=================================================

@Project -> File   ：NingMengProject -> PostZentao.py

@IDE    ：PyCharm

@Author ：孤问万古愁

@Date   ：2020/3/14 13:13

@Desc   ：

=================================================='''

import requests

import hashlib

import re

# 禅道登录流程：

# 1、打开登陆页面的时候页面产生一个zentaosid的唯一标识，以及一个verifyRand随机值

# 2、对输入的密码进行md5加密

# 3、已经加密的密码+verifyRand再一次md5加密

# 4、然后post请求登陆地址根据唯一的zentaosid验密

class GetToken():

    """获取禅道的token值及登录后的cookies"""

    def __init__(self, username='admin', pwd='admin123456'):  # 用户名、密码设置默认值

        self.username = username

        self.pwd = pwd

    def get_token(self):

        loginUrl = "http://127.0.0.1:8088/zentao/user-login.html"

        headers_1 = {"Content-Type": "application/x-www-form-urlencoded"}    # 定义headers

        get_sid = requests.get(loginUrl)   # get方法请求登录页面，用于获取SID

        get_sid.encoding = 'utf-8'

        verifyRand = re.findall("id='verifyRand' value='(\d+)'", get_sid.text)[0]  # 获取verifyRand值

        SID = get_sid.cookies["zentaosid"]

        # print(SID)

        hlFirst = hashlib.md5()

        hlFirst.update(self.pwd.encode(encoding='utf-8'))  # 第一次对密码进行加密

        # print('Md5 第一次加密结果 = ' + hlFirst.hexdigest())

        passwordResult = hlFirst.hexdigest() + verifyRand

        # print("passwordResult=" + passwordResult)

        hlLast = hashlib.md5()

        hlLast.update(passwordResult.encode(encoding='utf-8'))  # 第二次加密

        # print('Md5 第二次加密结果 = ' + hlLast.hexdigest())

        # 定义请求参数body

        bodyRequest = {"account": self.username, "password": hlLast.hexdigest(), "passwordStrength": 1,

                       "referer": "/zentao/", "verifyRand": verifyRand, "keepLogin": 1}

        # 定义cookies

        loginCookies = dict(zentaosid=SID, lang='zh-cn', keepLogin='on')

        loginRequest = requests.post(loginUrl, data=bodyRequest, cookies=loginCookies)

        token = loginRequest.cookies['zp']   # 从cookies中获取token

        # 测试一下：访问我的地盘页面

        test = requests.get("http://127.0.0.1:8088/zentao/my/",cookies=loginRequest.cookies)

        # print(test.text)

        # print(token)

        # # print(loginRequest.text)

        return token,loginRequest.cookies

if __name__ == '__main__':

    # GetToken().get_token()

    # print(GetToken().get_token())

    url="http://127.0.0.1:8088/zentao/product-create.html"

    text_1=requests.get(url,cookies=GetToken().get_token()[1]).text

    # print(text_1)

    uid = re.findall("var kuid = '(.*?)';", text_1)[0]

    headers_1 = {"Content-Type": "application/x-www-form-urlencoded"}

    body_te="name=&code=&line=0&PO=admin&QD=&RD=&type=normal&status=normal&desc=&acl=open&uid=5e6f56b6cb28f"

    body_add_product={"name":"禅道接口自动化","code":"ZenTaoInterface","line":0,"PO":"admin","QD":"","RD":"","type":"normal","status":"normal","desc":"","acl":"open","uid":uid}

    result = requests.post(url,data=body_add_product,headers=headers_1,cookies=GetToken().get_token()[1])

    print(result.text)