From c6a4a5fac45d8b87195f677ed11d37240a992c19 Mon Sep 17 00:00:00 2001 From: shirshasa Date: Mon, 19 Apr 2021 11:18:26 +0300 Subject: [PATCH] epub converter: update access.py --- src/access.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/src/access.py b/src/access.py index f481aea..322d28b 100644 --- a/src/access.py +++ b/src/access.py @@ -134,6 +134,28 @@ class Access: return img_url + def send_image_by_bytes(self, img_file_path, img_content: bytes, doc_id): + if self.is_time_for_refreshing(): + self.refresh_token() + + self.refreshing.wait() + + files = { + 'image': (os.path.basename(img_file_path), img_content) + } + response = requests.post(f'{self.url}/doc-convert/image', files=files, headers=self.headers) + + if response.status_code == 400: + self.update_status(doc_id, self.ERROR) + raise Exception(f'400 Bad request: {response.json()["message"]}.') + elif response.status_code == 200: + img_url = response.json()['imageUrl'] + else: + self.update_status(doc_id, self.ERROR) + raise Exception(f'{response.status_code}') + + return img_url + def send_book(self, doc_id, content): if self.is_time_for_refreshing(): self.refresh_token()