# HG changeset patch
# User dfornika
# Date 1636064854 0
# Node ID 71668d1d05007be1d2ebca6df2f7810352a2f5e8
# Parent  4f1ca9374be251f2821f481018b12ae180e36a5b
"planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_build_bracken_database/ commit 57b36cfbdd1663aef43d03b76e37364cc5bdeef3-dirty"

diff -r 4f1ca9374be2 -r 71668d1d0500 data_manager/bracken_build_database.py
--- a/data_manager/bracken_build_database.py	Thu Nov 04 22:11:33 2021 +0000
+++ b/data_manager/bracken_build_database.py	Thu Nov 04 22:27:34 2021 +0000
@@ -13,7 +13,7 @@
 DATA_TABLE_NAME = "bracken_databases"
 
 
-def bracken_build_database(target_directory, bracken_build_args, database_name, data_table_name=DATA_TABLE_NAME):
+def bracken_build_database(target_directory, bracken_build_args, database_name, prebuilt=false, data_table_name=DATA_TABLE_NAME):
 
     database_value = str(uuid.uuid4())
 
@@ -21,14 +21,15 @@
 
     database_path = os.path.join(bracken_build_args['kraken_database'], 'database' + str(bracken_build_args['read_len']) + 'mers.kmer_distrib')
 
-    bracken_build_args_list = [
-        '-t', bracken_build_args['threads'],
-        '-k', bracken_build_args['kmer_len'],
-        '-l', bracken_build_args['read_len'],
-        '-d', bracken_build_args['kraken_database'],
-    ]
+    if not prebuilt:
+        bracken_build_args_list = [
+            '-t', bracken_build_args['threads'],
+            '-k', bracken_build_args['kmer_len'],
+            '-l', bracken_build_args['read_len'],
+            '-d', bracken_build_args['kraken_database'],
+        ]
 
-    subprocess.check_call(['bracken-build'] + bracken_build_args_list)
+        subprocess.check_call(['bracken-build'] + bracken_build_args_list)
 
     data_table_entry = {
         "data_tables": {
@@ -53,6 +54,7 @@
     parser.add_argument('--read-len', dest='read_len', help='Read length')
     parser.add_argument('--kraken-db', dest='kraken_database', help='Kraken Database')
     parser.add_argument('--database-name', dest='database_name', help='Database Name')
+    parser.add_argument('--prebuilt', dest='prebuilt', help='Use pre-built DB')
     args = parser.parse_args()
 
     with open(args.data_manager_json) as fh:
@@ -60,12 +62,19 @@
 
     target_directory = data_manager_input['output_data'][0]['extra_files_path']
 
-    bracken_build_args = {
-        'threads': args.threads,
-        'kmer_len': args.kmer_len,
-        'read_len': args.read_len,
-        'kraken_database': args.kraken_database,
-    }
+    if args.prebuilt:
+        bracken_build_args = {
+            'threads': args.threads,
+            'read_len': args.read_len,
+            'kraken_database': args.kraken_database,
+        }
+    else:
+        bracken_build_args = {
+            'threads': args.threads,
+            'kmer_len': args.kmer_len,
+            'read_len': args.read_len,
+            'kraken_database': args.kraken_database,
+        }
 
     try:
         os.mkdir(target_directory)
@@ -81,6 +90,7 @@
         target_directory,
         bracken_build_args,
         args.database_name,
+        args.prebuilt,
     )
 
     with open(args.data_manager_json, 'w') as fh:
diff -r 4f1ca9374be2 -r 71668d1d0500 data_manager/bracken_build_database.xml
--- a/data_manager/bracken_build_database.xml	Thu Nov 04 22:11:33 2021 +0000
+++ b/data_manager/bracken_build_database.xml	Thu Nov 04 22:27:34 2021 +0000
@@ -16,7 +16,12 @@
           '${out_file}'
           --kraken-db '$db_dir'
           --threads \${GALAXY_SLOTS:-1}
-          --kmer-len ${kmer_len}
+          #if $check_prebuilt.prebuilt == "no"
+              --kmer-len ${kmer_len}
+          #end if
+          # if $check_prebuilt.prebuilt == "yes"
+              --prebuilt
+          #end if
           --read-len ${read_len}
           --database-name '${database_name}'
     ]]>