Add check_file method

abraia/abraia.py

@@ -87,6 +87,15 @@ def upload_file(self, src, path=''):
             return file_path({'name': name, 'source': f"{self.userid}/{path}"}, self.userid)
         return file_path(resp['file'], self.userid)
 
+    def check_file(self, path):
+        url = f"{API_URL}/files/{self.userid}/{path}"
+        resp = requests.head(url, auth=self.auth)
+        if resp.status_code == 404:
+            return False
+        if resp.status_code in [307, 400, 403]:
+            return True
+        raise APIError(resp.text, resp.status_code)
+
     def move_file(self, old_path, new_path):
         json = {'store': f"{self.userid}/{old_path}"}
         url = f"{API_URL}/files/{self.userid}/{new_path}"

requirements.txt

@@ -2,6 +2,6 @@ click>=8.0.3
 numpy>=1.21.0
 Pillow>=8.0.0
 tifffile>=2020.9.4
-python-dotenv>=0.19.2
+python-dotenv>=0.21.1
 requests>=2.27.1
 tqdm>=4.62.3

tests/test_abraia.py

@@ -21,6 +21,13 @@ def test_upload_file():
     assert path == 'tiger.jpg'
 
 
+def test_check_file():
+    path = 'lion.jpg'
+    assert abraia.check_file(path)
+    path = '_lion.jpg'
+    assert not abraia.check_file(path)
+
+
 def test_download_file():
     stream = abraia.download_file('tiger.jpg')
     assert isinstance(stream, BytesIO)