python&pycryptodome:AES加密

最新推荐文章于 2024-09-19 12:06:30 发布

C域

最新推荐文章于 2024-09-19 12:06:30 发布

阅读量193

点赞数

分类专栏： pythonj接口文章标签：加密解密

本文链接：https://blog.youkuaiyun.com/qq_35404734/article/details/106598422

版权

pythonj接口专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用AES加密算法进行数据加密的方法，并通过Python的unittest模块进行单元测试验证。详细展示了如何设置密钥、初始化向量以及自定义填充方式，实现数据加密并将其转换为Base64编码。此外，通过一个具体示例，演示了如何发送加密后的数据到指定URL，并对接收到的响应进行断言检查，确保加密和解密过程正确无误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

'''
self.base_url = b'http://127.0.0.1:8000/'
self.key = b'W7v4D60fds2Cmk2U'
self.iv = b'1172311105789011'
phone = '18611001100'
url = b'api/sec_get_guest_list/'
'''
from Crypto.Cipher import AES
import requests,json,base64,unittest

class AESTest(unittest.TestCase):
	def setUp(self):
		self.base_url= b'http://127.0.0.1:8000/'
		self.key = b'W7v4D60fds2Cmk2U'
		self.iv = b'1172311105789011'

		b = 16
		self.pad = lambda s:s + (b - len(s) % b)*chr(b - len(s) % b)

	def encryptAES(self,src):
		cryptor = AES.new(self.key,AES.MODE_CBC,self.iv)
		src = self.pad(src).encode('utf-8')
		cipher_text =cryptor.encrypt(src)
		cipher_text_b64 = base64.urlsafe_b64encode(cipher_text)
		return cipher_text_b64

	def test_guest_success(self):
		url = b'api/sec_get_guest_list/'
		p = {'eid':1,'phone':'18611001100','asd':'drgedrgvejoi'}
		p_json = json.dumps(p)
		encode = self.encryptAES(p_json)
		print(1)
		print(encode)
		r = requests.post(self.base_url+url,data={'data':encode})

		result = r.json()

		self.assertEqual(result['status'],200)
		self.assertEqual(result['message'],'success')
		self.assertEqual(result['data']['realname'],'alan')
		self.assertEqual(result['data']['phone'],'18611001100')

if __name__ == '__main__':
	unittest.main()