Commit 6336cc21 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

scratch for reset

parent 96923b68
Loading
Loading
Loading
Loading
+23 −3
Original line number Diff line number Diff line
@@ -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,
)