robotframework实战三--自定义关键字

时间：2017-11-24 20:18:52 阅读：137 评论：0 收藏：0 [点我收藏+]

标签：rand type log div stat convert self param encrypt

在rf的实战1中，我的登录获取验证码就使用了自定义关键字，具体怎么做的，如下

1.新建文件夹

新建一个文件夹，我的MyLibrary,并且存放在site-packages下

技术分享图片

2.编写代码

在python的编译器中把需要的代码编写完成，并且运行成功,以下代码就是获取页面的session，生成随机数，拿到session+key+随机数，生成MD5，在和请求一起get请求，返回验证码，代码在本地编译没有问题

import random
import hashlib
import types
import requests
key="XXXXXXXXX"
class verificationCode(object):
    def __init__(self):
        self.encryptiontest = ‘‘
    def code(self,cookie):
     # 随机数
        letter = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
        requestId = []
        requestId.append(‘‘.join(random.sample(list(letter), 6)))
        randomNumber=‘‘.join(requestId)
        data = "a:1:{s:10:\"session_id\";s:26:\"" + cookie + "\";}"
        getsign = key + randomNumber + data  # 获得密匙
        # MD5加密
        m = hashlib.md5()
        m.update(getsign)
        self.encryptiontest=m.hexdigest()
        #get请求
        url = "https://rosewholesale.com/auto/api.php"
        textmod = {‘XXXX‘: ‘XXX‘, ‘act‘: ‘XXX‘, ‘requestId‘: randomNumber, ‘sign‘: self.encryptiontest, ‘data‘: data,‘ftc‘: ‘XXXX‘}
        r = requests.get(url, params=textmod)
        print "code:",r.status_code
        getcontent=r.text
        print u‘返回的内容:‘,getcontent
        converturl = str(getcontent)
        splitStr = converturl[-76:-70]
        print u‘验证码为:‘,splitStr
        return  splitStr

if __name__ == ‘__main__‘:
    verif=verificationCode()
    verif.code(‘aq3k8qphile1nrknarpnjb9eg0‘)