PyKMIP/kmip/demos/pie/set_attribute.py
Peter Hamilton e5281cbd35 Add ProxyKmipClient demos for the new attribute operations
This change adds ProxyKmipClient demos for the new Set, Modify,
and Delete Attribute operations. The demos are not intended for
general use but instead to show how the client can be used to
execute these new operations against a KMIP server. See the
demo scripts themselves for more information.

Partially implements #547
2019-12-13 14:04:07 -05:00

63 lines
2.0 KiB
Python

# Copyright (c) 2019 The Johns Hopkins University/Applied Physics Laboratory
# All Rights Reserved.
#
# 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
import sys
from kmip.core.factories import attributes
from kmip.core import enums
from kmip.demos import utils
from kmip.pie import client
# NOTE: This demo script shows how to set the Sensitive attribute on
# the user-specified object. The server must support KMIP 2.0, since
# the SetAttribute operation is KMIP 2.0+ only and the Sensitive
# attribute is KMIP 1.4+ only. Otherwise, the client call to
# set_attribute will fail.
if __name__ == '__main__':
logger = utils.build_console_logger(logging.INFO)
parser = utils.build_cli_parser(enums.Operation.SET_ATTRIBUTE)
opts, args = parser.parse_args(sys.argv[1:])
if opts.uuid is None:
logger.error("No UUID provided, existing early from demo.")
sys.exit()
factory = attributes.AttributeFactory()
with client.ProxyKmipClient(
config=opts.config,
config_file=opts.config_file,
kmip_version=enums.KMIPVersion.KMIP_2_0
) as c:
try:
object_id = c.set_attribute(
unique_identifier=opts.uuid,
attribute_name="Sensitive",
attribute_value=True
)
logger.info(
"Successfully set the 'Sensitive' attribute on object: "
"{}".format(
object_id
)
)
except Exception as e:
logger.error(e)