forked from platformio/platformio-core
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
|
# Copyright (c) 2014-present PlatformIO <contact@platformio.org>
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
import requests.adapters
|
||
|
|
from requests.packages.urllib3.util.retry import Retry # pylint:disable=import-error
|
||
|
|
|
||
|
|
from platformio import app, util
|
||
|
|
from platformio.exception import PlatformioException
|
||
|
|
|
||
|
|
|
||
|
|
class RESTClientError(PlatformioException):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class RESTClient(object):
|
||
|
|
def __init__(self, base_url):
|
||
|
|
if base_url.endswith("/"):
|
||
|
|
base_url = base_url[:-1]
|
||
|
|
self.base_url = base_url
|
||
|
|
self._session = requests.Session()
|
||
|
|
self._session.headers.update({"User-Agent": app.get_user_agent()})
|
||
|
|
retry = Retry(
|
||
|
|
total=5,
|
||
|
|
backoff_factor=1,
|
||
|
|
method_whitelist=list(Retry.DEFAULT_METHOD_WHITELIST) + ["POST"],
|
||
|
|
status_forcelist=[500, 502, 503, 504],
|
||
|
|
)
|
||
|
|
adapter = requests.adapters.HTTPAdapter(max_retries=retry)
|
||
|
|
self._session.mount(base_url, adapter)
|
||
|
|
|
||
|
|
def send_request(self, method, path, **kwargs):
|
||
|
|
# check internet before and resolve issue with 60 seconds timeout
|
||
|
|
util.internet_on(raise_exception=True)
|
||
|
|
try:
|
||
|
|
response = getattr(self._session, method)(self.base_url + path, **kwargs)
|
||
|
|
except (requests.exceptions.ConnectionError, requests.exceptions.Timeout) as e:
|
||
|
|
raise RESTClientError(e)
|
||
|
|
return self.raise_error_from_response(response)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def raise_error_from_response(response, expected_codes=(200, 201, 202)):
|
||
|
|
if response.status_code in expected_codes:
|
||
|
|
try:
|
||
|
|
return response.json()
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
try:
|
||
|
|
message = response.json()["message"]
|
||
|
|
except (KeyError, ValueError):
|
||
|
|
message = response.text
|
||
|
|
raise RESTClientError(message)
|