From c9174a22d2194f6bd4636655eec368c85ae53c64 Mon Sep 17 00:00:00 2001 From: GinHa <15201596918@163.com> Date: Fri, 3 Jul 2026 16:33:48 +0800 Subject: [PATCH] =?UTF-8?q?=E9=94=99=E8=AF=AF=E4=BF=A1=E6=81=AF=E6=88=AA?= =?UTF-8?q?=E6=96=AD=E8=BF=BD=E5=8A=A05000=E5=AD=97=E7=AC=A6=20|=20?= =?UTF-8?q?=E6=97=A5=E5=BF=97=E4=BF=A1=E6=81=AF=E4=B8=8D=E5=9C=A8=E5=8A=A0?= =?UTF-8?q?=E5=AF=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- video-gen-api/app/middleware/logging.py | 2 +- video-gen-api/app/services/error_codes.py | 2 +- video-gen-api/app/services/image_gen.py | 4 ++-- video-gen-api/app/services/llm.py | 4 ++-- video-gen-api/app/services/log_config.py | 7 +++++-- video-gen-api/app/services/video_gen.py | 4 ++-- 6 files changed, 13 insertions(+), 10 deletions(-) diff --git a/video-gen-api/app/middleware/logging.py b/video-gen-api/app/middleware/logging.py index 21b066e5..08ba7f84 100644 --- a/video-gen-api/app/middleware/logging.py +++ b/video-gen-api/app/middleware/logging.py @@ -56,7 +56,7 @@ class RequestLoggingMiddleware(BaseHTTPMiddleware): "method": request.method, "path": request.url.path, "query_params": dict(request.query_params), - "request_body": encrypt_data(request_body) if request_body else "", + "request_body": encrypt_data(request_body, True) if request_body else "", "status": response.status_code, "duration_ms": duration_ms, "ip": request.client.host if request.client else "-", diff --git a/video-gen-api/app/services/error_codes.py b/video-gen-api/app/services/error_codes.py index bc4d6a61..dab0720d 100644 --- a/video-gen-api/app/services/error_codes.py +++ b/video-gen-api/app/services/error_codes.py @@ -159,4 +159,4 @@ def extract_error_message(exc: Exception, service_type: str = "video") -> str: return f"{service_type}生成失败: {code}" except (json.JSONDecodeError, TypeError, ValueError): pass - return raw[:200] if len(raw) > 200 else raw \ No newline at end of file + return raw[:5000] if len(raw) > 5000 else raw \ No newline at end of file diff --git a/video-gen-api/app/services/image_gen.py b/video-gen-api/app/services/image_gen.py index f5529af3..e09f668a 100644 --- a/video-gen-api/app/services/image_gen.py +++ b/video-gen-api/app/services/image_gen.py @@ -30,7 +30,7 @@ def _log_image_request(engine: ProviderImageEngineLike, record_id: str, request_ today = datetime.now().strftime(LOG_DATE_FORMAT) log_file = os.path.join(LOG_DIR, f"{today}.log") request_str = json.dumps(request_data, ensure_ascii=False) - request_encrypted = encrypt_data(request_data) + request_encrypted = encrypt_data(request_data, True) entry = { "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "type": "image_gen_request", @@ -54,7 +54,7 @@ def _log_image_response(record_id: str, response_data: dict, error: str | None = os.makedirs(LOG_DIR, exist_ok=True) today = datetime.now().strftime(LOG_DATE_FORMAT) log_file = os.path.join(LOG_DIR, f"{today}.log") - response_encrypted = encrypt_data(response_data) if response_data else "" + response_encrypted = encrypt_data(response_data, True) if response_data else "" entry = { "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "type": "image_gen_response", diff --git a/video-gen-api/app/services/llm.py b/video-gen-api/app/services/llm.py index 8f1447bb..41c71dcc 100644 --- a/video-gen-api/app/services/llm.py +++ b/video-gen-api/app/services/llm.py @@ -37,8 +37,8 @@ def _log_ai_request_response(config, request_data: dict, response_data: dict | N os.makedirs(LOG_DIR, exist_ok=True) today = datetime.now().strftime(LOG_DATE_FORMAT) log_file = os.path.join(LOG_DIR, f"{today}.log") - request_encrypted = encrypt_data(_sanitize_for_log(request_data)) - response_encrypted = encrypt_data(_sanitize_for_log(response_data)) if response_data else "" + request_encrypted = encrypt_data(_sanitize_for_log(request_data), True) + response_encrypted = encrypt_data(_sanitize_for_log(response_data), True) if response_data else "" entry = { "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "model_name": config.name, diff --git a/video-gen-api/app/services/log_config.py b/video-gen-api/app/services/log_config.py index c73593f5..c584275e 100644 --- a/video-gen-api/app/services/log_config.py +++ b/video-gen-api/app/services/log_config.py @@ -18,10 +18,13 @@ ENCRYPTION_KEY = b'videogen@202605!' -def encrypt_data(data: dict) -> str: +def encrypt_data(data: dict, is_encrypt: bool = False) -> str: data_str = json.dumps(data, ensure_ascii=False, sort_keys=True) + if is_encrypt: + return data_str + data_bytes = data_str.encode("utf-8") - + padder = padding.PKCS7(128).padder() padded_data = padder.update(data_bytes) + padder.finalize() diff --git a/video-gen-api/app/services/video_gen.py b/video-gen-api/app/services/video_gen.py index 147cc741..3c12e3ef 100644 --- a/video-gen-api/app/services/video_gen.py +++ b/video-gen-api/app/services/video_gen.py @@ -30,7 +30,7 @@ def _log_video_request(engine: ProviderVideoEngineLike, record_id: str, request_ today = datetime.now().strftime(LOG_DATE_FORMAT) log_file = os.path.join(LOG_DIR, f"{today}.log") request_str = json.dumps(request_data, ensure_ascii=False) - request_encrypted = encrypt_data(request_data) + request_encrypted = encrypt_data(request_data, True) entry = { "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "type": "video_gen_request", @@ -54,7 +54,7 @@ def _log_video_response(record_id: str, response_data: dict, error: str | None = os.makedirs(LOG_DIR, exist_ok=True) today = datetime.now().strftime(LOG_DATE_FORMAT) log_file = os.path.join(LOG_DIR, f"{today}.log") - response_encrypted = encrypt_data(response_data) if response_data else "" + response_encrypted = encrypt_data(response_data, True) if response_data else "" entry = { "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),