From 52a95659e916faf356f615ff254b6373a00e543d Mon Sep 17 00:00:00 2001
From: simon987 <fortier.simon@protonmail.com>
Date: Sat, 16 Mar 2019 11:14:06 -0400
Subject: [PATCH] bugfix

---
 api.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/api.py b/api.py
index 8b18cf1..f5a5103 100644
--- a/api.py
+++ b/api.py
@@ -3,6 +3,7 @@ import hmac
 import json
 from email.utils import formatdate
 from typing import BinaryIO
+from urllib.parse import urlparse
 
 import requests
 import websocket
@@ -14,7 +15,7 @@ API_TIMEOUT = 10
 class WsBucketApi:
 
     def __init__(self, url: str, secret: str):
-        self._url = url
+        self._url = urlparse(url)
         self._secret = secret.encode("utf8")
 
     def allocate(self, token: str, max_size: int, file_name: str):
@@ -50,7 +51,7 @@ class WsBucketApi:
         retries = 0
         while retries < MAX_HTTP_RETRIES:
             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"))
                 return response
             except Exception as e: