9c91c1df41
set/unset command classes should inherit from cliff.Command class. Also, this patch adds functional tests for compute agent. Change-Id: I25eafffd1167f82aa0d430628c22dee7516b1e19 Partial-Bug: 1546065
147 lines
4.2 KiB
Python
147 lines
4.2 KiB
Python
# Copyright 2013 OpenStack Foundation
|
|
#
|
|
# 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.
|
|
#
|
|
|
|
"""Agent action implementations"""
|
|
|
|
import six
|
|
|
|
from openstackclient.common import command
|
|
from openstackclient.common import utils
|
|
|
|
|
|
class CreateAgent(command.ShowOne):
|
|
"""Create compute agent command"""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(CreateAgent, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
"os",
|
|
metavar="<os>",
|
|
help="Type of OS")
|
|
parser.add_argument(
|
|
"architecture",
|
|
metavar="<architecture>",
|
|
help="Type of architecture")
|
|
parser.add_argument(
|
|
"version",
|
|
metavar="<version>",
|
|
help="Version")
|
|
parser.add_argument(
|
|
"url",
|
|
metavar="<url>",
|
|
help="URL")
|
|
parser.add_argument(
|
|
"md5hash",
|
|
metavar="<md5hash>",
|
|
help="MD5 hash")
|
|
parser.add_argument(
|
|
"hypervisor",
|
|
metavar="<hypervisor>",
|
|
help="Type of hypervisor",
|
|
default="xen")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
compute_client = self.app.client_manager.compute
|
|
args = (
|
|
parsed_args.os,
|
|
parsed_args.architecture,
|
|
parsed_args.version,
|
|
parsed_args.url,
|
|
parsed_args.md5hash,
|
|
parsed_args.hypervisor
|
|
)
|
|
agent = compute_client.agents.create(*args)._info.copy()
|
|
return zip(*sorted(six.iteritems(agent)))
|
|
|
|
|
|
class DeleteAgent(command.Command):
|
|
"""Delete compute agent command"""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(DeleteAgent, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
"id",
|
|
metavar="<id>",
|
|
help="ID of agent to delete")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
compute_client = self.app.client_manager.compute
|
|
compute_client.agents.delete(parsed_args.id)
|
|
|
|
|
|
class ListAgent(command.Lister):
|
|
"""List compute agent command"""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(ListAgent, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
"--hypervisor",
|
|
metavar="<hypervisor>",
|
|
help="Type of hypervisor")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
compute_client = self.app.client_manager.compute
|
|
columns = (
|
|
"Agent ID",
|
|
"Hypervisor",
|
|
"OS",
|
|
"Architecture",
|
|
"Version",
|
|
"Md5Hash",
|
|
"URL"
|
|
)
|
|
data = compute_client.agents.list(parsed_args.hypervisor)
|
|
return (columns,
|
|
(utils.get_item_properties(
|
|
s, columns,
|
|
) for s in data))
|
|
|
|
|
|
class SetAgent(command.Command):
|
|
"""Set compute agent command"""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(SetAgent, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
"id",
|
|
metavar="<id>",
|
|
help="ID of the agent")
|
|
parser.add_argument(
|
|
"version",
|
|
metavar="<version>",
|
|
help="Version of the agent")
|
|
parser.add_argument(
|
|
"url",
|
|
metavar="<url>",
|
|
help="URL")
|
|
parser.add_argument(
|
|
"md5hash",
|
|
metavar="<md5hash>",
|
|
help="MD5 hash")
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
compute_client = self.app.client_manager.compute
|
|
args = (
|
|
parsed_args.id,
|
|
parsed_args.version,
|
|
parsed_args.url,
|
|
parsed_args.md5hash
|
|
)
|
|
compute_client.agents.update(*args)
|