zhenxun_bot/zhenxun/builtin_plugins/sign_in/__init__.py

from nonebot.plugin import PluginMetadata
from nonebot_plugin_session import EventSession
from nonebot_plugin_apscheduler import scheduler
from nonebot_plugin_alconna import (
    Args,
    Query,
    Option,
    Alconna,
    Arparma,
    AlconnaQuery,
    on_alconna,
    store_true,
)

from zhenxun.services.log import logger
from zhenxun.utils.depends import UserName
from zhenxun.utils.message import MessageUtils
from zhenxun.configs.utils import PluginCdBlock, RegisterConfig, PluginExtraData

from ._data_source import SignManage
from .utils import clear_sign_data_pic
from .goods_register import driver  # noqa: F401

__plugin_meta__ = PluginMetadata(
    name="签到",
    description="每日签到，证明你在这里",
    usage="""
    每日签到
    会影响色图概率和开箱次数，以及签到的随机道具获取
    指令:
        签到
        我的签到
        好感度排行 ?[num=10]
        好感度总排行 ?[num=10]
    * 签到时有 3% 概率 * 2 *
    """.strip(),
    extra=PluginExtraData(
        author="HibiKier",
        version="0.1",
        configs=[
            RegisterConfig(
                module="send_setu",
                key="INITIAL_SETU_PROBABILITY",
                value=0.7,
                help="初始色图概率，总概率 = 初始色图概率 + 好感度",
                default_value=0.7,
                type=float,
            ),
            RegisterConfig(
                key="MAX_SIGN_GOLD",
                value=200,
                help="签到好感度加成额外获得的最大金币数",
                default_value=200,
                type=int,
            ),
            RegisterConfig(
                key="SIGN_CARD1_PROB",
                value=0.2,
                help="签到好感度双倍加持卡Ⅰ掉落概率",
                default_value=0.2,
                type=float,
            ),
            RegisterConfig(
                key="SIGN_CARD2_PROB",
                value=0.09,
                help="签到好感度双倍加持卡Ⅲ掉落概率",
                default_value=0.09,
                type=float,
            ),
            RegisterConfig(
                key="SIGN_CARD3_PROB",
                value=0.05,
                help="签到好感度双倍加持卡Ⅲ掉落概率",
                default_value=0.05,
                type=float,
            ),
            RegisterConfig(
                key="IMAGE_STYLE",
                value="zhenxun",
                help="签到图片样式, [normal, zhenxun]",
                default_value="zhenxun",
            ),
        ],
        limits=[PluginCdBlock()],
    ).dict(),
)


_sign_matcher = on_alconna(
    Alconna(
        "签到",
        Option("--my", action=store_true, help_text="我的签到"),
        Option(
            "-l|--list",
            Args["num?", int],
            help_text="好感度排行",
        ),
        Option("-g|--global", action=store_true, help_text="全局排行"),
    ),
    priority=5,
    block=True,
)

_sign_matcher.shortcut(
    "我的签到",
    command="签到",
    arguments=["--my"],
    prefix=True,
)

_sign_matcher.shortcut(
    "好感度排行",
    command="签到",
    arguments=["--list"],
    prefix=True,
)

_sign_matcher.shortcut(
    "好感度总排行",
    command="签到",
    arguments=["--global", "--list"],
    prefix=True,
)


@_sign_matcher.assign("$main")
async def _(session: EventSession, arparma: Arparma, nickname: str = UserName()):
    if session.id1:
        if path := await SignManage.sign(session, nickname):
            logger.info("签到成功", arparma.header_result, session=session)
            await MessageUtils.build_message(path).finish()
    return MessageUtils.build_message("用户id为空...").send()


@_sign_matcher.assign("my")
async def _(session: EventSession, arparma: Arparma, nickname: str = UserName()):
    if session.id1:
        if image := await SignManage.sign(session, nickname, True):
            logger.info("查看我的签到", arparma.header_result, session=session)
            await MessageUtils.build_message(image).finish()
    return MessageUtils.build_message("用户id为空...").send()


@_sign_matcher.assign("list")
async def _(
    session: EventSession, arparma: Arparma, num: Query[int] = AlconnaQuery("num", 10)
):
    if num.result > 50:
        await MessageUtils.build_message("排行榜人数不能超过50哦...").finish()
    gid = session.id3 or session.id2
    if not arparma.find("global") and not gid:
        await MessageUtils.build_message(
            "私聊中无法查看 '好感度排行'，请发送 '好感度总排行'"
        ).finish()
    if session.id1:
        if arparma.find("global"):
            gid = None
        if image := await SignManage.rank(session.id1, num.result, gid):
            logger.info("查看签到排行", arparma.header_result, session=session)
            await MessageUtils.build_message(image).finish()
    return MessageUtils.build_message("用户id为空...").send()


@scheduler.scheduled_job(
    "interval",
    hours=1,
)
async def _():
    try:
        clear_sign_data_pic()
        logger.info("清理日常签到图片数据数据完成...", "签到")
    except Exception as e:
        logger.error("清理日常签到图片数据数据失败...", e=e)