Mercurial > repos > mvdbeek > add_input_name_as_column
view add_input_name_as_column.py @ 2:ea4ad41e01fc draft default tip
"planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/add_input_name_as_column commit 9292c57c34283543e86ecc65f805977224f6fc7b"
author | iuc |
---|---|
date | Wed, 25 Mar 2020 11:09:05 +0000 |
parents | 07d2cbf43b51 |
children |
line wrap: on
line source
import argparse import io def Parser(): the_parser = argparse.ArgumentParser(description="add label to last column of file") the_parser.add_argument('--input', required=True, action="store", type=str, help="input tabular file") the_parser.add_argument('--output', required=True, action="store", type=str, help="output file path") the_parser.add_argument('--label', required=True, action="store", type=str, help="label to add in last column") the_parser.add_argument('--header', action="store", type=str, help="column label for last column") the_parser.add_argument('--prepend', action='store_true', default=False, help='Prepend column instead of appending' ) args = the_parser.parse_args() return args args = Parser() with io.open(args.input, encoding="utf-8") as input, io.open(args.output, 'w', encoding="utf-8") as output: for i, line in enumerate(input): line = line.strip('\n') if (i == 0) and args.header: new_entry = args.header else: new_entry = args.label if args.prepend: line = "%s\t%s\n" % (new_entry, line) else: line = "%s\t%s\n" % (line, new_entry) output.write(line)