scratch for reset

csst_dag/cli/reset.py

@@ -24,11 +24,31 @@ parser = argparse.ArgumentParser(
 
 # data related parameters
 parser.add_argument("--dataset", type=str, help="Dataset name")
+parser.add_argument("--instrument", type=str, help="Dataset name", default=None)
+parser.add_argument("--obs-type", type=str, help="Observation type", default=None)
+parser.add_argument("--obs-group", type=str, help="Observation group", default=None)
+parser.add_argument("--obs-id", type=str, help="Observation ID", default=None)
+parser.add_argument("--detector", type=str, help="Detector", default=None)
 parser.add_argument(
-    "--final-prc-status", type=int, help="Final processing status", default=-2
+    "--initial-prc-status", type=int, help="Initial processing status", default=-2
+)
+parser.add_argument(
+    "--final-prc-status", type=int, help="Final processing status", default=-1024
 )
-
 args = parser.parse_args()
 print("CLI parameters: ", args)
 
-recs = level0.find(dataset=args.dataset)
+qr_level0 = level0.find(
+    dataset=args.dataset,
+    instrument=args.instrument,
+    obs_type=args.obs_type,
+    obs_group=args.obs_group,
+    obs_id=args.obs_id,
+    detector=args.detector,
+    prc_status=args.initial_prc_status,
+)
+
+r_update = level0.update_prc_status(
+    [_["_id"] for _ in qr_level0.data],
+    prc_status=args.final_prc_status,
+)