add support for fits: open, getheader, getval, getdata

csst_common/fits.py

+from astropy.io import fits
+from csst_fs.s3_config import load_s3_options
+
+s3_options = load_s3_options()
+
+
+def 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)
+
+
+def getheader(name, **kwargs) -> fits.HDUList:
+    if name.startswith("s3://"):
+        # read FITS file from s3
+        return fits.getheader(name, use_fsspec=True, fsspec_kwargs=s3_options, **kwargs)
+    else:
+        # read FITS file from local
+        return fits.getheader(name, **kwargs)
+
+
+def getval(name, **kwargs) -> fits.HDUList:
+    if name.startswith("s3://"):
+        # read FITS file from s3
+        return fits.getval(name, use_fsspec=True, fsspec_kwargs=s3_options, **kwargs)
+    else:
+        # read FITS file from local
+        return fits.getval(name, **kwargs)
+
+
+def getdata(name, **kwargs) -> fits.HDUList:
+    if name.startswith("s3://"):
+        # read FITS file from s3
+        return fits.getdata(name, use_fsspec=True, fsspec_kwargs=s3_options, **kwargs)
+    else:
+        # read FITS file from local
+        return fits.getdata(name, **kwargs)

csst_common/io.py

@@ -21,21 +21,9 @@ from copy import deepcopy
 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(