From 2cd3d217a7bc57cc7a11566d8ed60f48c9ea9e9e Mon Sep 17 00:00:00 2001 From: simon987 Date: Sat, 23 Mar 2019 15:17:18 -0400 Subject: [PATCH] Update api --- api.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api.py b/api.py index 84359a2..98d224a 100644 --- a/api.py +++ b/api.py @@ -19,11 +19,13 @@ class WsBucketApi: self._secret = secret.encode("utf8") self._ws_scheme = ws_scheme - def allocate(self, token: str, max_size: int, file_name: str): + def allocate(self, token: str, max_size: int, file_name: str, upload_hook: str, to_dispose_date: int): return self._http_post("/slot", { "token": token, "max_size": max_size, "file_name": file_name, + "upload_hook": upload_hook, + "to_dispose_date": to_dispose_date, }) def read(self, token: str): @@ -31,7 +33,7 @@ class WsBucketApi: def upload(self, token: str, stream: BinaryIO, max_size: int): ws = websocket.WebSocket() - ws.connect(self._ws_scheme + "://" + self._url + "/upload", header={ + ws.connect(self._ws_scheme + "://" + self._url.netloc + "/upload", header={ "X-Upload-Token": token, })