From 2cee53f8fbfed4af0aa9fb19fce383b5fff0fff9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Boris=20Juri=C4=8D?= <499542@mail.muni.cz> Date: Fri, 1 Apr 2022 12:35:37 +0000 Subject: [PATCH] Update src/poly_a.py --- src/poly_a.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/poly_a.py b/src/poly_a.py index b045744..6ca8baf 100644 --- a/src/poly_a.py +++ b/src/poly_a.py @@ -72,13 +72,12 @@ def generate_poly_a( def main(): - """Append poly A tails to fasta sequences and write to file - - This function iterates over `input` fasta sequences from a specified file, - appends tails of given `length` and composition specified by `weights`, + """Append poly A tails to fasta sequences and write to file. + + This function iterates over `input` fasta sequences from a specified file, + appends tails of given `length` and composition specified by `weights`, and writes them to file specified as `output`. """ - parser = argparse.ArgumentParser(description='Process some integers.') parser.add_argument('-i', '--input', required=True, help='Fasta file to add tails to') @@ -91,7 +90,7 @@ def main(): args = parser.parse_args() new_records = [] - for fasta in SeqIO.parse(open(args.input),'fasta'): + for fasta in SeqIO.parse(open(args.input), 'fasta'): new_seq = fasta.seq + Seq(generate_poly_a(args.length, args.weights)) new_records.append(SeqRecord(new_seq, id=fasta.id, description=fasta.description)) SeqIO.write(new_records, args.output, "fasta") -- GitLab