Skip to content
Snippets Groups Projects
Unverified Commit 358181f3 authored by Dan Michael O. Heggø's avatar Dan Michael O. Heggø
Browse files

Refactor: Always call requests.request

parent 114266bd
No related branches found
No related tags found
No related merge requests found
...@@ -352,6 +352,7 @@ class Site(object): ...@@ -352,6 +352,7 @@ class Site(object):
data (dict): Post data data (dict): Post data
files (dict): Files to upload files (dict): Files to upload
retry_on_error (bool): Retry on connection error retry_on_error (bool): Retry on connection error
http_method (str): The HTTP method, defaults to 'POST'
Returns: Returns:
The raw text response. The raw text response.
...@@ -372,12 +373,16 @@ class Site(object): ...@@ -372,12 +373,16 @@ class Site(object):
while True: while True:
try: try:
args = {'files': files, 'headers': headers}
for k, v in self.requests.items():
args[k] = v
if http_method == 'GET': if http_method == 'GET':
stream = self.connection.get(url, params=data, files=files, args['params'] = data
headers=headers, **self.requests)
else: else:
stream = self.connection.post(url, data=data, files=files, args['data'] = data
headers=headers, **self.requests)
stream = self.connection.request(http_method, url, **args)
if stream.headers.get('x-database-lag'): if stream.headers.get('x-database-lag'):
wait_time = int(stream.headers.get('retry-after')) wait_time = int(stream.headers.get('retry-after'))
log.warning('Database lag exceeds max lag. ' log.warning('Database lag exceeds max lag. '
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment