qqbot/special_date.py

import datetime
import holidays
import json
import random

def get_v50(file_path='v50.json'):
    """
    随机获取json文件中的一条文案
    """
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            data = json.load(file)
            text_list = data.get('data', [])
            if not text_list:
                return "错误：文件中没有找到文案数据"
            return random.choice(text_list)
    except FileNotFoundError:
        return "错误：未找到指定的json文件"
    except json.JSONDecodeError as e:
        return f"错误：json文件格式不正确{e}"
    except Exception as e:
        return f"错误：发生未知错误 - {str(e)}"

def get_holidays():
    """
    获取多个地区的节日
    目前只写了US和CN的节日
    """
    today = datetime.date.today()
    year = today.year
    all_holidays = []

    us_holidays = holidays.country_holidays('US', years=today.year)
    china_holidays = holidays.country_holidays('CN', years=today.year)

    all_holidays.append(china_holidays)
    all_holidays.append(us_holidays)

    return all_holidays

def get_week():
    """
    获取今日的日期
    """
    today = datetime.date.today()
    # 获取星期几（0-6，0表示星期一，6表示星期日）
    weekday_num = today.weekday()
    # 定义星期几的列表
    weekdays = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期日"]
    return weekdays[weekday_num]