#874 added 'vs migrate' command

Author: allmightyspiff

SoftLayer/CLI/dedicatedhost/detail.py

@@ -19,7 +19,7 @@
 @click.option('--guests', is_flag=True, help='Show guests on dedicated host')
 @environment.pass_env
 def cli(env, identifier, price=False, guests=False):
-    """Get details for a virtual server."""
+    """Get details for a dedicated host."""
     dhost = SoftLayer.DedicatedHostManager(env.client)
 
     table = formatting.KeyValueTable(['name', 'value'])

SoftLayer/CLI/routes.py

@@ -46,6 +46,7 @@
     ('virtual:credentials', 'SoftLayer.CLI.virt.credentials:cli'),
     ('virtual:capacity', 'SoftLayer.CLI.virt.capacity:cli'),
     ('virtual:placementgroup', 'SoftLayer.CLI.virt.placementgroup:cli'),
+    ('virtual:migrate', 'SoftLayer.CLI.virt.migrate:cli'),
 
     ('dedicatedhost', 'SoftLayer.CLI.dedicatedhost'),
     ('dedicatedhost:list', 'SoftLayer.CLI.dedicatedhost.list:cli'),

SoftLayer/CLI/virt/migrate.py

@@ -0,0 +1,80 @@
+"""Manage Migrations of Virtual Guests"""
+# :license: MIT, see LICENSE for more details.
+import click
+
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import formatting
+from SoftLayer import utils
+
+
+@click.command()
+@click.option('--guest', '-g', type=click.INT, help="Guest ID to immediately migrate.")
+@click.option('--all', '-a', 'migrate_all', is_flag=True, default=False,
+              help="Migrate ALL guests that require migration immediately.")
+@click.option('--host', '-h', type=click.INT,
+              help="Dedicated Host ID to migrate to. Only works on guests that are already on a dedicated host.")
+@environment.pass_env
+def cli(env, guest, migrate_all, host):
+    """Manage VSIs that require migration. Can migrate Dedicated Host VSIs as well."""
+
+    vsi = SoftLayer.VSManager(env.client)
+    pending_filter = {'virtualGuests': {'pendingMigrationFlag': {'operation': 1}}}
+    dedicated_filer = {'virtualGuests': {'dedicatedHost': {'id': {'operation': 'not null'}}}}
+    mask = """mask[
+                id, hostname, domain, datacenter, pendingMigrationFlag, powerState,
+                primaryIpAddress,primaryBackendIpAddress, dedicatedHost
+            ]"""
+
+    # No options, just print out a list of guests that can be migrated
+    if not (guest or migrate_all):
+        require_migration = vsi.list_instances(filter=pending_filter, mask=mask)
+        require_table = formatting.Table(['id', 'hostname', 'domain', 'datacenter'], title="Require Migration")
+
+        for vsi_object in require_migration:
+            require_table.add_row([
+                vsi_object.get('id'),
+                vsi_object.get('hostname'),
+                vsi_object.get('domain'),
+                utils.lookup(vsi_object, 'datacenter', 'name')
+            ])
+
+        if require_migration:
+            env.fout(require_table)
+        else:
+            click.secho("No guests require migration at this time", fg='green')
+
+        migrateable = vsi.list_instances(filter=dedicated_filer, mask=mask)
+        migrateable_table = formatting.Table(['id', 'hostname', 'domain', 'datacenter', 'Host Name', 'Host Id'],
+                                             title="Dedicated Guests")
+        for vsi_object in migrateable:
+            migrateable_table.add_row([
+                vsi_object.get('id'),
+                vsi_object.get('hostname'),
+                vsi_object.get('domain'),
+                utils.lookup(vsi_object, 'datacenter', 'name'),
+                utils.lookup(vsi_object, 'dedicatedHost', 'name'),
+                utils.lookup(vsi_object, 'dedicatedHost', 'id')
+            ])
+        env.fout(migrateable_table)
+    # Migrate all guests with pendingMigrationFlag=True
+    elif migrate_all:
+        require_migration = vsi.list_instances(filter=pending_filter, mask="mask[id]")
+        for vsi_object in require_migration:
+            migrate(vsi, guest)
+    # Just migrate based on the options
+    else:
+        migrate(vsi, guest, host)
+
+
+def migrate(vsi_manager, vsi_id, host_id=None):
+    """Handles actually migrating virtual guests and handling the exception"""
+
+    try:
+        if host_id:
+            vsi_manager.migrate_dedicated(vsi_id, host_id)
+        else:
+            vsi_manager.migrate(vsi_id)
+        click.secho("Started a migration on {}".format(vsi_id), fg='green')
+    except SoftLayer.exceptions.SoftLayerAPIError as ex:
+        click.secho("Failed to migrate {}. {}".format(vsi_id, str(ex)), fg='red')

SoftLayer/managers/vs.py

@@ -1158,3 +1158,23 @@ def get_local_disks(self, instance_id):
         """
         mask = 'mask[diskImage]'
         return self.guest.getBlockDevices(mask=mask, id=instance_id)
+
+    def migrate(self, instance_id):
+        """Calls SoftLayer_Virtual_Guest::migrate
+
+        Only actually does anything if the virtual server requires a migration.
+        Will return an exception otherwise.
+
+        :param int instance_id: Id of the virtual server
+        """
+        return self.guest.migrate(id=instance_id)
+
+    def migrate_dedicated(self, instance_id, host_id):
+        """Calls SoftLayer_Virtual_Guest::migrate
+
+        Only actually does anything if the virtual server requires a migration.
+        Will return an exception otherwise.
+
+        :param int instance_id: Id of the virtual server
+        """
+        return self.guest.migrateDedicatedHost(host_id, id=instance_id)