simon987/ws_bucket_client
1
0
Fork 0
mirror of https://github.com/simon987/ws_bucket_client.git synced 2025-04-19 10:16:42 +00:00
Code Issues Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
simon987 2019-03-16 11:14:06 -04:00
parent 35f86961f0
commit 52a95659e9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api.py
View File

@ -3,6 +3,7 @@ import hmac
import json import json
from email.utils import formatdate from email.utils import formatdate
from typing import BinaryIO from typing import BinaryIO
from urllib.parse import urlparse
import requests import requests
import websocket import websocket
@ -14,7 +15,7 @@ API_TIMEOUT = 10
class WsBucketApi: class WsBucketApi:
def __init__(self, url: str, secret: str): def __init__(self, url: str, secret: str):
self._url = url self._url = urlparse(url)
self._secret = secret.encode("utf8") self._secret = secret.encode("utf8")
def allocate(self, token: str, max_size: int, file_name: str): def allocate(self, token: str, max_size: int, file_name: str):
@ -50,7 +51,7 @@ class WsBucketApi:
retries = 0 retries = 0
while retries < MAX_HTTP_RETRIES: while retries < MAX_HTTP_RETRIES:
try: try:
response = requests.post("http://" + self._url + endpoint, timeout=API_TIMEOUT, response = requests.post(self._url.scheme + self._url.host + endpoint, timeout=API_TIMEOUT,
headers=headers, data=body.encode("utf8")) headers=headers, data=body.encode("utf8"))
return response return response
except Exception as e: except Exception as e: