ansible-collections-openstack/plugins/modules/group_assignment.py
Jakob Meng d5ab2bf33f Refactored {group,role}_assignment modules
Change-Id: I6ec79eb203d0f68661b54bc89a194c366b3574c6
2023-01-26 13:36:30 +01:00

93 lines
2.5 KiB
Python

#!/usr/bin/python
# -*- coding: utf-8 -*-
# Copyright (c) 2015 Hewlett-Packard Development Company, L.P.
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
DOCUMENTATION = r'''
---
module: group_assignment
short_description: Assign OpenStack identity users to groups
author: OpenStack Ansible SIG
description:
- Add and remove OpenStack identity (Keystone) users to/from groups.
options:
group:
description:
- Name or ID for the group.
required: true
type: str
state:
description:
- Should the user be present or absent in the group.
choices: [present, absent]
default: present
type: str
user:
description:
- Name or ID for the user.
required: true
type: str
extends_documentation_fragment:
- openstack.cloud.openstack
'''
EXAMPLES = r'''
- name: Add demo_user user to demo_group group
openstack.cloud.group_assignment:
cloud: mycloud
user: demo_user
group: demo_group
'''
from ansible_collections.openstack.cloud.plugins.module_utils.openstack import OpenStackModule
class IdentityGroupAssignment(OpenStackModule):
argument_spec = dict(
group=dict(required=True),
state=dict(default='present', choices=['absent', 'present']),
user=dict(required=True),
)
module_kwargs = dict(
supports_check_mode=True
)
def run(self):
user_name_or_id = self.params['user']
user = self.conn.identity.find_user(user_name_or_id,
ignore_missing=False)
group_name_or_id = self.params['group']
group = self.conn.identity.find_group(group_name_or_id,
ignore_missing=False)
is_user_in_group = \
self.conn.identity.check_user_in_group(user, group)
state = self.params['state']
if self.ansible.check_mode:
self.exit_json(
changed=(
(state == 'present' and not is_user_in_group)
or (state == 'absent' and is_user_in_group)))
if state == 'present' and not is_user_in_group:
self.conn.identity.add_user_to_group(user, group)
self.exit_json(changed=True)
elif state == 'absent' and is_user_in_group:
self.conn.identity.remove_user_from_group(user, group)
self.exit_json(changed=True)
else:
self.exit_json(changed=False)
def main():
module = IdentityGroupAssignment()
module()
if __name__ == '__main__':
main()