Merge pull request !11 from WeidenthalerMatthias/feature.delete

README.md

@@ -71,6 +71,17 @@ fs.isfile('s3://csst-prod/test.txt')
 fs.isfile('s3://csst-prod/test.txt', s3_options=s3_options)
 ```
 
+### Delete a file from local or s3
+```python
+from csst_fs import fs
+# local or on s3, depending on the given path
+fs.delete('requirements.txt') # uses os.remove
+fs.delete('test', dir_fd=1)
+fs.delete('s3://csst-prod/test.txt') # uses fsspec.delete
+fs.delete('s3://csst-prod/test.txt', recursive=True, maxdepth=3)
+fs.delete('s3://csst-prod/test.txt', s3_options=s3_options)
+```
+
 ## astropy直接读写s3的写法适配
 ### fits.open
 #### 老写法

csst_fs/fs.py

 from .s3_fs import is_s3_path
 from .s3_fs import isfile as s3_isfile
+from .s3_fs import delete as s3_delete
+from .local_fs import delete as local_delete
 from .local_fs import isfile as local_isfile
 
 def isfile(path: str, *args, **kwargs) -> bool:
@@ -7,3 +9,9 @@ def isfile(path: str, *args, **kwargs) -> bool:
         return s3_isfile(path, *args, **kwargs)
     else:
         return local_isfile(path, *args, **kwargs)
+
+def delete(path, **kwargs):
+    if is_s3_path(path):
+        s3_delete(path, **kwargs)
+    else:
+        local_delete(path, **kwargs)

csst_fs/local_fs.py

@@ -2,4 +2,7 @@ import os
 
 def isfile(path: str) -> bool:
     return os.path.isfile(path)
+
+def delete(path: str, *, dir_fd=None, **kwargs):
+    os.remove(path, dir_fd=dir_fd)
     
\ No newline at end of file

csst_fs/s3_fs.py

@@ -20,6 +20,10 @@ def open(file: str, mode: str='r', s3_options=load_s3_options()):
 def isfile(path: str, s3_options: dict=load_s3_options()) -> bool:
     s3_fs = fsspec.filesystem('s3', **s3_options)
     return s3_fs.isfile(path)
+
+def delete(path: str, *, recursive=False, maxdepth=None, s3_options: dict=load_s3_options(), **kwargs):
+    s3_fs = fsspec.filesystem('s3', **s3_options)
+    s3_fs.delete(path, recursive, maxdepth)
     
 def is_s3_path(path: str):
     if(path.startswith("s3")):