Tkinter教程45:给软件添加一机一码,登入授权验证(hashlib+hmac+到期时间)

# -*- coding: utf-8 -*-# @Author : 小红牛# 微信公众号：wdPythonimport tkinter as tkfrom tkinter import ttk, messageboximport jsonimport hmacimport hashlibimport osimport uuidimport platformimport subprocess# ==================== 固定密钥 ====================SECRET_KEY = b"wdpython2026_key888"# ==================== 机器码生成 ====================def get_machine_code():    mac = uuid.getnode()    mac_str = '{:012x}'.format(mac).upper()    disk_serial = ""    if platform.system() == "Windows":        try:            cmd = "wmic diskdrive get SerialNumber"            result = subprocess.run(cmd, shell=True, capture_output=True, text=True)            lines = [line.strip() for line in result.stdout.splitlines() if line.strip()]            if len(lines) > 1:                disk_serial = lines[1].strip()        except:            pass    raw = f"68{mac_str}60{disk_serial}30{platform.node()}"    return hashlib.sha256(raw.encode()).hexdigest()[:32]def generate_license(username, expire):    if not username or not expire:        return None    message = f"{username}{expire}".encode('utf-8')    sign = hmac.new(SECRET_KEY, message, hashlib.sha256).hexdigest()    return {"username": username, "expire": expire, "sign": sign}# ==================== GUI ====================class LicenseGeneratorApp:    def __init__(self, master):        self.master = master        master.title("授权码生成器")        master.geometry("500x240")        master.resizable(False, False)        # 居中        master.update_idletasks()        w, h = master.winfo_width(), master.winfo_height()        x = (master.winfo_screenwidth() // 2) - (w // 2)        y = (master.winfo_screenheight() // 2) - (h // 2)        master.geometry(f"+{x}+{y}")        # 主框架（pack 布局）        main_frame = ttk.Frame(master, padding=20)        main_frame.pack(fill=tk.BOTH, expand=True)        # ---- 机器码行 ----        row1 = ttk.Frame(main_frame)        row1.pack(fill=tk.X, pady=5)        ttk.Label(row1, text="用户机器码（32位）：").pack(side=tk.LEFT)        self.username_entry = ttk.Entry(row1, width=40)        self.username_entry.pack(side=tk.LEFT, padx=5)        # ---- 过期日期行 ----        row2 = ttk.Frame(main_frame)        row2.pack(fill=tk.X, pady=5)        ttk.Label(row2, text="过期日期（YYYYMMDD）：").pack(side=tk.LEFT)        self.expire_entry = ttk.Entry(row2, width=36)        self.expire_entry.pack(side=tk.LEFT, padx=5)        self.expire_entry.insert(0, "20261231")        # ---- 提示 ----        ttk.Label(main_frame, text="提示：日期格式为 8 位数字，如 20261231", foreground="gray").pack(anchor="w", pady=5)        # ---- 按钮 ----        btn_frame = ttk.Frame(main_frame)        btn_frame.pack(pady=15)        ttk.Button(btn_frame, text="生成授权码", command=self.generate, width=15).pack(side=tk.LEFT, padx=15)        ttk.Button(btn_frame, text="刷新机器码", command=self.read_machine_code, width=10).pack(side=tk.LEFT, padx=5)        # ---- 状态栏 ----        self.status_var = tk.StringVar()        self.status_var.set("就绪")        ttk.Label(main_frame, textvariable=self.status_var, foreground="blue").pack(anchor="w", pady=10)        # 启动自动读取        self.read_machine_code()    def read_machine_code(self):        code = get_machine_code()        self.username_entry.delete(0, tk.END)        self.username_entry.insert(0, code)        self.status_var.set(f"已读取机器码：{code}")    def generate(self):        username = self.username_entry.get().strip()        expire = self.expire_entry.get().strip()        if not username:            messagebox.showerror("错误", "请先获取机器码")            return        if len(username) != 32:            messagebox.showerror("错误", "机器码应为32位字符")            return        if not expire.isdigit() or len(expire) != 8:            messagebox.showerror("错误", "日期必须为8位数字（YYYYMMDD）")            return        lic_data = generate_license(username, expire)        if not lic_data:            messagebox.showerror("错误", "生成授权数据失败")            return        # 直接写入当前目录（脚本所在目录）        file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "授权码.json")        try:            with open(file_path, 'w', encoding='utf-8') as f:                json.dump(lic_data, f, indent=2, ensure_ascii=False)            self.status_var.set(f"✅ 授权文件已生成：{file_path}")            messagebox.showinfo("成功", f"授权文件已保存至：\n{file_path}")        except Exception as e:            messagebox.showerror("保存失败", f"写入文件时出错：{e}")            self.status_var.set("保存失败")if __name__ == "__main__":    root = tk.Tk()    app = LicenseGeneratorApp(root)    root.mainloop()