Browse Source

add headers for requests

chenhaiyang 4 years ago
parent
commit
1f2d3eb262
1 changed files with 10 additions and 4 deletions
  1. 10 4
      app/services/service.py

+ 10 - 4
app/services/service.py

@@ -30,15 +30,21 @@ class Service:
         self._client = client
 
     @api_exception
-    async def _get(self, url: URL, params: Optional[Dict] = None) -> Dict:
-        raw_response = await self._client.get(url, params=params)
+    async def _get(self, url: URL, params: Optional[Dict] = None, headers: Optional[Dict] = None) -> Dict:
+        raw_response = await self._client.get(url, params=params, headers=headers)
         # logger.debug(f'{url} - elapsed: {raw_response.elapsed.total_seconds()}')
 
         return raw_response.json()
 
     @api_exception
-    async def _post(self, url: URL, params: Optional[Dict] = None, payload: Optional[Dict] = None) -> Dict:
-        raw_response = await self._client.post(url, params=params, json=payload)
+    async def _post(
+            self,
+            url: URL,
+            params: Optional[Dict] = None,
+            payload: Optional[Dict] = None,
+            headers: Optional[Dict] = None
+    ) -> Dict:
+        raw_response = await self._client.post(url, params=params, json=payload, headers=headers)
         # logger.debug(f'{url} - elapsed: {raw_response.elapsed.total_seconds()}')
 
         return raw_response.json()