fb22e98262
Change-Id: I63fd903761f852aad03a6f24cebaab0ae7c52a55 Depends-On: I44bf27413af2133451cbd84d908631ce6a2fe1f7
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
# 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.
|
|
|
|
from keystoneauth1 import adapter
|
|
from oslo_utils import importutils
|
|
from osprofiler import web
|
|
|
|
from aodhclient import exceptions
|
|
|
|
|
|
def Client(version, *args, **kwargs):
|
|
module = 'aodhclient.v%s.client' % version
|
|
module = importutils.import_module(module)
|
|
client_class = getattr(module, 'Client')
|
|
return client_class(*args, **kwargs)
|
|
|
|
|
|
class SessionClient(adapter.Adapter):
|
|
def request(self, url, method, **kwargs):
|
|
kwargs.setdefault('headers', kwargs.get('headers', {}))
|
|
# NOTE(sileht): The standard call raises errors from
|
|
# keystoneauth, where we need to raise the aodhclient errors.
|
|
raise_exc = kwargs.pop('raise_exc', True)
|
|
kwargs['headers'].update(web.get_trace_id_headers())
|
|
resp = super(SessionClient, self).request(url,
|
|
method,
|
|
raise_exc=False,
|
|
**kwargs)
|
|
|
|
if raise_exc and resp.status_code >= 400:
|
|
raise exceptions.from_response(resp, url, method)
|
|
return resp
|