import subprocess
import tkinter as tk
from tkinter import messagebox
 
def add_windows_credential(target_name, username="guest", password=""):
    """
    使用 cmdkey 命令添加 Windows 憑據(jù)
    """
    try:
        # 構(gòu)建 cmdkey 命令
        cmd = f'cmdkey /add:{target_name} /user:{username}'
        if password:
            cmd += f' /pass:{password}'
        
        # 執(zhí)行命令
        result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
        
        # 檢查命令執(zhí)行結(jié)果
        if result.returncode == 0:
            messagebox.showinfo("成功", f"成功添加憑據(jù)！\n目標(biāo)計(jì)算機(jī): {target_name}")
            return True
        else:
            messagebox.showerror("錯(cuò)誤", f"添加憑據(jù)失敗: {result.stderr}")
            return False
            
    except Exception as e:
        messagebox.showerror("錯(cuò)誤", f"添加憑據(jù)時(shí)發(fā)生錯(cuò)誤: {str(e)}")
        return False
 
class CredentialApp:
    def __init__(self, root):
        self.root = root
        self.root.title("Windows憑據(jù)添加工具")
        self.root.geometry("300x150")
        
        # 創(chuàng)建主框架，用于居中顯示內(nèi)容
        main_frame = tk.Frame(root)
        main_frame.pack(expand=True)
        
        # 創(chuàng)建輸入框和標(biāo)簽
        tk.Label(main_frame, text="請輸入目標(biāo)計(jì)算機(jī)名:", font=('Arial', 10)).pack(pady=10)
        self.computer_entry = tk.Entry(main_frame, width=25)
        self.computer_entry.pack(pady=5)
        
        # 按鈕框架
        button_frame = tk.Frame(main_frame)
        button_frame.pack(pady=20)
        
        # 添加確定和取消按鈕
        tk.Button(button_frame, text="確定", width=10, command=self.add_credential).pack(side=tk.LEFT, padx=10)
        tk.Button(button_frame, text="取消", width=10, command=self.root.quit).pack(side=tk.LEFT, padx=10)
    
    def add_credential(self):
        computer_name = self.computer_entry.get().strip()
        
        if not computer_name:
            messagebox.showwarning("警告", "請輸入計(jì)算機(jī)名！")
            return
            
        if add_windows_credential(computer_name):
            self.computer_entry.delete(0, tk.END)  # 清空輸入框
 
def main():
    root = tk.Tk()
    app = CredentialApp(root)
    root.mainloop()
 
if __name__ == "__main__":
    main()