Python编程：灵活计算年假天数的完整指南_计算时间差考虑法定节假日-CSDN博客

本文链接：https://blog.csdn.net/wuShiJingZuo/article/details/135018948

更多资料获取

📚 个人网站：ipengtao.com

在工作和生活的快节奏中，灵活计算年假天数是管理时间的重要一环。本篇博客将深入介绍如何使用Python编写一个强大而灵活的年假计算工具。从基本的日期操作开始，逐步引入复杂的规则和算法，打造一个满足各种场景的年假计算器。

基础日期操作

首先，需要理解Python中的日期处理。使用datetime模块可以轻松进行日期的表示和计算。

from datetime import datetime, timedelta

# 计算两个日期之间的天数差
start_date = datetime(2023, 1, 1)
end_date = datetime(2023, 12, 31)
days_difference = (end_date - start_date).days
print(f"天数差：{days_difference} 天")

添加法定假期和周末规则

接下来，考虑法定假期和周末的影响。定义一个包含法定假期和周末的列表，并在计算天数差时排除这些日期。

holidays = [datetime(2023, 5, 1), datetime(2023, 10, 1)]  # 法定假期
weekend_days = [5, 6]  # 周末，假设星期天是第0天

def calculate_workdays(start_date, end_date, holidays, weekend_days):
    current_date = start_date
    workdays = 0

    while current_date <= end_date:
        if current_date.weekday() not in weekend_days and current_date not in holidays:
            workdays += 1

        current_date += timedelta(days=1)

    return workdays

result = calculate_workdays(start_date, end_date, holidays, weekend_days)
print(f"实际工作天数：{result} 天")

加入灵活的年假规则

现在，引入更灵活的年假规则，例如根据工龄和入职时间计算年假天数。定义一个函数，根据具体规则计算年假。

def calculate_annual_leave(start_date, end_date, holidays, weekend_days, seniority_years):
    base_workdays = calculate_workdays(start_date, end_date, holidays, weekend_days)
    annual_leave_days = 0

    if seniority_years >= 5:
        annual_leave_days = base_workdays + 5
    elif seniority_years >= 2:
        annual_leave_days = base_workdays + 3
    else:
        annual_leave_days = base_workdays

    return annual_leave_days

seniority = 3
annual_leave_result = calculate_annual_leave(start_date, end_date, holidays, weekend_days, seniority)
print(f"年假天数（包括法定假期和周末）：{annual_leave_result} 天")

考虑灵活的年假调整规则

有些公司在特殊情况下可能对年假进行调整。例如，当年假年初结余较多时，可以设定超过一定天数的部分需要在一定时期内使用完。添加这一规则，使年假计算更灵活。

def adjust_annual_leave(annual_leave_days, carry_over_limit, expiration_days):
    if annual_leave_days > carry_over_limit:
        adjusted_leave_days = carry_over_limit + expiration_days
        print(f"年假调整：调整后年假天数为 {adjusted_leave_days} 天")
        return adjusted_leave_days
    else:
        return annual_leave_days

carry_over_limit = 10
expiration_days = 5
adjusted_result = adjust_annual_leave(annual_leave_result, carry_over_limit, expiration_days)
print(f"调整后的年假天数：{adjusted_result} 天")

考虑特殊情况和个性化配置

在实际应用中，有时候会有一些特殊情况或个性化的需求。例如，某些员工可能有额外的假期，或者公司在特定日期可能放假。为了满足这些需求，可以添加更多的参数和规则。

def calculate_annual_leave_with_special_cases(start_date, end_date, holidays, weekend_days, seniority_years, special_leave_days=0, special_leave_dates=[]):
    base_workdays = calculate_workdays(start_date, end_date, holidays, weekend_days)
    annual_leave_days = 0

    if seniority_years >= 5:
        annual_leave_days = base_workdays + 5
    elif seniority_years >= 2:
        annual_leave_days = base_workdays + 3
    else:
        annual_leave_days = base_workdays

    # 加上额外的假期
    annual_leave_days += special_leave_days

    # 减去特殊日期的假期
    for special_date in special_leave_dates:
        if start_date <= special_date <= end_date and special_date.weekday() not in weekend_days:
            annual_leave_days -= 1

    return annual_leave_days

# 示例：某员工有2天额外假期，12月1日公司放假
special_leave_days = 2
special_leave_dates = [datetime(2023, 12, 1)]

result_with_special_cases = calculate_annual_leave_with_special_cases(start_date, end_date, holidays, weekend_days, seniority, special_leave_days, special_leave_dates)
print(f"考虑特殊情况后的年假天数：{result_with_special_cases} 天")

结合GUI界面实现友好交互

为了使年假计算更加用户友好，可以考虑使用GUI界面。借助tkinter等库，我们可以创建一个简单而直观的界面，让用户输入相关信息，然后得到年假计算结果。

import tkinter as tk
from tkinter import ttk

def calculate_annual_leave_gui():
    def on_calculate():
        start_date = datetime.strptime(start_date_entry.get(), "%Y-%m-%d")
        end_date = datetime.strptime(end_date_entry.get(), "%Y-%m-%d")
        seniority = int(seniority_entry.get())

        result = calculate_annual_leave_with_special_cases(start_date, end_date, holidays, weekend_days, seniority)
        result_label.config(text=f"年假天数：{result} 天")

    # 创建GUI窗口
    window = tk.Tk()
    window.title("年假计算器")

    # 添加控件
    start_date_label = ttk.Label(window, text="入职日期：")
    start_date_label.grid(row=0, column=0, padx=10, pady=10)
    start_date_entry = ttk.Entry(window)
    start_date_entry.grid(row=0, column=1, padx=10, pady=10)

    end_date_label = ttk.Label(window, text="截止日期：")
    end_date_label.grid(row=1, column=0, padx=10, pady=10)
    end_date_entry = ttk.Entry(window)
    end_date_entry.grid(row=1, column=1, padx=10, pady=10)

    seniority_label = ttk.Label(window, text="工龄（年）：")
    seniority_label.grid(row=2, column=0, padx=10, pady=10)
    seniority_entry = ttk.Entry(window)
    seniority_entry.grid(row=2, column=1, padx=10, pady=10)

    calculate_button = ttk.Button(window, text="计算年假", command=on_calculate)
    calculate_button.grid(row=3, column=0, columnspan=2, pady=10)

    result_label = ttk.Label(window, text="")
    result_label.grid(row=4, column=0, columnspan=2, pady=10)

    # 运行GUI
    window.mainloop()

# 运行GUI
calculate_annual_leave_gui()