mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
99 lines
4.5 KiB
Python
99 lines
4.5 KiB
Python
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
||
from telegram.ext import CallbackContext, ConversationHandler, filters
|
||
|
||
from model.base import ServiceEnum
|
||
from model.genshinhelper import YuanShen, Genshin
|
||
from plugins.base import BasePlugins
|
||
from service import BaseService
|
||
from service.base import UserInfoData
|
||
|
||
|
||
class SignCommandData:
|
||
user_info: UserInfoData = UserInfoData()
|
||
chat_id: int = 0
|
||
reply_to_message_id: int = 0
|
||
|
||
|
||
class Sign(BasePlugins):
|
||
def __init__(self, service: BaseService):
|
||
super().__init__(service)
|
||
self._sign_y = YuanShen()
|
||
self._sing_g = Genshin()
|
||
|
||
CHECK_SERVER, COMMAND_RESULT = range(10400, 10402)
|
||
|
||
async def _start_sign(self, uid: int, cookies: dict, service: ServiceEnum) -> str:
|
||
if service == ServiceEnum.MIHOYO:
|
||
sign_api = self._sign_y
|
||
else:
|
||
sign_api = self._sing_g
|
||
sign_give = await sign_api.get_sign_give(cookies=cookies)
|
||
if sign_give.error:
|
||
return f"获取签到信息失败，API返回信息为 {sign_give.message}"
|
||
is_sign = await sign_api.is_sign(uid, cookies=cookies)
|
||
if is_sign.error:
|
||
return f"获取签到状态失败，API返回信息为 {is_sign.message}"
|
||
total_sign_day = is_sign.data["total_sign_day"]
|
||
award_name = sign_give.data["awards"][total_sign_day]["name"]
|
||
award_cnt = sign_give.data["awards"][total_sign_day]["cnt"]
|
||
today = is_sign.data["today"]
|
||
if not is_sign.data["is_sign"]:
|
||
sign = await sign_api.sign(uid, cookies=cookies)
|
||
if sign.code == 0:
|
||
result = "OK"
|
||
elif sign.code == -5003:
|
||
result = "今天旅行者已经签到过了~"
|
||
else:
|
||
result = f"签到失败 返回错误代码为 {sign.code}"
|
||
else:
|
||
result = "今天旅行者已经签到过了~"
|
||
message = f"###### {today} ######\n" \
|
||
f"UID: {uid}\n" \
|
||
f"今日奖励: {award_name} × {award_cnt}\n" \
|
||
f"签到结果: {result}"
|
||
return message
|
||
|
||
async def command_start(self, update: Update, context: CallbackContext) -> int:
|
||
user = update.effective_user
|
||
message = update.message
|
||
sign_command_data: SignCommandData = context.chat_data.get("sign_command_data")
|
||
if sign_command_data is None:
|
||
sign_command_data = SignCommandData()
|
||
context.chat_data["sign_command_data"] = sign_command_data
|
||
user_info = await self.service.user_service_db.get_user_info(user.id)
|
||
if user_info.service == ServiceEnum.NULL:
|
||
message = "请选择你要签到的服务器"
|
||
keyboard = [
|
||
[
|
||
InlineKeyboardButton("miHoYo", callback_data="miHoYo"),
|
||
InlineKeyboardButton("HoYoLab", callback_data="HoYoLab")
|
||
]
|
||
]
|
||
sign_command_data.user_info = user_info
|
||
await update.message.reply_text(message, reply_markup=InlineKeyboardMarkup(keyboard))
|
||
sign_command_data.chat_id = update.message.chat_id
|
||
sign_command_data.reply_to_message_id = update.message.message_id
|
||
return self.COMMAND_RESULT
|
||
else:
|
||
sign = await self._start_sign(user_info.mihoyo_game_uid, user_info.mihoyo_cookie, user_info.service)
|
||
reply_message = await message.reply_text(sign)
|
||
if filters.ChatType.GROUPS.filter(update.callback_query.message):
|
||
self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
|
||
return ConversationHandler.END
|
||
|
||
async def command_result(self, update: Update, context: CallbackContext) -> int:
|
||
sign_command_data: SignCommandData = context.chat_data["sign_command_data"]
|
||
user_info = sign_command_data.user_info
|
||
query = update.callback_query
|
||
await query.answer()
|
||
message = "签到失败"
|
||
if query.data == "miHoYo":
|
||
message = await self._start_sign(user_info.mihoyo_game_uid, user_info.mihoyo_cookie, ServiceEnum.MIHOYO)
|
||
if query.data == "HoYoLab":
|
||
message = await self._start_sign(user_info.hoyoverse_game_uid, user_info.hoyoverse_cookie,
|
||
ServiceEnum.HOYOLAB)
|
||
await query.edit_message_text(message)
|
||
if filters.ChatType.GROUPS.filter(update.callback_query.message):
|
||
self._add_delete_message_job(context, query.message.chat_id, query.message.message_id)
|
||
return ConversationHandler.END
|