bt_blackked

1 files changed, 56 insertions, 51 deletions

diff --git a/plugins/modules/vyos_lldp_interface.py b/plugins/modules/vyos_lldp_interface.py
index aca3224..23d1eab 100644
--- a/plugins/modules/vyos_lldp_interface.py
+++ b/plugins/modules/vyos_lldp_interface.py
@@ -20,11 +20,12 @@
 #
 
 ANSIBLE_METADATA = {
-    'metadata_version': '1.1',
-    'status': ['preview'],
-    'supported_by': 'network'
+    "metadata_version": "1.1",
+    "status": ["preview"],
+    "supported_by": "network",
 }
 
+
 DOCUMENTATION = """
 ---
 module: vyos_lldp_interface
@@ -94,16 +95,19 @@ from copy import deepcopy
 
 from ansible.module_utils.basic import AnsibleModule
 from ansible.module_utils.network.common.utils import remove_default_spec
-from ansible_collections.vyos.vyos.plugins.module_utils.network. \
-  vyos.vyos import get_config, load_config
+from ansible_collections.vyos.vyos.plugins.module_utils.network.vyos.vyos import (
+    get_config,
+    load_config,
+)
 
-from ansible_collections.vyos.vyos.plugins.module_utils.network. \
-  vyos.vyos import vyos_argument_spec
+from ansible_collections.vyos.vyos.plugins.module_utils.network.vyos.vyos import (
+    vyos_argument_spec,
+)
 
 
 def search_obj_in_list(name, lst):
     for o in lst:
-        if o['name'] == name:
+        if o["name"] == name:
             return o
 
     return None
@@ -114,28 +118,28 @@ def map_obj_to_commands(updates, module):
     want, have = updates
 
     for w in want:
-        name = w['name']
-        state = w['state']
+        name = w["name"]
+        state = w["state"]
 
         obj_in_have = search_obj_in_list(name, have)
 
-        if state == 'absent' and obj_in_have:
-            commands.append('delete service lldp interface ' + name)
-        elif state in ('present', 'enabled'):
+        if state == "absent" and obj_in_have:
+            commands.append("delete service lldp interface " + name)
+        elif state in ("present", "enabled"):
             if not obj_in_have:
-                commands.append('set service lldp interface ' + name)
-            elif obj_in_have and obj_in_have[
-                    'state'] == 'disabled' and state == 'enabled':
-                commands.append('delete service lldp interface ' + name +
-                                ' disable')
-        elif state == 'disabled':
+                commands.append("set service lldp interface " + name)
+            elif (
+                obj_in_have
+                and obj_in_have["state"] == "disabled"
+                and state == "enabled"
+            ):
+                commands.append("delete service lldp interface " + name + " disable")
+        elif state == "disabled":
             if not obj_in_have:
-                commands.append('set service lldp interface ' + name)
-                commands.append('set service lldp interface ' + name +
-                                ' disable')
-            elif obj_in_have and obj_in_have['state'] != 'disabled':
-                commands.append('set service lldp interface ' + name +
-                                ' disable')
+                commands.append("set service lldp interface " + name)
+                commands.append("set service lldp interface " + name + " disable")
+            elif obj_in_have and obj_in_have["state"] != "disabled":
+                commands.append("set service lldp interface " + name + " disable")
 
     return commands
 
@@ -150,13 +154,13 @@ def map_config_to_obj(module):
         splitted_line = i.split()
 
         if len(splitted_line) > 5:
-            new_obj = {'name': splitted_line[4]}
+            new_obj = {"name": splitted_line[4]}
 
             if splitted_line[5] == "'disable'":
-                new_obj['state'] = 'disabled'
+                new_obj["state"] = "disabled"
         else:
-            new_obj = {'name': splitted_line[4][1:-1]}
-            new_obj['state'] = 'present'
+            new_obj = {"name": splitted_line[4][1:-1]}
+            new_obj["state"] = "present"
 
         obj.append(new_obj)
 
@@ -166,7 +170,7 @@ def map_config_to_obj(module):
 def map_params_to_obj(module):
     obj = []
 
-    aggregate = module.params.get('aggregate')
+    aggregate = module.params.get("aggregate")
     if aggregate:
         for item in aggregate:
             for key in item:
@@ -175,10 +179,7 @@ def map_params_to_obj(module):
 
             obj.append(item.copy())
     else:
-        obj.append({
-            'name': module.params['name'],
-            'state': module.params['state']
-        })
+        obj.append({"name": module.params["name"], "state": module.params["state"]})
 
     return obj
 
@@ -188,50 +189,54 @@ def main():
     """
     element_spec = dict(
         name=dict(),
-        state=dict(default='present',
-                   choices=['present', 'absent', 'enabled', 'disabled']))
+        state=dict(
+            default="present", choices=["present", "absent", "enabled", "disabled"]
+        ),
+    )
 
     aggregate_spec = deepcopy(element_spec)
-    aggregate_spec['name'] = dict(required=True)
+    aggregate_spec["name"] = dict(required=True)
 
     # remove default in aggregate spec, to handle common arguments
     remove_default_spec(aggregate_spec)
 
-    argument_spec = dict(aggregate=dict(type='list',
-                                        elements='dict',
-                                        options=aggregate_spec), )
+    argument_spec = dict(
+        aggregate=dict(type="list", elements="dict", options=aggregate_spec)
+    )
 
     argument_spec.update(element_spec)
     argument_spec.update(vyos_argument_spec)
 
-    required_one_of = [['name', 'aggregate']]
-    mutually_exclusive = [['name', 'aggregate']]
+    required_one_of = [["name", "aggregate"]]
+    mutually_exclusive = [["name", "aggregate"]]
 
-    module = AnsibleModule(argument_spec=argument_spec,
-                           required_one_of=required_one_of,
-                           mutually_exclusive=mutually_exclusive,
-                           supports_check_mode=True)
+    module = AnsibleModule(
+        argument_spec=argument_spec,
+        required_one_of=required_one_of,
+        mutually_exclusive=mutually_exclusive,
+        supports_check_mode=True,
+    )
 
     warnings = list()
 
-    result = {'changed': False}
+    result = {"changed": False}
 
     if warnings:
-        result['warnings'] = warnings
+        result["warnings"] = warnings
 
     want = map_params_to_obj(module)
     have = map_config_to_obj(module)
 
     commands = map_obj_to_commands((want, have), module)
-    result['commands'] = commands
+    result["commands"] = commands
 
     if commands:
         commit = not module.check_mode
         load_config(module, commands, commit=commit)
-        result['changed'] = True
+        result["changed"] = True
 
     module.exit_json(**result)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()