refactor(user_utils): 简化拼音转换逻辑，移除备用方案，统一使用pypinyin库进行转换；优化正则表达式格式化
This commit is contained in:
Wenjie Zhang
parent
6d7e3ea2ef
commit
767a3d4762
@ -5,67 +5,17 @@
|
||||
|
||||
import re
|
||||
|
||||
try:
|
||||
from pypinyin import lazy_pinyin, Style
|
||||
PYPINYIN_AVAILABLE = True
|
||||
except ImportError:
|
||||
PYPINYIN_AVAILABLE = False
|
||||
from pypinyin import lazy_pinyin, Style
|
||||
|
||||
|
||||
def to_pinyin(text: str) -> str:
|
||||
"""
|
||||
将中文转换为拼音
|
||||
优先使用pypinyin库，如果不可用则使用简化版本
|
||||
使用pypinyin库进行转换
|
||||
"""
|
||||
if PYPINYIN_AVAILABLE:
|
||||
# 使用pypinyin进行转换
|
||||
pinyin_list = lazy_pinyin(text, style=Style.NORMAL)
|
||||
return ''.join(pinyin_list)
|
||||
else:
|
||||
# 简化的拼音映射表（备用方案）
|
||||
pinyin_map = {
|
||||
'张': 'zhang', '王': 'wang', '李': 'li', '刘': 'liu', '陈': 'chen',
|
||||
'杨': 'yang', '赵': 'zhao', '黄': 'huang', '周': 'zhou', '吴': 'wu',
|
||||
'徐': 'xu', '孙': 'sun', '胡': 'hu', '朱': 'zhu', '高': 'gao',
|
||||
'林': 'lin', '何': 'he', '郭': 'guo', '马': 'ma', '罗': 'luo',
|
||||
'梁': 'liang', '宋': 'song', '郑': 'zheng', '谢': 'xie', '韩': 'han',
|
||||
'唐': 'tang', '冯': 'feng', '于': 'yu', '董': 'dong', '萧': 'xiao',
|
||||
'程': 'cheng', '曹': 'cao', '袁': 'yuan', '邓': 'deng', '许': 'xu',
|
||||
'傅': 'fu', '沈': 'shen', '曾': 'zeng', '彭': 'peng', '吕': 'lv',
|
||||
'苏': 'su', '卢': 'lu', '蒋': 'jiang', '蔡': 'cai', '贾': 'jia',
|
||||
'丁': 'ding', '魏': 'wei', '薛': 'xue', '叶': 'ye', '阎': 'yan',
|
||||
'余': 'yu', '潘': 'pan', '杜': 'du', '戴': 'dai', '夏': 'xia',
|
||||
'钟': 'zhong', '汪': 'wang', '田': 'tian', '任': 'ren', '姜': 'jiang',
|
||||
'范': 'fan', '方': 'fang', '石': 'shi', '姚': 'yao', '谭': 'tan',
|
||||
'廖': 'liao', '邹': 'zou', '熊': 'xiong', '金': 'jin', '陆': 'lu',
|
||||
'郝': 'hao', '孔': 'kong', '白': 'bai', '崔': 'cui', '康': 'kang',
|
||||
'毛': 'mao', '邱': 'qiu', '秦': 'qin', '江': 'jiang', '史': 'shi',
|
||||
'顾': 'gu', '侯': 'hou', '邵': 'shao', '孟': 'meng', '龙': 'long',
|
||||
'万': 'wan', '段': 'duan', '漕': 'cao', '钱': 'qian', '汤': 'tang',
|
||||
# 添加一些常用字
|
||||
'小': 'xiao', '大': 'da', '明': 'ming', '强': 'qiang', '文': 'wen',
|
||||
'华': 'hua', '建': 'jian', '国': 'guo', '军': 'jun', '伟': 'wei',
|
||||
'超': 'chao', '雷': 'lei', '鹏': 'peng', '俊': 'jun', '敏': 'min',
|
||||
'志': 'zhi', '勇': 'yong', '杰': 'jie', '涛': 'tao', '斌': 'bin',
|
||||
'辉': 'hui', '阳': 'yang', '磊': 'lei', '刚': 'gang',
|
||||
'飞': 'fei', '翔': 'xiang', '波': 'bo', '峰': 'feng', '凯': 'kai',
|
||||
'静': 'jing', '丽': 'li', '秀': 'xiu', '红': 'hong', '霞': 'xia',
|
||||
'洁': 'jie', '燕': 'yan', '艳': 'yan', '娟': 'juan', '玲': 'ling',
|
||||
'琳': 'lin', '萍': 'ping', '芳': 'fang', '莉': 'li', '婷': 'ting',
|
||||
'慧': 'hui', '娜': 'na', '丹': 'dan', '青': 'qing', '倩': 'qian',
|
||||
}
|
||||
|
||||
result = []
|
||||
for char in text:
|
||||
if char in pinyin_map:
|
||||
result.append(pinyin_map[char])
|
||||
elif char.isalpha():
|
||||
result.append(char.lower())
|
||||
elif char.isdigit():
|
||||
result.append(char)
|
||||
# 忽略其他字符
|
||||
|
||||
return ''.join(result)
|
||||
# 使用pypinyin进行转换
|
||||
pinyin_list = lazy_pinyin(text, style=Style.NORMAL)
|
||||
return "".join(pinyin_list)
|
||||
|
||||
|
||||
def validate_username(username: str) -> tuple[bool, str]:
|
||||
@ -89,7 +39,7 @@ def validate_username(username: str) -> tuple[bool, str]:
|
||||
|
||||
# 检查是否包含不允许的字符
|
||||
# 允许中文、英文、数字、下划线
|
||||
if not re.match(r'^[\u4e00-\u9fa5a-zA-Z0-9_]+$', username):
|
||||
if not re.match(r"^[\u4e00-\u9fa5a-zA-Z0-9_]+$", username):
|
||||
return False, "用户名只能包含中文、英文、数字和下划线"
|
||||
|
||||
return True, ""
|
||||
@ -112,15 +62,15 @@ def generate_user_id(username: str) -> str:
|
||||
user_id = to_pinyin(username)
|
||||
|
||||
# 3. 处理特殊字符，只保留字母、数字和下划线
|
||||
user_id = re.sub(r'[^a-zA-Z0-9_]', '', user_id)
|
||||
user_id = re.sub(r"[^a-zA-Z0-9_]", "", user_id)
|
||||
|
||||
# 4. 确保不以数字开头
|
||||
if user_id and user_id[0].isdigit():
|
||||
user_id = 'u' + user_id
|
||||
user_id = "u" + user_id
|
||||
|
||||
# 5. 如果为空或太短，使用默认前缀
|
||||
if len(user_id) < 2:
|
||||
user_id = 'user' + str(hash(username) % 10000).zfill(4)
|
||||
user_id = "user" + str(hash(username) % 10000).zfill(4)
|
||||
|
||||
# 6. 长度限制
|
||||
if len(user_id) > 20:
|
||||
@ -158,6 +108,7 @@ def generate_unique_user_id(username: str, existing_user_ids: list[str]) -> str:
|
||||
if counter > 9999:
|
||||
# 使用时间戳作为后缀
|
||||
import time
|
||||
|
||||
candidate = f"{base_user_id}{int(time.time()) % 10000}"
|
||||
return candidate
|
||||
|
||||
@ -176,10 +127,10 @@ def is_valid_phone_number(phone: str) -> bool:
|
||||
return False
|
||||
|
||||
# 移除空格和特殊字符
|
||||
phone = re.sub(r'[\s\-\(\)]', '', phone)
|
||||
phone = re.sub(r"[\s\-\(\)]", "", phone)
|
||||
|
||||
# 中国大陆手机号格式：1开头，第二位是3-9，总共11位
|
||||
pattern = r'^1[3-9]\d{9}$'
|
||||
pattern = r"^1[3-9]\d{9}$"
|
||||
|
||||
return bool(re.match(pattern, phone))
|
||||
|
||||
@ -198,10 +149,10 @@ def normalize_phone_number(phone: str) -> str:
|
||||
return ""
|
||||
|
||||
# 移除所有非数字字符
|
||||
phone = re.sub(r'\D', '', phone)
|
||||
phone = re.sub(r"\D", "", phone)
|
||||
|
||||
# 如果是中国大陆手机号，确保格式正确
|
||||
if len(phone) == 11 and phone.startswith('1'):
|
||||
if len(phone) == 11 and phone.startswith("1"):
|
||||
return phone
|
||||
|
||||
return phone
|
||||
|
||||
Loading…
Reference in New Issue
Block a user