40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
"""Remove (clear) plot plugin for garden system."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from .garden_plugin_base import WPSGardenBase
|
|
|
|
|
|
class WPSGardenRemove(WPSGardenBase):
|
|
def wake_up(self) -> None:
|
|
super().wake_up()
|
|
self.register_plugin("remove")
|
|
self.register_plugin("铲除")
|
|
|
|
async def callback(self, message: str, chat_id: int, user_id: int) -> Optional[str]:
|
|
payload = self.parse_message_after_at(message).strip()
|
|
if not payload:
|
|
return await self.send_markdown_message("❌ 指令格式：`铲除 <地块序号>`", chat_id, user_id)
|
|
tokens = [token.strip() for token in payload.split() if token.strip()]
|
|
if not tokens or not tokens[0].isdigit():
|
|
return await self.send_markdown_message("❌ 指令格式：`铲除 <地块序号>`", chat_id, user_id)
|
|
plot_index = int(tokens[0])
|
|
if plot_index <= 0:
|
|
return await self.send_markdown_message("❌ 地块序号必须为正整数", chat_id, user_id)
|
|
|
|
success = self.service().clear_plot(user_id=user_id, plot_index=plot_index)
|
|
if not success:
|
|
return await self.send_markdown_message("❌ 指定地块不存在或已为空", chat_id, user_id)
|
|
|
|
message_body = (
|
|
"# 🧹 铲除完成\n"
|
|
f"- 已清空地块 {plot_index}\n"
|
|
"- 可以重新种植新的作物"
|
|
)
|
|
return await self.send_markdown_message(message_body, chat_id, user_id)
|
|
|
|
|
|
__all__ = ["WPSGardenRemove"]
|