feat: Use HEAD HTTP method to check if a document exists

Author: Darkheir

arango/collection.py

@@ -508,7 +508,7 @@ def has(
         handle, body, headers = self._prep_from_doc(document, rev, check_rev)
 
         request = Request(
-            method="get",
+            method="head",
             endpoint=f"/_api/document/{handle}",
             headers=headers,
             read=self.name,
@@ -519,9 +519,11 @@ def response_handler(resp: Response) -> bool:
                 return False
             if resp.status_code == 412:
                 raise DocumentRevisionError(resp, request)
+            if resp.status_code == 404:
+                return False
             if not resp.is_success:
                 raise DocumentInError(resp, request)
-            return bool(resp.body)
+            return True
 
         return self._execute(request, response_handler)
 

tests/test_document.py

@@ -1390,7 +1390,7 @@ def test_document_has(col, bad_col, docs):
 
         with assert_raises(DocumentRevisionError) as err:
             col.has(doc_input, rev=bad_rev, check_rev=True)
-        assert err.value.error_code == 1200
+        assert err.value.error_code == 412
 
     # Test existing documents with bad revision
     for doc_input in [
@@ -1400,15 +1400,15 @@ def test_document_has(col, bad_col, docs):
     ]:
         with assert_raises(DocumentRevisionError) as err:
             col.has(doc_input)
-        assert err.value.error_code == 1200
+        assert err.value.error_code == 412
 
         with assert_raises(DocumentRevisionError) as err:
             col.has(doc_input, rev=bad_rev)
-        assert err.value.error_code == 1200
+        assert err.value.error_code == 412
 
         with assert_raises(DocumentRevisionError) as err:
             col.has(doc_input, rev=bad_rev, check_rev=True)
-        assert err.value.error_code == 1200
+        assert err.value.error_code == 412
 
         assert doc_input in col
         assert col.has(doc_input, rev=rev, check_rev=True) is True
@@ -1487,12 +1487,12 @@ def test_document_has(col, bad_col, docs):
     # Test get with bad database
     with assert_raises(DocumentInError) as err:
         bad_col.has(doc_key)
-    assert err.value.error_code in {11, 1228}
+    assert err.value.error_code == 401
 
     # Test contains with bad database
     with assert_raises(DocumentInError) as err:
         assert doc_key in bad_col
-    assert err.value.error_code in {11, 1228}
+    assert err.value.error_code == 401
 
 
 def test_document_get(col, bad_col, docs):