Python阻止以特定格式打开文件

Question

在代码执行过程中，是否有任何方法可以防止用户以pdf格式打开文件？ 我的代码在循环中打开/保存多个文件，打开一个新文件可能会导致错误。 操作系统是Windows。 谢谢

Answer 1

使其成为打开文件的函数

如果允许扩展，则检查扩展

阅读

否则返回或可能引发自定义异常

from os import path

def openfile(filename):
    dont_read_extensions = [".pdf", ".jpg"]
    extension = path.splitext(filename)[1]
    if extension in dont_read_extensions:
        return #or you can raise custom exception
    else:
        with open(filename,"r") as f:
             #do file operations
             return # you can return with result