178 lines
3.9 KiB
Python
178 lines
3.9 KiB
Python
import os
|
||
from datetime import datetime
|
||
from zoneinfo import ZoneInfo
|
||
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.message import MessageUtils
|
||
|
||
from .dbService import g_pDBService
|
||
|
||
|
||
class CToolManager:
|
||
@classmethod
|
||
async def isRegisteredByUid(cls, uid: str) -> bool:
|
||
result = await g_pDBService.user.isUserExist(uid)
|
||
|
||
if not result:
|
||
await MessageUtils.build_message(
|
||
"尚未开通农场，快at我发送 开通农场 开通吧"
|
||
).send()
|
||
return False
|
||
|
||
return True
|
||
|
||
@classmethod
|
||
def sanitize_username(cls, username: str, max_length: int = 15) -> str:
|
||
"""
|
||
安全处理用户名
|
||
功能：
|
||
1. 移除首尾空白
|
||
2. 过滤危险字符
|
||
3. 转义单引号
|
||
4. 处理空值
|
||
5. 限制长度
|
||
"""
|
||
# 处理空值
|
||
if not username:
|
||
return "神秘农夫"
|
||
|
||
# 基础清洗
|
||
cleaned = username.strip()
|
||
|
||
# 允许的字符白名单（可自定义扩展）
|
||
safe_chars = {
|
||
"_",
|
||
"-",
|
||
"!",
|
||
"@",
|
||
"#",
|
||
"$",
|
||
"%",
|
||
"^",
|
||
"&",
|
||
"*",
|
||
"(",
|
||
")",
|
||
"+",
|
||
"=",
|
||
".",
|
||
",",
|
||
"~",
|
||
"·",
|
||
" ",
|
||
"a",
|
||
"b",
|
||
"c",
|
||
"d",
|
||
"e",
|
||
"f",
|
||
"g",
|
||
"h",
|
||
"i",
|
||
"j",
|
||
"k",
|
||
"l",
|
||
"m",
|
||
"n",
|
||
"o",
|
||
"p",
|
||
"q",
|
||
"r",
|
||
"s",
|
||
"t",
|
||
"u",
|
||
"v",
|
||
"w",
|
||
"x",
|
||
"y",
|
||
"z",
|
||
"A",
|
||
"B",
|
||
"C",
|
||
"D",
|
||
"E",
|
||
"F",
|
||
"G",
|
||
"H",
|
||
"I",
|
||
"J",
|
||
"K",
|
||
"L",
|
||
"M",
|
||
"N",
|
||
"O",
|
||
"P",
|
||
"Q",
|
||
"R",
|
||
"S",
|
||
"T",
|
||
"U",
|
||
"V",
|
||
"W",
|
||
"X",
|
||
"Y",
|
||
"Z",
|
||
"0",
|
||
"1",
|
||
"2",
|
||
"3",
|
||
"4",
|
||
"5",
|
||
"6",
|
||
"7",
|
||
"8",
|
||
"9",
|
||
}
|
||
# 添加常用中文字符（Unicode范围）
|
||
safe_chars.update(chr(c) for c in range(0x4E00, 0x9FFF + 1))
|
||
|
||
# 过滤危险字符
|
||
filtered = [
|
||
c if c in safe_chars or 0x4E00 <= ord(c) <= 0x9FFF else "" for c in cleaned
|
||
]
|
||
|
||
# 合并处理结果
|
||
safe_str = "".join(filtered)
|
||
|
||
# 转义单引号（双重保障）
|
||
escaped = safe_str.replace("'", "''")
|
||
|
||
# 处理空结果
|
||
if not escaped:
|
||
return "神秘农夫"
|
||
|
||
# 长度限制
|
||
return escaped[:max_length]
|
||
|
||
@classmethod
|
||
def renameFile(cls, currentFilePath: str, newFileName: str) -> bool:
|
||
"""重命名文件，如果目标文件名已存在则先删除再重命名
|
||
|
||
Args:
|
||
currentFilePath (str): 当前文件的完整路径
|
||
newFileName (str): 重命名后的文件名
|
||
|
||
Returns:
|
||
bool: 重命名成功返回 True，否则返回 False
|
||
"""
|
||
try:
|
||
dirPath = os.path.dirname(currentFilePath)
|
||
newFilePath = os.path.join(dirPath, newFileName)
|
||
|
||
if os.path.exists(newFilePath):
|
||
os.remove(newFilePath)
|
||
|
||
os.rename(currentFilePath, newFilePath)
|
||
return True
|
||
except Exception as e:
|
||
logger.warning(f"文件重命名失败: {e}")
|
||
return False
|
||
|
||
@classmethod
|
||
def dateTime(cls) -> datetime:
|
||
tz = ZoneInfo("Asia/Shanghai")
|
||
return datetime.now(tz)
|
||
|
||
|
||
g_pToolManager = CToolManager()
|