Add CLI support for passing processed_date

Author: Adam Robbins-Pianka

qiita_db/commands.py

@@ -6,6 +6,7 @@
 # The full license is in the file LICENSE, distributed with this software.
 # -----------------------------------------------------------------------------
 
+from dateutil.parser import parse
 import pandas as pd
 from functools import partial
 try:
@@ -128,8 +129,9 @@ def load_processed_data_cmd(fps, fp_types, processed_params_table_name,
         The ID of the row in the processed_params_ table
     preprocessed_data_id : int, optional
         Defaults to ``None``. The ID of the row in the preprocessed_data table.
-    processed_date : datetime, optional
+    processed_date : str, optional
         Defaults to ``None``. The date and time to use as the processing date.
+        Must be interpretable as a datetime object
 
     Returns
     -------
@@ -148,6 +150,9 @@ def load_processed_data_cmd(fps, fp_types, processed_params_table_name,
     else:
         preprocessed_data = None
 
+    if processed_date is not None:
+        processed_date = parse(processed_date)
+
     return ProcessedData.create(processed_params_table_name,
                                 processed_params_id, list(zip(fps, fp_types)),
                                 preprocessed_data, processed_date)

scripts/qiita_db

@@ -9,7 +9,6 @@
 # -----------------------------------------------------------------------------
 
 import click
-from datetime import datetime
 
 from qiita_db.util import (get_filetypes, get_filepath_types,
                            get_processed_params_tables)
@@ -58,9 +57,10 @@ def load_raw_data(fp, fp_type, filetype, study):
 @click.option('--preprocessed_data_id', type=int, default=None, help='The '
               'ID of the row in the preprocessed_data table from which '
               'this processed data was created')
-@click.option('--processed_date', type=datetime, default=None,
-              help='The date to use as the processed_date. If None, then '
-              'the current date and time will be used.')
+@click.option('--processed_date', type=str, default=None,
+              help='The date to use as the processed_date. Must be '
+              'interpretable  as a datetime. If None, then the current date '
+              'and time will be used.')
 def load_processed_data(fp, fp_type, processed_params_table,
                        processed_params_id, preprocessed_data_id,
                        processed_date):