# -*- 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):
  '''計算天數(shù)方法
  '''
  try:
    # 用以存儲天數(shù)
    nums = 0
    # 獲取輸入框中的數(shù)據(jù)
    year, month, day = [int(elem) for elem in date.get().split('-')]
    # 判斷月份，規(guī)定在 1-12 之間
    if 1 <= month <= 12:
      # 遍歷計算天數(shù)
      for month_x in range(1, month + 1):
        # 計算每月的天數(shù)
        _, month_day = calendar.monthrange(year, month_x)
        # 遍歷的月份等于當前月份，對天數(shù)進行規(guī)整
        if month_x == month:
          # 文本輸入框給出的天數(shù)不符合，則拋出異常
          if day > month_day:
            raise MyException("信息輸入錯誤，注意天數(shù)！")
          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 這樣的格式輸入。注意月份，天數(shù)！")
root = Tk()
root.title("計算天數(shù)")
# 設置框架
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)
# 設置內(nèi)邊距
for child in mainframe.winfo_children():
  child.grid_configure(padx=5, pady=5)
date_entry.focus()
root.bind('<Return>', calculate)
root.mainloop()