first commit
This commit is contained in:
73
models/emoji_db.py
Normal file
73
models/emoji_db.py
Normal file
@@ -0,0 +1,73 @@
|
||||
from models.connect_db import connect_db
|
||||
from models.logger import setup_logger
|
||||
|
||||
logger = setup_logger()
|
||||
connection = connect_db()
|
||||
|
||||
def update_description(emoji_uuid: str, description: str) -> bool:
|
||||
"""
|
||||
更新表情包描述词
|
||||
|
||||
Args:
|
||||
emoji_uuid: 表情包UUID
|
||||
description: 新的描述词
|
||||
|
||||
Returns:
|
||||
成功返回True,失败返回False
|
||||
"""
|
||||
try:
|
||||
sql = """
|
||||
UPDATE emoji
|
||||
SET description = %s
|
||||
WHERE uuid = %s
|
||||
"""
|
||||
|
||||
params = (description, emoji_uuid)
|
||||
|
||||
with connection.cursor() as cursor:
|
||||
affected_rows = cursor.execute(sql, params)
|
||||
|
||||
if affected_rows > 0:
|
||||
logger.info(f"成功更新表情包描述,UUID: {emoji_uuid}")
|
||||
return True
|
||||
else:
|
||||
logger.warning(f"未找到要更新的表情包,UUID: {emoji_uuid}")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"更新表情包描述失败: {e}")
|
||||
return False
|
||||
|
||||
|
||||
def check_emoji(uuid: str) -> bool:
|
||||
"""
|
||||
检查UUID是否存在于emoji表中
|
||||
|
||||
Args:
|
||||
uuid: 要检查的UUID字符串
|
||||
|
||||
Returns:
|
||||
uuid 在数据库中存在则为 True,否则 False
|
||||
"""
|
||||
# 参数验证
|
||||
if not uuid or not isinstance(uuid, str):
|
||||
logger.warning(f"无效的UUID参数: {uuid}")
|
||||
return False
|
||||
|
||||
try:
|
||||
with connection.cursor() as cursor:
|
||||
# 执行查询
|
||||
sql = "SELECT 1 FROM emoji WHERE uuid = %s LIMIT 1"
|
||||
cursor.execute(sql, (uuid,))
|
||||
|
||||
# 如果查询到结果则返回True
|
||||
result = cursor.fetchone()
|
||||
return result is not None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"检查UUID失败: {uuid}, 错误: {e}")
|
||||
return False
|
||||
|
||||
if __name__ == "__main__":
|
||||
record = check_emoji("ffe1663c-44e1-4719-a5ba-01485f70a87e")
|
||||
print(record)
|
||||
Reference in New Issue
Block a user