Python菜单程序，如何获取选项调用指定函数的函数？

def menu():
    print("菜单选项:")
    print("1. 选项一")
    print("2. 选项二")
    print("3. 选项三")
    choice = input("请输入选项号码: ")
    return choice

def option_one():
    print("执行选项一的功能")

def option_two():
    print("执行选项二的功能")

def option_three():
    print("执行选项三的功能")

def main():
    while True:
        choice = menu()
        if choice == "1":
            option_one()
        elif choice == "2":
            option_two()
        elif choice == "3":
            option_three()
        else:
            print("无效的选项，请重新输入")

if __name__ == "__main__":
    main()