Commit 08f9b3f8 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add append_meta

parent 3d72590f

csst_common/io.py

+24 −1
Original line number Original line Diff line number Diff line
@@ -42,6 +42,7 @@ REQUIRED_KEYS_IN_META = dict( 
    image="__image__",

    image="__image__",

    build="__build__",

    build="__build__",

    pmapname="__pmapname__",

    pmapname="__pmapname__",

    ref_cat="__ref_cat__",

    # 额外的观测筛选参数

    # 额外的观测筛选参数

    obs_date="1970-01-01T00:00:00",

    obs_date="1970-01-01T00:00:00",

    create_date="1970-01-01T00:00:00",

    create_date="1970-01-01T00:00:00",
@@ -81,10 +82,32 @@ def generate_meta(**kwargs) -> fits.Header: 
        keyword="META",

        keyword="META",

        value=json.dumps(kwargs, separators=(",", ":"), ensure_ascii=False),

        value=json.dumps(kwargs, separators=(",", ":"), ensure_ascii=False),

    )

    )

    meta_header = fits.Header(cards=[meta_card])

    meta_header = reformat_header(fits.Header(cards=[meta_card]), comment="INGEST META")

    return meta_header

    return meta_header









def append_meta(hdulist: fits.HDUList, **kwargs) -> fits.HDUList:

    """

    Append meta to hdulist.



    Parameters

    ----------

    hdulist : fits.HDUList

        HDUList.

    kwargs : Any

        Meta data.



    Returns

    -------

    fits.HDUList

        HDUList with meta.



    """

    meta_header = generate_meta(**kwargs)

    hdulist[0].header = append_header(hdulist[0].header, meta_header)

    return hdulist





def verify_checksum(file_path) -> bool:

def verify_checksum(file_path) -> bool:

    """

    """

    Verify a .fits file via checksum.

    Verify a .fits file via checksum.