获取文件的后缀名有好几种方法：

第一种：splittext()方法

os.path.splitext(path)[-1]
 

第二种：endswith()方法

path = "test_user_info.py"
bool = path.endswith(".py")
print(bool)
 

第三种：判断后缀名是否在字符串中（这种会存在误判，若是.pyx后缀，一样会打印True，前面两种不会）

path = "test_user_info.py"
if ".py" in path:
    print(True)
 

第四种：用split方法切割，但是吧这种只是拿到了py没有点，所以再加上点也是可以的

path = "test_user_info.py"
suffix = path.split(".")[1]
print("suffix: {}".format(suffix))