annotate data_manager/data_manager_refgenie_pull.py @ 4:c60d0cbdc01a draft

"planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/main/data_managers/data_manager_refgenie_pull commit 2b81f6b9cadf15796ef7f1f6a0d12f05d3f405d7"
author blankenberglab
date Fri, 01 Oct 2021 14:24:12 +0000
parents 659bfadcd8c6
children 0445344b7c3f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
1 #!/usr/bin/env python
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
2
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
3
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
4 import argparse
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
5 import logging
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
6 from base64 import urlsafe_b64encode
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
7 from urllib.parse import urljoin
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
8
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
9 import refgenconf
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
10 import requests
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
11
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
12
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
13 log = logging.getLogger("tools.iuc.data_managers.data_manager_refgenie_pull")
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
14
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
15
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
16 def galaxy_code_get_refgenie_assets(refgenie_config_file):
4
c60d0cbdc01a "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/main/data_managers/data_manager_refgenie_pull commit 2b81f6b9cadf15796ef7f1f6a0d12f05d3f405d7"
blankenberglab
parents: 0
diff changeset
17 try:
c60d0cbdc01a "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/main/data_managers/data_manager_refgenie_pull commit 2b81f6b9cadf15796ef7f1f6a0d12f05d3f405d7"
blankenberglab
parents: 0
diff changeset
18 rgc = refgenconf.RefGenConf(refgenie_config_file, writable=False, skip_read_lock=True)
c60d0cbdc01a "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/main/data_managers/data_manager_refgenie_pull commit 2b81f6b9cadf15796ef7f1f6a0d12f05d3f405d7"
blankenberglab
parents: 0
diff changeset
19 except refgenconf.exceptions.RefgenconfError as e:
c60d0cbdc01a "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/main/data_managers/data_manager_refgenie_pull commit 2b81f6b9cadf15796ef7f1f6a0d12f05d3f405d7"
blankenberglab
parents: 0
diff changeset
20 return[{'name': str(e), 'value': 'ERROR', 'options': [], 'selected': False}]
0
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
21 rval = []
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
22 for urlname, genomes in rgc.listr().items():
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
23 urlname_64 = urlsafe_b64encode(bytes(urlname, 'utf8')).decode('utf8')
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
24 ul = []
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
25 for genome, assets in genomes.items():
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
26 al = []
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
27 for name in assets:
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
28 al.append({'name': name, 'value': '%s/%s/%s' % (urlname_64, genome, name), 'options': [], 'selected': False})
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
29 ul.append({'name': genome, 'value': genome, 'options': al, 'selected': False})
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
30 rval.append({'name': urlname, 'value': urlname_64, 'options': ul, 'selected': False})
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
31 return rval
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
32
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
33
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
34 if __name__ == '__main__':
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
35 parser = argparse.ArgumentParser()
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
36 parser.add_argument('-n', '--names', dest='names', action='store', default=None, help='Table names to reload')
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
37 parser.add_argument('-u', '--url', dest='url', action='store', default=None, help='Base url for reload')
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
38 parser.add_argument('-k', '--key', dest='key', action='store', default=None, help='Galaxy API Key')
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
39
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
40 args = parser.parse_args()
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
41 if not args.names:
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
42 tables = requests.get(urljoin(args.url, "api/tool_data"), params={'key': args.key}).json()
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
43 args.names = [d.get('name') for d in tables]
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
44 for name in args.names:
659bfadcd8c6 "planemo upload for repository https://github.com/blankenberglab/galaxy-tools-blankenberg/tree/master/data_managers/data_manager_refgenie_pull commit 8d3537d4a138f34f831a629d6be37e3ee6581bc2"
blankenberglab
parents:
diff changeset
45 print(requests.get(urljoin(args.url, "api/tool_data/%s/reload" % (name)), params={'key': args.key}).json())