🎨 添加收获次数判断作物升星,一二三四五星作物写入库中

This commit is contained in:
Mualamx
2025-10-26 13:46:07 +08:00
parent 15c482f41f
commit a59ac4ded6
5 changed files with 33 additions and 124 deletions
+18
View File
@@ -75,6 +75,24 @@ class CUserPlantCountDB(CSqlManager):
rows = await cursor.fetchall()
return {row["plant"]: row["count"] for row in rows}
@classmethod
async def getUserPlantCountByPlantName(cls, uid: str, plant: str) -> int:
"""根据用户Uid和植物名称获取该对应的收获次数
Args:
uid (str): 用户uid
plant (str): 植物名称
Returns:
int: 收获次数
"""
cursor = await cls.m_pDB.execute(
"SELECT count FROM userPlantCount WHERE uid = ? AND plant = ?",
(uid, plant),
)
row = await cursor.fetchone()
return row["count"] if row else 0
@classmethod
async def updateUserPlantCountByName(cls, uid: str, plant: str, count: int) -> bool:
"""根据植物名称更新植物的收获次数
+15 -1
View File
@@ -384,7 +384,7 @@ class CUserSoilDB(CSqlManager):
if soilInfo and soilInfo.get("plantName"):
return False
# 获取植物配置
# 获取植物配置""" """
plantCfg = await g_pDBService.plant.getPlantByName(plantName)
if not plantCfg:
logger.error(f"未知植物: {plantName}")
@@ -404,6 +404,20 @@ class CUserSoilDB(CSqlManager):
async with cls._transaction():
prev = soilInfo or {}
await cls._deleteUserSoil(uid, soilIndex)
# 根据此植物收获次数对其星级进行升级
if g_pDBService.plant.existsPlant("一星" + plantName):
harvestCounts = (
await g_pDBService.userPlantCount.getUserPlantCountByPlantName(
uid, plantName
)
)
if harvestCounts >= 350:
plantName = "一星" + plantName
if harvestCounts >= 700:
plantName = "二星" + plantName
if harvestCounts >= 1200:
plantName = "三星" + plantName
await cls._insertUserSoil(
{
"uid": uid,
-75
View File
@@ -1230,80 +1230,5 @@ class CFarmManager:
return f"兑换{num}点券成功,当前点券:{number},赠送点券:{giftPoints},当前农场币:{point}"
@classmethod
async def getUserPlantCountImage(
cls, uid: str, filterKey: str | int = 1, num: int = 1
) -> bytes:
"""获取用户作物收获次数图片
Args:
filterKey (str|int):
- 字符串: 根据关键字筛选种子名称
- 整数: 翻至对应页(无筛选)
num (int, optional): 当 filterKey 为字符串时,用于指定页码。Defaults to 1.
Returns:
bytes: 返回商店图片bytes
"""
# 解析参数:区分筛选关键字和页码
filterStr = None
if isinstance(filterKey, int):
page = filterKey
else:
filterStr = filterKey
page = num
# 表头定义
columnName = [
"-",
"种子名称",
"收获次数",
]
# 查询用户作物收获次数(返回 dict: {plant: count})并根据筛选关键字过滤
plants = await g_pDBService.userPlantCount.getUserPlantCountByUid(uid) or {}
filteredPlants = []
for name, cnt in plants.items():
# 字符串筛选
if filterStr and filterStr not in name:
continue
filteredPlants.append({"plant": name, "count": cnt})
# 计算分页
totalCount = len(filteredPlants)
pageCount = math.ceil(totalCount / 15) if totalCount else 1
startIndex = (page - 1) * 15
pageItems = filteredPlants[startIndex : startIndex + 15]
# 构建数据行
dataList = []
for plant in pageItems:
# 图标处理
icon = ""
iconPath = g_sResourcePath / f"plant/{plant['plant']}/icon.png"
if iconPath.exists():
icon = (iconPath, 33, 33)
dataList.append(
[
icon,
plant["plant"], # 种子名称
plant["count"], # 收获次数
]
)
# 页码标题
title = f"收获次数 页数: {page}/{pageCount}"
# 渲染表格并返回图片bytes
result = await ImageTemplate.table_page(
title,
"",
columnName,
dataList,
)
return result.pic2bytes()
g_pFarmManager = CFarmManager()