Use the with statement to simplify the build script.

llvm-svn: 132221
This commit is contained in:
Johnny Chen
2011-05-27 21:58:22 +00:00
parent 041ec4aada
commit bf9cf5ec1e

View File

@@ -20,52 +20,30 @@
# That's what this python script does.
#
import os
include_python = '#include <Python.h>'
include_python_ifdef = '''#if defined (__APPLE__)
#include <Python/Python.h>
#else
#include <Python.h>
#endif
'''
input_dir_name = os.environ["SRCROOT"]
full_input_name = input_dir_name + "/source/LLDBWrapPython.cpp"
full_output_name = full_input_name + ".edited"
try:
f_in = open (full_input_name, 'r')
except IOError:
print "Error: Unable to open file for reading: " + full_input_name
else:
try:
f_out = open (full_output_name, 'w')
except IOError:
print "Error: Unable to open file for writing: " + full_output_name
else:
include_line_found = False
with open(full_input_name, 'r') as f_in:
with open(full_output_name, 'w') as f_out:
include_python_found = False
for line in f_in:
if not include_python_found:
if line.startswith(include_python):
# Write out the modified lines.
f_out.write(include_python_ifdef)
include_python_found = True
continue
try:
line = f_in.readline()
except IOError:
print "Error occurred while reading file."
else:
while line:
#
#
if not include_line_found:
if (line.find ("#include <Python.h>") == 0):
f_out.write ("#if defined (__APPLE__)\n");
f_out.write ("#include <Python/Python.h>\n");
f_out.write ("#else\n");
f_out.write (line);
f_out.write ("#endif\n");
include_line_found = True
else:
f_out.write (line)
else:
f_out.write (line)
try:
line = f_in.readline()
except IOError:
print "Error occurred while reading file."
try:
f_in.close()
f_out.close()
except:
print "Error occurred while closing files"
# Otherwise, copy the line verbatim to the output file.
f_out.write(line)