📅  最后修改于: 2023-12-03 14:57:53.190000             🧑  作者: Mango
在编写一个可以处理不同文件类型的程序时,我们需要让用户指明文件的类型。为了让这个过程更加友好,我们可以编写一个输入文件类型的引导程序。这个引导程序可以提示用户输入文件类型,并将输入值与我们程序可以处理的文件类型做比较。
# 支持的所有文件类型
SUPPORTED_FILE_TYPES = ["txt", "csv", "xls", "json"]
# 引导程序
def input_file_type():
while True:
user_input = input("请输入文件类型:")
if user_input.lower() in SUPPORTED_FILE_TYPES:
return user_input.lower()
else:
print("不支持的文件类型,请重新输入。")
# 调用示例
file_type = input_file_type()
print("用户选择的文件类型为:", file_type)
上述代码通过while循环不断提示用户输入文件类型,直到用户输入一个在支持文件类型列表中的值为止。当用户输入的值不在支持文件类型列表中时,程序会给出错误提示并重新等待用户输入。
用户输入的文件类型最终会以小写形式返回,以便与程序中使用的文件类型列表做比较。在实际使用中,我们可以根据用户的选择来执行相应的文件处理程序。