pyhon搜索目录下的文件或文件夹

# !/usr/bin/env python
# -*-coding:utf-8-*-
"""
Copyright(c)2018,浙江大华
file:     sambaCenter.py
author:   27711
date:     2018/11/24
version:  1.0
describe:

"""

import os
import subprocess

class CFileInfo(object):
    def __int__(self):
        self.fileType = 0  # 0:dir 1:file
        self.fileName = ""
        self.fileSize = 0  # Byte# ,if dir 0
        self.parentDir = ""  # "/home/10001"


class CSambaCenter(object):
    def __init__(self):
         pass

    def scanFiles(self, userId, dir="/home", preFix=None, postFix=None):
        """
        搜索指定目录下的匹配前缀或后缀的文件和文件夹
        :param userId: 用户ID 如10001
        :param dir:指定目录,若为文件则返回文件详情
        :param preFix:前缀如:“test_”
        :param postFix:后续如:".py"
        :return:目录和文件的list详情
        """
        try:
            if preFix is None and postFix is None:
                return self.getAllDirs(userId, dir)

            listFilesInfo = []
            for dirpath, dirs, files in os.walk(dir):
                for dirname in dirs:
                    if dirname[0] == '.':  # 剔除隐藏
                        pass
                    if postFix:
                        if dirname.endswith(postFix):
                            fileInfo = CFileInfo()
                            fileInfo.fileName = dirname
                            fileInfo.parentDir = dirpath
                            listFilesInfo.append(fileInfo)
                    elif preFix:
                        if dirname.startswith(preFix):
                            fileInfo = CFileInfo()
                            fileInfo.fileName = dirname
                            fileInfo.parentDir = dirpath
                            listFilesInfo.append(fileInfo)

                for file in files:
                    if file[0] == '.':  # 剔除隐藏
                        pass
                    if postFix:
                        if file.endswith(postFix):
                            fileInfo = CFileInfo()
                            fileInfo.fileType = 1
                            fileInfo.fileSize = os.path.getsize(file)
                            fileInfo.fileName = file
                            fileInfo.parentDir = dirpath
                            listFilesInfo.append(fileInfo)
                    elif preFix:
                        if file.startswith(preFix):
                            fileInfo = CFileInfo()
                            fileInfo.fileType = 1
                            fileInfo.fileSize = os.path.getsize(file)
                            fileInfo.fileName = file
                            fileInfo.parentDir = dirpath
                            listFilesInfo.append(fileInfo)

            return listFilesInfo
        except Exception as e:
            print(e)
            return None


if __name__ == '__main__':
    pass

 

剔除隐藏的文件和文件夹,for dirpath, dirs, files in os.walk(dir),通过python库接口方便,方案性能上需要再优化,若文件较多,不建议次方案

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐