Function bulk_enroll_users

apps/api/src/services/admin/admin.py:1358–1467

Enroll a batch of users in a course. Returns summary of results.

(
    token_user: APITokenUser,
    course_uuid: str,
    user_ids: List[int],
    request: Request,
    db_session: AsyncSession,
)

async def bulk_enroll_users(
    token_user: APITokenUser,
    course_uuid: str,
    user_ids: List[int],
    request: Request,
    db_session: AsyncSession,
) -> dict:
    """Enroll a batch of users in a course. Returns summary of results."""

    course = (await db_session.execute(
        select(Course).where(
            Course.course_uuid == course_uuid,
            Course.org_id == token_user.org_id,
        )
    )).scalars().first()
    if not course:
        raise HTTPException(status_code=404, detail="Course not found")

    # Pre-fetch memberships, existing enrollments, and trails in 3 queries total
    member_ids = set(
        (await db_session.execute(
            select(UserOrganization.user_id).where(
                UserOrganization.user_id.in_(user_ids),
                UserOrganization.org_id == token_user.org_id,
            )
        )).scalars().all()
    )
    already_enrolled_ids = set(
        (await db_session.execute(
            select(TrailRun.user_id).where(
                TrailRun.course_id == course.id,
                TrailRun.user_id.in_(user_ids),
                TrailRun.org_id == token_user.org_id,
            )
        )).scalars().all()
    )

    enrolled: List[int] = []
    already_enrolled: List[int] = []
    skipped: List[int] = []
    to_enroll: List[int] = []

    for user_id in user_ids:
        if user_id not in member_ids:
            skipped.append(user_id)
        elif user_id in already_enrolled_ids:
            already_enrolled.append(user_id)
        else:
            to_enroll.append(user_id)

    if not to_enroll:
        return {"enrolled": enrolled, "already_enrolled": already_enrolled, "skipped": skipped}

    now = datetime.now()
    trails_by_user: dict[int, Trail] = {
        t.user_id: t
        for t in (await db_session.execute(
            select(Trail).where(