nonebot-plugin-farm 1.2.0__py3-none-any.whl

nonebot_plugin_farm/__init__.py

@@ -0,0 +1,55 @@
+from nonebot import get_driver
+from nonebot.plugin import PluginMetadata
+
+from nonebot import logger
+
+from .command import diuse_farm, diuse_register, reclamation
+from .config import g_pConfigManager
+from .json import g_pJsonManager
+from .database.database import g_pSqlManager
+from .dbService import g_pDBService
+from .farm.farm import g_pFarmManager
+from .farm.shop import g_pShopManager
+from .request import g_pRequestManager
+
+__plugin_meta__ = PluginMetadata(
+    name="真寻农场",
+    description="快乐的农场时光",
+    usage="""
+    你也要种地?
+    指令：
+        at 开通农场
+        我的农场
+        我的农场币
+        种子商店 [页数]
+        购买种子 [作物/种子名称] [数量]
+        我的种子
+        播种 [作物/种子名称] [数量] (数量不填默认将最大可能播种
+        收获
+        铲除
+        我的作物
+        出售作物 [作物/种子名称] [数量] (不填写作物名将售卖仓库种全部作物 填作物名不填数量将指定作物全部出售
+        偷菜 at (每人每天只能偷5次
+        开垦
+        购买农场币 [数量] 数量为消耗金币的数量
+        更改农场名 [新农场名]
+    """.strip()
+)
+driver = get_driver()
+
+
+# 构造函数
+@driver.on_startup
+async def start():
+    # 初始化数据库
+    await g_pSqlManager.init()
+
+    # 初始化读取Json
+    await g_pJsonManager.init()
+
+    await g_pDBService.init()
+
+# 析构函数
+@driver.on_shutdown
+async def shutdown():
+    await g_pSqlManager.cleanup()

nonebot_plugin_farm/command.py

@@ -0,0 +1,425 @@
+from nonebot.adapters import Event, MessageTemplate
+from nonebot.rule import to_me
+from nonebot_plugin_alconna import (Alconna, AlconnaMatch, AlconnaQuery, Args,
+                                    Arparma, At, Match, MultiVar, Option,
+                                    Query, Subcommand, on_alconna, store_true)
+from nonebot_plugin_uninfo import Uninfo
+from nonebot_plugin_waiter import waiter
+
+from nonebot import logger
+from zhenxun_utils.message import MessageUtils
+
+from .dbService import g_pDBService
+from .farm.farm import g_pFarmManager
+from .farm.shop import g_pShopManager
+
+
+async def isRegisteredByUid(uid: str) -> bool:
+    result = await g_pDBService.user.isUserExist(uid)
+
+    if not result:
+        await MessageUtils.build_message("尚未开通农场，快at我发送 开通农场 开通吧").send()
+        return False
+
+    return True
+
+
+diuse_register = on_alconna(
+    Alconna("开通农场"),
+    priority=5,
+    rule=to_me(),
+    block=True,
+)
+
+@diuse_register.handle()
+async def handle_register(session: Uninfo):
+    uid = str(session.user.id)
+    user = await g_pDBService.user.getUserInfoByUid(uid)
+
+    if user:
+        await MessageUtils.build_message("🎉 您已经开通农场啦~").send(reply_to=True)
+        return
+
+    try:
+        # 获取原始用户名并安全处理
+        raw_name = str(session.user.name)
+        safe_name = sanitize_username(raw_name)
+
+        # 初始化用户信息
+        success = await g_pDBService.user.initUserInfoByUid(
+            uid=uid,
+            name=safe_name,
+            exp=0,
+            point=500
+        )
+
+        msg = (
+            "✅ 农场开通成功！\n💼 初始资金：500农场币"
+            if success
+            else "⚠️ 开通失败，请稍后再试"
+        )
+        logger.info(f"用户注册 {'成功' if success else '失败'}：{uid}")
+
+    except Exception as e:
+        msg = "⚠️ 系统繁忙，请稍后再试"
+        logger.error(f"注册异常 | UID:{uid} | 错误：{str(e)}")
+
+    await MessageUtils.build_message(msg).send(reply_to=True)
+
+def sanitize_username(username: str, max_length: int = 15) -> str:
+    """
+    安全处理用户名
+    功能：
+    1. 移除首尾空白
+    2. 过滤危险字符
+    3. 转义单引号
+    4. 处理空值
+    5. 限制长度
+    """
+    # 处理空值
+    if not username:
+        return "神秘农夫"
+
+    # 基础清洗
+    cleaned = username.strip()
+
+    # 允许的字符白名单（可自定义扩展）
+    safe_chars = {
+        '_', '-', '!', '@', '#', '$', '%', '^', '&', '*', '(', ')',
+        '+', '=', '.', ',', '~', '·', ' ',
+        'a','b','c','d','e','f','g','h','i','j','k','l','m',
+        'n','o','p','q','r','s','t','u','v','w','x','y','z',
+        'A','B','C','D','E','F','G','H','I','J','K','L','M',
+        'N','O','P','Q','R','S','T','U','V','W','X','Y','Z',
+        '0','1','2','3','4','5','6','7','8','9',
+    }
+    # 添加常用中文字符（Unicode范围）
+    safe_chars.update(chr(c) for c in range(0x4E00, 0x9FFF+1))
+
+    # 过滤危险字符
+    filtered = [
+        c if c in safe_chars or 0x4E00 <= ord(c) <= 0x9FFF
+        else ''
+        for c in cleaned
+    ]
+
+    # 合并处理结果
+    safe_str = ''.join(filtered)
+
+    # 转义单引号（双重保障）
+    escaped = safe_str.replace("'", "''")
+
+    # 处理空结果
+    if not escaped:
+        return "神秘农夫"
+
+    # 长度限制
+    return escaped[:max_length]
+
+diuse_farm = on_alconna(
+    Alconna(
+        "我的农场",
+        Option("--all", action=store_true),
+        Subcommand("my-point", help_text="我的农场币"),
+        Subcommand("seed-shop", Args["num?", int], help_text="种子商店"),
+        Subcommand("buy-seed", Args["name?", str]["num?", int], help_text="购买种子"),
+        Subcommand("my-seed", help_text="我的种子"),
+        Subcommand("sowing", Args["name?", str]["num?", int], help_text="播种"),
+        Subcommand("harvest", help_text="收获"),
+        Subcommand("eradicate", help_text="铲除"),
+        Subcommand("my-plant", help_text="我的作物"),
+        Subcommand("sell-plant", Args["name?", str]["num?", int], help_text="出售作物"),
+        Subcommand("stealing", Args["target?", At], help_text="偷菜"),
+        Subcommand("buy-point", Args["num?", int], help_text="购买农场币"),
+        #Subcommand("sell-point", Args["num?", int], help_text="转换金币")
+        Subcommand("change-name", Args["name?", str], help_text="更改农场名")
+    ),
+    priority=5,
+    block=True,
+)
+
+@diuse_farm.assign("$main")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    image = await g_pFarmManager.drawFarmByUid(uid)
+    await MessageUtils.build_message(image).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "我的农场币",
+    command="我的农场",
+    arguments=["my-point"],
+    prefix=True,
+)
+
+@diuse_farm.assign("my-point")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+    point = await g_pDBService.user.getUserPointByUid(uid)
+
+    if point < 0:
+        await MessageUtils.build_message("尚未开通农场，快at我发送 开通农场 开通吧").send()
+        return False
+
+    await MessageUtils.build_message(f"你的当前农场币为: {point}").send(reply_to=True)
+
+diuse_farm.shortcut(
+    "种子商店(.*?)",
+    command="我的农场",
+    arguments=["seed-shop"],
+    prefix=True,
+)
+
+@diuse_farm.assign("seed-shop")
+async def _(session: Uninfo, num: Query[int] = AlconnaQuery("num", 1)):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    image = await g_pShopManager.getSeedShopImage(num.result)
+    await MessageUtils.build_message(image).send()
+
+diuse_farm.shortcut(
+    "购买种子(?P<name>.*?)",
+    command="我的农场",
+    arguments=["buy-seed", "{name}"],
+    prefix=True,
+)
+
+@diuse_farm.assign("buy-seed")
+async def _(session: Uninfo, name: Match[str], num: Query[int] = AlconnaQuery("num", 1),):
+    if not name.available:
+        await MessageUtils.build_message(
+            "请在指令后跟需要购买的种子名称"
+        ).finish(reply_to=True)
+
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pShopManager.buySeed(uid, name.result, num.result)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "我的种子",
+    command="我的农场",
+    arguments=["my-seed"],
+    prefix=True,
+)
+
+@diuse_farm.assign("my-seed")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.getUserSeedByUid(uid)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "播种(?P<name>.*?)",
+    command="我的农场",
+    arguments=["sowing", "{name}"],
+    prefix=True,
+)
+
+@diuse_farm.assign("sowing")
+async def _(session: Uninfo, name: Match[str], num: Query[int] = AlconnaQuery("num", -1),):
+    if not name.available:
+        await MessageUtils.build_message(
+            "请在指令后跟需要播种的种子名称"
+        ).finish(reply_to=True)
+
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.sowing(uid, name.result, num.result)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+
+diuse_farm.shortcut(
+    "收获",
+    command="我的农场",
+    arguments=["harvest"],
+    prefix=True,
+)
+
+@diuse_farm.assign("harvest")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.harvest(uid)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "铲除",
+    command="我的农场",
+    arguments=["eradicate"],
+    prefix=True,
+)
+
+@diuse_farm.assign("eradicate")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.eradicate(uid)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+
+diuse_farm.shortcut(
+    "我的作物",
+    command="我的农场",
+    arguments=["my-plant"],
+    prefix=True,
+)
+
+@diuse_farm.assign("my-plant")
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.getUserPlantByUid(uid)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+
+reclamation = on_alconna(
+    Alconna("开垦"),
+    priority=5,
+    block=True,
+)
+
+@reclamation.handle()
+async def _(session: Uninfo):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    condition = await g_pFarmManager.reclamationCondition(uid)
+    condition += "\n 回复是将执行开垦"
+    await MessageUtils.build_message(condition).send(reply_to=True)
+
+    @waiter(waits=["message"], keep_session=True)
+    async def check(event: Event):
+        return event.get_plaintext()
+
+    resp = await check.wait(timeout=60)
+    if resp is None:
+        await MessageUtils.build_message("等待超时").send(reply_to=True)
+        return
+    if not resp == "是":
+        return
+
+    res = await g_pFarmManager.reclamation(uid)
+    await MessageUtils.build_message(res).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "出售作物(?P<name>.*?)",
+    command="我的农场",
+    arguments=["sell-plant", "{name}"],
+    prefix=True,
+)
+
+@diuse_farm.assign("sell-plant")
+async def _(session: Uninfo, name: Match[str], num: Query[int] = AlconnaQuery("num", -1),):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pShopManager.sellPlantByUid(uid, name.result, num.result)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "偷菜",
+    command="我的农场",
+    arguments=["stealing"],
+    prefix=True,
+)
+
+@diuse_farm.assign("stealing")
+async def _(session: Uninfo, target: Match[At]):
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    if not target.available:
+        await MessageUtils.build_message("请在指令后跟需要at的人").finish(reply_to=True)
+
+    tar = target.result
+    result = await g_pDBService.user.isUserExist(tar.target)
+
+    if not result:
+        await MessageUtils.build_message("目标尚未开通农场，快邀请ta开通吧").send()
+        return None
+
+    result = await g_pFarmManager.stealing(uid, tar.target)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+diuse_farm.shortcut(
+    "购买农场币(.*?)",
+    command="我的农场",
+    arguments=["buy-point"],
+    prefix=True,
+)
+
+@diuse_farm.assign("buy-point")
+async def _(session: Uninfo, num: Query[int] = AlconnaQuery("num", 0)):
+    if num.result <= 0:
+        await MessageUtils.build_message(
+            "请在指令后跟需要购买农场币的数量"
+        ).finish(reply_to=True)
+
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    result = await g_pFarmManager.buyPointByUid(uid, num.result)
+    await MessageUtils.build_message(result).send(reply_to=True)
+
+
+diuse_farm.shortcut(
+    "更改农场名(?P<name>)",
+    command="我的农场",
+    arguments=["change-name", "{name}"],
+    prefix=True,
+)
+
+@diuse_farm.assign("change-name")
+async def _(session: Uninfo, name: Match[str]):
+    if not name.available:
+        await MessageUtils.build_message(
+            "请在指令后跟需要更改的用户名"
+        ).finish(reply_to=True)
+
+    uid = str(session.user.id)
+
+    if await isRegisteredByUid(uid) == False:
+        return
+
+    safeName = sanitize_username(name.result)
+
+    result = await g_pDBService.user.updateUserNameByUid(uid, safeName)
+
+    if result == True:
+        await MessageUtils.build_message("更新用户名成功！").send(reply_to=True)
+    else:
+        await MessageUtils.build_message("更新用户名失败！").send(reply_to=True)

nonebot_plugin_farm/config/item.json

@@ -0,0 +1,22 @@
+{
+    "version": 0.01,
+    "zhuShi":
+    [
+        "name: 名称",
+        "level: 等级",
+        "price: 单价",
+        "icon: 显示图片",
+        "sell: 是否可以上架交易行"
+    ],
+    "item":
+    {
+        "muBan":
+        {
+            "name": "木板",
+            "level": 0,
+            "price": 5000,
+            "icon": "Test",
+            "sell": true
+        }
+    }
+}

nonebot_plugin_farm/config/level.json

@@ -0,0 +1,172 @@
+{
+    "version": 0.01,
+    "soil":[1, 1, 1, 2, 3, 5, 7, 9, 11, 13,
+        15, 17, 19, 21, 23, 25, 27, 29, 31, 33,
+        36, 39, 42, 45, 48, 51, 54, 57, 60, 70
+    ],
+    "reclamation":
+    {
+        "4":
+        {
+            "level":  2,
+            "point": 800,
+            "item": []
+        },
+        "5":
+        {
+            "level":  3,
+            "point": 1300,
+            "item": []
+        },
+        "6":
+        {
+            "level":  5,
+            "point": 3200,
+            "item": []
+        },
+        "7":
+        {
+            "level":  7,
+            "point": 5500,
+            "item": []
+        },
+        "8":
+        {
+            "level":  9,
+            "point": 12000,
+            "item": []
+        },
+        "9":
+        {
+            "level":  11,
+            "point": 19800,
+            "item": []
+        },
+        "10":
+        {
+            "level": 13,
+            "point": 30000,
+            "item": []
+        },
+        "11":
+        {
+            "level": 15,
+            "point": 50000,
+            "item": []
+        },
+        "12":
+        {
+            "level": 17,
+            "point": 70000,
+            "item": []
+        },
+        "13":
+        {
+            "level": 19,
+            "point": 90000,
+            "item": []
+        },
+        "14":
+        {
+            "level": 21,
+            "point": 130000,
+            "item": []
+        },
+        "15":
+        {
+            "level": 23,
+            "point": 150000,
+            "item": []
+        },
+        "16":
+        {
+            "level": 25,
+            "point": 180000,
+            "item": []
+        },
+        "17":
+        {
+            "level": 27,
+            "point": 230000,
+            "item": []
+        },
+        "18":
+        {
+            "level": 29,
+            "point": 300000,
+            "item": []
+        },
+        "19":
+        {
+            "level": 31,
+            "point": 500000,
+            "item": []
+        },
+        "20":
+        {
+            "level": 33,
+            "point": 850000,
+            "item": []
+        },
+        "21":
+        {
+            "level": 36,
+            "point": 1100000,
+            "item": []
+        },
+        "22":
+        {
+            "level": 39,
+            "point": 1300000,
+            "item": []
+        },
+        "23":
+        {
+            "level": 42,
+            "point": 1500000,
+            "item": []
+        },
+        "24":
+        {
+            "level": 45,
+            "point": 1700000,
+            "item": []
+        },
+        "25":
+        {
+            "level": 48,
+            "point": 2000000,
+            "item": []
+        },
+        "26":
+        {
+            "level": 51,
+            "point": 2200000,
+            "item": []
+        },
+        "27":
+        {
+            "level": 54,
+            "point": 2800000,
+            "item": []
+        },
+        "28":
+        {
+            "level": 57,
+            "point": 3500000,
+            "item": []
+        },
+        "29":
+        {
+            "level": 60,
+            "point": 4300000,
+            "item": []
+        },
+        "30":
+        {
+            "level": 70,
+            "point": 5200000,
+            "item": []
+        }
+    }
+}