增减签名原文传入

watermark_generate/controller/watermark_generate_controller.py

@@ -50,6 +50,7 @@ def watermark_embed():
     model_value = data.get('model_value')
     model_type = data.get('model_type')
     embed_type = data.get('embed_type')
+    row_data = data.get('row_data')
 
     if embed_type is None or embed_type == '':  # 通过传入参数控制嵌入方式，默认为黑盒水印嵌入
         embed_type = 'blackbox'
@@ -60,6 +61,8 @@ def watermark_embed():
         raise BusinessException(message='模型值不可为空', code=-1)
     if model_type is None:
         raise BusinessException(message='模型类型不可为空', code=-1)
+    if row_data is None:
+        raise BusinessException(message='签名原文不可为空', code=-1)
 
     file_path = os.path.dirname(model_file)  # 获取文件路径
     file_name = os.path.basename(model_file)  # 获取文件名
@@ -74,8 +77,8 @@ def watermark_embed():
         zip_ref.extractall(extract_to_path)
     # 生成密码标签
     logger.info(f"generate secret label ...")
-    ts = str(int(time.time()))
-    secret_label, public_key = secret_label_func.generate_secret_label(ts)
+    # ts = str(int(time.time()))
+    secret_label, public_key = secret_label_func.generate_secret_label(row_data)
     logger.debug(f"generate secret label: {secret_label} , public key: {public_key}")
 
     # 修改模型文件代码，并将public_key写入至文件保存至修改后的工程文件目录中