Merge pull request !4 from WeidenthalerMatthias/feature.open

README.md

@@ -52,6 +52,16 @@ s3_fs.info('s3://csst-prod/gaia/data')
 s3_fs.info('s3://csst-prod/gaia/test/requirements.txt', s3_options=s3_options)
 ```
 
+### Open for read/write
+```python
+from csst_fs import s3_fs
+# open single file (s3 or local)
+with s3_fs.open('s3://csst-prod/gaia/data') as file:
+    file.read()
+with s3_fs.open('s3://csst-prod/gaia/test/requirements.txt', s3_options=s3_options, mode='w') as file:
+    file.write("CSST")
+```
+
 ## astropy直接读写s3的写法适配
 ### fits.open
 #### 老写法

csst_fs/fsspec_fileobj.py

@@ -5,5 +5,5 @@ def open_fileobj(path: str, s3_options, mode='w'):
         s3_fs = fsspec.filesystem('s3', **s3_options)
         fileobj = s3_fs.open(path, mode=mode).__enter__()
     else:
-        fileobj = path
+        fileobj = open(path, mode=mode).__enter__()
     return fileobj
\ No newline at end of file

csst_fs/s3_fs.py

 import fsspec
 from .s3_config import load_s3_options
+from .fsspec_fileobj import open_fileobj
 
 def put(lpath, rpath, recursive=False, callback=fsspec.callbacks.DEFAULT_CALLBACK, maxdepth=None, s3_options=load_s3_options(), **kwargs,): 
     s3_fs = fsspec.filesystem('s3', **s3_options)
@@ -12,3 +13,6 @@ def get(rpath, lpath, recursive=False, callback=fsspec.callbacks.DEFAULT_CALLBAC
 def info(path, bucket=None, key=None, refresh=False, version_id=None, s3_options=load_s3_options()) -> dict:
     s3_fs = fsspec.filesystem('s3', **s3_options)
     return s3_fs.info(path, bucket, key, refresh, version_id)
+
+def open(file: str, mode: str='r', s3_options=load_s3_options()):
+    return open_fileobj(path=file, s3_options=s3_options, mode=mode)
\ No newline at end of file