add --demo

csst_dag/cli/msc.py

@@ -34,6 +34,7 @@ parser.add_argument(
 parser.add_argument(
     "--final-prc-status", type=int, help="Final processing status", default=-2
 )
+parser.add_argument("--demo", type=bool, help="Demo mode", default=False)
 
 args = parser.parse_args()
 print("CLI parameters: ", args)
@@ -61,7 +62,7 @@ for obs_type, dag_ids in DAG_LOOP_MAP.items():
             batch_id=args.batch_id,
             initial_prc_status=args.initial_prc_status,
             final_prc_status=args.final_prc_status,
-            demo=True,
+            demo=args.demo,
             priority=args.priority,
         )
         print(f"{len(msgs)} tasks.")

csst_dag/dag/l1.py

@@ -151,9 +151,8 @@ class CsstL1(BaseDAG):
     ):
         assert kwargs.keys() <= SCHEDULE_KWARGS, f"Unknown kwargs: {kwargs.keys()}"
         # no need to query plan
-
         #
-        plan.write_file(local_path="plan.json")
+        # plan.write_file(local_path="plan.json")
         # plan.find(
         #     instrument="MSC",
         #     dataset=dataset,

gen_msg.py

-import json
-from astropy import time
-import numpy as np
-
-
-def gen_dag_runid(digits=10):
-    now = time.Time.now()
-    dagrunid = now.iso[:10].replace("-", "")
-    for i in range(digits):
-        dagrunid += f"{np.random.randint(low=0, high=9)}"
-    return dagrunid
-
-
-def gen_msg(dag_id="csst-msc-l1-mbi", obsid="11009101682009", chipid="09"):
-
-    data = {
-        "dag_id": dag_id,
-        "dag_run_id": gen_dag_runid(),
-        "message": {
-            "obsid": obsid,
-            "chipid": chipid,
-        },
-    }
-    msg = json.dumps(data, ensure_ascii=False, indent=None)
-    print(f"http://localhost:3000/scalebox/run/level0?id={data['dag_run_id']}")
-    # msg = msg.replace('"', '\\"')
-    return msg
-
-
-gen_msg()
-# def gen_cmd(dag_id="csst-msc-l1-mbi", obsid="11009101682009", chipid="09"):
-#     msg = gen_msg(dag_id=dag_id, obsid=obsid, chipid=chipid)
-#     print()
-
-
-# http://localhost:3000/scalebox/run/level0?id=202411286117850576