✨ 添加点券商店,与种子商店分开显示对应种子
This commit is contained in:
parent
fea8ca9287
commit
698f803694
47
command.py
47
command.py
@ -99,6 +99,7 @@ diuse_farm = on_alconna(
|
|||||||
Subcommand("point-to-vipPoint", Args["num?", int], help_text="点券兑换"),
|
Subcommand("point-to-vipPoint", Args["num?", int], help_text="点券兑换"),
|
||||||
Subcommand("my-vipPoint", help_text="我的点券"),
|
Subcommand("my-vipPoint", help_text="我的点券"),
|
||||||
Subcommand("farm-help", help_text="农场帮助"),
|
Subcommand("farm-help", help_text="农场帮助"),
|
||||||
|
Subcommand("vipSeed-shop", Args["res?", MultiVar(str)], help_text="种子商店"),
|
||||||
),
|
),
|
||||||
priority=5,
|
priority=5,
|
||||||
block=True,
|
block=True,
|
||||||
@ -713,3 +714,49 @@ async def _(session: Uninfo):
|
|||||||
image = BuildImage(background=savePath)
|
image = BuildImage(background=savePath)
|
||||||
|
|
||||||
await MessageUtils.build_message(image).send(reply_to=True)
|
await MessageUtils.build_message(image).send(reply_to=True)
|
||||||
|
|
||||||
|
|
||||||
|
diuse_farm.shortcut(
|
||||||
|
"点券商店(.*?)",
|
||||||
|
command="我的农场",
|
||||||
|
arguments=["vipSeed-shop"],
|
||||||
|
prefix=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@diuse_farm.assign("vipSeed-shop")
|
||||||
|
async def _(session: Uninfo, res: Match[tuple[str, ...]]):
|
||||||
|
uid = str(session.user.id)
|
||||||
|
|
||||||
|
if not await g_pToolManager.isRegisteredByUid(uid):
|
||||||
|
return
|
||||||
|
|
||||||
|
if res.result is inspect._empty:
|
||||||
|
raw = []
|
||||||
|
else:
|
||||||
|
raw = res.result
|
||||||
|
|
||||||
|
filterKey: str | int | None = None
|
||||||
|
page: int = 1
|
||||||
|
|
||||||
|
if len(raw) >= 1 and raw[0] is not None:
|
||||||
|
first = raw[0]
|
||||||
|
if isinstance(first, str) and first.isdigit():
|
||||||
|
page = int(first)
|
||||||
|
else:
|
||||||
|
filterKey = first
|
||||||
|
|
||||||
|
if (
|
||||||
|
len(raw) >= 2
|
||||||
|
and raw[1] is not None
|
||||||
|
and isinstance(raw[1], str)
|
||||||
|
and raw[1].isdigit()
|
||||||
|
):
|
||||||
|
page = int(raw[1])
|
||||||
|
|
||||||
|
if filterKey is None:
|
||||||
|
image = await g_pShopManager.getVipSeedShopImage(page)
|
||||||
|
else:
|
||||||
|
image = await g_pShopManager.getVipSeedShopImage(filterKey, page)
|
||||||
|
|
||||||
|
await MessageUtils.build_message(image).send()
|
||||||
|
|||||||
100
farm/shop.py
100
farm/shop.py
@ -33,7 +33,6 @@ class CShopManager:
|
|||||||
"-",
|
"-",
|
||||||
"种子名称",
|
"种子名称",
|
||||||
"农场币",
|
"农场币",
|
||||||
"点券",
|
|
||||||
"解锁等级",
|
"解锁等级",
|
||||||
"果实单价",
|
"果实单价",
|
||||||
"收获经验",
|
"收获经验",
|
||||||
@ -47,6 +46,9 @@ class CShopManager:
|
|||||||
plants = await g_pDBService.plant.listPlants()
|
plants = await g_pDBService.plant.listPlants()
|
||||||
filteredPlants = []
|
filteredPlants = []
|
||||||
for plant in plants:
|
for plant in plants:
|
||||||
|
# 只留下农场币购买的种子
|
||||||
|
if plant["isVip"] == 1:
|
||||||
|
continue
|
||||||
# 跳过未解锁购买的种子
|
# 跳过未解锁购买的种子
|
||||||
if plant["isBuy"] == 0:
|
if plant["isBuy"] == 0:
|
||||||
continue
|
continue
|
||||||
@ -78,7 +80,6 @@ class CShopManager:
|
|||||||
icon,
|
icon,
|
||||||
plant["name"], # 种子名称
|
plant["name"], # 种子名称
|
||||||
plant["buy"], # 农场币种子单价
|
plant["buy"], # 农场币种子单价
|
||||||
plant["vipBuy"], # 点券种子单价
|
|
||||||
plant["level"], # 解锁等级
|
plant["level"], # 解锁等级
|
||||||
plant["price"], # 果实单价
|
plant["price"], # 果实单价
|
||||||
plant["experience"], # 收获经验
|
plant["experience"], # 收获经验
|
||||||
@ -227,5 +228,100 @@ class CShopManager:
|
|||||||
name=name, point=totalPoint, num=currentPoint + totalPoint
|
name=name, point=totalPoint, num=currentPoint + totalPoint
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
async def getVipSeedShopImage(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["isVip"] == 0:
|
||||||
|
continue
|
||||||
|
# 跳过未解锁购买的种子
|
||||||
|
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["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()
|
||||||
|
|
||||||
|
|
||||||
g_pShopManager = CShopManager()
|
g_pShopManager = CShopManager()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user