为了数清还有几天到周末,我用python绘制了日历

发布于:2024-11-04 ⋅ 阅读:(77) ⋅ 点赞:(0)

日历的秘密

昨天,在看小侄子写作业的时候,发现了一个秘密:他在“演算纸”(计算数学题用的草纸)上画了非常多的日历。对此我感到了非常的困惑,“这是做什么的?”

后来,经过了我不懈的追问,小侄子终于说出了实情,“在学校太无聊,所以经常画日历,看看还有几天放假。”

哦!原来是这样!虽然这种行为不值得提倡,但是,小侄子想要放假的心情是可以理解的。所以,我决定帮帮他,研究一下怎么用python绘制出日历。

计算节假日

节假日

众所周知,一周有7天,其中周一到周五是工作日,周六周日是双休日。

每年元旦休息1天,春节休息3天,中秋,端午,清明节各休息1天,劳动节休息1天,国庆节休息3天,总共休息11天。

看起来,只要掌握了节日的规律,就可以知道假期了。但是,由于“调休”,周六周日也不总是休息日,其他工作日也可能变成休息日,所以,不能简单的这样做。

那么,要如何才能够计算出哪些日子是假日呢?答案是,必须要等到每年公布节假日才行。然后,根据公布的结果,进行手动标注。这听起来很麻烦,有没有什么更好的办法可以一劳永逸的解决问题呢?

使用中国日历库

安装:pip install chinesecalendar

升级:pip install -U chinesecalendar

中国日历是一个现有的库,可以方便的提供每一天是否是中国的节假日,是什么假日。其实现原理是,该日历库的作者为我们标注好了每一年的节假日,只需要调用即可。

import datetime
import calendar
import chinese_calendar as cc


def day_in_month(year, month):
    info = []
    days_in_month = calendar.monthrange(year, month)[1]
    for day in range(1, days_in_month + 1):
        date = datetime.datetime(year, month, day)
        weekday = date.weekday()
        on_holiday, holiday_name = cc.get_holiday_detail(date)
        info.append({
            "day": f"{year}-{month}-{day}",
            "weekday": weekday + 1,
            "on_holiday": on_holiday,
            "holiday_name": holiday_name
        })
    return info

print(day_in_month(2024, 11))

不过需要注意的是:由于该库中的节假日都是作者添加的,所以,在不更新升级的情况下,节假日是不会增多的。例如,现在还没有公布2025年的节假日,那么该库就不能处理2025年的节假日。需要在该库更新以后,手动更新该库,才可以用上最新实现,这听起来不太友好。毕竟,如果在网络受限的环境中,或者没有及时更新,即使已经到了新的年份,旧有的程序也不能很好工作。

使用爬虫程序

另一种想法就是,通过发送网络请求,从网络信息上得到最新的节假日信息。这样,信息总是新的,而且,也不需要更新库,就可以一直使用旧有程序(假设网络地址持续有效,且结构没有发生更改)。

当然,我是非常不推荐这种方法的,与之前的方法不同,该方法虽然可以在不主动更新的情况下持续有效,但是没办法用于没有网络的环境。当然更重要的是,未经许可的爬虫行为并非是一种好的行为,如无必要应该尽量减少或者避免。

有一位作者也发布了一个库,用于爬虫的方式得到节假日信息,该库叫做china_calendar,爬取信息的地址选择的便民查询

import china_calendar


is_holiday = china_calendar.is_holiday("2025-01-28")
print(is_holiday)

倒计时工具

计算时间差

虽然倒计时并不算是日历的一部分,但是可以快速帮助我们得知距离某天还有多久(比如说,还有几天放假)。首先,我们可以计算两个日期的时间差。

from datetime import datetime


def calculate_days_difference(date1, date2):
    difference = date2 - date1
    return difference.days


date1 = datetime(2024, 11, 11)
date2 = datetime(2025, 1, 28)

days_diff = calculate_days_difference(date1, date2)
print(f"两日期相差{days_diff}天")

计算倒计时

当然,如果是计算某个日期距离今天的时间差,就变成了倒计时功能了。

def countdown_to_date(target_date):
    today = datetime.now().date()
    difference = target_date.date() - today
    return difference.days


date = datetime(2025, 1, 28)
days_diff = countdown_to_date(date)
print(f"距离2025年除夕还有{days_diff}天")

绘制日历

画一个日历并不困难,但是想要画的好看,美观,需要考虑的就比较多了。不过,我们也可以用一个现成的日历实现,比如说fullcalendar,其可以在html的页面中,通过js代码,快速绘制一个功能丰富的日历。

我们可以通过script标签引入,然后,通过适当的js代码,就可以得到功能丰富的日历了,比如说,其官方网站提供的例子:

节假日信息的日历

结合之前的chinese_calendar库,我们就可以轻松实现一个带有节假日信息的日历了,我们将背景颜色设置为红色,当作假期。并且,通过flask作为后端,搭建一个web页面。

import datetime
import calendar

import chinese_calendar as cc
from flask import Flask, render_template, request, jsonify


app = Flask(__name__)

def add_holiday_info(year, month):
    info = []
    days_in_month = calendar.monthrange(year, month)[1]
    for day in range(1, days_in_month + 1):
        date = datetime.datetime(year, month, day)
        on_holiday, holiday_name = cc.get_holiday_detail(date)
        if on_holiday:
            info.append({
                "start": f"{year}-{month:02d}-{day:02d}",
                "title": "",
                "display": "background",
                "backgroundColor": "#ff0000"
            })

            if holiday_name:
                info.append({
                    "start": f"{year}-{month:02d}-{day:02d}",
                    "title": holiday_name
                })
    return info


def add_surrounding_year(year):
    info = []
    last_year = year - 1
    next_year = year + 1

    for month in range(1, 13):
        try:
            info.extend(add_holiday_info(last_year, month))
            info.extend(add_holiday_info(year, month))
            info.extend(add_holiday_info(next_year, month))
        except NotImplementedError as e:
            pass

    return info


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/api/events")
def get_events():
    year = int(request.args.get("year"))
    return jsonify(add_surrounding_year(year))

同样,我们需要创建一个index.html作为页面

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>带有假期显示的日历</title>
    <script src='https://cdn.jsdelivr.net/npm/fullcalendar@6.1.15/index.global.min.js'></script>
    <style>
        #calendar {
            max-width: 900px;
            margin: 0 auto;
        }
    </style>
    <script>
        document.addEventListener('DOMContentLoaded', function() {
            var calendarEl = document.getElementById('calendar');

            var calendar = new FullCalendar.Calendar(calendarEl, {
              initialView: 'dayGridMonth',
              locale: 'zh-cn',

              datesSet: function(info) {
                  var year = info.start.getFullYear();
                  var month = ('0' + (info.start.getMonth())).slice(-2);

                  fetch('/api/events?year=' + year + '&month=' + month)
                    .then(response => response.json())
                    .then(events => {
                        calendar.removeAllEvents();
                        calendar.addEventSource(events);
                    })
                    .catch(error => console.error('Error fetching events:', error));
              }
            });

            calendar.render();
        });
    </script>
</head>
<body>
    <div id="calendar"></div>
</body>
</html>

很好,现在,我们就得到了一个带有节假日信息的日历了!虽然看起来好像和常见的日历不太一样。不过正确且清晰。

 添加额外信息

日历上只有节假日信息,肯定不能让人满意,我们还可以在日历上添加一些自己的备注信息,让其在日历上显示出来。对于这个需求,我们不需要改动页面,只需要添加备注信息即可。

import datetime
import calendar

import chinese_calendar as cc
from flask import Flask, render_template, request, jsonify


app = Flask(__name__)


extra_info = [
    ("2024-12-12", "双十二购物"),
    ("2024-12-21", "冬至活动"),
    ("2024-12-24", "约女神看电影"),
    ("2024-12-25", "抓捕麋鹿老人")
]


def add_extra_info():
    info = []
    for i in extra_info:
        info.append({
            "start": i[0],
            "title": i[1]
        })
    return info


def add_holiday_info(year, month):
    info = []
    days_in_month = calendar.monthrange(year, month)[1]
    for day in range(1, days_in_month + 1):
        date = datetime.datetime(year, month, day)
        on_holiday, holiday_name = cc.get_holiday_detail(date)
        if on_holiday:
            info.append({
                "start": f"{year}-{month:02d}-{day:02d}",
                "title": "",
                "display": "background",
                "backgroundColor": "#ff0000"
            })
    return info


def add_surrounding_year(year):
    info = []
    last_year = year - 1
    next_year = year + 1

    for month in range(1, 13):
        try:
            info.extend(add_holiday_info(last_year, month))
            info.extend(add_holiday_info(year, month))
            info.extend(add_holiday_info(next_year, month))
        except NotImplementedError as e:
            pass

    return info


def add_all_info(year):
    info = []
    info.extend(add_extra_info())
    info.extend(add_surrounding_year(year))
    return info


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/api/events")
def get_events():
    year = int(request.args.get("year"))
    return jsonify(add_all_info(year))

经过简单的修改以后,我们成功添加了备注信息。

引用与致谢

chinese_calendar的pypi地址

 china_calendar的pypi地址

 fullcalendar的官方地址

P.S.虽然我们成功绘制了日历,但是小侄子看了并不感到开心,“在学校里画日历,也是一种消磨时间的好办法,你都用程序画好了,我在学校干嘛?”,当然是要好好学习啊喂! 


网站公告

今日签到

点亮在社区的每一天
去签到