marcus/scour
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Rework command-line argument detection, use standard input if -i is not present, use standard output if -o is not present

Browse source 
:
This commit is contained in:
JSCHILL1 2009-04-03 19:58:33 -05:00
parent 713228b6c0
commit f3dd5cccb9
1 changed files with 26 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

60
scour.py
View file

@ -22,6 +22,8 @@
# TODOs:
#
# 4) Accept file from stdin if no -i arg present
# 5) Write file to stdout if no -o present
# 6) Read input file into memory using an XML library
# 7) Implement a function that will remove all unreferenced id attributes from
# from an SVG document (xlink:href="#someid", fill="url(#someid)", etc)
@ -34,44 +36,40 @@ APP = 'Scour'
VER = '0.01'
COPYRIGHT = 'Copyright Jeff Schiller, 2009'
print APP + ' ' + VER
print APP , VER
print COPYRIGHT
# parse command-line arguments
args = sys.argv[1:]
if( len(args) != 4 ):
print 'Error! Invalid number of arguments'
input = sys.stdin
output = sys.stdout
if( len(args) == 2):
if( args[0] == '-i' ):
input = open(args[1], 'r')
elif( args[0] == '-o' ):
output = open(args[1], 'w')
else:
sys.stderr.write('Invalid syntax\n')
quit()
elif( len(args) == 4 ):
if( args[0] == '-i' and args[2] == '-o' ):
input = open(args[1], 'r')
output = open(args[3], 'w')
elif( args[0] == '-o' and args[2] == 'i' ):
output = open(args[1], 'w')
input = open(args[3], 'r')
else:
sys.stderr.write('Invalid syntax\n')
quit()
elif( len(args) != 0 ):
sys.stderr.write('Invalid syntax\n')
quit()
infile = ''
outfile = ''
# Get input file
if( args[0] == '-i' ):
infile = args[1]
elif( args[2] == '-i' ):
infile = args[3]
# Get output file
if( args[0] == '-o' ):
outfile = args[1]
elif(args[2] == '-o' ):
outfile = args[3]
if( infile == '' ):
print 'Error! -i argument missing'
quit()
if( outfile == '' ):
print 'Error! -o argument is missing'
quit()
# Open input file for reading, throws error if file does not exist
input = open(infile, 'r')
# Open output file for writing, overwriting file if necessary
output = open(outfile, 'w')
# simply write the input to the output for now
output.write(input.read());
# Close input and output files