Switch KeystoneFederationProtocolInfo module to OpenStackModule

Switch keystone federation_protocol_info module to the general OpenStackModule.

Change-Id: Ia42c602ad7ea01dcb27d77370cc2617ec51aaaf3
This commit is contained in:
Artem Goncharov 2021-05-20 15:52:44 +02:00
parent 6b3bf3bba0
commit 2120814356

View File

@ -45,13 +45,19 @@ EXAMPLES = '''
RETURN = ''' RETURN = '''
''' '''
from ansible.module_utils.basic import AnsibleModule from ansible_collections.openstack.cloud.plugins.module_utils.openstack import OpenStackModule
from ansible_collections.openstack.cloud.plugins.module_utils.openstack import openstack_full_argument_spec
from ansible_collections.openstack.cloud.plugins.module_utils.openstack import openstack_module_kwargs
from ansible_collections.openstack.cloud.plugins.module_utils.openstack import openstack_cloud_from_module
def normalize_protocol(protocol): class IdentityFederationProtocolInfoModule(OpenStackModule):
argument_spec = dict(
name=dict(aliases=['id']),
idp_id=dict(required=True, aliases=['idp_name']),
)
module_kwargs = dict(
supports_check_mode=True
)
def normalize_protocol(self, protocol):
""" """
Normalizes the protocol definitions so that the outputs are consistent with the Normalizes the protocol definitions so that the outputs are consistent with the
parameters parameters
@ -67,43 +73,25 @@ def normalize_protocol(protocol):
_protocol['idp_id'] = protocol['idp_id'] _protocol['idp_id'] = protocol['idp_id']
return _protocol return _protocol
def run(self):
def main():
""" Module entry point """ """ Module entry point """
argument_spec = openstack_full_argument_spec( name = self.params.get('name')
name=dict(aliases=['id']), idp = self.params.get('idp_id')
idp_id=dict(required=True, aliases=['idp_name']),
)
module_kwargs = openstack_module_kwargs(
)
module = AnsibleModule(
argument_spec,
supports_check_mode=True,
**module_kwargs
)
name = module.params.get('name')
idp = module.params.get('idp_id')
sdk, cloud = openstack_cloud_from_module(module, min_version="0.44")
if name: if name:
try: protocol = self.conn.identity.get_federation_protocol(idp, name)
protocol = cloud.identity.get_federation_protocol(idp, name) protocol = self.normalize_protocol(protocol)
protocol = normalize_protocol(protocol) self.exit_json(changed=False, protocols=[protocol])
except sdk.exceptions.ResourceNotFound:
module.fail_json(msg='Failed to find protocol')
except sdk.exceptions.OpenStackCloudException as ex:
module.fail_json(msg='Failed to get protocol: {0}'.format(str(ex)))
module.exit_json(changed=False, protocols=[protocol])
else: else:
try: protocols = list(map(self.normalize_protocol, self.conn.identity.federation_protocols(idp)))
protocols = list(map(normalize_protocol, cloud.identity.federation_protocols(idp))) self.exit_json(changed=False, protocols=protocols)
except sdk.exceptions.OpenStackCloudException as ex:
module.fail_json(msg='Failed to list protocols: {0}'.format(str(ex)))
module.exit_json(changed=False, protocols=protocols) def main():
module = IdentityFederationProtocolInfoModule()
module()
if __name__ == '__main__': if __name__ == '__main__':