Mercurial > repos > rnateam > viennarna_rnafold

annotate rnafold_SHAPE.py @ 2:84aad99aca1f draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
author rnateam
date Tue, 16 May 2017 16:30:38 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
1 ### overcoming the problem of SHAPE data working with a single line.
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
2 ### creating multiple multiple files containg SHAPE data for a single sequence and running RNAfold for every
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
3 ### single sequence.
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
4
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
5 import os
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
6 import sys
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
7 from os import system
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
8 from Bio import SeqIO
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
9 import re
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
10 from subprocess import Popen, PIPE
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
11
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
12 params_list = sys.argv[1:]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
13 param_list_no_shape = [s for s in params_list if not "--shape=" in s ]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
14 shape_file = [s for s in params_list if "--shape=" in s ]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
15 assert (len(shape_file) == 1)
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
16
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
17 shape_file = shape_file[0]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
18 shape_file = shape_file.replace('--shape=', '')
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
19
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
20 params_no_shape = " ".join(str(x) for x in param_list_no_shape)
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
21
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
22 pattern = re.compile("^>.*$")
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
23 id_line = ""
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
24 with open(shape_file, 'r') as f:
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
25 content = f.read()
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
26 lines = content.split('\n')
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
27 for line in lines:
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
28 if pattern.match(line):
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
29 id_line = line.split()[0]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
30 id_line = id_line[1:]
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
31 continue
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
32 else:
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
33 with open(id_line +'.tmp', "a") as clFile:
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
34 clFile.write(line + "\n")
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
35
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
36 input_file = sys.stdin
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
37
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
38 for record in SeqIO.parse(input_file, "fasta"):
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
39 seq = ">{}\n{}".format(record.id,record.seq)
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
40 cmd = " RNAfold --shape=" + record.id + '.tmp ' + params_no_shape
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
41 p = Popen(cmd , stdin=PIPE, shell=True, stdout=PIPE, stderr=PIPE)
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
42 out,err = p.communicate(seq.encode())
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
43 if err:
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
44 raise RuntimeError("Error in calling RNAfold\n{}\n{}\n".format(out, err))
84aad99aca1f planemo upload for repository https://github.com/bgruening/galaxytools/tree/master/tools/rna_tools/vienna_rna commit b'bd13ffd1c3e126a6dc59dd3c478347ec1b5824a3\n'
rnateam
parents:
diff changeset
45 print (out.decode('utf-8').strip())