added new feature with enabled/disabled

Author: Ramkishor Chaladi

SoftLayer/CLI/routes.py

@@ -403,6 +403,8 @@
     ('user:device-access', 'SoftLayer.CLI.user.device_access:cli'),
     ('user:vpn-manual', 'SoftLayer.CLI.user.vpn_manual:cli'),
     ('user:vpn-subnet', 'SoftLayer.CLI.user.vpn_subnet:cli'),
+    ('user:vpn-enable', 'SoftLayer.CLI.user.vpn_enable_or_disable:vpn_enable'),
+    ('user:vpn-disable', 'SoftLayer.CLI.user.vpn_enable_or_disable:vpn_disable'),
     ('user:remove-access', 'SoftLayer.CLI.user.remove_access:cli'),
     ('user:grant-access', 'SoftLayer.CLI.user.grant_access:cli'),
     ('user:vpn-password', 'SoftLayer.CLI.user.vpn_password:cli'),

SoftLayer/CLI/user/vpn_enable_or_disable.py

@@ -0,0 +1,67 @@
+"""Enable or Disable vpn for a user."""
+# :license: MIT, see LICENSE for more details.
+
+
+import click
+
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import helpers
+from SoftLayer.CLI import exceptions
+from SoftLayer.CLI import formatting
+
+
+@click.command(cls=SoftLayer.CLI.command.SLCommand, )
+@click.argument('user')
+@environment.pass_env
+def vpn_enable(env, user):
+    """Enable vpn for a user.
+    
+    Example::
+        slcli user vpn-enable 1234567
+    """
+
+    mgr = SoftLayer.UserManager(env.client)
+    user_id = helpers.resolve_id(mgr.resolve_ids, user, 'username')
+
+    if not (env.skip_confirmations or
+            formatting.confirm(f'This will enable vpn with the id {user_id}. '
+                               'Continue?')):
+        raise exceptions.CLIAbort('Aborted.')
+    
+    result = mgr.vpn_enable_or_disable(user_id, True)
+    message = f"{user} vpn is successfully enabled"
+
+    if result:
+        click.secho(message, fg='green')
+    else:
+        click.secho(f"{user} vpn enable is not successful", fg='red')
+
+
+
+
+@click.command(cls=SoftLayer.CLI.command.SLCommand, )
+@click.argument('user')
+@environment.pass_env
+def vpn_disable(env, user):
+    """Disable vpn for a user.
+
+    Example::
+        slcli user vpn-disable 1234567
+    """
+
+    mgr = SoftLayer.UserManager(env.client)
+    user_id = helpers.resolve_id(mgr.resolve_ids, user, 'username')
+
+    if not (env.skip_confirmations or
+            formatting.confirm(f'This will disable vpn with the id {user_id}. '
+                               'Continue?')):
+        raise exceptions.CLIAbort('Aborted.')
+    
+    result = mgr.vpn_enable_or_disable(user_id, False)
+    message = f"{user} vpn is successfully disabled"
+
+    if result:
+        click.secho(message, fg='green')
+    else:
+        click.secho(f"{user} vpn disable is not successfully", fg='red')

SoftLayer/managers/user.py

@@ -75,6 +75,17 @@ def get_current_user(self, objectmask=None):
             objectmask = "mask[userStatus[name], parent[id, username]]"
         return self.account_service.getCurrentUser(mask=objectmask)
 
+    # def hide_values(self, list_of_dicts, values_to_hide):
+    #     hide_list = []
+    #     for item in list_of_dicts:
+    #         hide_item = {
+    #             key: '' if value in values_to_hide 
+    #             else value
+    #             for key, value in item.items() 
+    #         }
+    #         hide_list.append(hide_item)
+    #     return hide_list
+
     def get_all_permissions(self):
         """Calls SoftLayer_User_CustomerPermissions_Permission::getAllObjects
 
@@ -84,6 +95,9 @@ def get_all_permissions(self):
         if self.all_permissions is None:
             permissions = self.client.call('User_Permission_Action', 'getAllObjects')
             self.all_permissions = sorted(permissions, key=itemgetter('keyName'))
+            # values_to_hide = ['ACCOUNT_SUMMARY_VIEW']
+            # self.all_permissions = self.hide_values(self.all_permissions, values_to_hide)
+            # print("ZZZZZ: ",type(itemgetter('keyName')))
         return self.all_permissions
 
     def get_all_notifications(self):
@@ -361,6 +375,17 @@ def vpn_manual(self, user_id, value):
         user_object = {'vpnManualConfig': value}
         return self.edit_user(user_id, user_object)
 
+    def vpn_enable_or_disable(self, user_id, value):
+        """Enable or Disable vs power on/off for a user.
+
+        :param int user_id: User to edit.
+        :param bool value: Value for vs power enable flag.
+        or
+        :param bool value: Value for vs power disable flag.
+        """
+        user_object = {'sslVpnAllowedFlag': value}
+        return self.edit_user(user_id, user_object)
+
     def vpn_subnet_add(self, user_id, subnet_ids):
         """Add subnets for a user.