mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-24 17:19:40 +00:00
543 lines
22 KiB
Python
543 lines
22 KiB
Python
import base64
|
||
from datetime import datetime
|
||
from typing import TYPE_CHECKING, List, Optional, Union
|
||
|
||
from pydantic import BaseModel, validator
|
||
from simnet import Region
|
||
from simnet.errors import BadRequest as SimnetBadRequest, InvalidCookies, TimedOut as SimnetTimedOut
|
||
from sqlalchemy.orm.exc import StaleDataError
|
||
from telegram.constants import ParseMode
|
||
from telegram.error import BadRequest, Forbidden
|
||
|
||
from core.plugin import Plugin
|
||
from core.services.task.models import Task as TaskUser, TaskStatusEnum
|
||
from core.services.task.services import TaskResinServices, TaskRealmServices, TaskExpeditionServices, TaskDailyServices
|
||
from gram_core.plugin.methods.migrate_data import IMigrateData, MigrateDataException
|
||
from plugins.tools.genshin import GenshinHelper, PlayerNotFoundError, CookiesNotFoundError
|
||
from utils.log import logger
|
||
|
||
if TYPE_CHECKING:
|
||
from simnet import GenshinClient
|
||
from simnet.models.genshin.chronicle.notes import Notes, NotesWidget
|
||
from telegram.ext import ContextTypes
|
||
from gram_core.services.task.services import TaskServices
|
||
|
||
|
||
class TaskDataBase(BaseModel):
|
||
noticed: Optional[bool] = False
|
||
|
||
|
||
class ResinData(TaskDataBase):
|
||
notice_num: Optional[int] = 140
|
||
|
||
@validator("notice_num")
|
||
def notice_num_validator(cls, v):
|
||
if v < 60 or v > 200:
|
||
raise ValueError("树脂提醒数值必须在 60 ~ 200 之间")
|
||
return v
|
||
|
||
|
||
class RealmData(TaskDataBase):
|
||
notice_num: Optional[int] = 2000
|
||
|
||
@validator("notice_num")
|
||
def notice_num_validator(cls, v):
|
||
if v < 100 or v > 2400:
|
||
raise ValueError("洞天宝钱提醒数值必须在 100 ~ 2400 之间")
|
||
return v
|
||
|
||
|
||
class ExpeditionData(TaskDataBase):
|
||
pass
|
||
|
||
|
||
class DailyData(TaskDataBase):
|
||
notice_hour: Optional[int] = 22
|
||
|
||
@validator("notice_hour")
|
||
def notice_hour_validator(cls, v):
|
||
if v < 0 or v > 23:
|
||
raise ValueError("每日任务提醒时间必须在 0 ~ 23 之间")
|
||
return v
|
||
|
||
|
||
class WebAppData(BaseModel):
|
||
resin: Optional[ResinData]
|
||
realm: Optional[RealmData]
|
||
expedition: Optional[ExpeditionData]
|
||
daily: Optional[DailyData]
|
||
|
||
|
||
class DailyNoteTaskUser:
|
||
def __init__(
|
||
self,
|
||
user_id: int,
|
||
resin_db: Optional[TaskUser] = None,
|
||
realm_db: Optional[TaskUser] = None,
|
||
expedition_db: Optional[TaskUser] = None,
|
||
daily_db: Optional[TaskUser] = None,
|
||
):
|
||
self.user_id = user_id
|
||
self.resin_db = resin_db
|
||
self.realm_db = realm_db
|
||
self.expedition_db = expedition_db
|
||
self.daily_db = daily_db
|
||
self.resin = ResinData(**self.resin_db.data) if self.resin_db else None
|
||
self.realm = RealmData(**self.realm_db.data) if self.realm_db else None
|
||
self.expedition = ExpeditionData(**self.expedition_db.data) if self.expedition_db else None
|
||
self.daily = DailyData(**self.daily_db.data) if self.daily_db else None
|
||
|
||
@property
|
||
def status(self) -> TaskStatusEnum:
|
||
return max(
|
||
[
|
||
self.resin_db.status if self.resin_db else TaskStatusEnum.STATUS_SUCCESS,
|
||
self.realm_db.status if self.realm_db else TaskStatusEnum.STATUS_SUCCESS,
|
||
self.expedition_db.status if self.expedition_db else TaskStatusEnum.STATUS_SUCCESS,
|
||
self.daily_db.status if self.daily_db else TaskStatusEnum.STATUS_SUCCESS,
|
||
]
|
||
)
|
||
|
||
@status.setter
|
||
def status(self, value: TaskStatusEnum):
|
||
if self.resin_db:
|
||
self.resin_db.status = value
|
||
if self.realm_db:
|
||
self.realm_db.status = value
|
||
if self.expedition_db:
|
||
self.expedition_db.status = value
|
||
if self.daily:
|
||
self.daily_db.status = value
|
||
|
||
@staticmethod
|
||
def js_bool(value: bool) -> str:
|
||
return "true" if value else "false"
|
||
|
||
@staticmethod
|
||
def set_model_noticed(model: TaskDataBase):
|
||
data = model.copy(deep=True)
|
||
data.noticed = True
|
||
return data
|
||
|
||
@property
|
||
def web_config(self) -> str:
|
||
return base64.b64encode(
|
||
(
|
||
WebAppData(
|
||
resin=self.set_model_noticed(self.resin) if self.resin else None,
|
||
realm=self.set_model_noticed(self.realm) if self.realm else None,
|
||
expedition=self.set_model_noticed(self.expedition) if self.expedition else None,
|
||
daily=self.set_model_noticed(self.daily) if self.daily else None,
|
||
).json()
|
||
).encode()
|
||
).decode()
|
||
|
||
def save(self):
|
||
if self.resin_db:
|
||
self.resin_db.data = self.resin.dict()
|
||
if self.realm_db:
|
||
self.realm_db.data = self.realm.dict()
|
||
if self.expedition_db:
|
||
self.expedition_db.data = self.expedition.dict()
|
||
if self.daily_db:
|
||
self.daily_db.data = self.daily.dict()
|
||
|
||
|
||
class DailyNoteSystem(Plugin):
|
||
def __init__(
|
||
self,
|
||
genshin_helper: GenshinHelper,
|
||
resin_service: TaskResinServices,
|
||
realm_service: TaskRealmServices,
|
||
expedition_service: TaskExpeditionServices,
|
||
daily_service: TaskDailyServices,
|
||
):
|
||
self.genshin_helper = genshin_helper
|
||
self.resin_service = resin_service
|
||
self.realm_service = realm_service
|
||
self.expedition_service = expedition_service
|
||
self.daily_service = daily_service
|
||
|
||
async def get_single_task_user(self, user_id: int) -> DailyNoteTaskUser:
|
||
resin_db = await self.resin_service.get_by_user_id(user_id)
|
||
realm_db = await self.realm_service.get_by_user_id(user_id)
|
||
expedition_db = await self.expedition_service.get_by_user_id(user_id)
|
||
daily_db = await self.daily_service.get_by_user_id(user_id)
|
||
return DailyNoteTaskUser(
|
||
user_id=user_id,
|
||
resin_db=resin_db,
|
||
realm_db=realm_db,
|
||
expedition_db=expedition_db,
|
||
daily_db=daily_db,
|
||
)
|
||
|
||
@staticmethod
|
||
def get_resin_notice(user: DailyNoteTaskUser, notes: Union["Notes", "NotesWidget"]) -> str:
|
||
notice = None
|
||
if user.resin_db and notes.max_resin > 0:
|
||
if notes.current_resin >= user.resin.notice_num:
|
||
if not user.resin.noticed:
|
||
notice = (
|
||
f"### 树脂提示 ####\n\n当前树脂为 {notes.current_resin} / {notes.max_resin} ，记得使用哦~\n"
|
||
f"预计全部恢复完成：{notes.resin_recovery_time.strftime('%Y-%m-%d %H:%M')}"
|
||
)
|
||
user.resin.noticed = True
|
||
else:
|
||
user.resin.noticed = False
|
||
return notice
|
||
|
||
@staticmethod
|
||
def get_realm_notice(user: DailyNoteTaskUser, notes: Union["Notes", "NotesWidget"]) -> str:
|
||
notice = None
|
||
if user.realm_db and notes.max_realm_currency > 0:
|
||
if notes.current_realm_currency >= user.realm.notice_num:
|
||
if not user.realm.noticed:
|
||
notice = (
|
||
f"### 洞天宝钱提示 ####\n\n"
|
||
f"当前存储为 {notes.current_realm_currency} / {notes.max_realm_currency} ，记得领取哦~"
|
||
)
|
||
user.realm.noticed = True
|
||
else:
|
||
user.realm.noticed = False
|
||
return notice
|
||
|
||
@staticmethod
|
||
def get_expedition_notice(user: DailyNoteTaskUser, notes: Union["Notes", "NotesWidget"]) -> str:
|
||
notice = None
|
||
if user.expedition_db and len(notes.expeditions) > 0:
|
||
all_finished = all(i.status == "Finished" for i in notes.expeditions)
|
||
if all_finished:
|
||
if not user.expedition.noticed:
|
||
notice = "### 探索派遣提示 ####\n\n所有探索派遣已完成，记得重新派遣哦~"
|
||
user.expedition.noticed = True
|
||
else:
|
||
user.expedition.noticed = False
|
||
return notice
|
||
|
||
@staticmethod
|
||
def get_daily_notice(user: DailyNoteTaskUser, notes: Union["Notes", "NotesWidget"]) -> str:
|
||
notice = None
|
||
now_hour = datetime.now().hour
|
||
if user.daily_db:
|
||
if now_hour == user.daily.notice_hour:
|
||
if (notes.completed_commissions != notes.max_commissions or (not notes.claimed_commission_reward)) and (
|
||
not user.daily.noticed
|
||
):
|
||
notice_ = "已领取奖励" if notes.claimed_commission_reward else "未领取奖励"
|
||
notice = (
|
||
f"### 每日任务提示 ####\n\n"
|
||
f"当前进度为 {notes.completed_commissions} / {notes.max_commissions} ({notice_}) ，记得完成哦~"
|
||
)
|
||
user.daily.noticed = True
|
||
else:
|
||
user.daily.noticed = False
|
||
return notice
|
||
|
||
@staticmethod
|
||
async def start_get_notes(
|
||
client: "GenshinClient",
|
||
user: DailyNoteTaskUser = None,
|
||
) -> List[str]:
|
||
if client.region == Region.OVERSEAS:
|
||
notes = await client.get_genshin_notes()
|
||
else:
|
||
notes = await client.get_genshin_notes_by_stoken()
|
||
if not user:
|
||
return []
|
||
notices = [
|
||
DailyNoteSystem.get_resin_notice(user, notes),
|
||
DailyNoteSystem.get_realm_notice(user, notes),
|
||
DailyNoteSystem.get_expedition_notice(user, notes),
|
||
DailyNoteSystem.get_daily_notice(user, notes),
|
||
]
|
||
user.save()
|
||
return notices
|
||
|
||
async def get_all_task_users(self) -> List[DailyNoteTaskUser]:
|
||
resin_list = await self.resin_service.get_all()
|
||
realm_list = await self.realm_service.get_all()
|
||
expedition_list = await self.expedition_service.get_all()
|
||
daily_list = await self.daily_service.get_all()
|
||
user_list = set()
|
||
for i in resin_list:
|
||
user_list.add(i.user_id)
|
||
for i in realm_list:
|
||
user_list.add(i.user_id)
|
||
for i in expedition_list:
|
||
user_list.add(i.user_id)
|
||
for i in daily_list:
|
||
user_list.add(i.user_id)
|
||
return [
|
||
DailyNoteTaskUser(
|
||
user_id=i,
|
||
resin_db=next((x for x in resin_list if x.user_id == i), None),
|
||
realm_db=next((x for x in realm_list if x.user_id == i), None),
|
||
expedition_db=next((x for x in expedition_list if x.user_id == i), None),
|
||
daily_db=next((x for x in daily_list if x.user_id == i), None),
|
||
)
|
||
for i in user_list
|
||
]
|
||
|
||
async def remove_task_user(self, user: DailyNoteTaskUser):
|
||
if user.resin_db:
|
||
await self.resin_service.remove(user.resin_db)
|
||
if user.realm_db:
|
||
await self.realm_service.remove(user.realm_db)
|
||
if user.expedition_db:
|
||
await self.expedition_service.remove(user.expedition_db)
|
||
if user.daily_db:
|
||
await self.daily_service.remove(user.daily_db)
|
||
|
||
async def update_task_user(self, user: DailyNoteTaskUser):
|
||
if user.resin_db:
|
||
try:
|
||
await self.resin_service.update(user.resin_db)
|
||
except StaleDataError:
|
||
logger.warning("用户 user_id[%s] 自动便签提醒 - 树脂数据过期，跳过更新数据", user.user_id)
|
||
if user.realm_db:
|
||
try:
|
||
await self.realm_service.update(user.realm_db)
|
||
except StaleDataError:
|
||
logger.warning("用户 user_id[%s] 自动便签提醒 - 洞天宝钱数据过期，跳过更新数据", user.user_id)
|
||
if user.expedition_db:
|
||
try:
|
||
await self.expedition_service.update(user.expedition_db)
|
||
except StaleDataError:
|
||
logger.warning("用户 user_id[%s] 自动便签提醒 - 探索派遣数据过期，跳过更新数据", user.user_id)
|
||
if user.daily_db:
|
||
try:
|
||
await self.daily_service.update(user.daily_db)
|
||
except StaleDataError:
|
||
logger.warning("用户 user_id[%s] 自动便签提醒 - 每日任务数据过期，跳过更新数据", user.user_id)
|
||
|
||
@staticmethod
|
||
async def check_need_note(web_config: WebAppData) -> bool:
|
||
need_verify = False
|
||
if web_config.resin and web_config.resin.noticed:
|
||
need_verify = True
|
||
if web_config.realm and web_config.realm.noticed:
|
||
need_verify = True
|
||
if web_config.expedition and web_config.expedition.noticed:
|
||
need_verify = True
|
||
if web_config.daily and web_config.daily.noticed:
|
||
need_verify = True
|
||
return need_verify
|
||
|
||
async def import_web_config_resin(self, user: DailyNoteTaskUser, web_config: WebAppData):
|
||
user_id = user.user_id
|
||
if web_config.resin.noticed:
|
||
if not user.resin_db:
|
||
resin = self.resin_service.create(
|
||
user_id,
|
||
user_id,
|
||
status=TaskStatusEnum.STATUS_SUCCESS,
|
||
data=ResinData(notice_num=web_config.resin.notice_num).dict(),
|
||
)
|
||
await self.resin_service.add(resin)
|
||
else:
|
||
user.resin.notice_num = web_config.resin.notice_num
|
||
user.resin.noticed = False
|
||
user.resin_db.status = TaskStatusEnum.STATUS_SUCCESS
|
||
else:
|
||
if user.resin_db:
|
||
await self.resin_service.remove(user.resin_db)
|
||
user.resin_db = None
|
||
user.resin = None
|
||
|
||
async def import_web_config_realm(self, user: DailyNoteTaskUser, web_config: WebAppData):
|
||
user_id = user.user_id
|
||
if web_config.realm.noticed:
|
||
if not user.realm_db:
|
||
realm = self.realm_service.create(
|
||
user_id,
|
||
user_id,
|
||
status=TaskStatusEnum.STATUS_SUCCESS,
|
||
data=RealmData(notice_num=web_config.realm.notice_num).dict(),
|
||
)
|
||
await self.realm_service.add(realm)
|
||
else:
|
||
user.realm.notice_num = web_config.realm.notice_num
|
||
user.realm.noticed = False
|
||
else:
|
||
if user.realm_db:
|
||
await self.realm_service.remove(user.realm_db)
|
||
user.realm_db = None
|
||
user.realm = None
|
||
|
||
async def import_web_config_expedition(self, user: DailyNoteTaskUser, web_config: WebAppData):
|
||
user_id = user.user_id
|
||
if web_config.expedition.noticed:
|
||
if not user.expedition_db:
|
||
expedition = self.expedition_service.create(
|
||
user_id,
|
||
user_id,
|
||
status=TaskStatusEnum.STATUS_SUCCESS,
|
||
data=ExpeditionData().dict(),
|
||
)
|
||
await self.expedition_service.add(expedition)
|
||
else:
|
||
user.expedition.noticed = False
|
||
user.expedition_db.status = TaskStatusEnum.STATUS_SUCCESS
|
||
else:
|
||
if user.expedition_db:
|
||
await self.expedition_service.remove(user.expedition_db)
|
||
user.expedition_db = None
|
||
user.expedition = None
|
||
|
||
async def import_web_config_daily(self, user: DailyNoteTaskUser, web_config: WebAppData):
|
||
user_id = user.user_id
|
||
if web_config.daily.noticed:
|
||
if not user.daily_db:
|
||
daily = self.daily_service.create(
|
||
user_id,
|
||
user_id,
|
||
status=TaskStatusEnum.STATUS_SUCCESS,
|
||
data=DailyData(notice_hour=web_config.daily.notice_hour).dict(),
|
||
)
|
||
await self.daily_service.add(daily)
|
||
else:
|
||
user.daily.noticed = False
|
||
user.daily_db.status = TaskStatusEnum.STATUS_SUCCESS
|
||
else:
|
||
if user.daily_db:
|
||
await self.daily_service.remove(user.daily_db)
|
||
user.daily_db = None
|
||
user.daily = None
|
||
|
||
async def import_web_config(self, user_id: int, web_config: WebAppData):
|
||
user = await self.get_single_task_user(user_id)
|
||
if web_config.resin:
|
||
await self.import_web_config_resin(user, web_config)
|
||
if web_config.realm:
|
||
await self.import_web_config_realm(user, web_config)
|
||
if web_config.expedition:
|
||
await self.import_web_config_expedition(user, web_config)
|
||
if web_config.daily:
|
||
await self.import_web_config_daily(user, web_config)
|
||
user.save()
|
||
await self.update_task_user(user)
|
||
|
||
async def do_get_notes_job(self, context: "ContextTypes.DEFAULT_TYPE"):
|
||
include_status: List[TaskStatusEnum] = [
|
||
TaskStatusEnum.STATUS_SUCCESS,
|
||
TaskStatusEnum.TIMEOUT_ERROR,
|
||
]
|
||
task_list = await self.get_all_task_users()
|
||
for task_db in task_list:
|
||
if task_db.status not in include_status:
|
||
continue
|
||
user_id = task_db.user_id
|
||
logger.debug("自动便签提醒 - 请求便签信息 user_id[%s]", user_id)
|
||
try:
|
||
async with self.genshin_helper.genshin(user_id) as client:
|
||
text = await self.start_get_notes(client, task_db)
|
||
except InvalidCookies:
|
||
text = "自动便签提醒执行失败，Cookie无效"
|
||
task_db.status = TaskStatusEnum.INVALID_COOKIES
|
||
except SimnetBadRequest as exc:
|
||
text = f"自动便签提醒执行失败，API返回信息为 {str(exc)}"
|
||
task_db.status = TaskStatusEnum.GENSHIN_EXCEPTION
|
||
except SimnetTimedOut:
|
||
logger.info("用户 user_id[%s] 请求便签超时", user_id)
|
||
continue
|
||
except PlayerNotFoundError:
|
||
logger.info("用户 user_id[%s] 玩家不存在 关闭并移除自动便签提醒", user_id)
|
||
await self.remove_task_user(task_db)
|
||
continue
|
||
except CookiesNotFoundError:
|
||
logger.info("用户 user_id[%s] cookie 不存在 关闭并移除自动便签提醒", user_id)
|
||
await self.remove_task_user(task_db)
|
||
continue
|
||
except Exception as exc:
|
||
logger.error("执行自动便签提醒时发生错误 user_id[%s]", user_id, exc_info=exc)
|
||
text = "获取便签失败了呜呜呜 ~ 执行自动便签提醒时发生错误"
|
||
else:
|
||
task_db.status = TaskStatusEnum.STATUS_SUCCESS
|
||
for idx, task_user_db in enumerate(
|
||
[task_db.resin_db, task_db.realm_db, task_db.expedition_db, task_db.daily_db]
|
||
):
|
||
if task_user_db is None:
|
||
continue
|
||
notice_text = text[idx] if isinstance(text, list) else text
|
||
if not notice_text:
|
||
continue
|
||
if task_user_db.chat_id < 0:
|
||
notice_text = (
|
||
f'<a href="tg://user?id={task_user_db.user_id}">'
|
||
f"NOTICE {task_user_db.user_id}</a>\n\n{notice_text}"
|
||
)
|
||
try:
|
||
await context.bot.send_message(task_user_db.chat_id, notice_text, parse_mode=ParseMode.HTML)
|
||
except BadRequest as exc:
|
||
logger.error("执行自动便签提醒时发生错误 user_id[%s] Message[%s]", user_id, exc.message)
|
||
task_user_db.status = TaskStatusEnum.BAD_REQUEST
|
||
except Forbidden as exc:
|
||
logger.error("执行自动便签提醒时发生错误 user_id[%s] message[%s]", user_id, exc.message)
|
||
task_user_db.status = TaskStatusEnum.FORBIDDEN
|
||
except Exception as exc:
|
||
logger.error("执行自动便签提醒时发生错误 user_id[%s]", user_id, exc_info=exc)
|
||
continue
|
||
await self.update_task_user(task_db)
|
||
|
||
async def get_migrate_data(self, old_user_id: int, new_user_id: int, _) -> Optional["TaskMigrate"]:
|
||
return await TaskMigrate.create(
|
||
old_user_id,
|
||
new_user_id,
|
||
self.resin_service,
|
||
)
|
||
|
||
|
||
class TaskMigrate(IMigrateData):
|
||
old_user_id: int
|
||
new_user_id: int
|
||
task_services: "TaskServices"
|
||
need_migrate_tasks: List[TaskUser]
|
||
|
||
async def migrate_data_msg(self) -> str:
|
||
return f"定时任务数据 {len(self.need_migrate_tasks)} 条"
|
||
|
||
async def migrate_data(self) -> bool:
|
||
id_list = []
|
||
for task in self.need_migrate_tasks:
|
||
try:
|
||
await self.task_services.update(task)
|
||
except StaleDataError:
|
||
id_list.append(str(task.id))
|
||
if id_list:
|
||
raise MigrateDataException(f"任务数据迁移失败：id {','.join(id_list)}")
|
||
return True
|
||
|
||
@staticmethod
|
||
async def create_tasks(
|
||
old_user_id: int,
|
||
new_user_id: int,
|
||
task_services: "TaskServices",
|
||
) -> List[TaskUser]:
|
||
need_migrate, _ = await TaskMigrate.filter_sql_data(
|
||
TaskUser,
|
||
task_services.get_all_by_user_id,
|
||
old_user_id,
|
||
new_user_id,
|
||
(TaskUser.user_id, TaskUser.type),
|
||
)
|
||
for i in need_migrate:
|
||
i.user_id = new_user_id
|
||
return need_migrate
|
||
|
||
@classmethod
|
||
async def create(
|
||
cls,
|
||
old_user_id: int,
|
||
new_user_id: int,
|
||
task_services: "TaskServices",
|
||
) -> Optional["TaskMigrate"]:
|
||
need_migrate_tasks = await cls.create_tasks(old_user_id, new_user_id, task_services)
|
||
if not need_migrate_tasks:
|
||
return None
|
||
self = cls()
|
||
self.old_user_id = old_user_id
|
||
self.new_user_id = new_user_id
|
||
self.task_services = task_services
|
||
self.need_migrate_tasks = need_migrate_tasks
|
||
return self
|