platform_admin_api_client.py

from contextvars import ContextVar

from flask import current_app
from notifications_utils.local_vars import LazyLocalGetter
from werkzeug.local import LocalProxy

from app import memo_resetters
from app.notify_client import NotifyAdminAPIClient, api_client_request_session


class AdminApiClient(NotifyAdminAPIClient):
    """API client for looking up arbitrary API objects based on incomplete information, eg just a UUID"""

    def find_by_uuid(self, uuid_):
        return self.post(url="/platform-admin/find-by-uuid", data={"uuid": uuid_})

    def fetch_users_list(
        self,
        created_from_date: str | None = None,
        created_to_date: str | None = None,
        logged_from_date: str | None = None,
        logged_to_date: str | None = None,
        take_part_in_research: bool | None = None,
    ):
        return self.post(
            url="/platform-admin/users-list",
            data={
                "created_start": created_from_date,
                "created_end": created_to_date,
                "logged_in_start": logged_from_date,
                "logged_in_end": logged_to_date,
                "take_part_in_research": take_part_in_research,
            },
        )


_admin_api_client_context_var: ContextVar[AdminApiClient] = ContextVar("admin_api_client")
get_admin_api_client: LazyLocalGetter[AdminApiClient] = LazyLocalGetter(
    _admin_api_client_context_var,
    lambda: AdminApiClient(current_app, request_session=api_client_request_session),
)
memo_resetters.append(lambda: get_admin_api_client.clear())
admin_api_client = LocalProxy(get_admin_api_client)