826 lines
29 KiB
Python
826 lines
29 KiB
Python
import asyncio
|
||
import math
|
||
import random
|
||
from typing import Dict, List, Tuple
|
||
|
||
from zhenxun.configs.config import Config
|
||
from zhenxun.models.user_console import UserConsole
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils._build_image import BuildImage
|
||
from zhenxun.utils.enum import GoldHandle
|
||
from zhenxun.utils.image_utils import ImageTemplate
|
||
from zhenxun.utils.platform import PlatformUtils
|
||
|
||
from ..config import g_bIsDebug, g_sResourcePath
|
||
from ..dbService import g_pDBService
|
||
from ..event.event import g_pEventManager
|
||
from ..json import g_pJsonManager
|
||
from ..tool import g_pToolManager
|
||
|
||
|
||
class CFarmManager:
|
||
@classmethod
|
||
async def buyPointByUid(cls, uid: str, num: int) -> str:
|
||
if num <= 0:
|
||
return "你是怎么做到购买不是正数的农场币的"
|
||
|
||
user = await UserConsole.get_user(uid)
|
||
|
||
pro = float(Config.get_config("zhenxun_plugin_farm", "兑换倍数"))
|
||
tax = float(Config.get_config("zhenxun_plugin_farm", "手续费"))
|
||
|
||
#计算手续费
|
||
fee = math.floor(num * tax)
|
||
#实际扣费金额
|
||
deduction = num + fee
|
||
|
||
if user.gold < deduction:
|
||
return f"你的金币不足或不足承担手续费。当前手续费为{fee}"
|
||
|
||
await UserConsole.reduce_gold(uid, num, GoldHandle.PLUGIN , 'zhenxun_plugin_farm') # type: ignore
|
||
await UserConsole.reduce_gold(uid, fee, GoldHandle.PLUGIN , 'zhenxun_plugin_farm') # type: ignore
|
||
|
||
point = num * pro
|
||
|
||
p = await g_pDBService.user.getUserPointByUid(uid)
|
||
number = point + p
|
||
|
||
await g_pDBService.user.updateUserPointByUid(uid, int(number))
|
||
|
||
return f"充值{point}农场币成功，手续费{tax}金币，当前农场币：{number}"
|
||
|
||
@classmethod
|
||
async def drawFarmByUid(cls, uid: str) -> bytes:
|
||
"""绘制用户农场
|
||
|
||
Args:
|
||
uid (str): 用户UID
|
||
|
||
Returns:
|
||
bytes: 返回绘制结果
|
||
"""
|
||
img = BuildImage(background = g_sResourcePath / "background/background.jpg")
|
||
|
||
soilSize = g_pJsonManager.m_pSoil['size']
|
||
|
||
#TODO 缺少判断用户土地资源状况
|
||
soil = BuildImage(background = g_sResourcePath / "soil/普通土地.png")
|
||
await soil.resize(0, soilSize[0], soilSize[1])
|
||
|
||
grass = BuildImage(background = g_sResourcePath / "soil/草土地.png")
|
||
await grass.resize(0, soilSize[0], soilSize[1])
|
||
|
||
soilPos = g_pJsonManager.m_pSoil['soil']
|
||
|
||
userInfo = await g_pDBService.user.getUserInfoByUid(uid)
|
||
soilUnlock = int(userInfo['soil'])
|
||
|
||
x = 0
|
||
y = 0
|
||
isFirstExpansion = True #首次添加扩建图片
|
||
isFirstRipe = True
|
||
plant = None
|
||
for index in range(0, 30):
|
||
x = soilPos[str(index + 1)]['x']
|
||
y = soilPos[str(index + 1)]['y']
|
||
|
||
#如果土地已经到达对应等级
|
||
if index < soilUnlock:
|
||
await img.paste(soil, (x, y))
|
||
|
||
isPlant, plant, isRipe, offsetX, offsetY = await cls.drawSoilPlant(uid, index + 1)
|
||
|
||
if isPlant:
|
||
await img.paste(plant, (x + soilSize[0] // 2 - plant.width // 2 + offsetX,
|
||
y + soilSize[1] // 2 - plant.height // 2 + offsetY))
|
||
|
||
#1700 275
|
||
#首次添加可收获图片
|
||
if isRipe and isFirstRipe:
|
||
ripe = BuildImage(background = g_sResourcePath / "background/ripe.png")
|
||
|
||
await img.paste(ripe, (x + soilSize[0] // 2 - ripe.width // 2,
|
||
y - ripe.height // 2))
|
||
|
||
isFirstRipe = False
|
||
else:
|
||
await img.paste(grass, (x, y))
|
||
|
||
if isFirstExpansion:
|
||
isFirstExpansion = False
|
||
|
||
#首次添加扩建图片
|
||
expansion = BuildImage(background = g_sResourcePath / "background/expansion.png")
|
||
await expansion.resize(0, 69, 69)
|
||
await img.paste(expansion, (x + soilSize[0] // 2 - expansion.width // 2,
|
||
y + soilSize[1] // 2 - expansion.height))
|
||
|
||
#左上角绘制用户信息
|
||
#头像
|
||
image = await PlatformUtils.get_user_avatar(uid, "qq")
|
||
|
||
if image:
|
||
avatar = BuildImage(background = image)
|
||
|
||
await img.paste(avatar, (125, 85))
|
||
|
||
#头像框
|
||
frame = BuildImage(background = g_sResourcePath / "background/frame.png")
|
||
await img.paste(frame, (75, 44))
|
||
|
||
#用户名
|
||
nameImg = await BuildImage.build_text_image(userInfo['name'], size = 24, font_color = (77, 35, 4))
|
||
await img.paste(nameImg, (300, 92))
|
||
|
||
#经验值
|
||
level = await g_pDBService.user.getUserLevelByUid(uid)
|
||
|
||
beginX = 309
|
||
endX = 627
|
||
#绘制宽度计算公式为 (当前经验值 / 经验值上限) * 宽度
|
||
width = int((level[2] / level[1]) * (endX - beginX))
|
||
await img.rectangle((beginX, 188, beginX + width, 222), (171, 194, 41))
|
||
|
||
expImg = await BuildImage.build_text_image(f"{level[2]} / {level[1]}", size = 24, font_color = (102, 120, 19))
|
||
await img.paste(expImg, (390, 193))
|
||
|
||
#等级
|
||
levelImg = await BuildImage.build_text_image(str(level[0]), size = 32, font_color = (214, 111, 1))
|
||
await img.paste(levelImg, (660, 187))
|
||
|
||
#金币
|
||
pointImg = await BuildImage.build_text_image(str(userInfo['point']), size = 24, font_color = (253, 253, 253))
|
||
await img.paste(pointImg, (330, 255))
|
||
|
||
#点券 TODO
|
||
bondsImg = await BuildImage.build_text_image("0", size = 24, font_color = (253, 253, 253))
|
||
await img.paste(bondsImg, (570, 255))
|
||
|
||
#清晰度
|
||
definition = Config.get_config("zhenxun_plugin_farm", "绘制农场清晰度")
|
||
if definition == "medium":
|
||
await img.resize(0.6)
|
||
elif definition == "hight":
|
||
await img.resize(0.8)
|
||
elif definition == "original":
|
||
pass
|
||
else:
|
||
await img.resize(0.4)
|
||
|
||
return img.pic2bytes()
|
||
|
||
@classmethod
|
||
async def drawDetailFarmByUid(cls, uid: str) -> list:
|
||
info = []
|
||
|
||
farm = await cls.drawFarmByUid(uid)
|
||
|
||
info.append(BuildImage.open(farm))
|
||
|
||
dataList = []
|
||
columnName = [
|
||
"-",
|
||
"土地ID",
|
||
"作物名称",
|
||
"成熟时间",
|
||
"土地状态",
|
||
"被偷数量",
|
||
"剩余产出",
|
||
]
|
||
|
||
icon = ""
|
||
soilNumber = await g_pDBService.user.getUserSoilByUid(uid)
|
||
|
||
for i in range(1, soilNumber + 1):
|
||
soilInfo = await g_pDBService.userSoil.getUserSoil(uid, i)
|
||
|
||
if soilInfo:
|
||
if soilInfo["soilLevel"] == 1:
|
||
iconPath = g_sResourcePath / f"soil/TODO.png"
|
||
else:
|
||
iconPath = g_sResourcePath / f"soil/普通土地.png"
|
||
|
||
if iconPath.exists():
|
||
icon = (iconPath, 33, 33)
|
||
|
||
plantName = soilInfo.get("plantName", "-")
|
||
|
||
if plantName == "-":
|
||
matureTime = "-"
|
||
soilStatus = "-"
|
||
totalNumber = "-"
|
||
plantNumber = "-"
|
||
else:
|
||
matureTime = g_pToolManager.dateTime().fromtimestamp(int(soilInfo.get("matureTime", 0))).strftime("%Y-%m-%d %H:%M:%S")
|
||
soilStatus = await g_pDBService.userSoil.getUserSoilStatus(uid, i)
|
||
|
||
totalNumber = await g_pDBService.userSteal.getTotalStolenCount(uid, i)
|
||
planInfo = await g_pDBService.plant.getPlantByName(plantName)
|
||
|
||
if not planInfo:
|
||
plantNumber = f"None"
|
||
else:
|
||
plantNumber = f"{planInfo['harvest']}"
|
||
|
||
dataList.append(
|
||
[
|
||
icon,
|
||
i,
|
||
plantName,
|
||
matureTime,
|
||
soilStatus,
|
||
totalNumber,
|
||
plantNumber,
|
||
])
|
||
|
||
if len(dataList) >= 15:
|
||
result = await ImageTemplate.table_page(
|
||
"土地详细信息",
|
||
"",
|
||
columnName,
|
||
dataList,
|
||
)
|
||
|
||
info.append(result.copy())
|
||
dataList.clear()
|
||
|
||
if i >= soilNumber:
|
||
result = await ImageTemplate.table_page(
|
||
"土地详细信息",
|
||
"",
|
||
columnName,
|
||
dataList,
|
||
)
|
||
|
||
info.append(result.copy())
|
||
dataList.clear()
|
||
|
||
return info
|
||
|
||
@classmethod
|
||
async def drawSoilPlant(cls, uid: str, soilIndex: int) -> tuple[bool, BuildImage, bool, int, int]:
|
||
"""绘制植物资源
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
soilIndex (int): 土地索引 从1开始
|
||
|
||
Returns:
|
||
tuple[bool, BuildImage]: [绘制是否成功，资源图片, 是否成熟]
|
||
"""
|
||
|
||
plant = None
|
||
soilInfo = await g_pDBService.userSoil.getUserSoil(uid, soilIndex)
|
||
|
||
if not soilInfo or not soilInfo.get("plantName"):
|
||
return False, None, False #type: ignore
|
||
|
||
#是否枯萎
|
||
if int(soilInfo.get("wiltStatus", 0)) == 1:
|
||
plant = BuildImage(background = g_sResourcePath / f"plant/basic/9.png")
|
||
await plant.resize(0, 150, 212)
|
||
return True, plant, False, 0, 0
|
||
|
||
#获取作物详细信息
|
||
plantInfo = await g_pDBService.plant.getPlantByName(soilInfo['plantName'])
|
||
if not plantInfo:
|
||
logger.error(f"绘制植物资源失败: {soilInfo['plantName']}")
|
||
return False, None, False, 0, 0 #type: ignore
|
||
|
||
offsetX = plantInfo.get('officX', 0)
|
||
offsetY = plantInfo.get('officY', 0)
|
||
offsetW = plantInfo.get('officW', 0)
|
||
offsetH = plantInfo.get('officH', 0)
|
||
|
||
currentTime = g_pToolManager.dateTime().now()
|
||
matureTime = g_pToolManager.dateTime().fromtimestamp(int(soilInfo['matureTime']))
|
||
|
||
phase = await g_pDBService.plant.getPlantPhaseNumberByName(soilInfo['plantName'])
|
||
|
||
#如果当前时间大于成熟时间 说明作物成熟
|
||
if currentTime >= matureTime:
|
||
plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo['plantName']}/{phase}.png")
|
||
|
||
return True, plant, True, offsetX, offsetY
|
||
else:
|
||
#如果是多阶段作物 且没有成熟 #早期思路 多阶段作物 直接是倒数第二阶段图片
|
||
# if soilInfo['harvestCount'] >= 1:
|
||
# plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo['plantName']}/{plantInfo['phase'] - 1s}.png")
|
||
|
||
# return True, plant, False, offsetX, offsetY
|
||
|
||
#如果没有成熟 则根据当前阶段进行绘制
|
||
plantedTime = g_pToolManager.dateTime().fromtimestamp(int(soilInfo['plantTime']))
|
||
|
||
elapsedTime = currentTime - plantedTime
|
||
elapsedHour = elapsedTime.total_seconds() / 3600
|
||
|
||
currentStage = int(elapsedHour / (plantInfo['time'] / phase))
|
||
|
||
if currentStage <= 0:
|
||
if plantInfo['general'] == False:
|
||
plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo['plantName']}/0.png")
|
||
else:
|
||
plant = BuildImage(background = g_sResourcePath / f"plant/basic/0.png")
|
||
|
||
await plant.resize(0, 35 + offsetW, 58 + offsetH)
|
||
else:
|
||
plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo['plantName']}/{currentStage}.png")
|
||
|
||
return True, plant, False, offsetX, offsetY
|
||
|
||
@classmethod
|
||
async def getUserSeedByUid(cls, uid: str) -> bytes:
|
||
"""获取用户种子仓库"""
|
||
dataList = []
|
||
columnNames = [
|
||
"-",
|
||
"种子名称",
|
||
"数量",
|
||
"收获经验",
|
||
"收获数量",
|
||
"成熟时间（小时）",
|
||
"收获次数",
|
||
"是否可以上架交易行"
|
||
]
|
||
|
||
#从数据库获取结构化数据
|
||
seedRecords = await g_pDBService.userSeed.getUserSeedByUid(uid) or {}
|
||
|
||
if not seedRecords:
|
||
result = await ImageTemplate.table_page(
|
||
"种子仓库",
|
||
"播种示例：@小真寻 播种 大白菜 [数量]",
|
||
columnNames,
|
||
dataList,
|
||
)
|
||
return result.pic2bytes()
|
||
|
||
for seedName, count in seedRecords.items():
|
||
try:
|
||
plantInfo = await g_pDBService.plant.getPlantByName(seedName)
|
||
if not plantInfo:
|
||
continue
|
||
|
||
iconPath = g_sResourcePath / f"plant/{seedName}/icon.png"
|
||
icon = (iconPath, 33, 33) if iconPath.exists() else ""
|
||
sellable = "可以" if plantInfo['sell'] else "不可以"
|
||
|
||
dataList.append([
|
||
icon,
|
||
seedName,
|
||
count,
|
||
plantInfo['experience'],
|
||
plantInfo['harvest'],
|
||
plantInfo['time'],
|
||
plantInfo['crop'],
|
||
sellable
|
||
])
|
||
except KeyError:
|
||
continue
|
||
|
||
result = await ImageTemplate.table_page(
|
||
"种子仓库",
|
||
"播种示例：@小真寻 播种 大白菜 [数量]",
|
||
columnNames,
|
||
dataList,
|
||
)
|
||
return result.pic2bytes()
|
||
|
||
@classmethod
|
||
async def sowing(cls, uid: str, name: str, num: int = -1) -> str:
|
||
"""播种
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
name (str): 播种种子名称
|
||
num (int, optional): 播种数量
|
||
|
||
Returns:
|
||
str: 返回结果
|
||
"""
|
||
try:
|
||
#获取用户的种子数量
|
||
count = await g_pDBService.userSeed.getUserSeedByName(uid, name)
|
||
if count is None:
|
||
count = 0 #如果返回 None，则视为没有种子
|
||
|
||
if count <= 0:
|
||
return f"没有在你的仓库发现{name}种子，快去买点吧！"
|
||
|
||
#如果播种数量超过仓库种子数量
|
||
if count < num and num != -1:
|
||
return f"仓库中的{name}种子数量不足，当前剩余{count}个种子"
|
||
|
||
#获取用户土地数量
|
||
soilNumber = await g_pDBService.user.getUserSoilByUid(uid)
|
||
|
||
#如果播种数量为 -1，表示播种所有可播种的土地
|
||
if num == -1:
|
||
num = count
|
||
|
||
#发送播种前信号
|
||
await g_pEventManager.m_beforePlant.emit(uid=uid, name=name, num=num)
|
||
|
||
#记录是否成功播种
|
||
successCount = 0
|
||
for i in range(1, soilNumber + 1):
|
||
if count > 0 and num > 0:
|
||
success = await g_pDBService.userSoil.sowingByPlantName(uid, i, name)
|
||
if success:
|
||
#更新种子数量
|
||
num -= 1
|
||
count -= 1
|
||
|
||
#记录种子消耗数量
|
||
successCount += 1
|
||
|
||
#发送播种后信号
|
||
await g_pEventManager.m_afterPlant.emit(uid=uid, name=name, soilIndex=i)
|
||
|
||
|
||
#确保用户仓库数量更新
|
||
if successCount > 0:
|
||
await g_pDBService.userSeed.updateUserSeedByName(uid, name, count)
|
||
|
||
#根据播种结果给出反馈
|
||
if num == 0:
|
||
return f"播种{name}成功！仓库剩余{count}个种子"
|
||
else:
|
||
return f"播种数量超出开垦土地数量，已将可播种土地成功播种{name}！仓库剩余{count}个种子"
|
||
|
||
except Exception as e:
|
||
logger.warning(f"播种操作失败！", e=e)
|
||
return "播种失败，请稍后重试！"
|
||
|
||
@classmethod
|
||
async def harvest(cls, uid: str) -> str:
|
||
"""收获作物
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
|
||
Returns:
|
||
str: 返回
|
||
"""
|
||
try:
|
||
await g_pEventManager.m_beforeHarvest.emit(uid=uid)
|
||
|
||
soilNumber = await g_pDBService.user.getUserSoilByUid(uid)
|
||
|
||
harvestRecords = [] #收获日志记录
|
||
experience = 0 #总经验值
|
||
harvestCount = 0 #成功收获数量
|
||
|
||
for i in range(1, soilNumber + 1):
|
||
#如果没有种植
|
||
if not await g_pDBService.userSoil.isSoilPlanted(uid, i):
|
||
continue
|
||
|
||
soilInfo = await g_pDBService.userSoil.getUserSoil(uid, i)
|
||
if not soilInfo:
|
||
continue
|
||
|
||
#如果是枯萎状态
|
||
if soilInfo.get("wiltStatus", 1) == 1:
|
||
continue
|
||
|
||
plantInfo = await g_pDBService.plant.getPlantByName(soilInfo['plantName'])
|
||
if not plantInfo:
|
||
continue
|
||
|
||
currentTime = g_pToolManager.dateTime().now()
|
||
matureTime = g_pToolManager.dateTime().fromtimestamp(int(soilInfo['matureTime']))
|
||
|
||
if currentTime >= matureTime:
|
||
number = plantInfo['harvest']
|
||
|
||
#处理偷菜扣除数量
|
||
stealNum = await g_pDBService.userSteal.getTotalStolenCount(uid, i)
|
||
|
||
number -= stealNum
|
||
|
||
if number <= 0:
|
||
continue
|
||
|
||
harvestCount += 1
|
||
experience += plantInfo['experience']
|
||
|
||
harvestRecords.append(f"收获作物：{soilInfo['plantName']}，数量为：{number}，经验为：{plantInfo['experience']}")
|
||
|
||
await g_pDBService.userPlant.addUserPlantByUid(uid, soilInfo['plantName'], number)
|
||
|
||
#如果到达收获次数上限
|
||
if soilInfo['harvestCount'] + 1 >= plantInfo['crop']:
|
||
await g_pDBService.userSoil.updateUserSoil(uid, i, "wiltStatus", 1)
|
||
else:
|
||
phase = await g_pDBService.plant.getPlantPhaseByName(soilInfo['plantName'])
|
||
|
||
ts, hc = int(currentTime.timestamp()), soilInfo["harvestCount"] + 1
|
||
p1, p2, *rest = phase
|
||
|
||
await g_pDBService.userSoil.updateUserSoilFields(uid, i,
|
||
{
|
||
"harvestCount": hc,
|
||
"plantTime": ts - p1 - p2,
|
||
"matureTime": ts + p2 + sum(rest),
|
||
}
|
||
)
|
||
|
||
await g_pEventManager.m_afterHarvest.emit(uid=uid, name=soilInfo['plantName'], num=number, soilIndex=i)
|
||
|
||
if experience > 0:
|
||
exp = await g_pDBService.user.getUserExpByUid(uid)
|
||
await g_pDBService.user.updateUserExpByUid(uid, exp + experience)
|
||
harvestRecords.append(f"\t累计获得经验：{experience}")
|
||
|
||
if harvestCount <= 0:
|
||
return "没有可收获的作物哦~ 不要试图拔苗助长"
|
||
else:
|
||
return "\n".join(harvestRecords)
|
||
|
||
except Exception as e:
|
||
logger.warning(f"收获操作失败！", e=e)
|
||
return "收获失败，请稍后重试！"
|
||
|
||
@classmethod
|
||
async def eradicate(cls, uid: str) -> str:
|
||
"""铲除作物
|
||
TODO 缺少随意铲除作物 目前只能铲除荒废作物
|
||
Args:
|
||
uid (str): 用户Uid
|
||
|
||
Returns:
|
||
str: 返回
|
||
"""
|
||
soilNumber = await g_pDBService.user.getUserSoilByUid(uid)
|
||
|
||
await g_pEventManager.m_beforeEradicate.emit(uid=uid)
|
||
|
||
experience = 0
|
||
for i in range(1, soilNumber + 1):
|
||
#如果没有种植
|
||
if not await g_pDBService.userSoil.isSoilPlanted(uid, i):
|
||
continue
|
||
|
||
soilInfo = await g_pDBService.userSoil.getUserSoil(uid, i)
|
||
if not soilInfo:
|
||
continue
|
||
|
||
#如果不是枯萎状态
|
||
if soilInfo.get("wiltStatus", 0) == 0:
|
||
continue
|
||
|
||
experience += 3
|
||
|
||
if g_bIsDebug:
|
||
experience += 999
|
||
|
||
#批量更新数据库操作
|
||
await g_pDBService.userSoil.deleteUserSoil(uid, i)
|
||
|
||
#铲除作物会将偷菜记录清空
|
||
await g_pDBService.userSteal.deleteStealRecord(uid, i)
|
||
|
||
await g_pEventManager.m_afterEradicate.emit(uid=uid, soilIndex=i)
|
||
|
||
if experience > 0:
|
||
exp = await g_pDBService.user.getUserExpByUid(uid)
|
||
await g_pDBService.user.updateUserExpByUid(uid, exp + experience)
|
||
|
||
return f"成功铲除荒废作物，累计获得经验：{experience}"
|
||
else:
|
||
return "没有可以铲除的作物"
|
||
|
||
@classmethod
|
||
async def getUserPlantByUid(cls, uid: str) -> bytes:
|
||
"""获取用户作物仓库
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
|
||
Returns:
|
||
bytes: 返回图片
|
||
"""
|
||
data_list = []
|
||
column_name = [
|
||
"-",
|
||
"作物名称",
|
||
"数量",
|
||
"单价",
|
||
"总价",
|
||
"是否可以上架交易行"
|
||
]
|
||
|
||
plant = await g_pDBService.userPlant.getUserPlantByUid(uid)
|
||
|
||
if plant is None:
|
||
result = await ImageTemplate.table_page(
|
||
"作物仓库",
|
||
"出售示例：@小真寻 出售作物 大白菜 [数量]",
|
||
column_name,
|
||
data_list,
|
||
)
|
||
return result.pic2bytes()
|
||
|
||
sell = ""
|
||
for name, count in plant.items():
|
||
plantInfo = await g_pDBService.plant.getPlantByName(name)
|
||
if not plantInfo:
|
||
continue
|
||
|
||
icon = ""
|
||
icon_path = g_sResourcePath / f"plant/{name}/icon.png"
|
||
if icon_path.exists():
|
||
icon = (icon_path, 33, 33)
|
||
|
||
if plantInfo['sell'] == True:
|
||
sell = "可以"
|
||
else:
|
||
sell = "不可以"
|
||
|
||
number = int(count) * plantInfo['price']
|
||
|
||
data_list.append(
|
||
[
|
||
icon,
|
||
name,
|
||
count,
|
||
plantInfo['price'],
|
||
number,
|
||
sell
|
||
]
|
||
)
|
||
|
||
result = await ImageTemplate.table_page(
|
||
"作物仓库",
|
||
"出售示例：@小真寻 出售作物 大白菜 [数量]",
|
||
column_name,
|
||
data_list,
|
||
)
|
||
|
||
return result.pic2bytes()
|
||
|
||
@classmethod
|
||
async def stealing(cls, uid: str, target: str) -> str:
|
||
"""偷菜
|
||
|
||
Args:
|
||
uid (str): 用户Uid
|
||
target (str): 被偷用户Uid
|
||
|
||
Returns:
|
||
str: 返回
|
||
"""
|
||
#用户信息
|
||
userInfo = await g_pDBService.user.getUserInfoByUid(uid)
|
||
|
||
stealTime = userInfo['stealTime']
|
||
stealCount = int(userInfo['stealCount'])
|
||
|
||
if stealTime == '':
|
||
stealTime = g_pToolManager.dateTime().date().today().strftime('%Y-%m-%d')
|
||
stealCount = 5
|
||
elif g_pToolManager.dateTime().date().fromisoformat(stealTime) != g_pToolManager.dateTime().date().today():
|
||
stealTime = g_pToolManager.dateTime().date().today().strftime('%Y-%m-%d')
|
||
stealCount = 5
|
||
|
||
if stealCount <= 0:
|
||
return "你今天可偷次数到达上限啦，手下留情吧"
|
||
|
||
#获取用户解锁地块数量
|
||
soilNumber = await g_pDBService.user.getUserSoilByUid(uid)
|
||
harvestRecords: List[str] = []
|
||
isStealingNumber = 0
|
||
isStealingPlant = 0
|
||
|
||
for i in range(1, soilNumber + 1):
|
||
#如果没有种植
|
||
if not await g_pDBService.userSoil.isSoilPlanted(uid, i):
|
||
continue
|
||
|
||
soilInfo = await g_pDBService.userSoil.getUserSoil(uid, i)
|
||
if not soilInfo:
|
||
continue
|
||
|
||
#如果是枯萎状态
|
||
if soilInfo.get("wiltStatus", 1) == 1:
|
||
continue
|
||
|
||
#作物信息
|
||
plantInfo = await g_pDBService.plant.getPlantByName(soilInfo['plantName'])
|
||
if not plantInfo:
|
||
continue
|
||
|
||
currentTime = g_pToolManager.dateTime().now()
|
||
matureTime = g_pToolManager.dateTime().fromtimestamp(int(soilInfo['matureTime']))
|
||
|
||
if currentTime >= matureTime:
|
||
#如果偷过，则跳过该土地
|
||
if await g_pDBService.userSteal.hasStealed(target, i, uid):
|
||
continue
|
||
|
||
stealingNumber = plantInfo['harvest'] - await g_pDBService.userSteal.getTotalStolenCount(target, i)
|
||
randomNumber = random.choice([1, 2])
|
||
randomNumber = min(randomNumber, stealingNumber)
|
||
|
||
if randomNumber > 0:
|
||
await g_pDBService.userPlant.addUserPlantByUid(uid, soilInfo['plantName'], randomNumber)
|
||
|
||
harvestRecords.append(f"成功偷到作物：{soilInfo['plantName']}，数量为：{randomNumber}")
|
||
|
||
isStealingPlant += 1
|
||
|
||
#如果将作物偷完，就直接更新状态 并记录用户偷取过
|
||
if plantInfo['harvest'] - randomNumber + stealingNumber == 0:
|
||
#如果作物 是最后一阶段作物且偷完 则直接枯萎
|
||
if soilInfo['harvestCount'] + 1 >= plantInfo['crop']:
|
||
await g_pDBService.userSoil.updateUserSoil(target, i, "wiltStatus", 1)
|
||
else:
|
||
phase = await g_pDBService.plant.getPlantPhaseByName(soilInfo['plantName'])
|
||
|
||
ts, hc = int(currentTime.timestamp()), soilInfo["harvestCount"] + 1
|
||
p1, p2, *rest = phase
|
||
|
||
await g_pDBService.userSoil.updateUserSoilFields(uid, i,
|
||
{
|
||
"harvestCount": hc,
|
||
"plantTime": ts - p1 - p2,
|
||
"matureTime": ts + p2 + sum(rest),
|
||
}
|
||
)
|
||
|
||
await g_pDBService.userSteal.addStealRecord(target, i, uid, randomNumber, int(g_pToolManager.dateTime().now().timestamp()))
|
||
|
||
else:
|
||
await g_pDBService.userSteal.addStealRecord(target, i, uid, randomNumber, int(g_pToolManager.dateTime().now().timestamp()))
|
||
|
||
if isStealingPlant <= 0 and isStealingNumber <= 0:
|
||
return "目标没有作物可以被偷"
|
||
elif isStealingPlant <= 0 and isStealingNumber > 0:
|
||
return "你已经偷过目标啦，请手下留情"
|
||
else:
|
||
stealCount -= 1
|
||
|
||
await g_pDBService.user.updateStealCountByUid(uid, stealCount)
|
||
|
||
return "\n".join(harvestRecords)
|
||
|
||
@classmethod
|
||
async def reclamationCondition(cls, uid: str) -> str:
|
||
userInfo = await g_pDBService.user.getUserInfoByUid(uid)
|
||
rec = g_pJsonManager.m_pLevel['reclamation']
|
||
|
||
try:
|
||
if userInfo['soil'] >= 30:
|
||
return "你已经开垦了全部土地"
|
||
|
||
rec = rec[f"{userInfo['soil'] + 1}"]
|
||
|
||
level = rec['level']
|
||
point = rec['point']
|
||
item = rec['item']
|
||
|
||
str = ""
|
||
if len(item) == 0:
|
||
str = f"下次开垦所需条件：等级：{level}，农场币：{point}"
|
||
else:
|
||
str = f"下次开垦所需条件：等级：{level}，农场币：{point}，物品：{item}"
|
||
|
||
return str
|
||
except Exception as e:
|
||
return "获取开垦土地条件失败！"
|
||
|
||
@classmethod
|
||
async def reclamation(cls, uid: str) -> str:
|
||
userInfo = await g_pDBService.user.getUserInfoByUid(uid)
|
||
level = await g_pDBService.user.getUserLevelByUid(uid)
|
||
|
||
rec = g_pJsonManager.m_pLevel['reclamation']
|
||
|
||
try:
|
||
if userInfo['soil'] >= 30:
|
||
return "你已经开垦了全部土地"
|
||
|
||
rec = rec[f"{userInfo['soil'] + 1}"]
|
||
|
||
levelFileter = rec['level']
|
||
point = rec['point']
|
||
item = rec['item']
|
||
|
||
if level[0] < levelFileter:
|
||
return f"当前用户等级{level[0]}，升级所需等级为{levelFileter}"
|
||
|
||
if userInfo['point'] < point:
|
||
return f"当前用户农场币不足，升级所需农场币为{point}"
|
||
|
||
#TODO 缺少判断消耗的item
|
||
await g_pDBService.user.updateUserPointByUid(uid, userInfo['point'] - point)
|
||
await g_pDBService.user.updateUserSoilByUid(uid, userInfo['soil'] + 1)
|
||
|
||
return "开垦土地成功！"
|
||
except Exception as e:
|
||
return "执行开垦失败！"
|
||
|
||
g_pFarmManager = CFarmManager()
|