# HG changeset patch # User iuc # Date 1575905295 18000 # Node ID cffd8e2382cf9e8c74b76533d01129c6fedae899 "planemo upload commit 4e5a8d7629d90c41219b48b648dd4ab675b84af0" diff -r 000000000000 -r cffd8e2382cf data_manager/data_manager_cat.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/data_manager/data_manager_cat.py Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,116 @@ +#!/usr/bin/env python +from __future__ import print_function + +import argparse +import json +import os.path +import subprocess +import sys +import tarfile +import tempfile +import zipfile +try: + # For Python 3.0 and later + from urllib.request import urlopen +except ImportError: + # Fall back to Python 2 imports + from urllib2 import urlopen + + +def url_download(url, workdir): + file_path = os.path.join(workdir, 'download.dat') + src = None + dst = None + try: + src = urlopen(url) + with open(file_path, 'wb') as dst: + while True: + chunk = src.read(2**10) + if chunk: + dst.write(chunk) + else: + break + finally: + if src: + src.close() + if tarfile.is_tarfile(file_path): + fh = tarfile.open(file_path, 'r:*') + elif zipfile.is_zipfile(file_path): + fh = zipfile.ZipFile(file_path, 'r') + else: + return + fh.extractall(workdir) + os.remove(file_path) + + +def cat_prepare(install_dir, db_dir=None, tax_dir=None): + if db_dir and tax_dir: + cmd = ['CAT', 'prepare', '--existing', '-d', db_dir, '-t', tax_dir] + else: + cmd = ['CAT', 'prepare', '--fresh', '-q'] + cmd_stdout = tempfile.NamedTemporaryFile() + cmd_stderr = tempfile.NamedTemporaryFile() + return_code = subprocess.call(cmd, shell=False, cwd=install_dir, + stdout=cmd_stdout, stderr=cmd_stderr) + if return_code: + msg = "stdout:\n%s\nstderr:\n%s" % (cmd_stdout.read(), + cmd_stderr.read()) + cmd_stdout.close() + cmd_stderr.close() + raise Exception('Error: (%s), returncode=%s %s' + % (' '.join(cmd), return_code, msg)) + + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument('--config_file', required=True) + parser.add_argument('--install_path', default=None) + parser.add_argument('--db_url', default=None) + parser.add_argument('--database_folder', default=None) + parser.add_argument('--taxonomy_folder', default=None) + args = parser.parse_args() + + cat_path = None + cat_db = None + tax_db = None + if args.database_folder and args.taxonomy_folder: + cat_path = os.path.dirname(args.database_folder) + cat_db = os.path.basename(args.database_folder) + tax_db = os.path.basename(args.taxonomy_folder) + cat_prepare(os.getcwd(), + db_dir=args.database_folder, + tax_dir=args.taxonomy_folder) + elif not args.install_path: + sys.exit(1) + else: + if not os.path.exists(args.install_path): + os.makedirs(args.install_path) + if args.db_url: + url_download(args.db_url, args.install_path) + else: + cat_prepare(args.install_path) + for root, dirs, files in os.walk(args.install_path): + for dname in dirs: + if dname.endswith('CAT_database'): + cat_db = dname + elif dname.endswith('taxonomy'): + tax_db = dname + if cat_db and tax_db: + cat_path = root + break + cat_dir = os.path.basename(cat_path) + dm_dict = {} + dm_dict['data_tables'] = dm_dict.get('data_tables', {}) + data_table = 'cat_database' + dm_dict['data_tables'][data_table]\ + = dm_dict['data_tables'].get(data_table, []) + data_table_entry = dict(value=cat_dir, name=cat_dir, + database_folder=os.path.join(cat_dir, cat_db), + taxonomy_folder=os.path.join(cat_dir, tax_db)) + dm_dict['data_tables'][data_table].append(data_table_entry) + # save info to json file + open(args.config_file, 'w').write(json.dumps(dm_dict)) + + +if __name__ == "__main__": + main() diff -r 000000000000 -r cffd8e2382cf data_manager/data_manager_cat.xml --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/data_manager/data_manager_cat.xml Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,55 @@ + + Install a new CAT database + + cat + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + diff -r 000000000000 -r cffd8e2382cf data_manager_conf.xml --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/data_manager_conf.xml Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,27 @@ + + + + + + + + + + ${database_folder} + CAT/${database_folder} + + ${GALAXY_DATA_MANAGER_DATA_PATH}/CAT/${database_folder} + abspath + + + + ${taxonomy_folder} + CAT/${taxonomy_folder} + + ${GALAXY_DATA_MANAGER_DATA_PATH}/CAT/${taxonomy_folder} + abspath + + + + + diff -r 000000000000 -r cffd8e2382cf test-data/CAT_prepare_test.tar.gz Binary file test-data/CAT_prepare_test.tar.gz has changed diff -r 000000000000 -r cffd8e2382cf test-data/cat_database.loc --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test-data/cat_database.loc Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,7 @@ +## A typical download from https://tbb.bio.uu.nl/bastiaan/CAT_prepare/CAT_prepare_20190719.tar.gz +# ls CAT_prepare_20190719/ +# 2019-07-19.CAT_prepare.fresh.log +# 2019-07-19_CAT_database +# 2019-07-19_taxonomy +#value name database_folder taxonomy_folder +#2019-07-19_CAT_database 2019-07-19_CAT_database /opt/galaxy/tool-data/cat_database/CAT_prepare_20190719/2019-07-19_CAT_database /opt/galaxy/tool-data/cat_database/CAT_prepare_20190719/2019-07-19_taxonomy diff -r 000000000000 -r cffd8e2382cf tool-data/cat_database.loc.sample --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tool-data/cat_database.loc.sample Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,7 @@ +## A typical download from https://tbb.bio.uu.nl/bastiaan/CAT_prepare/CAT_prepare_20190719.tar.gz +# ls CAT_prepare_20190719/ +# 2019-07-19.CAT_prepare.fresh.log +# 2019-07-19_CAT_database +# 2019-07-19_taxonomy +#value name database_folder taxonomy_folder +#2019-07-19_CAT_database 2019-07-19_CAT_database /opt/galaxy/tool-data/cat_database/CAT_prepare_20190719/2019-07-19_CAT_database /opt/galaxy/tool-data/cat_database/CAT_prepare_20190719/2019-07-19_taxonomy diff -r 000000000000 -r cffd8e2382cf tool_data_table_conf.xml.sample --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tool_data_table_conf.xml.sample Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,7 @@ + + + + value, name, database_folder, taxonomy_folder + +
+
diff -r 000000000000 -r cffd8e2382cf tool_data_table_conf.xml.test --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tool_data_table_conf.xml.test Mon Dec 09 10:28:15 2019 -0500 @@ -0,0 +1,7 @@ + + + + value, name, database_folder, taxonomy_folder + +
+