http_client header

apache · Jul 5, 2024 · 8f8b5db · 8f8b5db
1 parent 173fbeb
commit 8f8b5db
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 13 deletions.
diff --git a/clients/client-python/gravitino/auth/default_oauth_to_token_provider.py b/clients/client-python/gravitino/auth/default_oauth_to_token_provider.py
@@ -22,7 +22,7 @@
 
 class DefaultOAuth2TokenProvider(OAuth2TokenProvider):
     """This class is the default implement of OAuth2TokenProvider."""
-    
+
     _credential: Optional[str]
     _scope: Optional[str]
     _path: Optional[str]

diff --git a/clients/client-python/gravitino/utils/http_client.py b/clients/client-python/gravitino/utils/http_client.py
@@ -72,6 +72,17 @@ def json(self):
 
 
 class HTTPClient:
+
+    FORMDATA_HEADER = {
+        "Content-Type": "application/x-www-form-urlencoded",
+        "Accept": "application/vnd.gravitino.v1+json",
+    }
+
+    JSON_HEADER = {
+        "Content-Type": "application/json",
+        "Accept": "application/vnd.gravitino.v1+json",
+    }
+
     def __init__(
         self,
         host,
@@ -136,24 +147,16 @@ def _request(
         method = method.upper()
         request_data = None
 
-        if headers:
-            self._update_headers(headers)
-
         if data:
             request_data = urlencode(data.to_dict()).encode()
-            headers = {
-                "Content-Type": "application/x-www-form-urlencoded",
-                "Accept": "application/vnd.gravitino.v1+json",
-            }
-            self._update_headers(headers)
+            self._update_headers(self.FORMDATA_HEADER)
         else:
             if json:
                 request_data = json.to_json().encode("utf-8")
 
-            headers = {
-                "Content-Type": "application/json",
-                "Accept": "application/vnd.gravitino.v1+json",
-            }
+            self._update_headers(self.JSON_HEADER)
+
+        if headers:
             self._update_headers(headers)
 
         opener = build_opener()