# HG changeset patch
# User dfornika
# Date 1636138537 0
# Node ID 37467f9b7ea31e60bda84baf06721b4575db2f46
# Parent  2650a19fe2601ab77c7df9100db16bfd7e9732f6
"planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/data_managers/data_manager_build_kraken2_database/ commit cec255ceca4b1dd39c4b5795cbfcf5f8e74fc9c7-dirty"
diff -r 2650a19fe260 -r 37467f9b7ea3 data_manager/kraken2_build_database.py
--- a/data_manager/kraken2_build_database.py	Thu Nov 04 17:25:24 2021 +0000
+++ b/data_manager/kraken2_build_database.py	Fri Nov 05 18:55:37 2021 +0000
@@ -51,6 +51,13 @@
         return self.value
 
 
+class PrebuiltDate(Enum):
+    date_20210517 = '2021-05-17'
+    date_20201202 = '2020-12-02'
+
+    def __str__(self):
+        return self.value
+
 class StandardPrebuiltSizes(Enum):
     full = 'full'
     gb_16 = '16'
@@ -119,6 +126,7 @@
 
     return data_table_entry
 
+
 def kraken2_build_standard_prebuilt(standard_prebuilt_size, target_directory, data_table_name=DATA_TABLE_NAME):
 
     now = datetime.datetime.utcnow().strftime("%Y-%m-%dT%H%M%SZ")
@@ -137,13 +145,14 @@
     ])
 
     database_path = database_value
-    
+
     size_to_url_str = {
         'full': '',
         '16': '_16gb',
         '8': '_8gb',
     }
-    date_url_str = '20210517'
+    # we may need to let the user choose the date when new DBs are posted.
+    date_url_str = args.prebuilt_date.replace('-', '')
     standard_prebuilt_size_url = size_to_url_str[standard_prebuilt_size]
     # download the pre-built database
     src = urlopen(
@@ -357,7 +366,8 @@
     parser.add_argument('--threads', dest='threads', default=1, help='threads')
     parser.add_argument('--database-type', dest='database_type', type=KrakenDatabaseTypes, choices=list(KrakenDatabaseTypes), required=True, help='type of kraken database to build')
     parser.add_argument('--minikraken2-version', dest='minikraken2_version', type=Minikraken2Versions, choices=list(Minikraken2Versions), help='MiniKraken2 version (only applies to --database-type minikraken)')
-    parser.add_argument('--standard-prebuilt-size', dest='standard_prebuilt_size', type=StandardPrebuiltSizes, choices=list(StandardPrebuiltSizes), help='Size of standard prebuilt database to download (only applies to --database-type standard_prebuilt)')
+    parser.add_argument('--standard-prebuilt-size', dest='standard_prebuilt_size', type=StandardPrebuiltSizes, choices=list(StandardPrebuiltSizes), help='Size of standard prebuilt database to download (only applies to --database-type standard_prebuilt. Options are: "8", "16", "full".)')
+    parser.add_argument('--prebuilt-date', dest='prebuilt_date', type=PrebuiltDates, choices=list(PrebuiltDates), help='Database build date (YYYY-MM-DD). Only applies to --database-type standard_prebuilt. Options are: "2021-05-17", "2020-12-02".)')
     parser.add_argument('--special-database-type', dest='special_database_type', type=SpecialDatabaseTypes, choices=list(SpecialDatabaseTypes), help='type of special database to build (only applies to --database-type special)')
     parser.add_argument('--custom-fasta', dest='custom_fasta', help='fasta file for custom database (only applies to --database-type custom)')
     parser.add_argument('--custom-database-name', dest='custom_database_name', help='Name for custom database (only applies to --database-type custom)')
diff -r 2650a19fe260 -r 37467f9b7ea3 data_manager/kraken2_build_database.xml
--- a/data_manager/kraken2_build_database.xml	Thu Nov 04 17:25:24 2021 +0000
+++ b/data_manager/kraken2_build_database.xml	Fri Nov 05 18:55:37 2021 +0000
@@ -65,11 +65,15 @@
             
                 
             
-	    
+            
                 
-                    
+                    
                     
-		    
+                    
+                
+		
+                    
+