ListPublicImages, StartVM, SuspendVM, ResumeVM, ListUserVM Views added.

helper.py

@@ -5,7 +5,7 @@ from decouple import config
 from pyotp import TOTP
 
 
-def check_otp(name, realm, seed):
+def check_otp(name, realm, token):
     try:
         data = {
             "auth_name": config('AUTH_NAME', ''),
@@ -13,7 +13,7 @@ def check_otp(name, realm, seed):
             "auth_realm": config('AUTH_REALM', ''),
             "name": name,
             "realm": realm,
-            "token": TOTP(seed).now()
+            "token": token
         }
     except binascii.Error:
         return 400
@@ -26,3 +26,15 @@ def check_otp(name, realm, seed):
         data=data
     )
     return response.status_code
+
+
+def add_otp_args(parser):
+    parser.add_argument("name", required=True)
+    parser.add_argument("realm", required=True)
+    parser.add_argument("token", required=True)
+    return parser
+
+
+def add_vmid_args(parser):
+    parser.add_argument("vmid", required=True)
+    return parser