import datetime
import tkinter as tk
from tkinter import ttk
from threading import Thread
import time
import os
 
 
class ShutdownApp:
    def __init__(self, root):
        self.time_left = None
        self.root = root
        self.root.title("定時關(guān)機應用")
        self.root.resizable(0, 0)
        screenwidth = self.root.winfo_screenwidth()
        screenheight = self.root.winfo_screenheight()
        width = 600
        height = 200
        size_geo = '%dx%d+%d+%d' % (width, height, (screenwidth - width) / 2, (screenheight - height) / 2)
        self.root.geometry(size_geo)
        self.root.iconphoto(False, tk.PhotoImage(file='C:/Users/wokao/Desktop/icon.png'))
        self.root["background"] = "#8DB6CD"
 
        self.notebook = tk.ttk.Notebook(self.root)
        self.framework1 = tk.Frame()
        self.framework2 = tk.Frame()
        self.notebook.add(self.framework1, text='默認模式')
        self.notebook.add(self.framework2, text='自定義模式')
        self.notebook.pack(padx=10, pady=5, fill=tk.BOTH, expand=True)
 
        # 選項卡1的界面
        tk.Label(self.framework1, text="選擇關(guān)機時長:").pack()
        cbox = ttk.Combobox(self.framework1)
        cbox['value'] = ('0.5小時', '1小時', '1.5小時', '2小時')
        cbox.current(1)
        self.selected_value = cbox.get()
        cbox.pack()
        self.start_button = tk.Button(self.framework1, text="開始", command=self.start_timer)
        self.start_button.pack()
        self.cancel_button = tk.Button(self.framework1, text="取消關(guān)機", state='disabled', command=self.cancel_timer)
        self.cancel_button.pack()
        self.timer_label = tk.Label(self.framework1, text="", bg="#8DB6CD")
        self.timer_label.pack()
 
        # 選項卡2的界面
        tk.Label(self.framework2, text="輸入關(guān)機時長（秒）:").pack()
        self.time_entry2 = tk.Entry(self.framework2)
        self.time_left2 = self.time_entry2.get()
        self.time_entry2.pack()
        self.start_button2 = tk.Button(self.framework2, text="開始", command=self.start_timer2)
        self.start_button2.pack()
        self.cancel_button2 = tk.Button(self.framework2, text="取消關(guān)機", state='disabled', command=self.cancel_timer2)
        self.cancel_button2.pack()
        self.timer_label2 = tk.Label(self.framework2, text="", bg="#8DB6CD")
        self.timer_label2.pack()
 
        self.timer_thread = None
        self.running = False
 
    # 選項卡1的功能實現(xiàn)
    def selected_time(self, selected_value):
        match selected_value:
            case '0.5小時':
                self.time_left = 1800
            case '1小時':
                self.time_left = 3600
            case '1.5小時':
                self.time_left = 5400
            case '2小時':
                self.time_left = 7200
 
    def start_timer(self):
        try:
            self.selected_time(self.selected_value)
        except ValueError:
            self.timer_label.config(text="請選擇關(guān)機倒計時時長！")
            return
 
        self.notebook.tab(1, state='disabled')
        self.running = True
        self.start_button.config(state='disabled')
        self.cancel_button.config(state='normal')
        self.timer_thread = Thread(target=self.run_timer)
        self.timer_thread.start()
 
    def run_timer(self):
        while self.time_left > 0 and self.running:
            timer = str(datetime.timedelta(seconds=int(self.time_left)))
            self.timer_label.config(text=f"關(guān)機倒計時: {timer} ", font=("黑體", 45), fg="white", bg="#8DB6CD")
            time.sleep(1)
            self.time_left -= 1
 
        self.timer_label.config(text="")
        if self.running:
            os.system("shutdown /s /t 1")  # 在Windows上執(zhí)行關(guān)機命令
 
    def cancel_timer(self):
        self.running = False
        self.start_button.config(state='normal')
        self.cancel_button.config(state='disabled')
        self.timer_label.config(text="已取消關(guān)機")
        self.notebook.tab(1, state='normal')
 
    # 選項卡2的功能實現(xiàn)
    def start_timer2(self):
        try:
            self.time_left2 = int(self.time_entry2.get())
        except ValueError:
            self.timer_label2.config(text="請輸入有效的數(shù)字！")
            return
 
        self.notebook.tab(0, state='disabled')
        self.running = True
        self.start_button2.config(state='disabled')
        self.cancel_button2.config(state='normal')
        self.timer_thread = Thread(target=self.run_timer2)
        self.timer_thread.start()
 
    def run_timer2(self):
        while self.time_left2 > 0 and self.running:
            self.timer_label2.config(text=f"關(guān)機倒計時: {self.time_left2} 秒", font=("黑體", 45),fg="white", bg="#8DB6CD")
            time.sleep(1)
            self.time_left2 -= 1
 
        self.timer_label2.config(text="")
        if self.running:
            os.system("shutdown /s /t 1")  # 在Windows上執(zhí)行關(guān)機命令
 
    def cancel_timer2(self):
        self.running = False
        self.start_button2.config(state='normal')
        self.cancel_button2.config(state='disabled')
        self.timer_label2.config(text="已取消關(guān)機")
        self.notebook.tab(0, state='normal')
 
 
if __name__ == "__main__":
    ui = tk.Tk()
    app = ShutdownApp(ui)
    ui.mainloop()