nxos module cleanup (#5065)

* Fix imports on nxos_bgp* modules

* Fix imports on nxos_evpn* modules

* Cleanup issues for nxos_facts

* Shuffle imports for nxos_template

* Fix imports on nxos_ospf* modules

* Fix nxos_hsrp

As get_hsrp_groups_in_devices is not actually called anywhere, I presume this
change is reasonable.

* Fix imports on nxos_interface* modules

* Update nxos_static_route imports

* update nxos_vrf

* Update nxos_config imports

network/nxos/_nxos_template.py

@@ -109,8 +109,9 @@ responses:
   type: list
   sample: ['...', '...']
 """
+import ansible.module_utils.nxos
 from ansible.module_utils.netcfg import NetworkConfig, dumps
-from ansible.module_utils.nxos import NetworkModule, NetworkError
+from ansible.module_utils.network import NetworkModule
 
 def get_config(module):
     config = module.params['config'] or dict()

network/nxos/nxos_bgp.py

@@ -360,13 +360,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_bgp_af.py

@@ -299,13 +299,12 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
 
 
 def to_list(val):

network/nxos/nxos_bgp_neighbor.py

@@ -244,13 +244,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_bgp_neighbor_af.py

@@ -321,13 +321,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_config.py

@@ -208,11 +208,11 @@ backup_path:
   type: path
   sample: /playbooks/ansible/backup/nxos_config.2016-07-16@22:28:34
 """
-import time
 
-from ansible.module_utils.netcfg import NetworkConfig, dumps
-from ansible.module_utils.nxos import NetworkModule, NetworkError
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
+from ansible.module_utils.network import NetworkModule, NetworkError
+from ansible.module_utils.netcfg import NetworkConfig, dumps
 
 def check_args(module, warnings):
     if module.params['force']:

network/nxos/nxos_evpn_global.py

@@ -72,13 +72,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_evpn_vni.py

@@ -126,14 +126,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
 from ansible.module_utils.shell import ShellError
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
 
 
 def to_list(val):

network/nxos/nxos_facts.py

@@ -172,9 +172,11 @@ vlan_list:
 """
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcli import CommandRunner, AddCommandError
-from ansible.module_utils.nxos import NetworkModule, NetworkError
+from ansible.module_utils.network import NetworkModule, NetworkError
+from ansible.module_utils.six import iteritems
 
 
 def add_command(runner, command, output=None):
@@ -193,6 +195,9 @@ class FactsBase(object):
         self.facts = dict()
         self.commands()
 
+    def commands(self):
+        raise NotImplementedError
+
     def transform_dict(self, data, keymap):
         transform = dict()
         for key, fact in keymap:
@@ -255,7 +260,7 @@ class Interfaces(FactsBase):
         ('state', 'state'),
         ('desc', 'description'),
         ('eth_bw', 'bandwidth'),
-        ('eth_duplex','duplex'),
+        ('eth_duplex', 'duplex'),
         ('eth_speed', 'speed'),
         ('eth_mode', 'mode'),
         ('eth_hw_addr', 'macaddress'),
@@ -511,7 +516,7 @@ def main():
         module.exit_json(out=module.from_json(runner.items))
 
     ansible_facts = dict()
-    for key, value in facts.iteritems():
+    for key, value in iteritems(facts):
         # this is to maintain capability with nxos_facts 2.1
         if key.startswith('_'):
             ansible_facts[key[1:]] = value

network/nxos/nxos_hsrp.py

@@ -118,20 +118,15 @@ changed:
     sample: true
 '''
 
-import collections
 import json
 
 # COMMON CODE FOR MIGRATION
-import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
 from ansible.module_utils.shell import ShellError
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
 
 
 def to_list(val):
@@ -406,7 +401,7 @@ def get_interface_mode(interface, intf_type, module):
     return mode
 
 
-def get_hsrp_groups_on_interfaces(device):
+def get_hsrp_groups_on_interfaces(device, module):
     command = 'show hsrp all'
     body = execute_show_command(command, module)
     hsrp = {}

network/nxos/nxos_interface.py

@@ -135,20 +135,15 @@ changed:
 '''
 
 import json
-import collections
 
 # COMMON CODE FOR MIGRATION
-import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
+from ansible.module_utils.network import NetworkModule
 from ansible.module_utils.shell import ShellError
 
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
-
 
 def to_list(val):
      if isinstance(val, (list, tuple)):
@@ -308,7 +303,7 @@ def is_default_interface(interface, module):
         body = execute_show_command(command, module,
                                     command_type='cli_show_ascii')[0]
     except IndexError:
-        body = []
+        body = ''
 
     if body:
         raw_list = body.split('\n')

network/nxos/nxos_interface_ospf.py

@@ -166,15 +166,12 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
+from ansible.module_utils.network import NetworkModule
 from ansible.module_utils.shell import ShellError
 
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
-
 
 def to_list(val):
      if isinstance(val, (list, tuple)):

network/nxos/nxos_ospf.py

@@ -80,13 +80,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_ospf_vrf.py

@@ -171,13 +171,11 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
+from ansible.module_utils.shell import ShellError
 
 
 def to_list(val):

network/nxos/nxos_static_route.py

@@ -111,13 +111,10 @@ changed:
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
-from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.netcfg import NetworkConfig, ConfigLine, dumps
+from ansible.module_utils.network import NetworkModule
 
 
 def to_list(val):
@@ -155,7 +152,7 @@ class CustomNetworkConfig(NetworkConfig):
         try:
             section = self.get_section_objects(path)
             if self._device_os == 'junos':
-                return self.to_lines(section)
+                return dumps(section, output='lines')
             return self.to_block(section)
         except ValueError:
             return list()
@@ -398,10 +395,10 @@ def network_from_string(address, mask, module):
 def normalize_prefix(module, prefix):
     splitted_prefix = prefix.split('/')
 
+    address = splitted_prefix[0]
     if len(splitted_prefix) > 2:
         module.fail_json(msg='Incorrect address format.', address=address)
     elif len(splitted_prefix) == 2:
-        address = splitted_prefix[0]
         mask = splitted_prefix[1]
         network = network_from_string(address, mask, module)
 

network/nxos/nxos_vrf.py

@@ -111,19 +111,15 @@ changed:
 '''
 
 import json
-import collections
 
 # COMMON CODE FOR MIGRATION
 import re
 
+import ansible.module_utils.nxos
 from ansible.module_utils.basic import get_exception
 from ansible.module_utils.netcfg import NetworkConfig, ConfigLine
 from ansible.module_utils.shell import ShellError
-
-try:
-    from ansible.module_utils.nxos import get_module
-except ImportError:
-    from ansible.module_utils.nxos import NetworkModule
+from ansible.module_utils.network import NetworkModule
 
 
 def to_list(val):
@@ -385,11 +381,10 @@ def get_commands_to_config_vrf(delta, vrf):
 
 def get_vrf_description(vrf, module):
     command_type = 'cli_show_ascii'
-    command = ('show run section vrf | begin ^vrf\scontext\s{0} '
-               '| end ^vrf.*'.format(vrf))
+    command = (r'show run section vrf | begin ^vrf\scontext\s{0} | end ^vrf.*'.format(vrf))
 
     description = ''
-    descr_regex = ".*description\s(?P<descr>[\S+\s]+).*"
+    descr_regex = r".*description\s(?P<descr>[\S+\s]+).*"
     body = execute_show_command(command, module, command_type)
 
     try:
@@ -507,7 +502,7 @@ def main():
             if existing.get('vni') and existing.get('vni') != '':
                 commands.insert(1, 'no vni {0}'.format(existing['vni']))
         if module.check_mode:
-            module.exit_json(changed=True, commands=cmds)
+            module.exit_json(changed=True, commands=commands)
         else:
             execute_config_command(commands, module)
             changed = True
@@ -526,4 +521,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()