从0开始python学习-51.pytest之接口加密封装

目录

MD5加密

base64加密

rsa加密

---

MD5加密

1. 封装加密方法

def md5_encode(self,data):data = str(data).encode("utf-8")md5_data = hashlib.md5(data).hexdigest()return md5_data

2. 写入需要使用加密的接口yaml用例

-request:method: posturl: http://192.168.0.1/md5logindata:username: ${md5_encode(admin)} # 这里直接调用MD5的方法即可password: ${md5_encode(123)}

base64加密

1. 封装加密方法

def base64_encode(self,data):data = str(data).encode("utf-8")base64_data = base64.b64encode(data).decode("utf-8")return base64_data

2. 写入需要使用加密的接口yaml用例

-request:method: posturl: http://192.168.0.1/base64logindata:username: ${base64_encode(admin)} password: ${base64_encode(123)}

rsa加密

需要先进行下载  pip install rsa

1. 生成rsa使用的公钥和私钥

def create_key(self):# 生成1024位的公钥和私钥-RSA密钥至少为500位，密钥长度随着保密级别提高public_key,private_key = rsa.newkeys(1024)# 将生成的公钥和密码写入文件with open("./public_key.pem","w+") as f :f.write(public_key.save_pkcs1().decode())with open("./private_key.pem","w+") as f :f.write(private_key.save_pkcs1().decode())

2. 封装加密方法

这里读取的时候注意相对路径的问题

def rsa_encode(self,data):# 读取公钥with open("./public_key.pem") as f:public_key = rsa.PublicKey.load_pkcs1(f.read().encode())# 把data转为utf-8格式data = str(data).encode("utf-8")# 把字符串加密成byte类型byte_data = rsa.encrypt(data,public_key)# 把字节转为字符串base64_data = base64.b64encode(byte_data).decode("utf-8")

3. 写入需要使用加密的接口yaml用例

-request:method: posturl: http://192.168.0.1/rsalogindata:username: ${rsa_encode(admin)} password: ${rsa_encode(123)}