add --dags to select DAGs

csst_dag/cli/csst_msc_l1.py

@@ -58,7 +58,7 @@ parser.add_argument(
 )
 # task parameters
 parser.add_argument("--batch-id", type=str, help="Batch ID", default="test-batch")
-parser.add_argument("--priority", type=str, help="Task priority", default=1)
+parser.add_argument("--priority", type=str, help="Task priority", default="1")
 # DAG parameters
 parser.add_argument("--pmapname", type=str, help="CCDS pmapname", default="")
 parser.add_argument(
@@ -72,6 +72,13 @@ parser.add_argument(
 )
 # force option
 parser.add_argument("--force", action="store_true", help="Force success", default=False)
+# submit top N
+parser.add_argument(
+    "--top-n", type=int, help="Submit top N tasks", default=-1
+)
+# select DAGs
+parser.add_argument('--dags', nargs='+', type=str, help="DAGs to select", default=None)
+
 
 args = parser.parse_args()
 
@@ -120,12 +127,16 @@ dag_run_list = []
 data_id_list = []
 n_dag_run_all = 0
 n_dag_run_success = 0
-for dag in [
+
+DEFAULT_DAGS = {
     "csst-msc-l1-mbi",
     "csst-msc-l1-ast",
     "csst-msc-l1-sls",
     "csst-msc-l1-qc0",
-]:
+}
+SELECTED_DAGS = DEFAULT_DAGS if args.dag is None else {dag for dag in args.dags if dag in DEFAULT_DAGS}
+print("Selected DAGs: ", SELECTED_DAGS)
+for dag in SELECTED_DAGS:
     this_task_list = CSST_DAGS[dag].schedule(
         dag_group_run=dag_group_run,
         plan_basis=plan_basis,