Add users

Author: avdata99

ckanext/activityinfo/blueprints/admin.py

@@ -1,7 +1,11 @@
 import logging
 from flask import Blueprint
 from ckan.plugins import toolkit
-from ckanext.activityinfo.utils import require_sysadmin_user, get_ai_resources
+from ckanext.activityinfo.utils import (
+    require_sysadmin_user,
+    get_ai_resources,
+    get_users_with_activity_info_token,
+)
 
 
 log = logging.getLogger(__name__)
@@ -17,7 +21,7 @@ def index():
     """
     log.info('ActivityInfo admin index view')
 
-    ai_users = []
+    ai_users = get_users_with_activity_info_token()
     ai_resources = get_ai_resources()
 
     ctx = {

ckanext/activityinfo/templates/activity_info/admin.html

@@ -6,6 +6,7 @@
 <div class="internal-div-section">
     <h1>Activity Info admin page</h1>
 
+    <h2>Users with Activity Info tokens</h2>
     <table id="activity-info-admin-users-list-table"
         class="table table-header table-hover table-bordered table-responsive">
         <thead>
@@ -26,6 +27,7 @@ <h1>Activity Info admin page</h1>
         </tbody>
     </table>
 
+    <h2>Activity Info linked resources</h2>
     <table id="activity-info-admin-resources"
         class="table table-header table-hover table-bordered table-responsive">
         <thead>

ckanext/activityinfo/utils.py

@@ -1,6 +1,8 @@
 import logging
 from functools import wraps
 from ckan.plugins import toolkit
+from ckan import model
+from sqlalchemy import and_
 
 
 log = logging.getLogger(__name__)
@@ -98,6 +100,36 @@ def get_ai_resources(limit=100):
     return ret
 
 
+def get_users_with_activity_info_token():
+    """
+    Get all users that have an ActivityInfo API key set in their plugin_extras.
+    Uses SQLAlchemy to query the JSONB plugin_extras column.
+
+    Returns:
+        A list of user objects with ActivityInfo API keys.
+    """
+    # Query users where plugin_extras -> 'activity_info' -> 'api_key' exists and is not null
+    # Use chained -> operators: plugin_extras -> 'activity_info' ->> 'api_key'
+    users = model.Session.query(model.User).filter(
+        and_(
+            model.User.state == 'active',
+            model.User.plugin_extras.isnot(None),
+            model.User.plugin_extras['activity_info'].isnot(None),
+            model.User.plugin_extras['activity_info']['api_key'].astext.isnot(None),
+            model.User.plugin_extras['activity_info']['api_key'].astext != '',
+        )
+    ).all()
+
+    final_users = []
+    for user in users:
+        final_users.append({
+            'id': user.id,
+            'name': user.name,
+        })
+
+    return final_users
+
+
 def require_sysadmin_user(func):
     '''
     Decorator for flask view functions. Returns 403 response if no user is logged in or if the login user is external
@@ -111,4 +143,4 @@ def view_wrapper(*args, **kwargs):
             return toolkit.abort(403, "Sysadmin user required")
         return func(*args, **kwargs)
 
-    return view_wrapper
+    return view_wrapper