Mercurial > repos > iuc > data_manager_snpeff
annotate data_manager/data_manager_snpEff_databases.py @ 5:87f6cff1bd43 draft
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit e32db0704e23d58358f749177988d2cb5f01c708
| author | iuc |
|---|---|
| date | Tue, 18 Apr 2017 12:55:49 -0400 |
| parents | a9cace1ee2b7 |
| children | fbb459120c29 |
| rev | line source |
|---|---|
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
1 #!/usr/bin/env python |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
2 import json |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
3 import optparse |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
4 import os |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
5 import subprocess |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
6 import sys |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
7 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
8 |
|
4
a9cace1ee2b7
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 5c6d595ecbf2e4e39c25662a165c9e475e01ecd1-dirty
iuc
parents:
3
diff
changeset
|
9 def fetch_databases(data_manager_dict, target_directory): |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
10 if not os.path.exists(target_directory): |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
11 os.makedirs(target_directory) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
12 databases_path = os.path.join( target_directory, 'databases.out' ) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
13 databases_output = open(databases_path, 'w') |
|
4
a9cace1ee2b7
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 5c6d595ecbf2e4e39c25662a165c9e475e01ecd1-dirty
iuc
parents:
3
diff
changeset
|
14 args = ['snpEff', 'databases'] |
|
a9cace1ee2b7
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 5c6d595ecbf2e4e39c25662a165c9e475e01ecd1-dirty
iuc
parents:
3
diff
changeset
|
15 return_code = subprocess.call(args=args, shell=False, stdout=databases_output.fileno()) |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
16 if return_code: |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
17 sys.exit( return_code ) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
18 databases_output.close() |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
19 data_manager_dict['data_tables'] = data_manager_dict.get( 'data_tables', {} ) |
|
5
87f6cff1bd43
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit e32db0704e23d58358f749177988d2cb5f01c708
iuc
parents:
4
diff
changeset
|
20 data_manager_dict['data_tables']['snpeffv_databases'] = data_manager_dict['data_tables'].get( 'snpeffv_databases', [] ) |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
21 data_table_entries = [] |
|
4
a9cace1ee2b7
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 5c6d595ecbf2e4e39c25662a165c9e475e01ecd1-dirty
iuc
parents:
3
diff
changeset
|
22 with open(databases_path, 'r') as fh: |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
23 for i, line in enumerate(fh): |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
24 fields = line.split('\t') |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
25 if len(fields) >= 2: |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
26 genome_version = fields[0].strip() |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
27 if genome_version.startswith("Genome") or genome_version.startswith("-"): |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
28 continue |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
29 # snpeff test genome |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
30 if genome_version == '30c2c903' or fields[1].strip() == 'TestCase' or fields[1].strip().startswith('Test_'): |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
31 continue |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
32 description = fields[1].strip() + ' : ' + genome_version |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
33 data_table_entries.append(dict(value=genome_version, name=description)) |
|
5
87f6cff1bd43
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit e32db0704e23d58358f749177988d2cb5f01c708
iuc
parents:
4
diff
changeset
|
34 data_manager_dict['data_tables']['snpeffv_databases'] = data_table_entries |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
35 return data_manager_dict |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
36 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
37 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
38 def main(): |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
39 parser = optparse.OptionParser() |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
40 (options, args) = parser.parse_args() |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
41 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
42 filename = args[0] |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
43 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
44 params = json.loads( open( filename ).read() ) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
45 target_directory = params[ 'output_data' ][0]['extra_files_path'] |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
46 os.mkdir( target_directory ) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
47 data_manager_dict = {} |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
48 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
49 # Create Defuse Reference Data |
|
4
a9cace1ee2b7
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 5c6d595ecbf2e4e39c25662a165c9e475e01ecd1-dirty
iuc
parents:
3
diff
changeset
|
50 data_manager_dict = fetch_databases( data_manager_dict, target_directory) |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
51 |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
52 # save info to json file |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
53 open( filename, 'wb' ).write( json.dumps( data_manager_dict ) ) |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
54 |
|
3
821c34cdacb6
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_snpeff commit 8652f36a3a3838dca989426961561e81432acf4f
iuc
parents:
0
diff
changeset
|
55 |
|
0
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
56 if __name__ == "__main__": |
|
111fc9457ec6
planemo upload for repository https://github.com/galaxyproject/tools-devteam/tree/master/data_managers/data_manager_snpeff commit 88c982c5adcd32b11d98428fc554a4fdfcc19584
iuc
parents:
diff
changeset
|
57 main() |
