Source code for scripts.join_sdp

#!/usr/bin/env python3

import argparse
import glob
import os
import sys

desc = """Combines several SDP parsed files to one.

[docs]def main(args): lines = [args.prefix + args.header + "\n"] for pattern in args.filenames: filenames = sorted(glob.glob(pattern)) if not filenames: raise IOError("Not found: " + pattern) for filename in filenames: base = os.path.basename(os.path.splitext(filename)[0]) lines.append(args.prefix + base + "\n") with open(filename, encoding="utf-8") as f: lines += f.readlines() f = sys.stdout if args.outfile is None else open(args.outfile, "w", encoding="utf-8") f.writelines(lines) if args.outfile is not None: f.close()
if __name__ == '__main__': argparser = argparse.ArgumentParser(description=desc) argparser.add_argument("filenames", nargs="+", help="SDP file names to join") argparser.add_argument("-o", "--outfile", help="output filename (standard output if unspecified)") argparser.add_argument("-H", "--header", default="SDP 2015", help="first line in the file, not including prefix") argparser.add_argument("-p", "--prefix", default="#", help="prefix for comment lines") main(argparser.parse_args())