47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def open_files(outdir, i):
|
||
|
|
return (open(os.path.join(outdir, 'test%02d.in' % i), 'w'),
|
||
|
|
open(os.path.join(outdir, 'test%02d.out' % i), 'w'))
|
||
|
|
|
||
|
|
def close_files(input, output):
|
||
|
|
print os.path.basename(input.name), os.path.basename(output.name)
|
||
|
|
input.close()
|
||
|
|
output.close()
|
||
|
|
|
||
|
|
def main():
|
||
|
|
if len(sys.argv) != 3:
|
||
|
|
print 'usage: %s input-file output-directory' % sys.argv[0]
|
||
|
|
return 2
|
||
|
|
|
||
|
|
infile = sys.argv[1]
|
||
|
|
outdir = sys.argv[2]
|
||
|
|
|
||
|
|
if not os.path.exists(outdir):
|
||
|
|
print >>sys.stderr, 'output directory %r does not exist!' % outdir
|
||
|
|
return 1
|
||
|
|
|
||
|
|
i = 0
|
||
|
|
input, output = open_files(outdir, i)
|
||
|
|
current = input
|
||
|
|
|
||
|
|
for line in open(infile):
|
||
|
|
if line == '====\n':
|
||
|
|
current = output
|
||
|
|
elif line == '========\n':
|
||
|
|
close_files(input, output)
|
||
|
|
i += 1
|
||
|
|
input, output = open_files(outdir, i)
|
||
|
|
current = input
|
||
|
|
else:
|
||
|
|
current.write(line)
|
||
|
|
|
||
|
|
close_files(input, output)
|
||
|
|
print >>sys.stderr, "%d test cases" % (i + 1)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
sys.exit(main() or 0)
|