Jelajahi Sumber

修改密码标签生成接口,并修改测试代码

liyan 9 bulan lalu
induk
melakukan
2baf322aed

+ 2 - 2
watermark_generate/controller/function_test.py

@@ -68,8 +68,8 @@ def test_generate_label():
     raw_data = data.get('raw_data')
     if raw_data is None:
         raise BusinessException(message='原文值不可为空', code=-1)
-    secret_label = secret_label_func.generate_secret_label(raw_data)
-    return jsonify({'secret_label': secret_label}), 200
+    secret_label, public_key = secret_label_func.generate_secret_label(raw_data)
+    return jsonify({'secret_label': secret_label, 'public_key': public_key}), 200
 
 
 @test.route('/test/verify_label', methods=['POST'])

+ 2 - 2
watermark_generate/tools/secret_label_func.py

@@ -14,8 +14,8 @@ def generate_secret_label(raw_data: str):
     """
     sign_data, public_key = sign_verify.get_sign(raw_data)
     # TODO 增加public_key保存公钥的流程
-    secret_label = json.dumps({'data': raw_data, 'sign': sign_data, 'public_key': public_key})
-    return secret_label
+    secret_label = json.dumps({'data': raw_data, 'sign': sign_data})
+    return secret_label, public_key
 
 
 def verify_secret_label(secret_label: str, public_key: str) -> bool: