Mercurial > repos > jjohnson > data_manager_eggnog_mapper_abspath

annotate data_manager/data_manager_eggnog.py @ 0:d4e2e8176db8 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
author jjohnson
date Thu, 07 Nov 2019 12:42:32 -0500
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
1 #!/usr/bin/env python
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
2
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
3 from __future__ import print_function
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
4
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
5 import argparse
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
6 import json
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
7 import os.path
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
8 import sqlite3
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
9 import sys
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
10 from sqlite3 import OperationalError
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
11
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
12
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
13 def _get_db_version(sqlitedb_path):
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
14 version = '4.5'
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
15 try:
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
16 query = 'select version from version'
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
17 conn = sqlite3.connect(sqlitedb_path)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
18 cur = conn.cursor()
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
19 cur.execute(query)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
20 version = cur.fetchone()[0]
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
21 except OperationalError as e:
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
22 print('Assuming eggnog version %s because %s %s' %
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
23 (version, sqlitedb_path, e), file=sys.stderr)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
24 return version
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
25
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
26
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
27 def main():
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
28 parser = argparse.ArgumentParser()
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
29 parser.add_argument('--config_file')
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
30 parser.add_argument('--install_path')
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
31 parser.add_argument('--dbs', default='')
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
32 args = parser.parse_args()
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
33
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
34 eggnog_db_path = os.path.join(args.install_path, 'eggnog.db')
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
35 if not os.path.exists(eggnog_db_path):
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
36 print('Can not open: %s' % eggnog_db_path, file=sys.stderr)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
37 exit(1)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
38 db_version = _get_db_version(eggnog_db_path)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
39
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
40 # params = json.loads(open(args.config_file).read())
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
41 dm_dict = {}
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
42 dm_dict['data_tables'] = dm_dict.get('data_tables', {})
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
43 data_table = 'eggnog_mapper_db'
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
44 dm_dict['data_tables'][data_table]\
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
45 = dm_dict['data_tables'].get(data_table, [])
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
46 data_table_entry = dict(value=db_version, name=db_version,
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
47 path=args.install_path)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
48 dm_dict['data_tables'][data_table].append(data_table_entry)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
49 data_table = 'eggnog_mapper_hmm_dbs'
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
50 dm_dict['data_tables'][data_table]\
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
51 = dm_dict['data_tables'].get(data_table, [])
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
52 if args.dbs:
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
53 dbs = [x.strip() for x in args.dbs.split(',')]
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
54 for db in dbs:
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
55 key = '%s_%s' % (db_version, db)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
56 data_table_entry = dict(key=key, db_version=db_version,
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
57 value=db, name=db, path=db)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
58 dm_dict['data_tables'][data_table].append(data_table_entry)
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
59
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
60 # save info to json file
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
61 open(args.config_file, 'wb').write(json.dumps(dm_dict))
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
62
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
63
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
64 if __name__ == "__main__":
d4e2e8176db8 planemo upload for repository https://github.com/galaxyproteomics/tools-galaxyp/tree/master/tools/eggnog_mapper/data_manager_eggnog_mapper_abspath commit 4ac0d4d79233ac5151ab32741183adbfecb70a45
jjohnson
parents:
diff changeset
65 main()