Add -I support.

2 files changed, 34 insertions, 8 deletions

diff --git a/src/scons.py b/src/scons.py
index 500910d..f515bea 100644
--- a/src/scons.py
+++ b/src/scons.py
@@ -59,6 +59,7 @@ class Taskmaster:
 local_help = None
 num_jobs = 1
 Scripts = []
+include_dirs = []
 
 # utility functions
 
@@ -315,7 +316,11 @@ def options_init():
 	short = 'i', long = ['ignore-errors'],
 	help = "Ignore errors from build actions.")
 
-    Option(func = opt_not_yet,
+    def opt_I(opt, arg):
+	global include_dirs
+	include_dirs = include_dirs + [arg]
+
+    Option(func = opt_I,
 	short = 'I', long = ['include-dir'], arg = 'DIRECTORY',
 	help = "Search DIRECTORY for imported Python modules.")
 
@@ -504,6 +509,8 @@ def main():
     #
     #sys.path = dirlist
 
+    sys.path = include_dirs + sys.path
+
     # initialize node factory
     init()
 
diff --git a/test/option--I.py b/test/option--I.py
index f205a76..80de467 100644
--- a/test/option--I.py
+++ b/test/option--I.py
@@ -10,17 +10,36 @@ test = TestCmd.TestCmd(program = 'scons.py',
                        workdir = '',
                        interpreter = 'python')
 
-test.write('SConstruct', "")
+test.subdir('sub1', 'sub2')
 
-test.run(chdir = '.', arguments = '-I foo')
+test.write(['sub1', 'foo.py'], """
+variable = "sub1/foo"
+""")
 
-test.fail_test(test.stderr() !=
-		"Warning:  the -I option is not yet implemented\n")
+test.write(['sub2', 'foo.py'], """
+variable = "sub2/foo"
+""")
 
-test.run(chdir = '.', arguments = '--include-dir=foo')
+test.write(['sub2', 'bar.py'], """
+variable = "sub2/bar"
+""")
 
-test.fail_test(test.stderr() !=
-		"Warning:  the --include-dir option is not yet implemented\n")
+test.write('SConstruct', """
+import foo
+print foo.variable
+import bar
+print bar.variable
+""")
+
+test.run(chdir = '.', arguments = '-I sub1 -I sub2')
+
+test.fail_test(test.stdout() != "sub1/foo\nsub2/bar\n")
+test.fail_test(test.stderr() != "")
+
+test.run(chdir = '.', arguments = '--include-dir=sub2 --include-dir=sub1')
+
+test.fail_test(test.stdout() != "sub2/foo\nsub2/bar\n")
+test.fail_test(test.stderr() != "")
 
 test.pass_test()