tkinter:如果我从下拉菜单中选择一个选项，它会禁用特定的输入框，我该怎么做？

import tkinter as tk

def disable_entry():
    if var.get() == "选项1":
        entry1.config(state=tk.DISABLED)
    elif var.get() == "选项2":
        entry2.config(state=tk.DISABLED)
    else:
        entry1.config(state=tk.NORMAL)
        entry2.config(state=tk.NORMAL)

root = tk.Tk()

var = tk.StringVar(root)
var.set("请选择选项")

option_menu = tk.OptionMenu(root, var, "选项1", "选项2", "选项3")
option_menu.pack()

entry1 = tk.Entry(root)
entry1.pack()

entry2 = tk.Entry(root)
entry2.pack()

var.trace("w", lambda *args: disable_entry())

root.mainloop()