mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-11-26 02:11:00 +00:00
65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
|
import asyncio
|
||
|
|
import datetime
|
||
|
|
|
||
|
|
from telegram import Update
|
||
|
|
from telegram.ext import CallbackContext
|
||
|
|
|
||
|
|
from core.plugin import handler, Plugin, job
|
||
|
|
from core.search.services import SearchServices
|
||
|
|
from utils.decorators.admins import bot_admins_rights_check
|
||
|
|
from utils.decorators.restricts import restricts
|
||
|
|
from utils.log import logger
|
||
|
|
|
||
|
|
__all__ = []
|
||
|
|
|
||
|
|
|
||
|
|
class SearchPlugin(Plugin):
|
||
|
|
def __init__(self, search: SearchServices = None):
|
||
|
|
self.search = search
|
||
|
|
self._lock = asyncio.Lock()
|
||
|
|
|
||
|
|
async def __async_init__(self):
|
||
|
|
async def load_data():
|
||
|
|
logger.info("Search 插件模块正在加载搜索条目")
|
||
|
|
async with self._lock:
|
||
|
|
await self.search.load_data()
|
||
|
|
logger.success("Search 插件加载模块搜索条目成功")
|
||
|
|
|
||
|
|
asyncio.create_task(load_data())
|
||
|
|
|
||
|
|
@job.run_repeating(interval=datetime.timedelta(hours=1), name="SaveEntryJob")
|
||
|
|
async def save_entry_job(self, _: CallbackContext):
|
||
|
|
if self._lock.locked():
|
||
|
|
logger.warning("条目数据正在保存 跳过本次定时任务")
|
||
|
|
else:
|
||
|
|
async with self._lock:
|
||
|
|
logger.info("条目数据正在自动保存")
|
||
|
|
await self.search.save_entry()
|
||
|
|
logger.success("条目数据自动保存成功")
|
||
|
|
|
||
|
|
@handler.command("save_entry", block=False)
|
||
|
|
@bot_admins_rights_check
|
||
|
|
@restricts()
|
||
|
|
async def save_entry(self, update: Update, _: CallbackContext):
|
||
|
|
user = update.effective_user
|
||
|
|
message = update.effective_message
|
||
|
|
logger.info("用户 %s[%s] 保存条目数据命令请求", user.full_name, user.id)
|
||
|
|
if self._lock.locked():
|
||
|
|
await message.reply_text("条目数据正在保存 请稍后重试")
|
||
|
|
else:
|
||
|
|
async with self._lock:
|
||
|
|
reply_text = await message.reply_text("正在保存数据")
|
||
|
|
await self.search.save_entry()
|
||
|
|
await reply_text.edit_text("数据保存成功")
|
||
|
|
|
||
|
|
@handler.command("remove_all_entry", block=False)
|
||
|
|
@bot_admins_rights_check
|
||
|
|
@restricts()
|
||
|
|
async def remove_all_entry(self, update: Update, _: CallbackContext):
|
||
|
|
user = update.effective_user
|
||
|
|
message = update.effective_message
|
||
|
|
logger.info("用户 %s[%s] 删除全部条目数据命令请求", user.full_name, user.id)
|
||
|
|
reply_text = await message.reply_text("正在删除全部条目数据")
|
||
|
|
await self.search.remove_all_entry()
|
||
|
|
await reply_text.edit_text("删除全部条目数据成功")
|