56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from __future__ import annotations
|
||
|
||
from typing import Optional, Sequence
|
||
|
||
from Plugins.WPSAPI import GuideEntry
|
||
|
||
from .WPSRedPacketBase import WPSRedPacketBase
|
||
|
||
|
||
class WPSRedPacketClaim(WPSRedPacketBase):
|
||
"""抢红包入口"""
|
||
|
||
def get_guide_subtitle(self) -> str:
|
||
return "通过红包ID领取积分奖励"
|
||
|
||
def collect_command_entries(self):
|
||
return (
|
||
GuideEntry(
|
||
title="抢红包",
|
||
identifier="抢红包 <红包ID> [口令]",
|
||
description="输入红包ID后领取剩余份额，口令红包需附带口令。",
|
||
),
|
||
)
|
||
|
||
def wake_up(self) -> None:
|
||
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()
|
||
tokens = [token.strip() for token in payload.split() if token.strip()]
|
||
if not tokens:
|
||
return await self.send_markdown_message(
|
||
self.format_error("指令格式：`抢红包 <红包ID> [口令]`"),
|
||
chat_id,
|
||
user_id,
|
||
)
|
||
|
||
packet_id = tokens[0]
|
||
extra_tokens: Sequence[str] = tokens[1:]
|
||
username = self.config_api().get_user_name(user_id)
|
||
service = self.red_packet_service()
|
||
success, message_text, _ = service.claim_packet(
|
||
chat_id,
|
||
packet_id,
|
||
user_id,
|
||
username,
|
||
extra_tokens,
|
||
)
|
||
if success:
|
||
return await self.send_markdown_message(message_text, chat_id, user_id)
|
||
return await self.send_markdown_message(message_text, chat_id, user_id)
|
||
|
||
|
||
__all__ = ["WPSRedPacketClaim"]
|
||
|