mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-06 03:46:10 +08:00
67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import logging
|
|
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
import logging
|
|
import nonebot
|
|
from nonebot import get_driver, logger
|
|
from nonebot.log import LoguruHandler
|
|
|
|
from .config import Config
|
|
from .platform import platform_manager
|
|
from .send import do_send_msgs
|
|
from .send import send_msgs
|
|
from .types import UserSubInfo
|
|
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
@get_driver().on_startup
|
|
async def _start():
|
|
scheduler.configure({"apscheduler.timezone": "Asia/Shanghai"})
|
|
scheduler.start()
|
|
|
|
# get_driver().on_startup(_start)
|
|
|
|
async def fetch_and_send(target_type: str):
|
|
config = Config()
|
|
target = config.get_next_target(target_type)
|
|
if not target:
|
|
return
|
|
logger.debug('try to fecth new posts from {}, target: {}'.format(target_type, target))
|
|
send_user_list = config.target_user_cache[target_type][target]
|
|
send_userinfo_list = list(map(
|
|
lambda user: UserSubInfo(
|
|
user,
|
|
lambda target: config.get_sub_category(target_type, target, user.user_type, user.user),
|
|
lambda target: config.get_sub_tags(target_type, target, user.user_type, user.user)
|
|
), send_user_list))
|
|
bot_list = list(nonebot.get_bots().values())
|
|
bot = bot_list[0] if bot_list else None
|
|
to_send = await platform_manager[target_type].fetch_new_post(target, send_userinfo_list)
|
|
for user, send_list in to_send:
|
|
for send_post in send_list:
|
|
logger.info('send to {}: {}'.format(user, send_post))
|
|
if not bot:
|
|
logger.warning('no bot connected')
|
|
else:
|
|
send_msgs(bot, user.user, user.user_type, await send_post.generate_messages())
|
|
|
|
for platform_name, platform in platform_manager.items():
|
|
if platform.schedule_type in ['cron', 'interval', 'date']:
|
|
logger.info(f'start scheduler for {platform_name} with {platform.schedule_type} {platform.schedule_kw}')
|
|
scheduler.add_job(
|
|
fetch_and_send, platform.schedule_type, **platform.schedule_kw,
|
|
args=(platform_name,))
|
|
|
|
scheduler.add_job(do_send_msgs, 'interval', seconds=0.3, coalesce=True)
|
|
|
|
class SchedulerLogFilter(logging.Filter):
|
|
|
|
def filter(self, record: logging.LogRecord) -> bool:
|
|
return not (record.name == "apscheduler" and 'skipped: maximum number of running instances reached' in record.getMessage())
|
|
|
|
aps_logger = logging.getLogger("apscheduler")
|
|
aps_logger.setLevel(30)
|
|
aps_logger.addFilter(SchedulerLogFilter())
|
|
aps_logger.handlers.clear()
|
|
aps_logger.addHandler(LoguruHandler())
|