55 lines
1.5 KiB
Python
Executable File
55 lines
1.5 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# Copyright (c) 2009 Petri Lehtinen <petri@digip.org>
|
|
#
|
|
# Jansson is free software; you can redistribute it and/or modify
|
|
# it under the terms of the MIT license. See LICENSE for details.
|
|
|
|
import os
|
|
import sys
|
|
|
|
def open_files(outdir, i, name):
|
|
basename = '%02d_%s' % (i, name)
|
|
print basename
|
|
input_path = os.path.join(outdir, basename + '.in')
|
|
output_path = os.path.join(outdir, basename + '.out')
|
|
return open(input_path, 'w'), open(output_path, 'w')
|
|
|
|
def main():
|
|
if len(sys.argv) != 3:
|
|
print 'usage: %s input-file output-directory' % sys.argv[0]
|
|
return 2
|
|
|
|
infile = os.path.normpath(sys.argv[1])
|
|
outdir = os.path.normpath(sys.argv[2])
|
|
|
|
if not os.path.exists(outdir):
|
|
print >>sys.stderr, 'output directory %r does not exist!' % outdir
|
|
return 1
|
|
|
|
n = 0
|
|
current = None
|
|
input, output = None, None
|
|
|
|
for line in open(infile):
|
|
if line.startswith('==== '):
|
|
n += 1
|
|
if input is not None and output is not None:
|
|
input.close()
|
|
output.close()
|
|
input, output = open_files(outdir, n, line[5:line.find(' ====\n')])
|
|
current = input
|
|
elif line == '====\n':
|
|
current = output
|
|
else:
|
|
current.write(line)
|
|
|
|
if input is not None and output is not None:
|
|
input.close()
|
|
output.close()
|
|
|
|
print >>sys.stderr, "%s: %d test cases" % (infile, n)
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main() or 0)
|