4723ef86f1
Change-Id: I40e6be2d8d5219bdf81bd9359d46d4f9464e1255 Depends-On: Ic934acbe1340a3e00361f2709f34725e0e4aa3ba
43 lines
1.7 KiB
Python
43 lines
1.7 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 pankoclient import exceptions
|
|
|
|
|
|
def Client(version, *args, **kwargs):
|
|
module = 'pankoclient.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 pankoclient 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, resp.json()
|