load_dir.py

来自「boost库提供标准的C++ API 配合dev c++使用,功能更加强大」· Python 代码 · 共 59 行

#!/usr/bin/python

"""
Traverses a directory and output the code that would
create the same directory structure during testing.
Assumes that the instance of Tester is called 't'. 
"""

import sys
import os
import stat
import string

def usage():
    print "usage: load_dir.py directory"

def remove_first_component(path):
    result = [path]
    while 1:
        s = os.path.split(result[0])
        if not s[0]:
            break
        result[:1] = list(s)
    return apply(os.path.join, result[1:])
    
    

def create_file(arg, dirname, fnames):
    for n in fnames:
        path = os.path.join(dirname, n)
        if not os.path.isdir(path):
            print "t.write(\"%s\", \"\"\"" % (remove_first_component(path),),
            f = open(path, "r")
            for l in f:
                print l,
            print '\n""")\n'

def main():
    if len(sys.argv) != 2:
        usage()
    else:
        path = sys.argv[1]

        if not os.access(path, os.F_OK):
            print "Path '%s' does not exist" % (path,)
            sys.exit(1)

        if not os.path.isdir(path):
            print "Path '%s' is not a directory" % (path,)
        
        os.path.walk(path, create_file, None)

if __name__ == '__main__':
    main()