Merge pull request #286 from aj-cruz/eos-vlan-fix

Fix Arista EOS Driver get_vlans

network_importer/drivers/arista_eos.py

@@ -40,13 +40,13 @@ def get_vlans(task: Task) -> Result:
 
         nr_device = task.host.get_connection("napalm", task.nornir.config)
         eos_device = nr_device.device
-        results = eos_device.run_commands(["show vlan"])
+        device_results = eos_device.run_commands(["show vlan"])
 
-        if not isinstance(results[0], dict) or "vlans" not in results[0]:
+        if not isinstance(device_results[0], dict) or "vlans" not in device_results[0]:
             LOGGER.warning("%s | No vlans information returned", task.host.name)
             return Result(host=task.host, result=False)
 
-        for vid, data in results[0]["vlans"].items():
-            results.vlans.append(Vlan(name=data["name"], id=vid))
+        for vid, data in device_results[0]["vlans"].items():
+            results.vlans.append(Vlan(name=data["name"], vid=int(vid)))
 
-        return Result(host=task.host, result=results)
+        return Result(host=task.host, result=results.dict())