104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright 2013 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 six
|
|
from six.moves.urllib import parse
|
|
|
|
from bileanclient.openstack.common.apiclient import base
|
|
|
|
OS_REQ_ID_HDR = 'x-openstack-request-id'
|
|
|
|
|
|
class User(base.Resource):
|
|
def __repr__(self):
|
|
return "<User %s>" % self._info
|
|
|
|
|
|
class UserManager(base.BaseManager):
|
|
resource_class = User
|
|
|
|
def _list(self, url, response_key, obj_class=None, body=None):
|
|
resp, body = self.client.get(url)
|
|
|
|
if obj_class is None:
|
|
obj_class = self.resource_class
|
|
|
|
data = body[response_key]
|
|
return ([obj_class(self, res, loaded=True) for res in data if res],
|
|
resp)
|
|
|
|
def list(self, **kwargs):
|
|
"""Retrieve a list of users.
|
|
|
|
:rtype: list of :class:`User`.
|
|
"""
|
|
def paginate(params, return_request_id=None):
|
|
'''Paginate users, even if more than API limit.'''
|
|
current_limit = int(params.get('limit') or 0)
|
|
url = '/users?%s' % parse.urlencode(params, True)
|
|
users, resp = self._list(url, 'users')
|
|
for user in users:
|
|
yield user
|
|
|
|
if return_request_id is not None:
|
|
return_request_id.append(resp.headers.get(OS_REQ_ID_HDR, None))
|
|
|
|
num_users = len(users)
|
|
remaining_limit = current_limit - num_users
|
|
if remaining_limit > 0 and num_users > 0:
|
|
params['limit'] = remaining_limit
|
|
params['marker'] = user.id
|
|
for user in paginate(params):
|
|
yield user
|
|
|
|
return_request_id = kwargs.get('return_req_id', None)
|
|
params = {}
|
|
if 'filters' in kwargs:
|
|
filters = kwargs.pop('filters')
|
|
params.update(filters)
|
|
|
|
for key, value in six.iteritems(kwargs):
|
|
if value:
|
|
params[key] = value
|
|
|
|
return paginate(params, return_request_id)
|
|
|
|
def get(self, user_id, return_request_id=None):
|
|
"""Get the details for a specific user.
|
|
|
|
:param user_id: ID of the user
|
|
:param return_request_id: If an empty list is provided, populate this
|
|
list with the request ID value from the header
|
|
x-openstack-request-id
|
|
"""
|
|
resp, body = self.client.get('/users/%s' % parse.quote(str(user_id)))
|
|
if return_request_id is not None:
|
|
return_request_id.append(resp.headers.get(OS_REQ_ID_HDR, None))
|
|
data = body.get('user')
|
|
return self.resource_class(self, data, loaded=True)
|
|
|
|
def action(self, user_id, **kwargs):
|
|
"""Perform specified action on user.
|
|
|
|
:param user_id: ID of the user
|
|
"""
|
|
url = '/users/%s/action' % parse.quote(str(user_id))
|
|
return_request_id = kwargs.pop('return_req_id', None)
|
|
resp, body = self.client.post(url, data=kwargs)
|
|
if return_request_id is not None:
|
|
return_request_id.append(resp.headers.get(OS_REQ_ID_HDR, None))
|
|
return self.resource_class(self, body.get('user'), loaded=True)
|