Update: access funcs

mgrctl/utils/api_users.py

@@ -10,38 +10,43 @@ class UserAPI(object):
         self.callback_class = callback_class
         self.callback = callback_class()
 
-    def get_users(self, role: str) -> dict:
+    def get_users(self, role: str) -> list:
         data = {}
         if role == 'admin':
             data = {"where": "((roles+CP+'%@admin%')+AND+(state+EQ+'active'))"}
-        return self.callback.call_api(
+        response = self.callback.call_api(
             url='/user',
             method='GET',
             data=data
         )
+        users = self._extract_users(users=response)
+        return users
 
-    def _format_users(self, users: dict) -> list:
+    def _extract_users(self, users: dict) -> list:
+        return users.get('list', [])
+
+    def _format_users(self, users: list) -> list:
         output = []
-        for user in users.get('list', []):
+        for user in users:
             output.append({
                 'id': user.get('id', ''),
                 'email': user.get('email', ''),
                 'roles': user.get('roles', []),
-                'state': user.get('state', '')
+                'state': user.get('state', ''),
+                # add more fields here...
             })
         return output
 
     def get_first_random_admin(self):
         users = self.get_users(role='admin')
         admin = {}
-        for user in users.get('list', []):
-            if '@admin' in admin.get('roles', []):
+        for user in users:
+            if '@admin' in user.get('roles', []):
                 admin = user
                 break
         return admin
 
-    def echo_users(self, role: str) -> None:
-        users = self.get_users(role)
+    def echo_users(self, users: list) -> None:
         output = self._format_users(users)
         click.echo(tabulate(output, headers='keys'))
 
@@ -66,6 +71,19 @@ class UserAPI(object):
     def echo_access_links(self, links: list) -> None:
         click.echo(tabulate(links, headers='keys'))
 
+    def gen_access_links_interactive(self) -> None:
+        users = self.get_users(role='admin')
+        self.echo_users(users)
+        try:
+            click.echo('Choose user id and count of keys')
+            _id = int(input('User ID: '))
+            count = int(input('Count of keys: '))
+            keys = self.get_access_keys(user=_id, count=count)
+            links = self.gen_access_links(keys)
+            self.echo_access_links(links)
+        except ValueError:
+            click.echo('Error: Invalid, value is not a valid integer')
+
     def gen_api_token(self, email=None, password=None):
         token = self.callback.get_auth_token(email, password)
         return token