add user remove-access command

Author: caberos

SoftLayer/CLI/routes.py

@@ -364,6 +364,7 @@
     ('user:delete', 'SoftLayer.CLI.user.delete:cli'),
     ('user:vpn-manual', 'SoftLayer.CLI.user.vpn_manual:cli'),
     ('user:vpn-subnet', 'SoftLayer.CLI.user.vpn_subnet:cli'),
+    ('user:remove-access', 'SoftLayer.CLI.user.remove_access:cli'),
 
     ('vlan', 'SoftLayer.CLI.vlan'),
     ('vlan:create', 'SoftLayer.CLI.vlan.create:cli'),

SoftLayer/CLI/user/remove_access.py

@@ -0,0 +1,38 @@
+"""User remove access to devices."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+
+import SoftLayer
+from SoftLayer.CLI import environment
+
+
+@click.command(cls=SoftLayer.CLI.command.SLCommand, )
+@click.argument('identifier')
+@click.option('--hardware', '-h',
+              help="Display hardware this user has access to.")
+@click.option('--virtual', '-v',
+              help="Display virtual guests this user has access to.")
+@click.option('--dedicated', '-l',
+              help="dedicated host ID ")
+@environment.pass_env
+def cli(env, identifier, hardware, virtual, dedicated):
+    """User details."""
+
+    mgr = SoftLayer.UserManager(env.client)
+    device = ''
+    result = False
+    if hardware:
+        device = hardware
+        result = mgr.remove_hardware_access(identifier, hardware)
+
+    if virtual:
+        device = virtual
+        result = mgr.remove_virtual_access(identifier, virtual)
+
+    if dedicated:
+        device = dedicated
+        result = mgr.remove_dedicated_access(identifier, dedicated)
+
+    if result:
+        click.secho("Remove to access to device: %s" % device, fg='green')

SoftLayer/fixtures/SoftLayer_User_Customer.py

@@ -86,3 +86,6 @@
 editObject = True
 addApiAuthenticationKey = True
 updateVpnUser = True
+removeDedicatedHostAccess = True
+removeHardwareAccess = True
+removeVirtualGuestAccess = True

SoftLayer/managers/user.py

@@ -392,6 +392,36 @@ def get_overrides_list(self, user_id, subnet_ids):
 
         return overrides_list
 
+    def remove_hardware_access(self, user_id, hardware_id):
+        """Remove hardware from a portal user’s hardware access list.
+
+        :param int user_id:
+        :param int hardware_id
+
+        :returns: true
+        """
+        return self.user_service.removeHardwareAccess(hardware_id, id=user_id)
+
+    def remove_virtual_access(self, user_id, virtual_id):
+        """Remove hardware from a portal user’s virtual guests access list.
+
+        :param int user_id:
+        :param int hardware_id
+
+        :returns: true
+        """
+        return self.user_service.removeVirtualGuestAccess(virtual_id, id=user_id)
+
+    def remove_dedicated_access(self, user_id, dedicated_id):
+        """Remove hardware from a portal user’s dedicated host access list.
+
+        :param int user_id:
+        :param int dedicated_id
+
+        :returns: true
+        """
+        return self.user_service.removeDedicatedHostAccess(dedicated_id, id=user_id)
+
 
 def _keyname_search(haystack, needle):
     for item in haystack:

docs/cli/users.rst

@@ -48,4 +48,8 @@ Version 5.6.0 introduces the ability to interact with user accounts from the cli
     :prog: user vpn-subnet
     :show-nested:
 
+.. click:: SoftLayer.CLI.user.remove_access:cli
+    :prog: user remove-access
+    :show-nested:
+
 

tests/CLI/modules/user_tests.py

@@ -336,3 +336,15 @@ def test_edit_notification_off_failure(self, click):
         click.secho.assert_called_with('Failed to update notifications: Test notification', fg='red')
         self.assert_no_fail(result)
         self.assert_called_with('SoftLayer_Email_Subscription', 'disable', identifier=111)
+
+    def test_remove_access_hardware(self):
+        result = self.run_command(['user', 'remove-access', '123456', '--hardware', '147258'])
+        self.assert_no_fail(result)
+
+    def test_remove_access_virtual(self):
+        result = self.run_command(['user', 'remove-access', '123456', '--virtual', '987456'])
+        self.assert_no_fail(result)
+
+    def test_remove_access_dedicated(self):
+        result = self.run_command(['user', 'remove-access', '123456', '--dedicated', '369852'])
+        self.assert_no_fail(result)

tests/managers/user_tests.py

@@ -294,3 +294,15 @@ def test_gather_notifications_fail(self):
                                self.manager.gather_notifications,
                                ['Test not exit'])
         self.assertEqual("Test not exit is not a valid notification name", str(ex))
+
+    def test_remove_hardware(self):
+        self.manager.remove_hardware_access(123456, 369852)
+        self.assert_called_with('SoftLayer_User_Customer', 'removeHardwareAccess')
+
+    def test_remove_virtual(self):
+        self.manager.remove_virtual_access(123456, 369852)
+        self.assert_called_with('SoftLayer_User_Customer', 'removeVirtualGuestAccess')
+
+    def test_remove_dedicated(self):
+        self.manager.remove_dedicated_access(123456, 369852)
+        self.assert_called_with('SoftLayer_User_Customer', 'removeDedicatedHostAccess')