Python3 获取&判断文件后缀

方法一、 str的endswith方法:

cases_path = '测试用例1.xlsx'
print(cases_path.endswith('.xlsx'))

返回结果: True

应用 判断后缀名的示例:

cases = '测试用例1.xlsx'
cases_path = cases.endswith('.xlsx')
if cases:
    print('文件后缀验证通过')
else:
    print("仅支持以下格式：.xlsx")

方法二 、os.path.splitext() 方法:

该方法返回两个元素, 第一个是路径去掉后缀的部分, 第二个是文件后缀:

import os
import openpyxl

class start():
    def load_cases(self,cases):
        if cases:
            case_paht = os.path.splitext(cases)[1]

            if case_paht in ['.xlsx', '.xlsm', '.xltx', '.xltm']:
                print('文件后缀验证通过')
            else:
                print("测试用例仅支持以下格式：.xlsx、.xlsm、.xltx、.xltm")
            print("文件名：", os.path.splitext(cases)[0])
            print("文件后缀：", case_paht)
        else:
            print("测试用例不能为空，请填写测试用例路径")

if __name__ == '__main__':
    st = start().load_cases('测试用例1.xlsx')