python递归解析、过滤目录、保留某个目录中的一系列子目录

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Time-stamp: <2014-03-09 14:50:09 Sunday by tyrion>

# @version 1.0
# @author tyrion

import os
import re

c_suffix = ['.c', '.C']
asm_suffix = ['.s', '.S']
filters = ["bm3803", "V8"]
excude_dirs = ["arch"]

c_file = []
asm_file = []
dirs = []
h_dirs = []

flag = False

def Parse_Directory_Structure(dir):
    for i in os.listdir(dir):
        sub_dir = os.path.join(dir, i)
        if (os.path.split(sub_dir)[1] in excude_dirs):
            continue
        if os.path.isdir(sub_dir):
            dirs.append(sub_dir)
            Parse_Directory_Structure(sub_dir)
        else:
            if (os.path.splitext(sub_dir)[1] in c_suffix):
                c_file.append(sub_dir)
            elif (os.path.splitext(sub_dir)[1] in asm_suffix):
                asm_file.append(sub_dir)


def Filter_Dirs(dirs, filter_index, filters = []):
    if (filter_index >= len(filters)):
        return dirs
    else:
        directory_prefixs = []
        tmp_dirs = []
        for value in dirs:
            if filters[filter_index] in value:
                directory_prefixs.append(value.split(filters[filter_index])[0])
            
        directory_prefixs = list(set(directory_prefixs))

        for value in dirs:
            flag = False
            for directory_prefix in directory_prefixs:
                if directory_prefix in value:
                    if filters[filter_index] in value:
                        tmp_dirs.append(value)
                    flag = True
                    break
                else:
                    continue
            if flag is True:
                continue
            else:
                tmp_dirs.append(value)

        return Filter_Dirs(tmp_dirs, filter_index+1, filters)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值