add fits_open to be compatible with s3

csst_common/io.py

@@ -12,18 +12,30 @@ Modified-History:
     2025-08-05, Bo Zhang, added generate_meta
 """
 
-import os
-import json
 import bisect
+import json
+import os
+import uuid
 import warnings
 from copy import deepcopy
-from typing import Optional, Any
-import uuid
+from typing import Optional
 
 from astropy.io import fits
+from csst_fs.s3_config import load_s3_options
 
 from .time import now
 
+s3_options = load_s3_options()
+
+
+def fits_open(name, **kwargs) -> fits.HDUList:
+    if name.startswith("s3://"):
+        # read fits file from s3
+        return fits.open(name, use_fsspec=True, fsspec_kwargs=s3_options, **kwargs)
+    else:
+        # read fits file from local
+        return fits.open(name, **kwargs)
+
 
 # meta字段和默认值
 REQUIRED_KEYS_IN_META = dict(

tests/test_fits_header_ops.py → tests/test_io.py

@@ -21,6 +21,7 @@ from csst_common.io import (
     generate_meta,
     append_meta,
     extract_meta,
+    fits_open,
 )
 
 
@@ -141,3 +142,7 @@ class TestFitsHeaderOps(unittest.TestCase):
         self.assertIn("META", set(hdulist[0].header.keys()))
         meta = extract_meta(hdulist)
         self.assertEqual(meta["obs_id"], "123456")
+
+    def test_fits_open(self):
+        hl = fits_open("/nfsdata/share/pipeline-unittest/csst_common/test_fits/hl.fits")
+        self.assertTrue(isinstance(hl, fits.HDUList))