| 1 | #!/usr/bin/env python
|
| 2 | """
|
| 3 | csv_concat.py
|
| 4 | """
|
| 5 |
|
| 6 | import sys
|
| 7 |
|
| 8 |
|
| 9 | def main(argv):
|
| 10 | first_header = None
|
| 11 | for path in argv[1:]:
|
| 12 | with open(path) as f:
|
| 13 | # Assume there's no quoting or escaping.
|
| 14 | header = f.readline()
|
| 15 | if first_header is None:
|
| 16 | sys.stdout.write(header)
|
| 17 | first_header = header.strip()
|
| 18 | else:
|
| 19 | h = header.strip()
|
| 20 | if h != first_header:
|
| 21 | raise RuntimeError(
|
| 22 | 'Invalid header in %r: %r. Expected %r' % (path, h,
|
| 23 | first_header))
|
| 24 | # Now print rest of lines
|
| 25 | for line in f:
|
| 26 | sys.stdout.write(line)
|
| 27 |
|
| 28 |
|
| 29 |
|
| 30 | if __name__ == '__main__':
|
| 31 | try:
|
| 32 | main(sys.argv)
|
| 33 | except RuntimeError as e:
|
| 34 | print >>sys.stderr, 'FATAL: %s' % e
|
| 35 | sys.exit(1)
|