171 lines
5.2 KiB
Python
171 lines
5.2 KiB
Python
import math
|
|
from re import I
|
|
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils._build_image import BuildImage
|
|
from zhenxun.utils.image_utils import ImageTemplate
|
|
|
|
from ..config import g_pJsonManager, g_sResourcePath
|
|
from ..database import g_pSqlManager
|
|
|
|
|
|
class CShopManager:
|
|
|
|
@classmethod
|
|
async def getSeedShopImage(cls, num: int = 1) -> bytes:
|
|
"""获取商店页面
|
|
|
|
Returns:
|
|
bytes: 返回商店图片bytes
|
|
"""
|
|
|
|
data_list = []
|
|
column_name = [
|
|
"-",
|
|
"种子名称",
|
|
"种子单价"
|
|
"解锁等级",
|
|
"果实单价",
|
|
"收获经验",
|
|
"收获数量",
|
|
"成熟时间（小时）",
|
|
"收获次数",
|
|
"再次成熟时间（小时）",
|
|
"是否可以上架交易行"
|
|
]
|
|
|
|
sell = ""
|
|
plants = list(g_pJsonManager.m_pPlant['plant'].items())
|
|
start = (num - 1) * 15
|
|
maxItems = min(len(plants) - start, 15)
|
|
items = plants[start:start + maxItems]
|
|
|
|
for key, plant in items:
|
|
icon = ""
|
|
icon_path = g_sResourcePath / f"plant/{key}/icon.png"
|
|
if icon_path.exists():
|
|
icon = (icon_path, 33, 33)
|
|
|
|
if plant['again'] == True:
|
|
sell = "可以"
|
|
else:
|
|
sell = "不可以"
|
|
|
|
data_list.append(
|
|
[
|
|
icon,
|
|
key,
|
|
plant['buy'],
|
|
plant['level'],
|
|
plant['price'],
|
|
plant['experience'],
|
|
plant['harvest'],
|
|
plant['time'],
|
|
plant['crop'],
|
|
plant['again'],
|
|
sell
|
|
]
|
|
)
|
|
|
|
count = math.ceil(len(g_pJsonManager.m_pPlant['plant']) / 15)
|
|
title = f"种子商店 页数: {num}/{count}"
|
|
|
|
result = await ImageTemplate.table_page(
|
|
title,
|
|
"购买示例：@小真寻 购买种子 大白菜 5",
|
|
column_name,
|
|
data_list,
|
|
)
|
|
|
|
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 "请输入购买数量！"
|
|
|
|
plantInfo = None
|
|
|
|
try:
|
|
plantInfo = g_pJsonManager.m_pPlant['plant'][name]
|
|
except Exception as e:
|
|
return "购买出错！请检查需购买的种子名称！"
|
|
|
|
level = await g_pSqlManager.getUserLevelByUid(uid)
|
|
|
|
if level[0] < int(plantInfo['level']):
|
|
return "你的等级不够哦，努努力吧"
|
|
|
|
point = await g_pSqlManager.getUserPointByUid(uid)
|
|
total = int(plantInfo['buy']) * num
|
|
|
|
logger.debug(f"用户：{uid}购买{name}，数量为{num}。用户农场币为{point}，购买需要{total}")
|
|
|
|
if point < total:
|
|
return "你的农场币不够哦~ 快速速氪金吧！"
|
|
else:
|
|
await g_pSqlManager.updateUserPointByUid(uid, point - total)
|
|
|
|
if await g_pSqlManager.addUserSeedByUid(uid, name, num) == False:
|
|
return "购买失败，执行数据库错误！"
|
|
|
|
return f"成功购买{name}，花费{total}农场币, 剩余{point - total}农场币"
|
|
|
|
@classmethod
|
|
async def sellPlantByUid(cls, uid: str, name: str = "", num: int = 1) -> str:
|
|
"""出售作物
|
|
|
|
Args:
|
|
uid (str): 用户Uid
|
|
|
|
Returns:
|
|
str:
|
|
"""
|
|
plantDict = await g_pSqlManager.getUserPlantByUid(uid)
|
|
if not plantDict:
|
|
return "你仓库没有可以出售的作物"
|
|
|
|
totalPoint = 0
|
|
|
|
if not name:
|
|
for plantName, count in plantDict.items():
|
|
plantInfo = g_pJsonManager.m_pPlant['plant'][plantName]
|
|
totalPoint += plantInfo['price'] * count
|
|
await g_pSqlManager.deleteUserPlantByName(uid, plantName)
|
|
else:
|
|
currentCount = plantDict.get(name)
|
|
if currentCount is None:
|
|
return f"出售作物{name}出错：你没有这种作物"
|
|
|
|
if num == -1:
|
|
sellCount = currentCount
|
|
else:
|
|
if num > currentCount:
|
|
return f"出售作物{name}出错：数量不足"
|
|
sellCount = num
|
|
|
|
plantInfo = g_pJsonManager.m_pPlant['plant'][name]
|
|
totalPoint = plantInfo['price'] * sellCount
|
|
await g_pSqlManager.addUserPlantByUid(uid, name, -sellCount)
|
|
|
|
point = await g_pSqlManager.getUserPointByUid(uid)
|
|
await g_pSqlManager.updateUserPointByUid(uid, point + totalPoint)
|
|
|
|
if not name:
|
|
return f"成功出售所有作物，获得农场币：{totalPoint}，当前农场币：{point + totalPoint}"
|
|
else:
|
|
return f"成功出售{name}，获得农场币：{totalPoint}，当前农场币：{point + totalPoint}"
|
|
|
|
g_pShopManager = CShopManager()
|