mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
218 lines
7.6 KiB
Python
218 lines
7.6 KiB
Python
import random
|
||
from datetime import datetime, timedelta
|
||
|
||
import nonebot
|
||
import pytz
|
||
from apscheduler.jobstores.base import ConflictingIdError, JobLookupError
|
||
from nonebot import Driver
|
||
from nonebot.adapters.onebot.v11 import ActionFailed
|
||
from nonebot.plugin import require
|
||
|
||
from configs.config import Config
|
||
from models.group_member_info import GroupInfoUser
|
||
from services.log import logger
|
||
from utils.message_builder import at
|
||
from utils.utils import get_bot, scheduler
|
||
|
||
from .._models import Genshin
|
||
|
||
driver: Driver = nonebot.get_driver()
|
||
|
||
|
||
require("query_memo")
|
||
|
||
from ..query_memo import get_memo
|
||
|
||
global_map = {}
|
||
|
||
|
||
class UserManager:
|
||
def __init__(self, max_error_count: int = 3):
|
||
self._data = []
|
||
self._overflow_data = []
|
||
self._error_count = {}
|
||
self.max_error_count = max_error_count
|
||
|
||
def append(self, o: str):
|
||
if o not in self._data:
|
||
self._data.append(o)
|
||
|
||
def remove(self, o: str):
|
||
if o in self._data:
|
||
self._data.remove(o)
|
||
|
||
def exists(self, o: str):
|
||
return o in self._data
|
||
|
||
def add_error_count(self, uid: str):
|
||
if uid in self._error_count.keys():
|
||
self._error_count[uid] += 1
|
||
else:
|
||
self._error_count[uid] = 1
|
||
|
||
def check(self, uid: str) -> bool:
|
||
if uid in self._error_count.keys():
|
||
return self._error_count[uid] == self.max_error_count
|
||
return False
|
||
|
||
def remove_error_count(self, uid):
|
||
if uid in self._error_count.keys():
|
||
del self._error_count[uid]
|
||
|
||
def add_overflow(self, uid: str):
|
||
if uid not in self._overflow_data:
|
||
self._overflow_data.append(uid)
|
||
|
||
def remove_overflow(self, uid: str):
|
||
if uid in self._overflow_data:
|
||
self._overflow_data.remove(uid)
|
||
|
||
def is_overflow(self, uid: str) -> bool:
|
||
return uid in self._overflow_data
|
||
|
||
|
||
user_manager = UserManager()
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
"""
|
||
启动时分配定时任务
|
||
"""
|
||
g_list = await Genshin.filter(resin_remind=True).all()
|
||
update_list = []
|
||
date = datetime.now(pytz.timezone("Asia/Shanghai")) + timedelta(seconds=30)
|
||
for u in g_list:
|
||
if u.resin_remind:
|
||
if u.resin_recovery_time:
|
||
if u.resin_recovery_time and u.resin_recovery_time > datetime.now(
|
||
pytz.timezone("Asia/Shanghai")
|
||
):
|
||
add_job(u.user_id, u.uid)
|
||
logger.info(
|
||
f"genshin_resin_remind add_job：USER：{u.user_id} UID：{u.uid}启动原神树脂提醒 "
|
||
)
|
||
else:
|
||
u.resin_recovery_time = None # type: ignore
|
||
update_list.append(u)
|
||
add_job(u.user_id, u.uid)
|
||
logger.info(
|
||
f"genshin_resin_remind add_job CHECK：USER：{u.user_id} UID：{u.uid}启动原神树脂提醒 "
|
||
)
|
||
else:
|
||
add_job(u.user_id, u.uid)
|
||
logger.info(
|
||
f"genshin_resin_remind add_job CHECK：USER：{u.user_id} UID：{u.uid}启动原神树脂提醒 "
|
||
)
|
||
if update_list:
|
||
await Genshin.bulk_update(update_list, ["resin_recovery_time"])
|
||
|
||
|
||
def add_job(user_id: str, uid: int):
|
||
# 移除
|
||
try:
|
||
scheduler.remove_job(f"genshin_resin_remind_{uid}_{user_id}")
|
||
except JobLookupError:
|
||
pass
|
||
date = datetime.now(pytz.timezone("Asia/Shanghai")) + timedelta(seconds=30)
|
||
try:
|
||
scheduler.add_job(
|
||
_remind,
|
||
"date",
|
||
run_date=date.replace(microsecond=0),
|
||
id=f"genshin_resin_remind_{uid}_{user_id}",
|
||
args=[user_id, uid],
|
||
)
|
||
except ConflictingIdError:
|
||
pass
|
||
|
||
|
||
async def _remind(user_id: int, uid: str):
|
||
user = await Genshin.get_or_none(user_id=str(user_id), uid=int(uid))
|
||
uid = str(uid)
|
||
if uid[0] in ["1", "2"]:
|
||
server_id = "cn_gf01"
|
||
elif uid[0] == "5":
|
||
server_id = "cn_qd01"
|
||
else:
|
||
return
|
||
data, code = await get_memo(uid, server_id)
|
||
now = datetime.now(pytz.timezone("Asia/Shanghai"))
|
||
next_time = None
|
||
if code == 200:
|
||
current_resin = int(data["current_resin"]) # 当前树脂
|
||
max_resin = int(data["max_resin"]) # 最大树脂
|
||
msg = f"你的已经存了 {current_resin} 个树脂了！不要忘记刷掉！"
|
||
# resin_recovery_time = data["resin_recovery_time"] # 树脂全部回复时间
|
||
if current_resin < max_resin:
|
||
user_manager.remove(uid)
|
||
user_manager.remove_overflow(uid)
|
||
if current_resin < max_resin - 40:
|
||
next_time = now + timedelta(minutes=(max_resin - 40 - current_resin) * 8)
|
||
elif max_resin - 40 <= current_resin < max_resin - 20:
|
||
next_time = now + timedelta(minutes=(max_resin - 20 - current_resin) * 8)
|
||
elif max_resin - 20 <= current_resin < max_resin:
|
||
next_time = now + timedelta(minutes=(max_resin - current_resin) * 8)
|
||
elif current_resin == max_resin:
|
||
custom_overflow_resin = Config.get_config(
|
||
"resin_remind", "CUSTOM_RESIN_OVERFLOW_REMIND"
|
||
)
|
||
if user_manager.is_overflow(uid) and custom_overflow_resin:
|
||
next_time = now + timedelta(minutes=custom_overflow_resin * 8)
|
||
user_manager.add_overflow(uid)
|
||
user_manager.remove(uid)
|
||
msg = f"你的树脂都溢出 {custom_overflow_resin} 个了！浪费可耻！"
|
||
else:
|
||
next_time = now + timedelta(minutes=40 * 8 + random.randint(5, 50))
|
||
|
||
if not user_manager.exists(uid) and current_resin >= max_resin - 40:
|
||
if current_resin == max_resin:
|
||
user_manager.append(uid)
|
||
bot = get_bot()
|
||
if bot:
|
||
if user_id in [x["user_id"] for x in await bot.get_friend_list()]:
|
||
await bot.send_private_msg(
|
||
user_id=user_id,
|
||
message=msg,
|
||
)
|
||
else:
|
||
if user:
|
||
group_id = user.bind_group
|
||
if not group_id:
|
||
if group_list := await GroupInfoUser.get_user_all_group(
|
||
user_id
|
||
):
|
||
group_id = group_list[0]
|
||
try:
|
||
await bot.send_group_msg(
|
||
group_id=group_id, message=at(user_id) + msg
|
||
)
|
||
except ActionFailed as e:
|
||
logger.error(f"树脂提醒推送发生错误 {type(e)}：{e}")
|
||
|
||
if not next_time:
|
||
if user_manager.check(uid) and Config.get_config(
|
||
"resin_remind", "AUTO_CLOSE_QUERY_FAIL_RESIN_REMIND"
|
||
):
|
||
if user:
|
||
user.resin_remind = False
|
||
user.resin_recovery_time = None
|
||
await user.save(update_fields=["resin_recovery_time", "resin_remind"])
|
||
next_time = now + timedelta(minutes=(20 + random.randint(5, 20)) * 8)
|
||
user_manager.add_error_count(uid)
|
||
else:
|
||
user_manager.remove_error_count(uid)
|
||
if user:
|
||
user.resin_recovery_time = next_time
|
||
await user.save(update_fields=["resin_recovery_time", "resin_remind"])
|
||
scheduler.add_job(
|
||
_remind,
|
||
"date",
|
||
run_date=next_time,
|
||
id=f"genshin_resin_remind_{uid}_{user_id}",
|
||
args=[user_id, uid],
|
||
)
|
||
logger.info(
|
||
f"genshin_resin_remind add_job：USER：{user_id} UID：{uid} " f"{next_time} 原神树脂提醒"
|
||
)
|