2864edd6d1
Split apiclient from freezer-agent repo Use cliff as a cli for freezer Implements bp: freezerclient
128 lines
4.0 KiB
Python
128 lines
4.0 KiB
Python
# (c) Copyright 2014-2016 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 logging
|
|
|
|
from cliff.command import Command
|
|
from cliff.lister import Lister
|
|
from cliff.show import ShowOne
|
|
|
|
from freezerclient import exceptions
|
|
from freezerclient.utils import doc_from_json_file
|
|
|
|
|
|
logging = logging.getLogger(__name__)
|
|
|
|
|
|
class ClientShow(ShowOne):
|
|
"""Show a single client"""
|
|
def get_parser(self, prog_name):
|
|
parser = super(ClientShow, self).get_parser(prog_name)
|
|
parser.add_argument(dest='client_id',
|
|
help='ID of the client')
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
client = self.app.client.clients.get(parsed_args.client_id)
|
|
|
|
if not client:
|
|
raise exceptions.ApiClientException('Client not found')
|
|
|
|
column = (
|
|
'Client ID',
|
|
'Client UUID',
|
|
'hostname',
|
|
'description'
|
|
)
|
|
data = (
|
|
client.get('client', {}).get('client_id'),
|
|
client.get('uuid'),
|
|
client.get('client', {}).get('hostname'),
|
|
client.get('client', {}).get('description', '')
|
|
)
|
|
|
|
return column, data
|
|
|
|
|
|
class ClientList(Lister):
|
|
"""List of clients registered in the api"""
|
|
def get_parser(self, prog_name):
|
|
parser = super(ClientList, self).get_parser(prog_name)
|
|
|
|
parser.add_argument(
|
|
'--limit',
|
|
dest='limit',
|
|
default=100,
|
|
help='Specify a limit for search query',
|
|
)
|
|
|
|
parser.add_argument(
|
|
'--offset',
|
|
dest='offset',
|
|
default=0,
|
|
help='',
|
|
)
|
|
|
|
parser.add_argument(
|
|
'--search',
|
|
dest='search',
|
|
default='',
|
|
help='Define a filter for the query',
|
|
)
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
clients = self.app.client.clients.list(limit=parsed_args.limit,
|
|
offset=parsed_args.offset,
|
|
search=parsed_args.search)
|
|
|
|
return (('Client ID', 'uuid', 'hostname', 'description'),
|
|
((client.get('client', {}).get('client_id'),
|
|
client.get('uuid'),
|
|
client.get('client', {}).get('hostname'),
|
|
client.get('client', {}).get('description', '')
|
|
) for client in clients))
|
|
|
|
|
|
class ClientDelete(Command):
|
|
"""Delete a client from the api"""
|
|
def get_parser(self, prog_name):
|
|
parser = super(ClientDelete, self).get_parser(prog_name)
|
|
parser.add_argument(dest='client_id',
|
|
help='ID of the client')
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
self.app.client.clients.delete(parsed_args.client_id)
|
|
logging.info('Client {0} deleted'.format(parsed_args.client_id))
|
|
|
|
|
|
class ClientRegister(Command):
|
|
"""Register a new client"""
|
|
def get_parser(self, prog_name):
|
|
parser = super(ClientRegister, self).get_parser(prog_name)
|
|
parser.add_argument('--file',
|
|
dest='file',
|
|
help='Path to json file with the client')
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
client = doc_from_json_file(parsed_args.file)
|
|
try:
|
|
client_id = self.app.client.clients.create(client)
|
|
except Exception as err:
|
|
raise exceptions.ApiClientException(err.message)
|
|
else:
|
|
logging.info("Client {0} registered".format(client_id))
|