os.walk without digging into directories below

前端 未结 20 1268
庸人自扰
庸人自扰 2020-12-04 06:21

How do I limit os.walk to only return files in the directory I provide it?

def _dir_list(self, dir_name, whitelist):
    outputList = []
    for         


        
20条回答
  •  温柔的废话
    2020-12-04 07:06

    root folder changes for every directory os.walk finds. I solver that checking if root == directory

    def _dir_list(self, dir_name, whitelist):
        outputList = []
        for root, dirs, files in os.walk(dir_name):
            if root == dir_name: #This only meet parent folder
                for f in files:
                    if os.path.splitext(f)[1] in whitelist:
                        outputList.append(os.path.join(root, f))
                    else:
                        self._email_to_("ignore")
        return outputList
    

提交回复
热议问题