仅使用python脚本中的os.walk（）处理特定目录

Question

I have this python script that concatenates a classpath variable. It concatenates the directories of all the files that endwith a ".properties" extension as well as a ".jar" extension.

However, some ".jar" files as well as ".properties" files are repeated in different directories. So I have decided to only search through 2 folders. Those named lib and those named properties . I'm using Python 2.4

#! /usr/bin/env python

import os
import sys
import glob

java_command = "/myapps/java/home/bin/java -classpath "

def any(s):
    for v in s:
        if v:
            return True
    return False


def run(project_dir, main_class, specific_args):
    classpath = []

    for root, dirs, files in os.walk(project_dir):
        classpath.extend(os.path.join(root, f) for f in files if f.endswith('.jar'))
        if any(f.endswith('.properties') for f in files):
            classpath.append(root)

    classpath_augment = ':'.join(classpath)

    args_passed_in = '%s %s %s %s' % (java_command, classpath_augment, main_class, specific_args)
    print args_passed_in
    code = os.system(args_passed_in)
    print code

Answer 1

只需通过生成器comp设置一组对象：

classpath = set(os.path.join(root, f) for f in files if f.endswith('.jar') or f.endswith('.properties'))