Python 簡易版小工具 | 計算天數

簡易版小工具 | 計算天數

---

需求

---

給定一個日期，格式如 「2020-2-12」，計算出這個日期是 2020 年的第幾天？

實現思路

---

1. 使用 tkinter 和 tkinter.ttk 對界面進行佈置；
2. 使用 calendar 計算天數；
3. 規範輸入日期的格式；
4. 對月份，天數進行邏輯判斷；
5. 輸入錯誤拋出異常提示。

代碼實現

---

# -*- coding: utf-8 -*-
'''
@File: calc_day_v2.py
@Time: 2020/02/12 20:33:22
@Author: 大夢三千秋
@Contact: yiluolion@126.com
'''

# Put the import lib here
from tkinter import *
import tkinter.messagebox as messagebox
from tkinter import ttk
import calendar

class MyException(BaseException):
    '''自定義異常類
    '''
    def __init__(self, message):
        self.message = message

def calculate(*args):
    '''計算天數方法
    '''
    try:
        # 用以存儲天數
        nums = 0
        # 獲取輸入框中的數據
        year, month, day = [int(elem) for elem in date.get().split('-')]
        # 判斷月份，規定在 1-12 之間
        if 1 <= month <= 12:
            # 遍歷計算天數
            for month_x in range(1, month + 1):
                # 計算每個月的天數
                _, month_day = calendar.monthrange(year, month_x)
                # 遍歷的月份等於當前月份，對天數進行規整
                if month_x == month:
                    # 文本輸入框給出的天數不符合，則拋出異常
                    if day > month_day:
                        raise MyException("信息輸入錯誤，注意天數！")
                    continue
                nums += month_day
            nums += day
            # 設置值到文本框
            days.set(nums)
            the_year.set(year)
        else:  # 月份超出範圍拋出異常
            raise MyException("信息輸入錯誤，注意月份！")
    except MyException as e:
        messagebox.showinfo(title="輸入信息錯誤", message=e)
    except Exception as e:
        # print(e)
        messagebox.showinfo(title="輸入信息錯誤", message="輸出格式錯誤，按照 2020-2-12 這樣的格式輸入。注意月份，天數！")


root = Tk()
root.title("計算天數")

# 設置框架
mainframe = ttk.Frame(root, padding="3 3 12 12")
mainframe.grid(column=0, row=0, sticky=(N, S, E, W))
root.columnconfigure(0, weight=1)
root.rowconfigure(0, weight=1)

date = StringVar()
the_year = StringVar()
days = StringVar()

# 文本框部件佈置
date_entry = ttk.Entry(mainframe, width=10, textvariable=date)
date_entry.grid(column=2, row=1, sticky=(W, E))

# 標籤及按鈕的佈置
ttk.Label(mainframe, text="例如：2020-2-12").grid(column=5, row=1, sticky=(W, E))
ttk.Label(mainframe, textvariable=days).grid(column=4, row=2, sticky=(W, E))
ttk.Label(mainframe, textvariable=the_year).grid(column=2, row=2, sticky=(W, E))
ttk.Button(mainframe, text="Calculate", command=calculate).grid(column=5, row=3)

ttk.Label(mainframe, text="日期：").grid(column=1, row=1, sticky=E)
ttk.Label(mainframe, text="這一天是").grid(column=1, row=2, sticky=E)
ttk.Label(mainframe, text="年的第").grid(column=3, row=2, sticky=E)
ttk.Label(mainframe, text="天").grid(column=5, row=2, sticky=W)

# 設置內邊距
for child in mainframe.winfo_children():
    child.grid_configure(padx=5, pady=5)

date_entry.focus()

root.bind('<Return>', calculate)

root.mainloop()

使用效果

---

正確輸入的效果以下：

未按照格式輸入，錯誤提示效果：

月份輸入錯誤，提示效果以下：

天數超出當月範圍的錯誤提示效果：

---

本篇的內容主要是對昨天的 tkinter 模塊的延展使用，實現一個計算天數的小工具。

---

歡迎關注微信公衆號《書所集錄》