strExtension-允许两种格式

时间:2019-07-12 20:14:11

标签: python

如何允许strExtension的格式如.pdf和.html。 如果更改为strExtension =“。*”,则该脚本无效。 如何使用两种格式? pdf和html吗?

# extension of files
strExtension = ".pdf" 
# for each file in source directory
for file in os.listdir(sourceDir):
    # if file ends with extension
if file.endswith(strExtension):

3 个答案:

答案 0 :(得分:2)

只需测试两种情况:

if file.endswith(strExtension1) or file.endswith(strExtension2):

或如乔恩·克莱门茨(Jon Clements)在下面评论:

if file.endswith((strExtension1,strExtension2)):

答案 1 :(得分:0)

str.endswith()不支持通配符。但是它支持元组,所以您可以这样做:

extensions = (".html", ".pdf")
if file.endswith(extensions):
    # do stuff

答案 2 :(得分:0)

这对我有用。感谢您的快速帮助。谢谢大家...

# extension of files
strExtension1 = ".pdf"
strExtension2 = ".html" 
# for each file in source directory
for file in os.listdir(sourceDir):
# if file ends with extension
if file.endswith(strExtension1) or file.endswith(strExtension2):