Merge pull request #400 from mykaul/multiple_ips

Add a function to get all the IPs of a VM
lago-project · Dec 26, 2016 · bfd635d · bfd635d
2 parents d62571f + dbb538e
commit bfd635d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/lago/plugins/vm.py b/lago/plugins/vm.py
@@ -361,6 +361,17 @@ def iscsi_name(self):
     def ip(self):
         return str(self.virt_env.get_net().resolve(self.name()))
 
+    def all_ips(self):
+        nets = {}
+        ips = []
+        nets = self.virt_env.get_nets()
+        for net in nets.values():
+            mapping = net.mapping()
+            for hostname, ip in mapping.items():
+                if hostname.startswith(self.name()):
+                    ips.append(str(ip))
+        return ips
+
     def ssh(
         self,
         command,

diff --git a/lago/virt.py b/lago/virt.py
@@ -345,6 +345,9 @@ def add_mapping(self, name, ip, save=True):
     def resolve(self, name):
         return self._spec['mapping'][name]
 
+    def mapping(self):
+        return self._spec['mapping']
+
     def _libvirt_name(self):
         return self._env.prefixed_name(self.name(), max_length=15)