python小例子教你找出文件后缀

大家都知道很多网站啊app中都有上传功能，那么用户在上传的时候有没有按要求上传显得非常重要，比如要求上传图片的他弄了个txt过来，要求上传视频的，他上传了一段木马脚本，那就惨了。

所以用户上传之后，首先要在前端做一个判别，看看用户到底上传的什么东西，这时候获取文件的后缀就相当的重要了。

我们来举个例子。

比如用户上传的是一个下面这样的文件：

soundme.txt.mp3

很显然这是一个mp3文件，它的后缀是.mp3

那应该咋个做呢？

主要使用到一个函数：

rfind()  #从右往左查找指定字符串，并返回下标值。

来一段简单的代码吧，python3.10下执行的。

#coding:utf-8

upload_file=input('请输入文件名：')
index=upload_file.rfind('.') #获取.的下标值，也就是找出最右边.是在什么位置
file_type=upload_file[index:] #点.右边的就是文件后缀] 
file_name=upload_file[:index]  #点.左边的就是文件名
print('文件名为{}\n文件后缀为{}'.format(file_name,file_type))

输出结果如下：

找出文件名后，再写一个判断语句即可判断用户到底上传了什么东西，可以在前端做一个简单的过滤。