From 7c3a3f052d6dbfaafe62feaa430c0d3c9f3c7a4c Mon Sep 17 00:00:00 2001 From: shoulinwei Date: Tue, 21 Feb 2023 22:39:46 +0800 Subject: [PATCH] get fits header --- csst_dfs_commons/utils/fits.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/csst_dfs_commons/utils/fits.py b/csst_dfs_commons/utils/fits.py index 2a30759..2b8fed4 100644 --- a/csst_dfs_commons/utils/fits.py +++ b/csst_dfs_commons/utils/fits.py @@ -2,13 +2,20 @@ from astropy_healpix import HEALPix from astropy.coordinates import ICRS from astropy import units as u from astropy.coordinates import SkyCoord +from astropy.io import fits -def get_header_value(key: str, header, default_value = None): +def get_header_value(key: str, headers, default_value = None): try: - v = header[key] - if type(v) == str: - return v.strip() - return v + ret_value = None + if not isinstance(headers, list): + headers = [headers] + for header in headers: + if key in header: + ret_value = header[key] + if type(ret_value) == str: + return ret_value.strip() + if ret_value is None: + return default_value except Exception as e: return default_value -- GitLab