mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
188 lines
7.9 KiB
Python
188 lines
7.9 KiB
Python
import asyncio
|
|
import contextlib
|
|
import time
|
|
from asyncio import sleep
|
|
from typing import List, Tuple
|
|
|
|
from simnet import Region
|
|
from telegram import Update, Message
|
|
from telegram.error import BadRequest, Forbidden
|
|
from telegram.ext import CallbackContext
|
|
from telegram.ext import filters
|
|
|
|
from core.plugin import Plugin, handler
|
|
from gram_core.basemodel import RegionEnum
|
|
from gram_core.services.cookies import CookiesService
|
|
from gram_core.services.cookies.models import CookiesStatusEnum
|
|
from gram_core.services.users.services import UserAdminService
|
|
from plugins.starrail.redeem.runner import RedeemRunner, RedeemResult, RedeemQueueFull
|
|
from plugins.tools.genshin import GenshinHelper
|
|
from utils.log import logger
|
|
|
|
|
|
REDEEM_TEXT = """#### 兑换结果 ####
|
|
时间：{} (UTC+8)
|
|
UID: {}
|
|
兑换码：{}
|
|
兑换结果：{}"""
|
|
REDEEM_ALL_TEXT = """#### 批量兑换 ####
|
|
|
|
兑换码：{}
|
|
正在兑换中，请稍等
|
|
{} / {}"""
|
|
REDEEM_ALL_FAIL_TEXT = """#### 批量兑换 ####
|
|
|
|
兑换码：{}
|
|
兑换成功：{}
|
|
兑换失败：{}"""
|
|
|
|
|
|
class Redeem(Plugin):
|
|
"""兑换码兑换"""
|
|
|
|
def __init__(
|
|
self,
|
|
genshin_helper: GenshinHelper,
|
|
user_admin_service: UserAdminService,
|
|
cookies_service: CookiesService,
|
|
):
|
|
self.genshin_helper = genshin_helper
|
|
self.user_admin_service = user_admin_service
|
|
self.max_code_in_pri_message = 5
|
|
self.max_code_in_pub_message = 3
|
|
self.redeem_runner = RedeemRunner(genshin_helper)
|
|
self.cookies_service = cookies_service
|
|
|
|
async def _callback(self, data: "RedeemResult") -> None:
|
|
code = data.code
|
|
uid = data.uid if data.uid else "未知"
|
|
msg = data.error if data.error else "成功"
|
|
today = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
|
|
text = REDEEM_TEXT.format(today, uid, code, msg)
|
|
reply_message = None
|
|
try:
|
|
reply_message = await data.message.edit_text(text)
|
|
except BadRequest:
|
|
try:
|
|
reply_message = await data.message.reply_text(text)
|
|
except BadRequest:
|
|
pass
|
|
if reply_message and filters.ChatType.GROUPS.filter(reply_message):
|
|
self.add_delete_message_job(reply_message)
|
|
|
|
async def redeem_one_code(self, update: Update, user_id: int, uid: int, code: str, chinese: bool):
|
|
if not code:
|
|
return
|
|
message = update.effective_message
|
|
reply_message = await message.reply_text("正在兑换中，请稍等")
|
|
|
|
task_data = RedeemResult(user_id=user_id, code=code, uid=uid, message=reply_message)
|
|
if chinese:
|
|
task_data.error = "此服务器暂不支持进行兑换哦~"
|
|
await self._callback(task_data)
|
|
return
|
|
priority = 1 if await self.user_admin_service.is_admin(user_id) else 2
|
|
try:
|
|
await self.redeem_runner.run(task_data, self._callback, priority)
|
|
except RedeemQueueFull:
|
|
await reply_message.edit_text("兑换队列已满，请稍后再试")
|
|
if filters.ChatType.GROUPS.filter(reply_message):
|
|
self.add_delete_message_job(reply_message)
|
|
|
|
async def redeem_codes(self, update: Update, user_id: int, codes: List[str]):
|
|
async with self.genshin_helper.genshin(user_id) as client:
|
|
chinese = client.region == Region.CHINESE
|
|
uid = client.player_id
|
|
tasks = []
|
|
for code in codes:
|
|
tasks.append(self.redeem_one_code(update, user_id, uid, code, chinese))
|
|
await asyncio.gather(*tasks)
|
|
|
|
@handler.command(command="redeem", cookie=True, block=False)
|
|
@handler.message(filters=filters.Regex("^兑换码兑换(.*)"), cookie=True, block=False)
|
|
async def command_start(self, update: Update, context: CallbackContext) -> None:
|
|
user_id = await self.get_real_user_id(update)
|
|
message = update.effective_message
|
|
limit = self.max_code_in_pri_message
|
|
if filters.ChatType.GROUPS.filter(message):
|
|
self.add_delete_message_job(message)
|
|
limit = self.max_code_in_pub_message
|
|
codes = [i for i in self.get_args(context) if i][:limit]
|
|
self.log_user(update, logger.info, "兑换码兑换命令请求 codes[%s]", codes)
|
|
if not codes:
|
|
return
|
|
await self.redeem_codes(update, user_id, codes)
|
|
|
|
@handler.command(command="start", filters=filters.Regex(r" redeem_(.*)"), block=False)
|
|
async def start_redeem(self, update: Update, context: CallbackContext) -> None:
|
|
user = update.effective_user
|
|
args = self.get_args(context)
|
|
codes = [i for i in args[0].split("_")[1:] if i][: self.max_code_in_pri_message]
|
|
logger.info("用户 %s[%s] 通过start命令 进入兑换码兑换流程 codes[%s]", user.full_name, user.id, codes)
|
|
await self.redeem_codes(update, user.id, codes)
|
|
|
|
async def _job_callback(self, data: "RedeemResult") -> None:
|
|
if data.error:
|
|
logger.warning("执行自动兑换兑换码时发生错误 user_id[%s] message[%s]", data.user_id, data.error)
|
|
data.count[1] += 1
|
|
return
|
|
data.count[0] += 1
|
|
user_id = data.user_id
|
|
code = data.code
|
|
uid = data.uid if data.uid else "未知"
|
|
msg = "成功"
|
|
today = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
|
|
text = REDEEM_TEXT.format(today, uid, code, msg)
|
|
try:
|
|
await self.application.bot.send_message(user_id, text)
|
|
except BadRequest as exc:
|
|
logger.warning("执行自动兑换兑换码时发生错误 user_id[%s] Message[%s]", user_id, exc.message)
|
|
except Forbidden as exc:
|
|
logger.warning("执行自动兑换兑换码时发生错误 user_id[%s] message[%s]", user_id, exc.message)
|
|
except Exception as exc:
|
|
logger.warning("执行自动兑换兑换码时发生错误 user_id[%s]", user_id, exc_info=exc)
|
|
|
|
async def job_redeem_one_code(self, user_id: int, code: str, count: List[int]):
|
|
task_data = RedeemResult(user_id=user_id, code=code, count=count)
|
|
priority = 1 if await self.user_admin_service.is_admin(user_id) else 2
|
|
try:
|
|
await self.redeem_runner.run(task_data, self._job_callback, priority, True)
|
|
except RedeemQueueFull:
|
|
await sleep(5)
|
|
await self.job_redeem_one_code(user_id, code, count)
|
|
|
|
async def do_redeem_job(self, message: "Message", code: str) -> Tuple[int, int]:
|
|
count = [0, 0]
|
|
task_list = await self.cookies_service.get_all(
|
|
region=RegionEnum.HOYOLAB, status=CookiesStatusEnum.STATUS_SUCCESS
|
|
)
|
|
task_len = len(task_list)
|
|
for idx, task_db in enumerate(task_list):
|
|
user_id = task_db.user_id
|
|
try:
|
|
await self.job_redeem_one_code(user_id, code, count)
|
|
except Exception as exc:
|
|
logger.warning("执行自动兑换兑换码时发生错误 user_id[%s]", user_id, exc_info=exc)
|
|
if idx % 10 == 0:
|
|
text = REDEEM_ALL_TEXT.format(code, idx, task_len)
|
|
with contextlib.suppress(Exception):
|
|
await message.edit_text(text)
|
|
return count[0], count[1]
|
|
|
|
@handler.command(command="redeem_all", admin=True, block=False)
|
|
async def redeem_all_command_start(self, update: Update, context: CallbackContext) -> None:
|
|
message = update.effective_message
|
|
codes = [i for i in self.get_args(context) if i]
|
|
self.log_user(update, logger.info, "兑换码批量兑换命令请求 codes[%s]", codes)
|
|
if not codes:
|
|
await message.reply_text("请输入兑换码")
|
|
return
|
|
code = codes[0]
|
|
reply = await message.reply_text("开始运行批量兑换任务，请等待...")
|
|
success, failed = await self.do_redeem_job(reply, code)
|
|
text = REDEEM_ALL_FAIL_TEXT.format(code, success, failed)
|
|
await message.reply_text(text)
|
|
self.add_delete_message_job(reply, delay=1)
|