目录
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)}