annotate assembly_post_processor.py @ 46:b7858b1f64f1 draft

Uploaded
author greg
date Wed, 03 May 2017 10:14:03 -0400
parents
children 9d58af3e459d
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
46
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
1 #!/usr/bin/env python
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
2 import argparse
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
3 import os
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
4 import subprocess
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
5
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
6 import utils
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
7
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
8 OUTPUT_DIR = 'assemblyPostProcessor_dir'
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
9
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
10 parser = argparse.ArgumentParser()
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
11 parser.add_argument('--dereplicate', dest='dereplicate', default=None, help='Remove duplicate sequences')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
12 parser.add_argument('--gap_trimming', dest='gap_trimming', type=float, default=0, help='Trim alignments')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
13 parser.add_argument('--gene_family_search', dest='gene_family_search', default=None, help='Targeted gene families')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
14 parser.add_argument('--method', dest='method', default=None, help='Protein clustering method')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
15 parser.add_argument('--min_length', dest='min_length', type=int, default=0, help='Minimum sequence length')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
16 parser.add_argument('--num_threads', dest='num_threads', type=int, help='Number of processors')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
17 parser.add_argument('--prediction_method', dest='prediction_method', help='Coding regions prediction method')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
18 parser.add_argument('--scaffold', dest='scaffold', default=None, help='Gene family scaffold')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
19 parser.add_argument('--score_matrices', dest='score_matrices', default=None, help='Scores matrices')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
20 parser.add_argument('--strand_specific', dest='strand_specific', default=None, help='Strand-specific assembly')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
21 parser.add_argument('--transcripts', dest='transcripts', help='Transcriptome assembly fasta file')
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
22
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
23 args = parser.parse_args()
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
24
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
25 # Build the command line.
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
26 cmd = 'AssemblyPostProcessor'
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
27 if args.dereplicate is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
28 cmd += ' --dereplicate'
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
29 if args.gap_trimming > 0:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
30 cmd += ' --gap_trimming %4f' % args.gap_trimming
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
31 if args.gene_family_search is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
32 cmd += ' --gene_family_search %s' % args.gene_family_search
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
33 if args.method is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
34 cmd += ' --method %s' % args.method
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
35 if args.min_length > 0:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
36 cmd += ' --min_length %d' % args.min_length
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
37 cmd += ' --num_threads %d' % args.num_threads
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
38 cmd += ' --prediction_method %s' % args.prediction_method
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
39 if args.scaffold is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
40 cmd += ' --scaffold %s' % args.scaffold
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
41 if args.score_matrices is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
42 cmd += ' --score_matrices %s' % args.score_matrices
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
43 if args.strand_specific is not None:
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
44 cmd += ' --strand_specific'
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
45 cmd += ' --transcripts %s' % args.transcripts
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
46
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
47 # Run the command.
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
48 proc = subprocess.Popen(args=cmd, stderr=subprocess.PIPE, stdout=subprocess.PIPE, shell=True)
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
49 rc = proc.wait()
b7858b1f64f1 Uploaded
greg
parents:
diff changeset
50 utils.check_execution_errors(rc, proc.stderr, proc.stdout)