Mercurial > repos > davidvanzessen > combined_immune_repertoire_pipeline
diff igblastmerge.py @ 9:8d83319a0f3d draft
Uploaded
author | davidvanzessen |
---|---|
date | Tue, 10 Dec 2013 05:53:08 -0500 |
parents | 4e3df2384422 |
children |
line wrap: on
line diff
--- a/igblastmerge.py Mon Nov 25 09:21:55 2013 -0500 +++ b/igblastmerge.py Tue Dec 10 05:53:08 2013 -0500 @@ -1,4 +1,3 @@ -import argparse import sys # error def stop_err( msg ): @@ -7,35 +6,45 @@ # main def main(): - parser = argparse.ArgumentParser() #docs.python.org/dev/library/argparse.html - parser.add_argument("--input", help="Input file(s)", nargs="+") - parser.add_argument("--id", help="Input file(s) id's", nargs="+") - parser.add_argument("--output", help="Output file") - - args = parser.parse_args() - try: - o = open(args.output, 'w') - i = open(args.input[-1], 'r') - separator = "\t" - newline = "\n" - header = "Sample" - line = i.readline() - o.write(line[:line.rfind(newline)] + separator + header + newline) #write the header - i.close() - - for cf,i in zip(args.input,args.id): - f = open(cf, 'r') - line = f.readline() - line = f.readline() #skip header - while line: - o.write(line[:line.rfind(newline)] + separator + i + newline) - line = f.readline() - f.close() - o.close() + args = sys.argv[1:-2] - except Exception, ex: - stop_err('Error running new_column.py\n' + str(ex)) - sys.exit(0) + try: + o = open(sys.argv[-1], 'w') + i = open(args[1], 'r') + separator = "\t" + newline = "\n" + line = i.readline() + #write the header + o.write(line[:line.rfind(newline)] + separator + "Sample" + separator + "Replicate" + newline) + i.close() + + current = 1 + sampleID = args[0] + count = 1 + + while True: + f = open(args[current], 'r') + line = f.readline() + line = f.readline() + while line: + o.write(line[:line.rfind(newline)] + separator + sampleID + separator + str(count) + newline) + line = f.readline() + f.close() + + if current >= (len(args) - 1): + break + if args[current + 1].find("/") is -1: + sampleID = args[current + 1] + current += 1 + count = 1 + else: + count += 1 + current += 1 + o.close() + + except Exception, ex: + stop_err('Error running new_column.py\n' + str(ex)) + sys.exit(0) if __name__ == "__main__": print sys.argv