Diuse/zhenxun_plugin_farm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✨ 新增收获逻辑

Browse Source 
✨ 新增播种逻辑
✨ 新增绘制作物逻辑
This commit is contained in:
Diuse
2025-03-20 18:06:38 +08:00
parent b1788f42c2
commit 59bb2068df
7 changed files with 454 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files
farm/farm.py
+271 -62
View File
@@ -1,4 +1,8 @@
from numpy import arange
import asyncio
import logging
from datetime import datetime
from io import StringIO
from typing import Dict, List, Tuple
from zhenxun.services.log import logger
from zhenxun.utils._build_image import BuildImage
@@ -22,15 +26,15 @@ class CFarmManager:
"""
soilNumber = await g_pSqlManager.getUserLevelByUid(uid)
img = BuildImage(background=g_sResourcePath / "background/background.jpg")
img = BuildImage(background = g_sResourcePath / "background/background.jpg")
soilSize = g_pJsonManager.m_pSoil['size'] # type: ignore
#TODO 缺少判断用户土地资源状况
soil = BuildImage(background=g_sResourcePath / "soil/普通土地.png")
soil = BuildImage(background = g_sResourcePath / "soil/普通土地.png")
await soil.resize(0, soilSize[0], soilSize[1])
grass = BuildImage(background=g_sResourcePath / "soil/草土地.png")
grass = BuildImage(background = g_sResourcePath / "soil/草土地.png")
await grass.resize(0, soilSize[0], soilSize[1])
soilPos = g_pJsonManager.m_pSoil['soil'] # type: ignore
@@ -38,7 +42,9 @@ class CFarmManager:
x = 0
y = 0
isFirst = True
isFirstExpansion = True #首次添加扩建图片
isFirstRipe = True
plant = None
for index, level in enumerate(soilUnlock):
x = soilPos[str(index + 1)]['x']
y = soilPos[str(index + 1)]['y']
@@ -47,25 +53,88 @@ class CFarmManager:
if soilNumber >= int(level):
await img.paste(soil, (x, y))
#TODO 缺少判断土地上是否有农作物
plant = BuildImage(background=g_sResourcePath / "plant/basic/0.png")
await plant.resize(0, 35, 58)
await img.paste(plant, (x + 100, y + 50))
isPlant, plant, isRipe= await cls.drawSoilPlant(uid, f"soil{str(index + 1)}")
if isPlant:
await img.paste(plant, (x + soilSize[0] // 2 - plant.width // 2,
y + soilSize[1] // 2 - plant.height // 2))
#首次添加可收获图片
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 isFirst:
isFirst = False
if isFirstExpansion:
isFirstExpansion = False
#首次添加扩建图片
expansion = BuildImage(background=g_sResourcePath / "background/expansion.png")
expansion = BuildImage(background = g_sResourcePath / "background/expansion.png")
await expansion.resize(0, 69, 69)
await img.paste(expansion, (x + 85, y + 20))
await img.paste(expansion, (x + soilSize[0] // 2 - expansion.width // 2,
y + soilSize[1] // 2 - expansion.height))
await img.resize(0.6)
return img.pic2bytes()
@classmethod
async def getUserPlantByUid(cls, uid: str) -> bytes:
async def drawSoilPlant(cls, uid: str, soilid: str) -> tuple[bool, BuildImage, bool]:
"""绘制植物资源
Args:
uid (str): 用户Uid
soilid (str): 土地id
Returns:
tuple[bool, BuildImage]: [绘制是否成功，资源图片, 是否成熟]
"""
plant = None
soilStatus, soilInfo = await g_pSqlManager.getUserSoilStatusBySoilID(uid, soilid)
if soilStatus == True:
return False, None, False
else:
soilInfo = soilInfo.split(',')
plantInfo = g_pJsonManager.m_pPlant['plant'][soilInfo[0]] # type: ignore
if int(soilInfo[3]) == 4:
plant = BuildImage(background = g_sResourcePath / f"plant/basic/9.png")
return True, plant, False
currentTime = datetime.now()
matureTime = datetime.fromtimestamp(int(soilInfo[2]))
if currentTime >= matureTime:
phase = int(plantInfo['phase'])
plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo[0]}/{phase - 1}.png")
return True, plant, True
else:
plantedTime = datetime.fromtimestamp(int(soilInfo[1]))
elapsedTime = currentTime - plantedTime
elapsedMinutes = elapsedTime.total_seconds() / 60
currentStage = int(elapsedMinutes / (plantInfo['time'] / (plantInfo['phase'] - 1)))
#TODO 缺少判断部分种子是否是通用0阶段图片
if currentStage <= 0:
plant = BuildImage(background = g_sResourcePath / f"plant/basic/0.png")
await plant.resize(0, 35, 58)
else:
plant = BuildImage(background = g_sResourcePath / f"plant/{soilInfo[0]}/{currentStage}.png")
return True, plant, False
@classmethod
async def getUserSeedByUid(cls, uid: str) -> bytes:
"""获取用户种子仓库
Args:
@@ -88,12 +157,194 @@ class CFarmManager:
"是否可以上架交易行"
]
seed = await g_pSqlManager.getUserSeedByUid(uid)
if seed == None:
result = await ImageTemplate.table_page(
"种子仓库",
"播种示例：@小真寻 播种 大白菜 [数量]",
column_name,
data_list,
)
return result.pic2bytes()
sell = ""
for item in seed.split(','):
if '|' in item:
seedName, count = item.split('|', 1) # 分割一次，避免多竖线问题
try:
plantInfo = g_pJsonManager.m_pPlant['plant'][seedName] # type: ignore
icon = ""
icon_path = g_sResourcePath / f"plant/{seedName}/icon.png"
if icon_path.exists():
icon = (icon_path, 33, 33)
if plantInfo['again'] == True:
sell = "可以"
else:
sell = "不可以"
data_list.append(
[
icon,
seedName,
count,
plantInfo['experience'],
plantInfo['harvest'],
plantInfo['time'],
plantInfo['crop'],
plantInfo['again'],
sell
]
)
except Exception as e:
continue
result = await ImageTemplate.table_page(
"种子仓库",
"播种示例：@小真寻 播种 大白菜 [数量]",
column_name,
data_list,
)
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:
"""
plant = await g_pSqlManager.getUserSeedByUid(uid)
if plant == None:
return "你的种子仓库是空的，快去买点吧！"
plantDict = {}
for item in plant.split(','):
if '|' in item:
seed_name, count = item.split('|', 1)
plantDict[seed_name] = int(count)
isAll = False
if num == -1:
isAll = True
soilNumber = await g_pSqlManager.getUserSoilByUid(uid)
for i in range(1, soilNumber + 1):
if plantDict[name] > 0:
if isAll or num > 0:
soilName = f"soil{i}"
success, message = await g_pSqlManager.getUserSoilStatusBySoilID(uid, soilName)
if success:
# 更新种子数量
num -= 1
plantDict[name] -= 1
if plantDict[name] == 0:
del plantDict[name]
# 更新数据库
await g_pSqlManager.updateUserSoilStatusByPlantName(uid, soilName, name)
await g_pSqlManager.updateUserSeedByUid(
uid,
','.join([f"{k}|{v}" for k, v in plantDict.items()])
)
if num > 0:
return f"播种数量超出解锁土地数量，已将可播种土地成功播种{name}！仓库还剩下{plantDict[name]}个种子"
else:
return f"播种{name}成功！仓库还剩下{plantDict[name]}个种子"
@classmethod
async def harvest(cls, uid: str) -> str:
"""收获作物
Args:
uid (str): 用户Uid
Returns:
str: 返回
"""
soilNumber = await g_pSqlManager.getUserLevelByUid(uid)
soilUnlock = g_pJsonManager.m_pLevel['soil'] # type: ignore
plant = {}
soilNames = [f"soil{i + 1}" for i, level in enumerate(soilUnlock) if soilNumber >= level]
soilStatuses = await asyncio.gather(*[
g_pSqlManager.getUserSoilStatusBySoilID(uid, name)
for name in soilNames
])
plant: Dict[str, int] = {}
harvest_records: List[str] = []
for (soil_name, (status, info)) in zip(soilNames, soilStatuses):
if not status:
soil_info = info.split(',')
plant_id = soil_info[0]
plant_info = g_pJsonManager.m_pPlant['plant'][plant_id] # type: ignore
current_time = datetime.now()
mature_time = datetime.fromtimestamp(int(soil_info[2]))
if current_time >= mature_time:
plant[plant_id] = plant.get(plant_id, 0) + plant_info['harvest']
harvest_records.append(f"收获作物：{plant_id}，数量为：{plant_info['harvest']}")
# 批量更新数据库操作
await g_pSqlManager.updateUserSoilStatusByPlantName(uid, soil_name, "", 4)
if not plant:
return "可收获作物为0哦~ 不要试图拔苗助长"
else:
# 批量更新用户作物数据
await g_pSqlManager.updateUserPlantByUid(
uid,
','.join([f"{k}|{v}" for k, v in plant.items()])
)
return "\n".join(harvest_records)
@classmethod
async def getUserPlantByUid(cls, uid: str) -> bytes:
"""获取用户作物仓库
Args:
uid (str): 用户Uid
Returns:
bytes: 返回图片
"""
data_list = []
column_name = [
"-",
"作物名称",
"数量",
"单价",
"总价",
"是否可以上架交易行"
]
plant = await g_pSqlManager.getUserPlantByUid(uid)
if plant == None:
result = await ImageTemplate.table_page(
"种子仓库",
"播种示例：@小真寻 播种 大白菜",
"作物仓库",
"播种示例：@小真寻 出售作物 大白菜 [数量]",
column_name,
data_list,
)
@@ -122,11 +373,8 @@ class CFarmManager:
icon,
plantName,
count,
plantInfo['experience'],
plantInfo['harvest'],
plantInfo['time'],
plantInfo['crop'],
plantInfo['again'],
plantInfo['price'],
count * int(plantInfo['price']),
sell
]
)
@@ -135,51 +383,12 @@ class CFarmManager:
continue
result = await ImageTemplate.table_page(
"种子商店",
"购买示例：@小真寻 购买种子 大白菜 5",
"作物仓库",
"播种示例：@小真寻 出售作物 大白菜 [数量]",
column_name,
data_list,
)
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:
"""
plant = await g_pSqlManager.getUserPlantByUid(uid)
if plant == None:
return "你的种子仓库是空的，快去买点吧！"
for item in plant.split(','):
if '|' in item:
plantName, count = item.split('|', 1) # 分割一次，避免多竖线问题
#判断仓库是否有当前播种种子
if plantName == name:
count = int(count)
#获取用户解锁多少块地
soilName = ""
soilNumber = await g_pSqlManager.getUserSoilByUid(uid)
#遍历地块，查看地块是否可以播种
for i in arange(1, soilNumber + 1):
if count > 0:
soilName = f"soil{str(i)}"
#如果可以播种
if await g_pSqlManager.getUserSoilStatusBySoilID(uid, soilName):
count -= 1
await g_pSqlManager.updateUserSoilStatusBySowing(uid, soilName, plantName)
return f"播种{plantName}成功！"
return "播种失败"
g_pFarmManager = CFarmManager()
farm/shop.py
+20 -7
View File
@@ -9,11 +9,13 @@ from ..database import g_pSqlManager
class CShopManager:
@classmethod
async def getPlantShopImage(cls) -> bytes:
"""by ATTomato
async def getSeedShopImage(cls) -> bytes:
"""获取商店页面
TODO: 缺少翻页功能
Returns:
bytes: _description_
bytes: 返回商店图片bytes
"""
data_list = []
@@ -68,7 +70,18 @@ class CShopManager:
return result.pic2bytes()
@classmethod
async def buyPlant(cls, uid: str, name: str, num: int = 1) -> str:
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 "请输入购买数量！"
@@ -89,7 +102,7 @@ class CShopManager:
if point < total:
return "你的农场币不够哦~ 快速速氪金吧！"
else:
p = await g_pSqlManager.getUserPlantByUid(uid)
p = await g_pSqlManager.getUserSeedByUid(uid)
if not p == None:
for item in p.split(','):
@@ -102,10 +115,10 @@ class CShopManager:
else:
userPlants[name] = num
plant_list = [f"{k}|{v}" for k, v in userPlants.items()]
plantList = [f"{k}|{v}" for k, v in userPlants.items()]
await g_pSqlManager.updateUserPointByUid(uid, point - total)
await g_pSqlManager.updateUserPlantByUid(uid, ','.join(plant_list))
await g_pSqlManager.updateUserSeedByUid(uid, ','.join(plantList))
return f"成功购买{name}，当前仓库数量为：{userPlants[name]}，花费{total}农场币, 剩余{point - total}农场币"