Mercurial > repos > jjohnson > ensembl_variant_report

annotate ensembl_variant_report.py @ 2:bf2687e2511e draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
author jjohnson
date Fri, 06 Jan 2017 16:28:35 -0500
parents c3a9e63e8c51
children 9e83cc05d384
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
1 #!/usr/bin/env python
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
2 """
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
3 Report variants to Ensembl Transcripts
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
4
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
5 FrameShift report line
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
6 # Gene Variant Position Reference Variation Prevalence Sequencing Depth Transcript AA Position AA change AA Length Stop Codon Stop Region AA Variation
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
7 TIGD6 chr5:149374879 AAG AG 1.00 13 ENSG00000164296|ENST00000296736 345 Q344 522 A-TGA-T KRWTSSRPST*
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
8
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
9 MissSense report line
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
10 # Gene Variant Position Reference Variation Prevalence Sequencing Depth Transcript AA Position AA change AA Length Stop Codon Stop Region AA Variation
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
11 FN1 chr2:216235089 G A 1.00 7394 ENSG00000115414|ENST00000354785 2261 V2261I 2478 G-TAA TGLTRGATYN_I_IVEALKDQQR
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
12 """
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
13 import sys
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
14 import os.path
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
15 import re
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
16 import time
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
17 import optparse
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
18 from ensemblref import EnsemblRef
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
19 from Bio.Seq import reverse_complement, translate
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
20
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
21
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
22 def __main__():
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
23 #Parse Command Line
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
24 parser = optparse.OptionParser()
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
25 #I/O
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
26 parser.add_option( '-i', '--input', dest='input', default=None, help='Tabular file with peptide_sequence column' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
27 parser.add_option( '-s', '--format', dest='format', default='tabular', choices=['tabular','snpeff'], help='Tabular file with peptide_sequence column' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
28 #Columns for tabular input
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
29 parser.add_option( '-C', '--chrom_column', type='int', dest='chrom_column', default=1, help='column ordinal with Ensembl transctip ID' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
30 parser.add_option( '-P', '--pos_column', type='int', dest='pos_column', default=2, help='column ordinal with Ensembl transctip ID' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
31 parser.add_option( '-R', '--ref_column', type='int', dest='ref_column', default=3, help='column ordinal with Ensembl transctip ID' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
32 parser.add_option( '-A', '--alt_column', type='int', dest='alt_column', default=4, help='column ordinal with Ensembl transctip ID' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
33 parser.add_option( '-T', '--transcript_column', type='int', dest='transcript_column', default=1, help='column ordinal with Ensembl transctip ID' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
34 parser.add_option( '-F', '--dpr_column', type='int', dest='dpr_column', default=1, help='column with VCF: DPR or AD' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
35 parser.add_option( '-D', '--dp_column', type='int', dest='dp_column', default=1, help='column with VCF: DP' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
36 parser.add_option( '-g', '--gene_model', dest='gene_model', default=None, help='GTF gene model file. Used to annotate NSJ peptide entries.')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
37 parser.add_option( '-2', '--twobit', dest='twobit', default=None, help='Reference genome in UCSC twobit format')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
38 #Output file
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
39 parser.add_option( '-o', '--output', dest='output', default=None, help='The output report (else write to stdout)' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
40 #filters
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
41 parser.add_option( '-d', '--min_depth', type='int', dest='min_depth', default=None, help='Minimum read depth to report' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
42 parser.add_option( '-f', '--min_freq', type='float', dest='min_freq', default=None, help='Minimum variant frequency to report' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
43 #peptide options
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
44 parser.add_option( '-l', '--leading_aa', type='int', dest='leading_aa', default=10, help='Number AAs before missense variant' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
45 parser.add_option( '-t', '--trailing_aa', type='int', dest='trailing_aa', default=10, help='Number AAs after missense variant' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
46 parser.add_option( '-r', '--readthrough', type='int', dest='readthrough', default=0, help='' )
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
47 #
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
48 parser.add_option('--debug', dest='debug', action='store_true', default=False, help='Print debugging messages')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
49 (options, args) = parser.parse_args()
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
50
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
51 ##INPUTS##
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
52 if options.input != None:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
53 try:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
54 inputPath = os.path.abspath(options.input)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
55 inputFile = open(inputPath, 'r')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
56 except Exception, e:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
57 print >> sys.stderr, "failed: %s" % e
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
58 exit(2)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
59 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
60 inputFile = sys.stdin
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
61
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
62 if options.output != None:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
63 try:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
64 outputPath = os.path.abspath(options.output)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
65 outputFile = open(outputPath, 'w')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
66 except Exception, e:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
67 print >> sys.stderr, "failed: %s" % e
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
68 exit(3)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
69 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
70 outputFile = sys.stdout
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
71
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
72 def parse_tabular():
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
73 ci = options.chrom_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
74 pi = options.pos_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
75 ri = options.ref_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
76 ai = options.alt_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
77 ti = options.transcript_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
78 di = options.dp_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
79 fi = options.dpr_column - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
80 for linenum,line in enumerate(inputFile):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
81 if options.debug:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
82 print >> sys.stderr, "%d: %s\n" % (linenum,line)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
83 if line.startswith('#'):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
84 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
85 if line.strip() == '':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
86 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
87 fields = line.rstrip('\r\n').split('\t')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
88 transcript = fields[ti]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
89 if not transcript:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
90 print >> sys.stderr, "%d: %s\n" % (linenum,line)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
91 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
92 chrom = fields[ci]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
93 pos = int(fields[pi])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
94 ref = fields[ri]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
95 alt = fields[ai]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
96 dp = int(fields[di])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
97 dpr = [int(x) for x in fields[fi].split(',')]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
98 yield (transcript,pos,ref,alt,dp,dpr)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
99
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
100 def parse_snpeff_vcf():
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
101 for linenum,line in enumerate(inputFile):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
102 if line.startswith('##'):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
103 if line.find('SnpEffVersion=') > 0:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
104 SnpEffVersion = re.search('SnpEffVersion="?(\d+\.\d+)',line).groups()[0]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
105 elif line.startswith('#CHROM'):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
106 pass
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
107 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
108 fields = line.strip('\r\n').split('\t')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
109 if options.debug: print >> sys.stderr, "\n%s" % (fields)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
110 (chrom, pos, id, ref, alts, qual, filter, info) = fields[0:8]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
111 pos = int(pos)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
112 qual = float(qual)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
113 for info_item in info.split(';'):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
114 if info_item.find('=') < 0: continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
115 (key, val) = info_item.split('=', 1)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
116 if key == 'DP':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
117 dp = int(val)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
118 if key == 'DPR':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
119 dpr = dpr = [int(x) for x in val.split(',')]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
120 if key in ['EFF','ANN']:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
121 for effect in val.split(','):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
122 if options.debug: print >> sys.stderr, "\n%s" % (effect.split('|'))
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
123 if key == 'ANN':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
124 (alt,eff,impact,gene_name,gene_id,feature_type,transcript,biotype,exon,c_hgvs,p_hgvs,cdna,cds,aa,distance,info) = effect.split('|')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
125 elif key == 'EFF':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
126 (eff, effs) = effect.rstrip(')').split('(')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
127 (impact, functional_class, codon_change, aa_change, aa_len, gene_name, biotype, coding, transcript, exon, alt) = effs.split('|')[0:11]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
128 yield (transcript,pos,ref,alt,dp,dpr)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
129
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
130
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
131 #Process gene model
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
132 ens_ref = None
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
133 if options.gene_model != None:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
134 try:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
135 geneModelFile = os.path.abspath(options.gene_model)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
136 twoBitFile = os.path.abspath(options.twobit)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
137 print >> sys.stderr, "Parsing ensembl ref: %s %s" % (options.gene_model,options.twobit)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
138 time1 = time.time()
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
139 ens_ref = EnsemblRef(geneModelFile,twoBitFile)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
140 time2 = time.time()
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
141 print >> sys.stderr, "Parsing ensembl ref: %d seconds" % (int(time2-time1))
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
142 except Exception, e:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
143 print >> sys.stderr, "Parsing gene model failed: %s" % e
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
144 exit(2)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
145 try:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
146 parse_input = parse_tabular if options.format == 'tabular' else parse_snpeff_vcf
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
147 for tid,pos1,ref,alt,dp,dpr in parse_input():
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
148 freq = float(dpr[1])/float(sum(dpr)) if dpr else None
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
149 if options.min_depth and dp is not None and dp < options.min_depth:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
150 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
151 if options.min_freq and freq is not None and freq < options.min_freq:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
152 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
153 ## transcript_id, pos, ref, alt, dp, dpr
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
154 tx = ens_ref.get_gtf_transcript(tid)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
155 coding = ens_ref.transcript_is_coding(tid)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
156 if not coding:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
157 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
158 frame_shift = len(ref) != len(alt)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
159 cds = ens_ref.get_cds(tid)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
160 pos0 = pos1 - 1 # zero based position
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
161 spos = pos0 if tx.gene.strand else pos0 + len(ref) - 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
162 alt_seq = alt if tx.gene.strand else reverse_complement(alt)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
163 cds_pos = ens_ref.genome_to_cds_pos(tid, spos)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
164 alt_cds = cds[:cds_pos] + alt_seq + cds[cds_pos+len(ref):]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
165 offset = 0
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
166 if tx.gene.strand:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
167 for i in range(min(len(ref),len(alt))):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
168 if ref[i] == alt[i]:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
169 offset = i
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
170 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
171 break
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
172 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
173 for i in range(-1,-min(len(ref),len(alt)) -1,-1):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
174 if ref[i] == alt[i]:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
175 offset = i
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
176 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
177 break
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
178 refpep = translate(cds[:len(cds)/3*3])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
179 pep = translate(alt_cds[:len(alt_cds)/3*3])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
180 peplen = len(pep)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
181 aa_pos = (cds_pos + offset) / 3
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
182 if aa_pos >= len(pep):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
183 print >> sys.stderr, "%d: %s\n" % (linenum,line)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
184 continue
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
185 if frame_shift:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
186 #find stop_codons
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
187 nstops = 0
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
188 stop_codons = []
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
189 for i in range(aa_pos,peplen):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
190 if refpep[i] != pep[i]:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
191 aa_pos = i
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
192 break
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
193 for i in range(aa_pos,peplen):
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
194 if pep[i] == '*':
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
195 nstops += 1
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
196 stop_codons.append("%s-%s%s" % (alt_cds[i*3-1],alt_cds[i*3:i*3+3],"-%s" % alt_cds[i*3+4] if len(alt_cds) > i*3 else ''))
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
197 if nstops > options.readthrough:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
198 reported_peptide = pep[aa_pos:i+1]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
199 reported_stop_codon = ','.join(stop_codons)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
200 break
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
201 else:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
202 reported_stop_codon = "%s-%s" % (alt_cds[peplen*3-4],alt_cds[peplen*3-3:peplen*3])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
203 reported_peptide = "%s_%s_%s" % (pep[max(aa_pos-options.leading_aa,0):aa_pos],
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
204 pep[aa_pos],
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
205 pep[aa_pos+1:min(aa_pos+1+options.trailing_aa,len(pep))])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
206 cs_pos = aa_pos * 3
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
207 aa_pos = (cds_pos + offset) / 3
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
208 ref_codon = cds[cs_pos:cs_pos+3]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
209 ref_aa = translate(ref_codon)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
210 alt_codon = alt_cds[cs_pos:cs_pos+3]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
211 alt_aa = translate(alt_codon)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
212 gene = tx.gene.names[0]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
213 report_fields = [tx.gene.names[0],
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
214 '%s:%d %s' % (tx.gene.contig,pos1,'+' if tx.gene.strand else '-'),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
215 ref,
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
216 alt,
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
217 "%1.2f" % freq if freq is not None else '',
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
218 str(dp),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
219 "%s|%s" % (tx.gene.gene_id,tx.cdna_id),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
220 "%d" % (aa_pos + 1),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
221 "%s%d%s" % (ref_aa,aa_pos + 1,alt_aa),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
222 "%d" % len(pep),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
223 reported_stop_codon,
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
224 reported_peptide
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
225 ]
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
226 if options.debug:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
227 report_fields.append("%d %d %d %d %s %s" % (cds_pos, offset, cs_pos,aa_pos,ref_codon,alt_codon))
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
228 outputFile.write('\t'.join(report_fields))
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
229 if options.debug:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
230 print >> sys.stderr, "%s %s\n%s\n%s\n%s %s" % (
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
231 cds[cs_pos-6:cs_pos], cds[cs_pos:cs_pos+15],
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
232 translate(cds[cs_pos-6:cs_pos+15]),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
233 translate(alt_cds[cs_pos-6:cs_pos+15]),
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
234 alt_cds[cs_pos-6:cs_pos], alt_cds[cs_pos:cs_pos+15])
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
235 outputFile.write('\n')
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
236 except Exception, e:
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
237 print >> sys.stderr, "failed: %s" % e
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
238 exit(1)
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
239
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
240
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
241 if __name__ == "__main__" : __main__()
c3a9e63e8c51 planemo upload for repository https://github.com/jj-umn/galaxytools/tree/master/ensembl_variant_report commit 239c1ee096e5fc3e2e929f7bf2d4afba5c677d4b-dirty
jjohnson
parents:
diff changeset
242