gen_msg.py 944 Bytes
Newer Older
BO ZHANG's avatar
tweaks  
BO ZHANG committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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