91 lines
2.0 KiB
Python
91 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
通用工具函数
|
||
包含任务ID生成、参数验证等通用功能
|
||
"""
|
||
|
||
import uuid
|
||
import hashlib
|
||
import time
|
||
import re
|
||
from typing import Optional
|
||
|
||
|
||
def generate_task_id(prefix: str = "") -> str:
|
||
"""生成唯一的任务ID
|
||
|
||
Args:
|
||
prefix: 任务ID前缀
|
||
|
||
Returns:
|
||
生成的任务ID
|
||
"""
|
||
timestamp = str(int(time.time() * 1000))
|
||
random_id = str(uuid.uuid4()).replace("-", "")
|
||
combined = timestamp + random_id
|
||
|
||
# 使用MD5哈希生成32位字符串
|
||
task_id = hashlib.md5(combined.encode()).hexdigest()
|
||
|
||
if prefix:
|
||
return f"{prefix}_{task_id}"
|
||
return task_id
|
||
|
||
|
||
def validate_text_input(text: str, max_length: int = 10000) -> tuple[bool, str]:
|
||
"""验证输入文本
|
||
|
||
Args:
|
||
text: 待验证的文本
|
||
max_length: 最大长度限制
|
||
|
||
Returns:
|
||
(is_valid, message): 验证结果和消息
|
||
"""
|
||
if not text or not text.strip():
|
||
return False, "文本内容不能为空"
|
||
|
||
text = text.strip()
|
||
|
||
if len(text) > max_length:
|
||
return False, f"文本长度超过限制，最大支持{max_length}个字符"
|
||
|
||
# 检查是否包含有效字符
|
||
if not re.search(r"[\u4e00-\u9fff\w\s]", text):
|
||
return False, "文本内容无效，请输入有效的中文、英文或数字"
|
||
|
||
return True, "验证通过"
|
||
|
||
|
||
def parse_language_code(lang_code: Optional[str]) -> str:
|
||
"""解析语言代码
|
||
|
||
Args:
|
||
lang_code: 语言代码（如 zh, zh-cn, en, ja等）
|
||
|
||
Returns:
|
||
标准化的语言代码
|
||
"""
|
||
if not lang_code:
|
||
return "zh" # 默认中文
|
||
|
||
lang_code = lang_code.lower().strip()
|
||
|
||
# 语言代码映射
|
||
lang_mapping = {
|
||
"zh": "zh",
|
||
"zh-cn": "zh",
|
||
"zh-tw": "zh",
|
||
"zh-hk": "zh",
|
||
"en": "en",
|
||
"en-us": "en",
|
||
"en-gb": "en",
|
||
"ja": "jp",
|
||
"jp": "jp",
|
||
"ko": "kr",
|
||
"kr": "kr",
|
||
"yue": "yue", # 粤语
|
||
}
|
||
|
||
return lang_mapping.get(lang_code, "zh")
|