#!/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()