zhenxun_bot/plugins/remind/__init__.py

from utils.init_result import image
from utils.utils import scheduler, get_bot
from services.log import logger
from models.group_remind import GroupRemind
from models.group_info import GroupInfo
from models.friend_user import FriendUser
from nonebot.adapters.cqhttp.exception import ActionFailed

__name__ = "早晚安 [Hidden]"


# 早上好
@scheduler.scheduled_job(
    'cron',
    # year=None,
    # month=None,
    # day=None,
    # week=None,
    # day_of_week="mon,tue,wed,thu,fri",
    hour=6,
    minute=1,
    # second=None,
    # start_date=None,
    # end_date=None,
    # timezone=None,
)
async def _():
    try:
        bot = get_bot()
        gl = await bot.get_group_list(self_id=bot.self_id)
        gl = [g['group_id'] for g in gl]
        for g in gl:
            if await GroupRemind.get_status(g, 'zwa'):
                result = image("zao.jpg", "zhenxun")
                try:
                    await bot.send_group_msg(group_id=g,
                                             message="早上好" + result)
                except ActionFailed:
                    logger.warning(f'{g} 群被禁言中，无法发送早安')
    except Exception as e:
        logger.error(f'早晚安错误 e:{e}')


# 睡觉了
@scheduler.scheduled_job(
    'cron',
    hour=23,
    minute=59,
)
async def _():
    try:
        bot = get_bot()
        gl = await bot.get_group_list(self_id=bot.self_id)
        gl = [g['group_id'] for g in gl]
        for g in gl:
            if await GroupRemind.get_status(g, 'zwa'):
                result = image("sleep.jpg", "zhenxun")
                try:
                    await bot.send_group_msg(group_id=g,
                                             message="小真寻要睡觉了，你们也要早点睡呀" + result)
                except ActionFailed:
                    logger.warning(f'{g} 群被禁言中，无法发送晚安')
    except Exception as e:
        logger.error(f'早晚安错误 e:{e}')


# 自动更新群组信息
@scheduler.scheduled_job(
    'cron',
    hour=3,
    minute=1,
)
async def _():
    try:
        bot = get_bot()
        gl = await bot.get_group_list(self_id=bot.self_id)
        gl = [g['group_id'] for g in gl]
        for g in gl:
            group_info = await bot.get_group_info(group_id=g)
            await GroupInfo.add_group_info(group_info['group_id'], group_info['group_name'],
                                           group_info['max_member_count'], group_info['member_count'])
            logger.info(f'自动更新群组 {g} 信息成功')
    except Exception as e:
        logger.error(f'自动更新群组信息错误 e:{e}')


# 自动更新好友信息
@scheduler.scheduled_job(
    'cron',
    hour=3,
    minute=1,
)
async def _():
    try:
        bot = get_bot()
        fl = await bot.get_friend_list(self_id=bot.self_id)
        for f in fl:
            if await FriendUser.add_friend_info(f['user_id'], f['nickname']):
                logger.info(f'自动更新好友 {f["user_id"]} 信息成功')
            else:
                logger.warning(f'自动更新好友 {f["user_id"]} 信息失败')
    except Exception as e:
        logger.error(f'自动更新群组信息错误 e:{e}')

#  一次性任务
# 固定时间触发，仅触发一次：
#
# from datetime import datetime
#
# @nonebot.scheduler.scheduled_job(
#     'date',
#     run_date=datetime(2021, 1, 1, 0, 0),
#     # timezone=None,
# )
# async def _():
#     await bot.send_group_msg(group_id=123456,
#                              message="2021，新年快乐！")

#  定期任务
#  从 start_date 开始到 end_date 结束，根据类似 Cron
#
# 的规则触发任务：
#
# @nonebot.scheduler.scheduled_job(
#     'cron',
#     # year=None,
#     # month=None,
#     # day=None,
#     # week=None,
#     day_of_week="mon,tue,wed,thu,fri",
#     hour=7,
#     # minute=None,
#     # second=None,
#     # start_date=None,
#     # end_date=None,
#     # timezone=None,
# )
# async def _():
#     await bot.send_group_msg(group_id=123456,
#                              message="起床啦！")

#  间隔任务
#
# interval 触发器
#
# 从 start_date 开始，每间隔一段时间触发，到 end_date 结束：
#
# @nonebot.scheduler.scheduled_job(
#     'interval',
#     # weeks=0,
#     # days=0,
#     # hours=0,
#     minutes=5,
#     # seconds=0,
#     # start_date=time.now(),
#     # end_date=None,
# )
# async def _():
#     has_new_item = check_new_item()
#     if has_new_item:
#         await bot.send_group_msg(group_id=123456,
#                                  message="XX有更新啦！")


# 动态的计划任务
# import datetime
#
# from apscheduler.triggers.date import DateTrigger # 一次性触发器
# # from apscheduler.triggers.cron import CronTrigger # 定期触发器
# # from apscheduler.triggers.interval import IntervalTrigger # 间隔触发器
# from nonebot import on_command, scheduler
#
# @on_command('赖床')
# async def _(session: CommandSession):
#     await session.send('我会在5分钟后再喊你')
#
#     # 制作一个“5分钟后”触发器
#     delta = datetime.timedelta(minutes=5)
#     trigger = DateTrigger(
#         run_date=datetime.datetime.now() + delta
#     )
#
#     # 添加任务
#     scheduler.add_job(
#         func=session.send,  # 要添加任务的函数，不要带参数
#         trigger=trigger,  # 触发器
#         args=('不要再赖床啦！',),  # 函数的参数列表，注意：只有一个值时，不能省略末尾的逗号
#         # kwargs=None,
#         misfire_grace_time=60,  # 允许的误差时间，建议不要省略
#         # jobstore='default',  # 任务储存库，在下一小节中说明
#     )