新增菜园系统
This commit is contained in:
330
Plugins/WPSGardenSystem/garden_service.py
Normal file
330
Plugins/WPSGardenSystem/garden_service.py
Normal file
@@ -0,0 +1,330 @@
|
||||
"""Garden service handling planting, harvesting, stealing, and selling."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import random
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Dict, List, Optional, Tuple
|
||||
|
||||
from PWF.Convention.Runtime.Architecture import Architecture
|
||||
from PWF.Convention.Runtime.GlobalConfig import ConsoleFrontColor, ProjectConfig
|
||||
from PWF.CoreModules.database import get_db
|
||||
from PWF.CoreModules.plugin_interface import PluginInterface
|
||||
from PWF.CoreModules.flags import get_internal_debug
|
||||
from pydantic import BaseModel
|
||||
|
||||
from .garden_models import (
|
||||
GARDEN_CROPS,
|
||||
GARDEN_FRUITS,
|
||||
GARDEN_MISC_ITEMS,
|
||||
GardenCropDefinition,
|
||||
get_garden_db_models,
|
||||
)
|
||||
|
||||
Timestamp = str
|
||||
|
||||
|
||||
def _local_now() -> datetime:
|
||||
return datetime.now()
|
||||
|
||||
|
||||
def _parse_local_iso(ts: str) -> datetime:
|
||||
return datetime.fromisoformat(ts)
|
||||
|
||||
|
||||
class GardenConfig(BaseModel):
|
||||
max_plots: int
|
||||
sale_multiplier: int
|
||||
fortune_coeff: float
|
||||
theft_threshold_ratio: float
|
||||
seed_store_limit: int
|
||||
|
||||
class Config:
|
||||
allow_mutation = False
|
||||
|
||||
@classmethod
|
||||
def load(cls) -> "GardenConfig":
|
||||
project_config: ProjectConfig = Architecture.Get(ProjectConfig)
|
||||
max_plots = int(project_config.FindItem("garden_max_plots_per_user", 4))
|
||||
sale_multiplier = int(project_config.FindItem("garden_sale_multiplier", 10))
|
||||
fortune_coeff = float(project_config.FindItem("garden_fortune_coeff", 0.03))
|
||||
theft_ratio = float(project_config.FindItem("garden_theft_threshold_ratio", 0.5))
|
||||
seed_store_limit = int(project_config.FindItem("garden_seed_store_limit", 5))
|
||||
project_config.SaveProperties()
|
||||
return cls(
|
||||
max_plots=max_plots,
|
||||
sale_multiplier=sale_multiplier,
|
||||
fortune_coeff=fortune_coeff,
|
||||
theft_threshold_ratio=theft_ratio,
|
||||
seed_store_limit=seed_store_limit,
|
||||
)
|
||||
|
||||
|
||||
class GardenService:
|
||||
def __init__(self) -> None:
|
||||
self._config = GardenConfig.load()
|
||||
self._db = get_db()
|
||||
self._logger: ProjectConfig = Architecture.Get(ProjectConfig)
|
||||
|
||||
@property
|
||||
def config(self) -> GardenConfig:
|
||||
return self._config
|
||||
|
||||
# region Query helpers
|
||||
def list_plots(self, user_id: int) -> List[Dict[str, object]]:
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"SELECT * FROM garden_plots WHERE user_id = ? ORDER BY plot_index ASC",
|
||||
(user_id,),
|
||||
)
|
||||
rows = cursor.fetchall()
|
||||
return [dict(row) for row in rows]
|
||||
|
||||
def get_plot(self, user_id: int, plot_index: int) -> Optional[Dict[str, object]]:
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"SELECT * FROM garden_plots WHERE user_id = ? AND plot_index = ?",
|
||||
(user_id, plot_index),
|
||||
)
|
||||
row = cursor.fetchone()
|
||||
return dict(row) if row else None
|
||||
|
||||
# endregion
|
||||
|
||||
# region Planting
|
||||
def plant(
|
||||
self,
|
||||
*,
|
||||
user_id: int,
|
||||
chat_id: int,
|
||||
seed_id: str,
|
||||
plot_index: Optional[int] = None,
|
||||
register_callback: Optional[
|
||||
Tuple[PluginInterface, str]
|
||||
] = None,
|
||||
) -> Tuple[int, datetime]:
|
||||
crop = GARDEN_CROPS.get(seed_id)
|
||||
if not crop:
|
||||
raise ValueError("未知的种子")
|
||||
plots = self.list_plots(user_id)
|
||||
used_indices = {int(plot["plot_index"]) for plot in plots}
|
||||
if len(used_indices) >= self._config.max_plots:
|
||||
raise ValueError("没有空闲土地")
|
||||
if plot_index is None:
|
||||
for idx in range(1, self._config.max_plots + 1):
|
||||
if idx not in used_indices:
|
||||
plot_index = idx
|
||||
break
|
||||
if plot_index is None:
|
||||
raise ValueError("无法分配地块")
|
||||
planted_at = _local_now()
|
||||
mature_at = planted_at + timedelta(minutes=crop.growth_minutes)
|
||||
debug_mode = get_internal_debug()
|
||||
if debug_mode:
|
||||
mature_at = planted_at
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"""
|
||||
INSERT INTO garden_plots (
|
||||
user_id, chat_id, plot_index, seed_id, seed_quality, planted_at, mature_at,
|
||||
is_mature, base_yield, extra_type, extra_payload, remaining_fruit, theft_users, scheduled_task_id
|
||||
) VALUES (?, ?, ?, ?, ?, ?, ?, 0, ?, ?, ?, ?, ?, NULL)
|
||||
""",
|
||||
(
|
||||
user_id,
|
||||
chat_id,
|
||||
plot_index,
|
||||
crop.seed_id,
|
||||
crop.tier,
|
||||
planted_at.isoformat(),
|
||||
mature_at.isoformat(),
|
||||
crop.base_yield,
|
||||
crop.extra_reward.kind,
|
||||
json.dumps(crop.extra_reward.payload) if crop.extra_reward else None,
|
||||
crop.base_yield,
|
||||
json.dumps([]),
|
||||
),
|
||||
)
|
||||
self._db.conn.commit()
|
||||
|
||||
task_id = None
|
||||
if register_callback:
|
||||
plugin, callback_name = register_callback
|
||||
delay_ms = 0 if debug_mode else int(crop.growth_minutes * 60 * 1000)
|
||||
task_id = plugin.register_clock(
|
||||
getattr(plugin, callback_name),
|
||||
delay_ms,
|
||||
kwargs={"user_id": user_id, "chat_id": chat_id, "plot_index": plot_index},
|
||||
)
|
||||
cursor.execute(
|
||||
"UPDATE garden_plots SET scheduled_task_id = ? WHERE user_id = ? AND plot_index = ?",
|
||||
(task_id, user_id, plot_index),
|
||||
)
|
||||
self._db.conn.commit()
|
||||
return plot_index, mature_at.isoformat()
|
||||
|
||||
def mark_mature(self, user_id: int, plot_index: int) -> None:
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"UPDATE garden_plots SET is_mature = 1, scheduled_task_id = NULL WHERE user_id = ? AND plot_index = ?",
|
||||
(user_id, plot_index),
|
||||
)
|
||||
self._db.conn.commit()
|
||||
|
||||
# endregion
|
||||
|
||||
# region Harvest
|
||||
def harvest(self, *, user_id: int, plot_index: int, fortune_value: float) -> Dict[str, object]:
|
||||
plot = self.get_plot(user_id, plot_index)
|
||||
if not plot:
|
||||
raise ValueError("指定地块不存在")
|
||||
if int(plot["is_mature"]) != 1:
|
||||
raise ValueError("作物尚未成熟")
|
||||
crop = GARDEN_CROPS.get(plot["seed_id"])
|
||||
if not crop:
|
||||
raise ValueError("未知作物")
|
||||
base_yield = int(plot["base_yield"])
|
||||
extra_reward = None
|
||||
if crop.extra_reward:
|
||||
base_rate = crop.extra_reward.base_rate
|
||||
probability = max(
|
||||
0.0,
|
||||
min(1.0, base_rate + fortune_value * self._config.fortune_coeff),
|
||||
)
|
||||
if random.random() <= probability:
|
||||
if crop.extra_reward.kind == "points":
|
||||
data = crop.extra_reward.payload
|
||||
max_points = min(
|
||||
data.get("max", base_yield * crop.seed_price),
|
||||
crop.seed_price * self._config.sale_multiplier,
|
||||
)
|
||||
min_points = data.get("min", 0)
|
||||
if max_points > 0:
|
||||
amount = random.randint(min_points, max_points)
|
||||
extra_reward = {"type": "points", "amount": amount}
|
||||
elif crop.extra_reward.kind == "item" and crop.extra_item_id:
|
||||
data = crop.extra_reward.payload
|
||||
min_qty = max(0, data.get("min", 0))
|
||||
max_qty = max(min_qty, data.get("max", min_qty))
|
||||
if max_qty > 0:
|
||||
quantity = random.randint(min_qty, max_qty)
|
||||
extra_reward = {
|
||||
"type": "item",
|
||||
"item_id": crop.extra_item_id,
|
||||
"quantity": quantity,
|
||||
}
|
||||
result = {
|
||||
"crop": crop,
|
||||
"base_yield": base_yield,
|
||||
"extra": extra_reward,
|
||||
}
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"DELETE FROM garden_plots WHERE user_id = ? AND plot_index = ?",
|
||||
(user_id, plot_index),
|
||||
)
|
||||
self._db.conn.commit()
|
||||
return result
|
||||
|
||||
def clear_plot(self, *, user_id: int, plot_index: int) -> bool:
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"DELETE FROM garden_plots WHERE user_id = ? AND plot_index = ?",
|
||||
(user_id, plot_index),
|
||||
)
|
||||
deleted = cursor.rowcount > 0
|
||||
if deleted:
|
||||
self._db.conn.commit()
|
||||
return deleted
|
||||
|
||||
# endregion
|
||||
|
||||
# region Steal
|
||||
def steal(self, *, thief_id: int, owner_id: int, plot_index: int) -> Dict[str, object]:
|
||||
plot = self.get_plot(owner_id, plot_index)
|
||||
if not plot:
|
||||
raise ValueError("目标地块不存在")
|
||||
if int(plot["is_mature"]) != 1:
|
||||
raise ValueError("目标作物尚未成熟")
|
||||
crop = GARDEN_CROPS.get(plot["seed_id"])
|
||||
if not crop:
|
||||
raise ValueError("未知作物")
|
||||
remaining = int(plot["remaining_fruit"])
|
||||
threshold = int(round(int(plot["base_yield"]) * self._config.theft_threshold_ratio))
|
||||
if remaining <= threshold:
|
||||
raise ValueError("果实剩余不足，无法偷取")
|
||||
theft_users = set(json.loads(plot["theft_users"]))
|
||||
if thief_id in theft_users:
|
||||
raise ValueError("你已经偷取过该作物")
|
||||
theft_users.add(thief_id)
|
||||
remaining -= 1
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"""
|
||||
UPDATE garden_plots SET remaining_fruit = ?, theft_users = ?
|
||||
WHERE user_id = ? AND plot_index = ?
|
||||
""",
|
||||
(
|
||||
remaining,
|
||||
json.dumps(list(theft_users)),
|
||||
owner_id,
|
||||
plot_index,
|
||||
),
|
||||
)
|
||||
self._db.conn.commit()
|
||||
return {
|
||||
"crop": crop,
|
||||
"stolen_quantity": 1,
|
||||
"remaining": remaining,
|
||||
"chat_id": plot["chat_id"],
|
||||
}
|
||||
|
||||
# endregion
|
||||
|
||||
# region Selling
|
||||
def sell_fruit(self, *, user_id: int, fruit_id: str, quantity: int) -> Tuple[int, int]:
|
||||
crop = GARDEN_FRUITS.get(fruit_id)
|
||||
if not crop:
|
||||
raise ValueError("未知果实")
|
||||
if quantity <= 0:
|
||||
raise ValueError("数量必须大于0")
|
||||
price_per = crop.seed_price * self._config.sale_multiplier
|
||||
total_points = price_per * quantity
|
||||
return total_points, price_per
|
||||
|
||||
# endregion
|
||||
|
||||
# region Maintenance
|
||||
def recover_overdue_plots(self) -> None:
|
||||
cursor = self._db.conn.cursor()
|
||||
cursor.execute(
|
||||
"SELECT user_id, plot_index, mature_at, is_mature FROM garden_plots WHERE is_mature = 0",
|
||||
)
|
||||
rows = cursor.fetchall()
|
||||
now = _local_now()
|
||||
updated = 0
|
||||
for row in rows:
|
||||
mature_at = _parse_local_iso(row["mature_at"])
|
||||
if mature_at <= now:
|
||||
self.mark_mature(row["user_id"], row["plot_index"])
|
||||
updated += 1
|
||||
if updated:
|
||||
self._logger.Log(
|
||||
"Info",
|
||||
f"{ConsoleFrontColor.GREEN}同步成熟地块 {updated} 个{ConsoleFrontColor.RESET}",
|
||||
)
|
||||
|
||||
# endregion
|
||||
|
||||
# region Utilities
|
||||
def format_display_time(self, iso_ts: str) -> str:
|
||||
try:
|
||||
dt = _parse_local_iso(iso_ts)
|
||||
return dt.strftime("%Y年%m月%d日 %H时%M分")
|
||||
except Exception:
|
||||
return iso_ts
|
||||
|
||||
# endregion
|
||||
|
||||
__all__ = ["GardenService", "GardenConfig", "get_garden_db_models"]
|
||||
Reference in New Issue
Block a user