Ver código fonte

修改模拟密钥生成

liyan 1 ano atrás
pai
commit
86abd041c6
1 arquivos alterados com 8 adições e 5 exclusões
  1. 8 5
      tool/secret_func.py

+ 8 - 5
tool/secret_func.py

@@ -3,10 +3,12 @@
 """
 import random
 import string
+import secrets
 
 # 模拟256长度的十六进制字符串
 mock_secret_key = '921999081bdd6fe50d3a5700e714915c31458929647e1d115bf180024cb67f7b337824246b8f74b0eeff021d5631ea9a1ec118297d759d01165eeabe4ee5d02519118ecc7d4d6bef43af09b5956b0adf92adcf99186a05a2f160c3071345a7093bc0cb476f9313db3330471cd764ddfeccd22d3fa090ecdd98cc4c0c083173e6'
-
+mock_secret_key_50 = '6e887a550f5ba6a7733ec341f1692027936895ca4041a50828'
+mock_secret_key_20 = '08b9dc4ab8f22541588a'
 
 def get_secret(len):
     """
@@ -14,7 +16,7 @@ def get_secret(len):
     :param len: 标签长度
     :return: 生成的密码标签
     """
-    return mock_secret_key
+    return mock_secret_key_20
 
 
 def verify_secret(secret):
@@ -23,15 +25,16 @@ def verify_secret(secret):
     :param secret: 密码标签
     :return: 验证结果
     """
-    return secret == mock_secret_key
+    return secret == mock_secret_key_20
 
 
 def generate_random_string(length):
     """生成指定长度的随机字符串"""
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
+def generate_hex_string(length):
+    return ''.join(secrets.choice('0123456789abcdef') for _ in range(length))
 
 if __name__ == '__main__':
-    random_string = generate_random_string(256)
-    print(len(random_string))
+    random_string = generate_hex_string(20)
     print(random_string)