chore: auto fixes from pre-commit.com hooks

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 1ba9939dbfa9 · 2025-03-12T17:07:11.000Z
for more information, see https://pre-commit.ci
diff --git a/plugins/module_utils/controller.py b/plugins/module_utils/controller.py
@@ -18,10 +18,10 @@
 
 class Controller:
     IDENTITY_FIELDS = {
-                        "users": "username",
-                        "role_user_assignments": "role_definition",
-                        "role_team_assignments": "role_definition"
-                    }
+        "users": "username",
+        "role_user_assignments": "role_definition",
+        "role_team_assignments": "role_definition",
+    }
     ENCRYPTED_STRING = "$encrypted$"
 
     def __init__(self, client: Client, module: AnsibleModule) -> None:
diff --git a/plugins/modules/role_team_assignment.py b/plugins/modules/role_team_assignment.py
@@ -8,10 +8,14 @@
 __metaclass__ = type
 
 
-ANSIBLE_METADATA = {'metadata_version': '1.1', 'status': ['preview'], 'supported_by': 'community'}
+ANSIBLE_METADATA = {
+    "metadata_version": "1.1",
+    "status": ["preview"],
+    "supported_by": "community",
+}
 
 
-DOCUMENTATION = '''
+DOCUMENTATION = """
 ---
 module: role_team_assignment
 author: "Tom Page (@Tompage1994)"
@@ -53,36 +57,36 @@
       type: str
 extends_documentation_fragment:
 - ansible.platform.auth
-'''
+"""
 
 
-EXAMPLES = '''
+EXAMPLES = """
 - name: Give Administrators organization admin role for org 1
   ansible.platform.role_team_assignment:
     role_definition: Organization Admin
     object_id: 1
     team: Administrators
     state: present
 ...
-'''
+"""
 
 from ansible.module_utils.basic import AnsibleModule
 
 from ..module_utils.arguments import AUTH_ARGSPEC
 from ..module_utils.client import Client
-from ..module_utils.common import lookup_resource_id
 from ..module_utils.controller import Controller
 from ..module_utils.errors import EDAError
 
+
 def main():
     # Any additional arguments that are not fields of the item can be added here
     argument_spec = dict(
-        team=dict(required=False, type='str'),
-        object_id=dict(required=False, type='int'),
-        role_definition=dict(required=True, type='str'),
-        object_ansible_id=dict(required=False, type='str'),
-        team_ansible_id=dict(required=False, type='str'),
-        state=dict(default='present', choices=['present', 'absent', 'exists']),
+        team=dict(required=False, type="str"),
+        object_id=dict(required=False, type="int"),
+        role_definition=dict(required=True, type="str"),
+        object_ansible_id=dict(required=False, type="str"),
+        team_ansible_id=dict(required=False, type="str"),
+        state=dict(default="present", choices=["present", "absent", "exists"]),
     )
 
     argument_spec.update(AUTH_ARGSPEC)
@@ -91,8 +95,8 @@ def main():
         argument_spec=argument_spec,
         supports_check_mode=True,
         mutually_exclusive=[
-            ('team', 'team_ansible_id'),
-            ('object_id', 'object_ansible_id'),
+            ("team", "team_ansible_id"),
+            ("object_id", "object_ansible_id"),
         ],
     )
 
@@ -104,64 +108,62 @@ def main():
         validate_certs=module.params.get("validate_certs"),
     )
 
-    team_param = module.params.get('team')
-    object_id = module.params.get('object_id')
-    role_definition_str = module.params.get('role_definition')
-    object_ansible_id = module.params.get('object_ansible_id')
-    team_ansible_id = module.params.get('team_ansible_id')
-    state = module.params.get('state')
+    team_param = module.params.get("team")
+    object_id = module.params.get("object_id")
+    role_definition_str = module.params.get("role_definition")
+    object_ansible_id = module.params.get("object_ansible_id")
+    team_ansible_id = module.params.get("team_ansible_id")
+    state = module.params.get("state")
 
     controller = Controller(client, module)
 
-    role_definition = controller.get_exactly_one('role_definitions', allow_none=False, name=role_definition_str)
-    team = controller.get_exactly_one('teams', allow_none=True, name=team_param)
+    role_definition = controller.get_exactly_one(
+        "role_definitions", allow_none=False, name=role_definition_str
+    )
+    team = controller.get_exactly_one("teams", allow_none=True, name=team_param)
 
-    new_item = {
-        'role_definition': role_definition['id']
-    }
+    new_item = {"role_definition": role_definition["id"]}
 
     if object_id is not None:
-        new_item['object_id'] = object_id
+        new_item["object_id"] = object_id
     if team is not None:
-        new_item['team'] = team['id']
+        new_item["team"] = team["id"]
     if object_ansible_id is not None:
-        new_item['object_ansible_id'] = object_ansible_id
+        new_item["object_ansible_id"] = object_ansible_id
     if team_ansible_id is not None:
-        new_item['team_ansible_id'] = team_ansible_id
+        new_item["team_ansible_id"] = team_ansible_id
 
     try:
         assignment = controller.get_one_or_many(
-                                'role_team_assignments',
-                                **{'data': new_item}
-                            )
+            "role_team_assignments", **{"data": new_item}
+        )
         assignment = assignment[0] if len(assignment) == 1 else None
     except EDAError as eda_err:
         module.fail_json(msg=str(eda_err))
 
-    if state == 'absent':
+    if state == "absent":
         try:
             ret = controller.delete_if_needed(
-                    assignment,
-                    endpoint='role_team_assignments'
-                )
+                assignment, endpoint="role_team_assignments"
+            )
         except EDAError as eda_err:
             module.fail_json(msg=str(eda_err))
 
-    elif state == 'present' and assignment is None:
+    elif state == "present" and assignment is None:
         try:
             ret = controller.create_if_needed(
-                        new_item=new_item,
-                        endpoint='role_team_assignments',
-                        item_type='role_team_assignment',
-                    )
+                new_item=new_item,
+                endpoint="role_team_assignments",
+                item_type="role_team_assignment",
+            )
         except EDAError as eda_err:
             module.fail_json(msg=str(eda_err))
 
     else:
-        ret = {'changed': False}
+        ret = {"changed": False}
 
     module.exit_json(**ret)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()
diff --git a/plugins/modules/role_user_assignment.py b/plugins/modules/role_user_assignment.py
@@ -8,10 +8,14 @@
 __metaclass__ = type
 
 
-ANSIBLE_METADATA = {'metadata_version': '1.1', 'status': ['preview'], 'supported_by': 'community'}
+ANSIBLE_METADATA = {
+    "metadata_version": "1.1",
+    "status": ["preview"],
+    "supported_by": "community",
+}
 
 
-DOCUMENTATION = '''
+DOCUMENTATION = """
 ---
 module: role_user_assignment
 author: "Tom Page (@Tompage1994)"
@@ -53,36 +57,36 @@
       type: str
 extends_documentation_fragment:
 - ansible.platform.auth
-'''
+"""
 
 
-EXAMPLES = '''
+EXAMPLES = """
 - name: Give Administrators organization admin role for org 1
   ansible.platform.role_user_assignment:
     role_definition: Organization Admin
     object_id: 1
     user: Administrators
     state: present
 ...
-'''
+"""
 
 from ansible.module_utils.basic import AnsibleModule
 
 from ..module_utils.arguments import AUTH_ARGSPEC
 from ..module_utils.client import Client
-from ..module_utils.common import lookup_resource_id
 from ..module_utils.controller import Controller
 from ..module_utils.errors import EDAError
 
+
 def main():
     # Any additional arguments that are not fields of the item can be added here
     argument_spec = dict(
-        user=dict(required=False, type='str'),
-        object_id=dict(required=False, type='int'),
-        role_definition=dict(required=True, type='str'),
-        object_ansible_id=dict(required=False, type='str'),
-        user_ansible_id=dict(required=False, type='str'),
-        state=dict(default='present', choices=['present', 'absent', 'exists']),
+        user=dict(required=False, type="str"),
+        object_id=dict(required=False, type="int"),
+        role_definition=dict(required=True, type="str"),
+        object_ansible_id=dict(required=False, type="str"),
+        user_ansible_id=dict(required=False, type="str"),
+        state=dict(default="present", choices=["present", "absent", "exists"]),
     )
 
     argument_spec.update(AUTH_ARGSPEC)
@@ -91,13 +95,13 @@ def main():
         argument_spec=argument_spec,
         supports_check_mode=True,
         mutually_exclusive=[
-            ('user', 'user_ansible_id'),
-            ('object_id', 'object_ansible_id'),
+            ("user", "user_ansible_id"),
+            ("object_id", "object_ansible_id"),
         ],
         required_one_of=[
-            ('user', 'user_ansible_id'),
-            ('object_id', 'object_ansible_id'),
-        ]
+            ("user", "user_ansible_id"),
+            ("object_id", "object_ansible_id"),
+        ],
     )
 
     client = Client(
@@ -108,64 +112,62 @@ def main():
         validate_certs=module.params.get("validate_certs"),
     )
 
-    user_param = module.params.get('user')
-    object_id = module.params.get('object_id')
-    role_definition_str = module.params.get('role_definition')
-    object_ansible_id = module.params.get('object_ansible_id')
-    user_ansible_id = module.params.get('user_ansible_id')
-    state = module.params.get('state')
+    user_param = module.params.get("user")
+    object_id = module.params.get("object_id")
+    role_definition_str = module.params.get("role_definition")
+    object_ansible_id = module.params.get("object_ansible_id")
+    user_ansible_id = module.params.get("user_ansible_id")
+    state = module.params.get("state")
 
     controller = Controller(client, module)
 
-    role_definition = controller.get_exactly_one('role_definitions', name=role_definition_str)
-    user = controller.get_exactly_one('users', name=user_param)
+    role_definition = controller.get_exactly_one(
+        "role_definitions", name=role_definition_str
+    )
+    user = controller.get_exactly_one("users", name=user_param)
 
-    new_item = {
-        'role_definition': role_definition['id']
-    }
+    new_item = {"role_definition": role_definition["id"]}
 
     if object_id is not None:
-        new_item['object_id'] = object_id
+        new_item["object_id"] = object_id
     if user is not None:
-        new_item['user'] = user['id'] if user else None
+        new_item["user"] = user["id"] if user else None
     if object_ansible_id is not None:
-        new_item['object_ansible_id'] = object_ansible_id
+        new_item["object_ansible_id"] = object_ansible_id
     if user_ansible_id is not None:
-        new_item['user_ansible_id'] = user_ansible_id
+        new_item["user_ansible_id"] = user_ansible_id
 
     try:
         assignment = controller.get_one_or_many(
-                                'role_user_assignments',
-                                **{'data': new_item}
-                            )
+            "role_user_assignments", **{"data": new_item}
+        )
         assignment = assignment[0] if len(assignment) == 1 else None
     except EDAError as eda_err:
         module.fail_json(msg=str(eda_err))
 
-    if state == 'absent':
+    if state == "absent":
         try:
             ret = controller.delete_if_needed(
-                    assignment,
-                    endpoint='role_user_assignments'
-                )
+                assignment, endpoint="role_user_assignments"
+            )
         except EDAError as eda_err:
             module.fail_json(msg=str(eda_err))
 
-    elif state == 'present' and assignment is None:
+    elif state == "present" and assignment is None:
         try:
             ret = controller.create_if_needed(
-                        new_item=new_item,
-                        endpoint='role_user_assignments',
-                        item_type='role_user_assignment',
-                    )
+                new_item=new_item,
+                endpoint="role_user_assignments",
+                item_type="role_user_assignment",
+            )
         except EDAError as eda_err:
             module.fail_json(msg=str(eda_err))
 
     else:
-        ret = {'changed': False}
+        ret = {"changed": False}
 
     module.exit_json(**ret)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()
diff --git a/tests/integration/targets/role_team_assignment/main.yml b/tests/integration/targets/role_team_assignment/main.yml
@@ -9,7 +9,7 @@
   block:
     - name: Set the team to use for the test
       set_fact:
-        team_name: "TBC"  # A team will need to exist in the Platform instance for this to work. 
+        team_name: "TBC"  # A team will need to exist in the Platform instance for this to work.
 
     # CREATE
     - name: Create role_team_assignment in check mode
