新增验签模拟接口

watermark_generate/controller/secret_controller.py

@@ -5,6 +5,7 @@ from flask import Blueprint, request
 from watermark_generate.tools import secret_func
 from watermark_generate.domain import *
 from watermark_generate.tools import logger_tool
+from watermark_generate.tools.secret_func import verify_signature
 
 secret = Blueprint('secret', __name__)
 logger = logger_tool.logger
@@ -61,8 +62,20 @@ def verify_crypto_label():
     verify_label_form = VerifyLabelFormSchema().load(request.json)
     label = verify_label_form.label
     info = verify_label_form.info
-    vert = verify_label_form.cert
+    cert = verify_label_form.cert
     result = secret_func.verify(label)
     logger.debug(f'验证结果：{result}')
     resp = VerifyLabelResp(code=0, msg='ok') if result else VerifyLabelResp(code=-1, msg='密码标签验证失败')
     return VerifyLabelRespSchema().dump(resp)
+
+
+@secret.route('/znwr/jit/ai/v1/verify-sign', methods=['POST'])
+def verify_signature_handle():
+    logger.debug(f'verify-sign request:{request.json}')
+    verify_signature_form = VerifySignatureFormSchema().load(request.json)
+    original = verify_signature_form.original
+    signature = verify_signature_form.signature
+    cert = verify_signature_form.cert
+    result = verify_signature(original=original, signature=signature, cert=cert)
+    resp = VerifyLabelResp(code=0, msg='ok') if result else VerifyLabelResp(code=-1, msg='验签失败')
+    return VerifyLabelRespSchema().dump(resp)

watermark_generate/domain/secret_domain.py

@@ -69,6 +69,17 @@ class Model:
         self.date = date
 
 
+class VerifySignatureForm:
+    """
+    验签功能请求体
+    """
+
+    def __init__(self, original, signature, cert):
+        self.original = original
+        self.signature = signature
+        self.cert = cert
+
+
 # 定义 OwnerSchema
 class OwnerSchema(ma.Schema):
     name = fields.String()
@@ -137,3 +148,13 @@ class VerifyLabelRespSchema(ma.Schema):
     @post_load
     def make_verify_label_resp(self, object, **kwargs):
         return VerifyLabelResp(**object)
+
+
+class VerifySignatureFormSchema(ma.Schema):
+    original = fields.String()
+    signature = fields.String()
+    cert = fields.String()
+
+    @post_load
+    def make_verify_signature(self, object, **kwargs):
+        return VerifySignatureForm(**object)