方法一、 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')