232 lines
7.5 KiB
Python
232 lines
7.5 KiB
Python
import math
|
||
|
||
from zhenxun.utils.image_utils import ImageTemplate
|
||
|
||
from ..config import g_sResourcePath, g_sTranslation
|
||
from ..dbService import g_pDBService
|
||
|
||
|
||
class CShopManager:
|
||
@classmethod
|
||
async def getSeedShopImage(cls, 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 = [
|
||
"-",
|
||
"种子名称",
|
||
"农场币",
|
||
"点券",
|
||
"解锁等级",
|
||
"果实单价",
|
||
"收获经验",
|
||
"收获数量",
|
||
"成熟时间（小时）",
|
||
"收获次数",
|
||
"是否可以上架交易行",
|
||
]
|
||
|
||
# 查询所有可购买作物，并根据筛选关键字过滤
|
||
plants = await g_pDBService.plant.listPlants()
|
||
filteredPlants = []
|
||
for plant in plants:
|
||
# 跳过未解锁购买的种子
|
||
if plant["isBuy"] == 0:
|
||
continue
|
||
# 字符串筛选
|
||
if filterStr and filterStr not in plant["name"]:
|
||
continue
|
||
filteredPlants.append(plant)
|
||
|
||
# 计算分页
|
||
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['name']}/icon.png"
|
||
if iconPath.exists():
|
||
icon = (iconPath, 33, 33)
|
||
|
||
# 交易行标记
|
||
sell = "可以" if plant["sell"] else "不可以"
|
||
|
||
dataList.append(
|
||
[
|
||
icon,
|
||
plant["name"], # 种子名称
|
||
plant["buy"], # 农场币种子单价
|
||
plant["vipBuy"], # 点券种子单价
|
||
plant["level"], # 解锁等级
|
||
plant["price"], # 果实单价
|
||
plant["experience"], # 收获经验
|
||
plant["harvest"], # 收获数量
|
||
plant["time"], # 成熟时间（小时）
|
||
plant["crop"], # 收获次数
|
||
sell, # 是否可上架交易行
|
||
]
|
||
)
|
||
|
||
# 页码标题
|
||
title = f"种子商店 页数: {page}/{pageCount}"
|
||
|
||
# 渲染表格并返回图片bytes
|
||
result = await ImageTemplate.table_page(
|
||
title,
|
||
"购买示例：@小真寻 购买种子 大白菜 5",
|
||
columnName,
|
||
dataList,
|
||
)
|
||
return result.pic2bytes()
|
||
|
||
@classmethod
|
||
async def buySeed(cls, uid: str, name: str, num: int = 1) -> str:
|
||
"""购买种子
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
name (str): 植物名称
|
||
num (int, optional): 购买数量
|
||
|
||
Returns:
|
||
str:
|
||
"""
|
||
|
||
if num <= 0:
|
||
return g_sTranslation["buySeed"]["notNum"]
|
||
|
||
plantInfo = await g_pDBService.plant.getPlantByName(name)
|
||
if not plantInfo:
|
||
return g_sTranslation["buySeed"]["error"]
|
||
|
||
level = await g_pDBService.user.getUserLevelByUid(uid)
|
||
|
||
if level[0] < int(plantInfo["level"]):
|
||
return g_sTranslation["buySeed"]["noLevel"]
|
||
|
||
"""
|
||
logger.debug(
|
||
f"用户：{uid}购买{name}，数量为{num}。用户农场币为{point}，购买需要{total}"
|
||
)
|
||
"""
|
||
if plantInfo["isVip"] == 1:
|
||
vipPoint = await g_pDBService.user.getUserVipPointByUid(uid)
|
||
total = int(plantInfo["vipBuy"]) * num
|
||
if vipPoint < total:
|
||
return g_sTranslation["buySeed"]["noVipPoint"]
|
||
await g_pDBService.user.updateUserVipPointByUid(uid, vipPoint - total)
|
||
else:
|
||
point = await g_pDBService.user.getUserPointByUid(uid)
|
||
total = int(plantInfo["buy"]) * num
|
||
if point < total:
|
||
return g_sTranslation["buySeed"]["noPoint"]
|
||
await g_pDBService.user.updateUserPointByUid(uid, point - total)
|
||
|
||
if not await g_pDBService.userSeed.addUserSeedByUid(uid, name, num):
|
||
return g_sTranslation["buySeed"]["errorSql"]
|
||
|
||
if plantInfo["isVip"] == 1:
|
||
return g_sTranslation["buySeed"]["vipSuccess"].format(
|
||
name=name, total=total, point=vipPoint - total
|
||
)
|
||
else:
|
||
return g_sTranslation["buySeed"]["success"].format(
|
||
name=name, total=total, point=point - total
|
||
)
|
||
|
||
@classmethod
|
||
async def sellPlantByUid(cls, uid: str, name: str = "", num: int = 1) -> str:
|
||
"""出售作物
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
|
||
Returns:
|
||
str:
|
||
"""
|
||
if not isinstance(name, str) or name.strip() == "":
|
||
name = ""
|
||
|
||
plant = await g_pDBService.userPlant.getUserPlantByUid(uid)
|
||
if not plant:
|
||
return g_sTranslation["sellPlant"]["no"]
|
||
|
||
point = 0
|
||
totalSold = 0
|
||
isAll = num == -1
|
||
|
||
if name == "":
|
||
for plantName, count in plant.items():
|
||
isLock = await g_pDBService.userPlant.checkPlantLockByName(
|
||
uid, plantName
|
||
)
|
||
|
||
if isLock:
|
||
continue
|
||
|
||
plantInfo = await g_pDBService.plant.getPlantByName(plantName)
|
||
if not plantInfo:
|
||
continue
|
||
|
||
point += plantInfo["price"] * count
|
||
await g_pDBService.userPlant.updateUserPlantByName(uid, plantName, 0)
|
||
else:
|
||
if name not in plant:
|
||
return g_sTranslation["sellPlant"]["error"].format(name=name)
|
||
available = plant[name]
|
||
sellAmount = available if isAll else min(available, num)
|
||
if sellAmount <= 0:
|
||
return g_sTranslation["sellPlant"]["error1"].format(name=name)
|
||
await g_pDBService.userPlant.updateUserPlantByName(
|
||
uid, name, available - sellAmount
|
||
)
|
||
totalSold = sellAmount
|
||
|
||
if name == "":
|
||
totalPoint = point
|
||
else:
|
||
plantInfo = await g_pDBService.plant.getPlantByName(name)
|
||
if not plantInfo:
|
||
price = 0
|
||
else:
|
||
price = plantInfo["price"]
|
||
|
||
totalPoint = totalSold * price
|
||
|
||
currentPoint = await g_pDBService.user.getUserPointByUid(uid)
|
||
await g_pDBService.user.updateUserPointByUid(uid, currentPoint + totalPoint)
|
||
|
||
if name == "":
|
||
return g_sTranslation["sellPlant"]["success"].format(
|
||
point=totalPoint, num=currentPoint + totalPoint
|
||
)
|
||
else:
|
||
return g_sTranslation["sellPlant"]["success1"].format(
|
||
name=name, point=totalPoint, num=currentPoint + totalPoint
|
||
)
|
||
|
||
|
||
g_pShopManager = CShopManager()
|