13c8e61dd9
change from double quote to single quote Change-Id: Id0b3f6c88b4338e5c86577bd584594fefc8413ae
74 lines
2.4 KiB
Python
74 lines
2.4 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.
|
|
|
|
import os
|
|
|
|
from keystoneauth1 import loading
|
|
from keystoneauth1 import plugin
|
|
|
|
|
|
class VitrageNoAuthPlugin(plugin.BaseAuthPlugin):
|
|
# noinspection PyMissingConstructor
|
|
def __init__(self, user_id, project_id, roles, endpoint):
|
|
self._user_id = user_id
|
|
self._project_id = project_id
|
|
self._endpoint = endpoint
|
|
self._roles = roles
|
|
|
|
def get_token(self, session, **kwargs):
|
|
return '<no-token-needed>'
|
|
|
|
def get_headers(self, session, **kwargs):
|
|
return {'x-user-id': self._user_id,
|
|
'x-project-id': self._project_id,
|
|
'x-roles': self._roles}
|
|
|
|
def get_user_id(self, session, **kwargs):
|
|
return self._user_id
|
|
|
|
def get_project_id(self, session, **kwargs):
|
|
return self._project_id
|
|
|
|
def get_endpoint(self, session, **kwargs):
|
|
return self._endpoint
|
|
|
|
|
|
class VitrageOpt(loading.Opt):
|
|
@property
|
|
def argparse_args(self):
|
|
return ['--%s' % o.name for o in self._all_opts]
|
|
|
|
@property
|
|
def argparse_default(self):
|
|
# select the first ENV that is not false-y or return None
|
|
for o in self._all_opts:
|
|
v = os.environ.get('VITRAGE_%s' % o.name.replace('-', '_').upper())
|
|
if v:
|
|
return v
|
|
return self.default
|
|
|
|
|
|
class VitrageNoAuthLoader(loading.BaseLoader):
|
|
plugin_class = VitrageNoAuthPlugin
|
|
|
|
def get_options(self):
|
|
options = super(VitrageNoAuthLoader, self).get_options()
|
|
options.extend([
|
|
VitrageOpt('user-id', help='User ID', required=True),
|
|
VitrageOpt('project-id', help='Project ID', required=True),
|
|
VitrageOpt('roles', help='Roles', default='admin'),
|
|
VitrageOpt('vitrage-endpoint', help='Vitrage endpoint',
|
|
dest="endpoint", required=True),
|
|
])
|
|
return options
|