nameList=['.doc','.md','.py']
name=input('请输入你需要上传的文件格式:')
if name in nameList:
print('恭喜!你输入的%s格式可以上传'%name)
else:
print('对不起,你输入的%s格式不允许上传'%name)
源码如上。
执行,输入.doc .md .py 结果如下:
请输入你需要上传的文件格式:.md
恭喜!你输入的.md格式可以上传
输入.txt 结果如下:
请输入你需要上传的文件格式:.txt
对不起,你输入的.txt格式不允许上传