Add "set" CLI command
Change-Id: Iabec422f15a00813a8676afc2da1d90292c42a1e
This commit is contained in:
Yuriy Taraday
parent
b9db11db4f
commit
065d95b598
@ -50,3 +50,4 @@ nailgun.extensions =
|
||||
tuning_box = tuning_box.nailgun:Extension
|
||||
tuning_box.cli =
|
||||
get = tuning_box.cli.get:Get
|
||||
set = tuning_box.cli.set:Set
|
||||
|
||||
120
tuning_box/cli/set.py
Normal file
120
tuning_box/cli/set.py
Normal file
@ -0,0 +1,120 @@
|
||||
# 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 json
|
||||
|
||||
import yaml
|
||||
|
||||
from tuning_box.cli import base
|
||||
|
||||
|
||||
class Set(base.ResourceCommand):
|
||||
url_last_part = 'values'
|
||||
|
||||
def get_parser(self, *args, **kwargs):
|
||||
parser = super(Set, self).get_parser(*args, **kwargs)
|
||||
parser.add_argument(
|
||||
'--key',
|
||||
type=str,
|
||||
help="Name of key to set in the resource",
|
||||
)
|
||||
parser.add_argument(
|
||||
'--value',
|
||||
type=str,
|
||||
help="Value for a key to set in the resource",
|
||||
)
|
||||
parser.add_argument(
|
||||
'--type',
|
||||
choices=('null', 'int', 'str', 'json', 'yaml', 'bool'),
|
||||
help="Tyep of value passed in --value",
|
||||
)
|
||||
parser.add_argument(
|
||||
'--format',
|
||||
choices=('json', 'yaml'),
|
||||
help="Format of data passed to stdin",
|
||||
)
|
||||
return parser
|
||||
|
||||
def verify_arguments(self, parsed_args):
|
||||
if parsed_args.key is None: # no key
|
||||
if parsed_args.value is not None or parsed_args.type is not None:
|
||||
raise Exception("--value and --type arguments make sense only "
|
||||
"with --key argument.")
|
||||
if parsed_args.format is None:
|
||||
raise Exception("Please specify format of data passed to stdin"
|
||||
" to replace whole resource data.")
|
||||
elif parsed_args.value is not None: # have key and value
|
||||
if parsed_args.format is not None:
|
||||
raise Exception("You shouldn't specify --format if you pass "
|
||||
"value in command line, specify --type "
|
||||
"instead.")
|
||||
if parsed_args.type == 'null':
|
||||
raise Exception("You shouldn't specify a value for 'null' type"
|
||||
" because there can be only one.")
|
||||
if parsed_args.type is None:
|
||||
raise Exception("Please specify type of value passed in "
|
||||
"--value argument to properly represent it"
|
||||
" in the storage.")
|
||||
elif parsed_args.type != 'null': # have key but no value
|
||||
if parsed_args.type is not None:
|
||||
raise Exception("--type specifies type for value provided in "
|
||||
"--value but there is not --value argument")
|
||||
if parsed_args.format is None:
|
||||
raise Exception("Please specify format of data passed to stdin"
|
||||
" to replace the key.")
|
||||
|
||||
def get_value_to_set(self, parsed_args):
|
||||
type_ = parsed_args.type
|
||||
if type_ == 'null':
|
||||
return None
|
||||
elif type_ == 'bool':
|
||||
if parsed_args.value.lower() in ('1', 'true'):
|
||||
return True
|
||||
elif parsed_args.value.lower() in ('0', 'false'):
|
||||
return False
|
||||
else:
|
||||
raise Exception(
|
||||
"Bad value for 'bool' type: '{}'. Should be one of '0', "
|
||||
"'1', 'false', 'true'.".format(parsed_args.value))
|
||||
elif type_ == 'int':
|
||||
return int(parsed_args.value)
|
||||
elif type_ == 'str':
|
||||
return parsed_args.value
|
||||
elif type_ == 'json':
|
||||
return json.loads(parsed_args.value)
|
||||
elif type_ == 'yaml':
|
||||
return yaml.safe_load(parsed_args.value)
|
||||
elif type_ is None:
|
||||
if parsed_args.format == 'json':
|
||||
return json.load(self.app.stdin)
|
||||
elif parsed_args.format == 'yaml':
|
||||
docs_gen = yaml.safe_load_all(self.app.stdin)
|
||||
doc = next(docs_gen)
|
||||
guard = object()
|
||||
if next(docs_gen, guard) is not guard:
|
||||
self.app.stderr.write("Warning: will use only first "
|
||||
"document from YAML stream")
|
||||
return doc
|
||||
assert False, "Shouldn't get here"
|
||||
|
||||
def take_action(self, parsed_args):
|
||||
self.verify_arguments(parsed_args)
|
||||
value = self.get_value_to_set(parsed_args)
|
||||
|
||||
client = self.get_client()
|
||||
resource_url = self.get_resource_url(parsed_args, self.url_last_part)
|
||||
if parsed_args.key is not None:
|
||||
resource = client.get(resource_url)
|
||||
resource[parsed_args.key] = value
|
||||
else:
|
||||
resource = value
|
||||
client.put(resource_url, resource)
|
||||
@ -41,3 +41,6 @@ class HTTPClient(object):
|
||||
|
||||
def get(self, url, params=None):
|
||||
return self.request('GET', url, params=params)
|
||||
|
||||
def put(self, url, body):
|
||||
return self.request('PUT', url, json=body)
|
||||
|
||||
@ -187,3 +187,46 @@ class TestGet(testscenarios.WithScenarios, _BaseCLITest):
|
||||
)
|
||||
self.cli.run(self.args.split())
|
||||
self.assertEqual(self.expected_result, self.cli.stdout.getvalue())
|
||||
|
||||
|
||||
class TestSet(testscenarios.WithScenarios, _BaseCLITest):
|
||||
scenarios = [
|
||||
(s[0],
|
||||
dict(zip(('args', 'expected_body', 'should_get', 'stdin'), s[1])))
|
||||
for s in [
|
||||
('json', ('--format json', {'a': 3}, False, '{"a": 3}')),
|
||||
('yaml', ('--format yaml', {'a': 3}, False, 'a: 3')),
|
||||
('key,json', ('--key b --format json', {'a': 1, 'b': {'a': 3}},
|
||||
True, '{"a": 3}')),
|
||||
('key,yaml', ('--key b --format yaml', {'a': 1, 'b': {'a': 3}},
|
||||
True, 'a: 3')),
|
||||
('key,null', ('--key b --type null', {'a': 1, 'b': None})),
|
||||
('key,str', ('--key b --type str --value 4', {'a': 1, 'b': '4'})),
|
||||
]
|
||||
]
|
||||
|
||||
args = None
|
||||
expected_body = None
|
||||
should_get = True
|
||||
stdin = None
|
||||
|
||||
def test_set(self):
|
||||
url = self.BASE_URL + '/environments/1/lvl1/value1/resources/1/values'
|
||||
self.req_mock.put(url)
|
||||
if self.should_get:
|
||||
self.req_mock.get(
|
||||
url,
|
||||
headers={'Content-Type': 'application/json'},
|
||||
json={'a': 1, 'b': True},
|
||||
)
|
||||
args = ("set --env 1 --level lvl1=value1 --resource 1 " +
|
||||
self.args).split()
|
||||
if self.stdin:
|
||||
self.cli.stdin.write(self.stdin)
|
||||
self.cli.stdin.seek(0)
|
||||
self.cli.run(args)
|
||||
req_history = self.req_mock.request_history
|
||||
if self.should_get:
|
||||
self.assertEqual('GET', req_history[0].method)
|
||||
self.assertEqual('PUT', req_history[-1].method)
|
||||
self.assertEqual(self.expected_body, req_history[-1].json())
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user