automatically set image and build

csst_common/io.py

@@ -39,8 +39,6 @@ REQUIRED_KEYS_IN_META = dict(
     qc_status=-1024,
     dag="__dag__",
     dag_run="__dag_run__",
-    image="__image__",
-    build="__build__",
     pmapname="__pmapname__",
     ref_cat="__ref_cat__",
     # 额外的观测筛选参数
@@ -49,6 +47,9 @@ REQUIRED_KEYS_IN_META = dict(
     filter="__filter__",
     ra=-3.141592653589793,
     dec=-3.141592653589793,
+    # Docker镜像名称和版本
+    docker_image="__docker_image__",  # 自动设置
+    build="__build__",  # 自动设置
 )
 
 
@@ -66,7 +67,9 @@ def generate_meta(**kwargs) -> fits.Header:
     fits.Header
         Meta data header.
     """
+    # copy REQUIRED_KEYS_IN_META
     meta = deepcopy(REQUIRED_KEYS_IN_META)
+    # update meta with kwargs
     for k in kwargs.keys():
         if k in meta.keys():
             # 类型必须兼容
@@ -77,6 +80,10 @@ def generate_meta(**kwargs) -> fits.Header:
             meta[k] = kwargs[k]
         else:
             raise KeyError(f"未知的meta参数: {k}")
+    # automatically set docker_image and build
+    meta["docker_image"] = os.getenv("PIPELINE_ID", default="__docker_image__")
+    meta["build"] = os.getenv("BUILD", default="__build__")
+    # convert meta to fits header
     meta_card = fits.Card(
         keyword="META",
         value=json.dumps(meta, separators=(",", ":"), ensure_ascii=False),