python时间约束

print(" 欢迎使用学生成绩管理系统 ")
while True:
print("系统功能如下：\n1.显示全部记录\n2.输入\n3.删除\n4.学号排序\n5.课程排序\n6.退出系统")
print('开始倒计时1分钟：')
task_time = 1
start_time = time.strftime('%H:%M:%S', time.localtime())
for i in range(1, task_time * 5):
print('剩余时间：%i秒' % (task_time * 5- i))
time.sleep(1)

time_start = time.time()
x = int(input("请选择功能，按下数字键:"))
time_end = time.time()
time_sum = time_end - time_start
if time_sum >= 5:
    print("输入超时")
    break

如何限制输入时间啊，这样会在input那卡住

你都不看下time_end是啥，打印下