python中的walk目录,用于连接classpath变量

时间:2013-09-18 20:34:57

标签: python python-2.7 os.walk

我编写了一个脚本来使用python连接一个名为classpath_augment的变量。我能够成功地将目录和包含的jar文件连接到classpath_augment变量,但是,我还需要将包含.properties文件的目录添加到类路径变量中。

我该怎么办?
以下是我的代码:

#! /usr/bin/env python

import os
import sys
import glob

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

def run(project_dir, main_class, specific_args):

        classpath_augment = ""

        for r, d, f in os.walk(project_dir):
                for files in f:
                        if (files.endswith(".jar")):
                                classpath_augment += os.path.join(r, files)+":"

        if (classpath_augment[-1] == ":"):
                classpath_augment = classpath_augment[:-1]

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

1 个答案:

答案 0 :(得分:0)

只需查看.properties个文件:

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)

    print java_command, classpath_augment, main_class, specific_args

我冒昧地简化了你的代码;使用列表首先收集所有类路径路径,然后使用str.join()创建最终字符串。这比逐个连接每个新路径更快。

如果您使用的是非常旧的Python版本且any()尚不可用,请使用for循环:

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

    for root, dirs, files in os.walk(project_dir):
        has_properties = False
        for f in files:
            if f.endswith('.jar'):
                classpath.append(os.path.join(root, f))
            if f.endswith('.properties'):
                has_properties = True
        if has_properties:
            classpath.append(root)

    classpath_augment = ':'.join(classpath)

    print java_command, classpath_augment, main_class, specific_args