Add shortcut for TRACE and QUERY(https://httpwg.org/http-extensions/d…

…raft-ietf-httpbis-safe-method-w-body.html)
lexiforest · Oct 10, 2024 · e93c239 · e93c239
1 parent 93e551d
commit e93c239
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/curl_cffi/requests/__init__.py b/curl_cffi/requests/__init__.py
@@ -173,3 +173,5 @@ def request(
 patch = partial(request, "PATCH")
 delete = partial(request, "DELETE")
 options = partial(request, "OPTIONS")
+trace = partial(request, "TRACE")
+query = partial(request, "QUERY")
diff --git a/curl_cffi/requests/session.py b/curl_cffi/requests/session.py
@@ -94,7 +94,7 @@ class BaseSessionParams(TypedDict, total=False):
     BaseSessionParams = TypedDict
 
 ThreadType = Literal["eventlet", "gevent"]
-HttpMethod = Literal["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "TRACE", "PATCH"]
+HttpMethod = Literal["GET", "POST", "PUT", "DELETE", "OPTIONS", "HEAD", "TRACE", "PATCH", "QUERY"]
 
 
 def _is_absolute_url(url: str) -> bool:
@@ -1080,6 +1080,8 @@ def cleanup(fut):
     patch = partialmethod(request, "PATCH")
     delete = partialmethod(request, "DELETE")
     options = partialmethod(request, "OPTIONS")
+    trace = partialmethod(request, "TRACE")
+    query = partialmethod(request, "QUERY")
 
 
 class AsyncSession(BaseSession):
@@ -1364,3 +1366,5 @@ def cleanup(fut):
     patch = partialmethod(request, "PATCH")
     delete = partialmethod(request, "DELETE")
     options = partialmethod(request, "OPTIONS")
+    trace = partialmethod(request, "TRACE")
+    query = partialmethod(request, "QUERY")