Updates controller module arg spec

nautobot · Aug 14, 2024 · 4c6d4a8 · 4c6d4a8
1 parent bc92c8c
commit 4c6d4a8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/plugins/modules/controller.py b/plugins/modules/controller.py
@@ -127,7 +127,11 @@
   type: str
 """
 
-from ansible_collections.networktocode.nautobot.plugins.module_utils.utils import NAUTOBOT_ARG_SPEC
+from ansible_collections.networktocode.nautobot.plugins.module_utils.utils import (
+    NAUTOBOT_ARG_SPEC,
+    TAGS_ARG_SPEC,
+    CUSTOM_FIELDS_ARG_SPEC,
+)
 from ansible_collections.networktocode.nautobot.plugins.module_utils.dcim import (
     NautobotDcimModule,
     NB_CONTROLLERS,
@@ -141,6 +145,8 @@ def main():
     Main entry point for module execution
     """
     argument_spec = deepcopy(NAUTOBOT_ARG_SPEC)
+    argument_spec.update(deepcopy(TAGS_ARG_SPEC))
+    argument_spec.update(deepcopy(CUSTOM_FIELDS_ARG_SPEC))
     argument_spec.update(
         dict(
             name=dict(required=True, type="str"),
@@ -152,8 +158,6 @@ def main():
             tenant=dict(required=False, type="raw"),
             platform=dict(required=False, type="raw"),
             status=dict(required=False, type="raw"),
-            tags=dict(required=False, type="list", elements="raw"),
-            custom_fields=dict(required=False, type="dict"),
             controller_device_redundancy_group=dict(required=False, type="str"),
         )
     )