Mercurial > repos > iuc > data_manager_bwameth_index_builder

annotate data_manager/bwameth_index_builder.py @ 5:f95ae6d21db4 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
"planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
author iuc
date Sun, 22 Nov 2020 12:42:43 +0000
parents ecd6bdddcdec
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
1 #!/usr/bin/env python
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
2 # Based heavily on the HISAT2 data manager wrapper
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
3
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
4 import argparse
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
5 import json
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
6 import os
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
7 import subprocess
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
8 import sys
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
9
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
10 DEFAULT_DATA_TABLE_NAME = "bwameth_indexes"
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
11
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
12
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
13 def build_bwameth_index(data_manager_dict, params, args):
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
14 data_table_name = args.data_table_name
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
15 target_directory = params['output_data'][0]['extra_files_path']
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
16 if not os.path.exists(target_directory):
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
17 os.mkdir(target_directory)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
18 fasta_base_name = os.path.basename(args.fasta_filename)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
19 sym_linked_fasta_filename = os.path.join(target_directory, fasta_base_name)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
20 os.symlink(os.path.abspath(args.fasta_filename), sym_linked_fasta_filename)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
21 cmd = ['bwameth.py', 'index', sym_linked_fasta_filename]
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
22 proc = subprocess.Popen(args=cmd, shell=False, cwd=target_directory)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
23 return_code = proc.wait()
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
24 if return_code:
3
d1c0b1f0c67e "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 8b77954efb8d46dfb3bec5421819076f3757ad06"
iuc
parents: 2
diff changeset
25 print("Error building index.", file=sys.stderr)
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
26 sys.exit(return_code)
2
0beac6016b41 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 823259bba9405d22dc2add24746057122c819ad3"
iuc
parents: 0
diff changeset
27 data_table_entry = dict(value=args.dbkey, dbkey=args.dbkey, name=args.name, path=sym_linked_fasta_filename)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
28 _add_data_table_entry(data_manager_dict, data_table_name, data_table_entry)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
29
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
30
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
31 def _add_data_table_entry(data_manager_dict, data_table_name, data_table_entry):
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
32 data_manager_dict['data_tables'] = data_manager_dict.get('data_tables', {})
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
33 data_manager_dict['data_tables'][data_table_name] = data_manager_dict['data_tables'].get(data_table_name, [])
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
34 data_manager_dict['data_tables'][data_table_name].append(data_table_entry)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
35 return data_manager_dict
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
36
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
37
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
38 def main():
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
39 # Parse Command Line
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
40 parser = argparse.ArgumentParser()
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
41 parser.add_argument('--output', default=None)
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
42 parser.add_argument('--fasta_filename', default=None)
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
43 parser.add_argument('--dbkey', default=None)
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
44 parser.add_argument('--name', default=None)
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
45 parser.add_argument('--description', default=None)
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
46 parser.add_argument('--data_table_name', default=DEFAULT_DATA_TABLE_NAME)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
47 args = parser.parse_args()
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
48
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
49 filename = args.output
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
50 with open(filename) as fh:
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
51 params = json.load(fh)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
52 data_manager_dict = {}
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
53
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
54 if args.dbkey in [None, '', '?']:
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
55 raise Exception('"%s" is not a valid dbkey. You must specify a valid dbkey.' % (args.dbkey))
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
56
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
57 # build the index
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
58 build_bwameth_index(data_manager_dict, params, args)
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
59
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
60 # save info to json file
5
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
61 with open(filename, 'w') as fh:
f95ae6d21db4 "planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 02d2967f77e3fa5a18aea63dc84aa9ab418dc165"
iuc
parents: 4
diff changeset
62 json.dump(data_manager_dict, fh, sort_keys=True)
0
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
63
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
64
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
65 if __name__ == "__main__":
fa17303fc8de planemo upload for repository https://github.com/galaxyproject/tools-iuc/data_managers/data_manager_bwameth_index_builder commit 30192223eeb60f33a42046921351c6ba3c80c90c
iuc
parents:
diff changeset
66 main()