reset.py

"""
Aim
---
Reset a dataset.

Example
-------
python -m csst_dag.cli.reset -h

python -m csst_dag.cli.reset \
    --dataset=csst-msc-c9-25sqdeg-v3 \
    --final-prc-status=-2
"""

from csst_dag.dag import CsstDAGs
from csst_dfs_client import level0
import argparse
import tqdm

parser = argparse.ArgumentParser(
    description="Scheduler for MSC L1 pipeline.",
    formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)

# 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(
    "--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)

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,
)