suspend/unsuspend user: ref #335

Author: usmannasir

cli/cliParser.py

@@ -61,5 +61,6 @@ def prepareArguments(self):
         parser.add_argument('--websitesLimit', help='Website limit while creating user.')
         parser.add_argument('--selectedACL', help='Select ACL while creating user.')
         parser.add_argument('--securityLevel', help='Set security level while creating user.')
+        parser.add_argument('--state', help='State value used in user suspension.')
 
         return parser.parse_args()

cli/cyberPanel.py

@@ -1399,15 +1399,37 @@ def main():
     elif args.function == "listUsers":
 
         from userManagment.views import fetchTableUsers
+        data = {}
+        data['userID'] = 1
+        response = fetchTableUsers(data)
+
+        print(response.content.decode())
+
+    elif args.function == "suspendUser":
+
+        completeCommandExample = 'cyberpanel suspendUser --userName cyberpanel --state SUSPEND'
+
+        if not args.userName:
+            print("\n\nPlease enter User Name. For example:\n\n" + completeCommandExample + "\n\n")
+            return
+
+        if not args.state:
+            print("\n\nPlease enter state value i.e SUSPEND/UnSuspend. For example:\n\n" + completeCommandExample + "\n\n")
+            return
+
+        from userManagment.views import controlUserState
 
         data = {}
+        data['accountUsername'] = args.userName
+        data['state'] = args.state
         data['userID'] = 1
 
-        response = fetchTableUsers(data)
+        response = controlUserState(data)
 
         print(response.content.decode())
 
 
 
+
 if __name__ == "__main__":
     main()

userManagment/views.py

@@ -1029,49 +1029,56 @@ def fetchTableUsers(request):
 
 def controlUserState(request):
     try:
-        val = request.session['userID']
         try:
-            if request.method == 'POST':
+            val = request.session['userID']
+        except:
+            val = request['userID']
+        try:
+            try:
                 data = json.loads(request.body)
-                accountUsername = data['accountUsername']
-                state = data['state']
+            except:
+                data = request
 
-                user = Administrator.objects.get(userName=accountUsername)
+            accountUsername = data['accountUsername']
+            state = data['state']
 
-                currentACL = ACLManager.loadedACL(val)
-                loggedUser = Administrator.objects.get(pk=val)
+            user = Administrator.objects.get(userName=accountUsername)
 
-                if currentACL['admin'] == 1:
-                    pass
-                elif user.owner == loggedUser.pk:
-                    pass
-                elif user.pk == loggedUser.pk:
-                    pass
-                else:
-                    data_ret = {'fetchStatus': 0, 'error_message': 'Un-authorized access.'}
-                    json_data = json.dumps(data_ret)
-                    return HttpResponse(json_data)
+            currentACL = ACLManager.loadedACL(val)
+            loggedUser = Administrator.objects.get(pk=val)
 
-                if state == 'SUSPEND':
-                    user.state = 'SUSPENDED'
-                else:
-                    user.state = 'ACTIVE'
+            if currentACL['admin'] == 1:
+                pass
+            elif user.owner == loggedUser.pk:
+                pass
+            elif user.pk == loggedUser.pk:
+                pass
+            else:
+                data_ret = {'fetchStatus': 0, 'error_message': 'Un-authorized access.'}
+                json_data = json.dumps(data_ret)
+                return HttpResponse(json_data)
 
-                user.save()
+            if state == 'SUSPEND':
+                user.state = 'SUSPENDED'
+            else:
+                user.state = 'ACTIVE'
 
-                extraArgs = {}
-                extraArgs['user'] = user
-                extraArgs['currentACL'] = ACLManager.loadedACL(user.pk)
-                extraArgs['state'] = state
+            user.save()
 
-                from userManagment.userManager import UserManager
+            extraArgs = {}
+            extraArgs['user'] = user
+            extraArgs['currentACL'] = ACLManager.loadedACL(user.pk)
+            extraArgs['state'] = state
 
-                um = UserManager('controlUserState', extraArgs)
-                um.start()
+            from userManagment.userManager import UserManager
+
+            um = UserManager('controlUserState', extraArgs)
+            um.start()
+
+            data_ret = {'status': 1}
+            json_data = json.dumps(data_ret)
+            return HttpResponse(json_data)
 
-                data_ret = {'status': 1}
-                json_data = json.dumps(data_ret)
-                return HttpResponse(json_data)
 
         except BaseException as msg:
             data_ret = {'status': 0, 'saveStatus': 0, 'error_message': str(msg)}