moyu.py

from graia.ariadne.app import Ariadne
from graia.ariadne.message.chain import MessageChain
from graia.saya import Saya, Channel
from graiax.shortcut.saya import crontab

from typing import Optional
from datetime import date, timedelta
from itertools import takewhile, dropwhile

saya = Saya.current()
channel = Channel.current()

channel.name("moyu")
channel.description("每天提醒你距离节假日/周末还有多久")
channel.author("I_love_study")

# 来自《国务院办公厅关于2023年部分节假日安排的通知》
holiday = {
    "元旦": [date(2022, 12, 31), date(2023, 1, 2)],
    "春节": [date(2023, 1, 21), date(2023, 1, 27)],
    "清明节": [date(2023, 4, 5), date(2023, 4, 5)],
    "劳动节": [date(2023, 4, 29), date(2023, 5, 3)],
    "端午节": [date(2023, 6, 22), date(2023, 6, 24)],
    "中秋节、国庆节": [date(2023, 9, 29), date(2023, 10, 6)]
}

# 周末上班
weekend_but_work = {
    # 春节
    date(2023, 1, 28),
    date(2023, 1, 29),
    # 劳动节
    date(2023, 4, 23),
    date(2023, 5, 6),
    # 端午
    date(2023, 6, 25),
    # 中秋节、国庆节
    date(2023, 10, 7),
    date(2023, 10, 8)
}

def next_weekend() -> Optional[date]:
    now = date.today()

    # 现在就是周末
    if now.isoweekday() in [6, 7]:
        return

    weekend = now + timedelta(days=6 - now.isoweekday())

    while True:
        # 是调休，不能算是周末
        if weekend in weekend_but_work:
            w = weekend.isoweekday()
            weekend += timedelta(days=1 if w == 6 else 6)
            continue

        for _, [f, t] in holiday.items():
            # 这是假期，超越了周末
            if f <= weekend <= t:
                if t.isoweekday() in [6, 7]:
                    t = t + timedelta(days=1)
                weekend = t + timedelta(days=6 - t.isoweekday())
                continue
        break

    return weekend


def in_holiday() -> bool:
    now = date.today()
    return any(f <= now <= t for _, [f, t] in holiday.items())

@crontab("0 8 * * *")
async def moyu(app: Ariadne):
    today = date.today()
    msg = MessageChain(
        f"早上好，摸鱼人！今天是{today.strftime('%Y年%m月%d日')}")

    # 看看今天是不是周末
    if today.isoweekday() in [6, 7]:
        msg += "\n今天是周末，"
        if today in weekend_but_work:
            msg += "但很可惜，是要上班desu"
        elif in_holiday():
            msg += "而且还是放假期间哦"
        else:
            msg += "好好休息一下吧"
    elif in_holiday():
        msg += "现在是假期哦~"

    end_holiday = list(dropwhile(lambda x: x[1][1] < today, holiday.items()))
    if weekend := next_weekend():
        i = len(list(takewhile(lambda x: x[1][1] < today, end_holiday)))
        end_holiday.insert(i, ("周末", [weekend]))

    msg += "\n\n"

    for n, (day, f) in enumerate(end_holiday, 1):
        if n > 3:
            break

        f = f[0]
        msg += f"距离{day}还有{(f - today).days}天\n"

    g = saya.access('group_setting').get_active_group(__name__)

    if 806724946 in g:
        g.remove(806724946)

    for sg in g:
        await app.send_group_message(sg, msg)