annotate test-data/util/shrink_tabix.py @ 4:ee21f3568be3 draft default tip

planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit f7bdf08922aaf4119aefe7041e754a69cf64aebd
author iuc
date Wed, 13 Jul 2022 15:14:18 +0000
parents 541cb8265fe4
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
1 from __future__ import print_function
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
2
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
3 import argparse
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
4
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
5 import pysam
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
6
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
7
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
8 def main(infile, ofile, region):
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
9 print(infile, '->', ofile)
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
10 with pysam.Tabixfile(infile) as i:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
11 fformat = i.format.lower()
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
12 if fformat == 'sam':
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
13 fformat = 'bed'
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
14 if ofile[-3:] == '.gz':
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
15 ofile = ofile[:-3]
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
16 with open(ofile, 'w') as o:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
17 try:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
18 region_it = i.fetch(region=region)
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
19 except ValueError:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
20 region_it = i.fetch(region='chr' + region)
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
21 for line in i.header:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
22 o.write(line + '\n')
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
23 for line in region_it:
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
24 o.write(str(line) + '\n')
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
25 pysam.tabix_index(ofile, preset=fformat, force=True)
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
26
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
27
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
28 if __name__ == '__main__':
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
29 p = argparse.ArgumentParser()
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
30 p.add_argument('infile')
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
31 p.add_argument(
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
32 '-r', '--region',
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
33 required=True,
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
34 help='the region of the input file to rewrite'
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
35 )
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
36 p.add_argument(
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
37 '-o', '--ofile',
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
38 required=True,
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
39 help="the name of the output file"
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
40 )
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
41 args = vars(p.parse_args())
541cb8265fe4 planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/gemini commit 62ed732cba355e695181924a8ed4cce49ca21c59
iuc
parents:
diff changeset
42 main(**args)