#627 added autoscale tag, to allow users to tag all guests in a group at once

Author: allmightyspiff

SoftLayer/CLI/autoscale/detail.py

@@ -87,6 +87,6 @@ def cli(env, identifier):
     for guest in guests:
         real_guest = guest.get('virtualGuest')
         member_table.add_row([
-            guest.get('id'), real_guest.get('hostname'), utils.clean_time(real_guest.get('provisionDate'))
+            real_guest.get('id'), real_guest.get('hostname'), utils.clean_time(real_guest.get('provisionDate'))
         ])
     env.fout(member_table)

SoftLayer/CLI/autoscale/edit.py

@@ -0,0 +1,27 @@
+"""Edits an Autoscale group."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+
+from SoftLayer.CLI import environment
+from SoftLayer.managers.autoscale import AutoScaleManager
+
+
+@click.command()
+@click.argument('identifier')
+# Suspend / Unsuspend
+# Name
+# Min/Max
+# template->userData
+# 'hostname', 'domain', 'startCpus', 'maxMemory', 'localDiskFlag'
+# 'blockDeviceTemplateGroup.globalIdentifier
+# blockDevices.diskImage.capacity
+# sshKeys.id
+# postInstallScriptUri
+@environment.pass_env   
+def cli(env, identifier):
+    """Edits an Autoscale group."""
+
+    autoscale = AutoScaleManager(env.client)
+    groups = autoscale.details(identifier)
+    click.echo(groups)

SoftLayer/CLI/autoscale/tag.py

@@ -0,0 +1,34 @@
+"""Tags all guests in an autoscale group."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+
+from SoftLayer.CLI import environment
+from SoftLayer.managers.autoscale import AutoScaleManager
+from SoftLayer.managers.vs import VSManager
+
+
+@click.command()
+@click.argument('identifier')
+@click.option('--tags', '-g', help="Tags to set for each guest in this group. Existing tags are overwritten. "
+                                   "An empty string will remove all tags")
+@environment.pass_env
+def cli(env, identifier, tags):
+    """Tags all guests in an autoscale group.
+
+    --tags "Use, quotes, if you, want whitespace"
+
+    --tags Otherwise,Just,commas
+    """
+
+    autoscale = AutoScaleManager(env.client)
+    vsmanager = VSManager(env.client)
+    mask = "mask[id,virtualGuestId,virtualGuest[tagReferences,id,hostname]]"
+    guests = autoscale.get_virtual_guests(identifier, mask=mask)
+    click.echo("New Tags: {}".format(tags))
+    for guest in guests:
+        real_guest = guest.get('virtualGuest')
+        click.echo("Setting tags for {}".format(real_guest.get('hostname')))
+        vsmanager.set_tags(tags, real_guest.get('id'),)
+    click.echo("Done")
+    # pp(guests)

SoftLayer/CLI/routes.py

@@ -308,6 +308,8 @@
     ('autoscale:detail', 'SoftLayer.CLI.autoscale.detail:cli'),
     ('autoscale:scale', 'SoftLayer.CLI.autoscale.scale:cli'),
     ('autoscale:logs', 'SoftLayer.CLI.autoscale.logs:cli'),
+    ('autoscale:tag', 'SoftLayer.CLI.autoscale.tag:cli'),
+    ('autoscale:edit', 'SoftLayer.CLI.autoscale.edit:cli')
 ]
 
 ALL_ALIASES = {

SoftLayer/fixtures/SoftLayer_Scale_Group.py

@@ -198,6 +198,8 @@
     ]
 }
 
+getVirtualGuestMembers = getObject['virtualGuestMembers']
+
 scale = [
     {
         "accountId": 31111,

SoftLayer/managers/autoscale.py

@@ -36,7 +36,7 @@ def details(self, identifier, mask=None):
             https://sldn.softlayer.com/reference/services/SoftLayer_Scale_Group/getObject/
         """
         if not mask:
-            mask = """mask[virtualGuestMembers[id,virtualGuest[hostname,domain,provisionDate]], terminationPolicy,
+            mask = """mask[virtualGuestMembers[id,virtualGuest[id,hostname,domain,provisionDate]], terminationPolicy,
                    virtualGuestMemberCount, virtualGuestMemberTemplate[sshKeys],
                    policies[id,name,createDate,cooldown,actions,triggers,scaleActions],
                    networkVlans[networkVlanId,networkVlan[networkSpace,primaryRouter[hostname]]],
@@ -93,3 +93,13 @@ def get_logs(self, identifier, mask=None, object_filter=None):
         """
         return self.client.call('SoftLayer_Scale_Group', 'getLogs', id=identifier, mask=mask, filter=object_filter,
                                 iter=True)
+
+    def get_virtual_guests(self, identifier, mask=None):
+        """Calls `SoftLayer_Scale_Group::getVirtualGuestMembers()`_
+
+        :param identifier: SoftLayer_Scale_Group Id
+        :param mask: optional SoftLayer_Scale_Member objectMask
+        .. _SoftLayer_Scale_Group::getVirtualGuestMembers():
+            https://sldn.softlayer.com/reference/services/SoftLayer_Scale_Group/getVirtualGuestMembers/
+        """
+        return self.client.call('SoftLayer_Scale_Group', 'getVirtualGuestMembers', id=identifier, mask=mask, iter=True)

docs/cli/autoscale.rst

@@ -26,6 +26,13 @@ For making changes to the triggers or the autoscale group itself, see the `Autos
     :prog: autoscale logs
     :show-nested:
 
+.. click:: SoftLayer.CLI.autoscale.tag:cli
+    :prog: autoscale tag
+    :show-nested:
+
+.. click:: SoftLayer.CLI.autoscale.edit:cli
+    :prog: autoscale edit
+    :show-nested:
 
 
 .. _Autoscale Portal: https://cloud.ibm.com/classic/autoscale

tests/CLI/modules/autoscale_tests.py

@@ -44,3 +44,11 @@ def test_autoscale_list(self):
     def test_autoscale_detail(self):
         result = self.run_command(['autoscale', 'detail', '12222222'])
         self.assert_no_fail(result)
+
+    def test_autoscale_tag(self):
+        result = self.run_command(['autoscale', 'tag', '12345'])
+        self.assert_no_fail(result)
+
+    def test_autoscale_edit(self):
+        result = self.run_command(['autoscale', 'edit', '12345'])
+        self.assert_no_fail(result)

tests/managers/autoscale_tests.py

@@ -93,3 +93,24 @@ def test_autoscale_getLogs(self):
             'getLogs',
             identifier=11111
         )
+
+    def test_autoscale_get_virtual_guests(self):
+        self.autoscale.get_virtual_guests(11111)
+
+        self.assert_called_with(
+            'SoftLayer_Scale_Group',
+            'getVirtualGuestMembers',
+            identifier=11111,
+            mask=None
+        )
+
+    def test_autoscale_get_virtual_guests_mask(self):
+        test_mask = "mask[id]"
+        self.autoscale.get_virtual_guests(11111, mask=test_mask)
+
+        self.assert_called_with(
+            'SoftLayer_Scale_Group',
+            'getVirtualGuestMembers',
+            identifier=11111,
+            mask=test_mask
+        )