Prepared software for more analysis options

src/nsbdiagnosistoolkit/cli/root.py

@@ -1,6 +1,8 @@
 import argparse
 import asyncio
+import datetime
 from os import path
+import os
 
 from nsbdiagnosistoolkit.cli import aggregator
 from nsbdiagnosistoolkit.engine.data.genomics import NamedString
@@ -10,6 +12,14 @@ from nsbdiagnosistoolkit.engine.local.fasta import read_fasta
 
 
 parser = argparse.ArgumentParser()
+parser.add_argument(
+    "--run-name", "-name",
+    dest="run_name",
+    required=False,
+    default=datetime.datetime.now().strftime(r"%Y%m%d%H%M%S"),
+    type=str,
+    help="The name of the run. Will use a date and time string if not provided."
+)
 parser.add_argument(
     "--fasta", "-fa", "-fst",
     nargs="+",
@@ -33,25 +43,27 @@ parser.add_argument(
     "--institut-pasteur-mlst",
     "-ipdbmlst",
     dest="institut_pasteur_db",
+    required=False,
+    default=None,
     type=str,
     help="The Institut Pasteur MLST database to use."
 )
 parser.add_argument(
-    "-csv",
-    dest="csv_path",
-    required=False,
-    default=None,
-    help="The destination to place the CSV output."
+    "out",
+    default="./.",
+    help="The output folder. Files will be named by the provided (or default) run name."
 )
 
 
 def cli():
     args = parser.parse_args()
     gen_strings = aggregator.aggregate_sequences(args.fastas, args.abifs)
-    mlst_profiles = aggregator.profile_all_genetic_strings(
-        gen_strings, args.institut_pasteur_db)
-    asyncio.run(write_mlst_profiles_as_csv(
-        asyncio.run(mlst_profiles), str(args.csv_path)))
+    os.makedirs(args.out)
+    if args.institut_pasteur_db is not None:
+        mlst_profiles = aggregator.profile_all_genetic_strings(
+            gen_strings, args.institut_pasteur_db)
+        asyncio.run(write_mlst_profiles_as_csv(
+            asyncio.run(mlst_profiles), str(path.join(args.out, "MLST_" + args.run_name + ".csv"))))
 
 
 if __name__ == "__main__":