78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
import json
|
|
from sqlalchemy import select, or_, func
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db.models import User
|
|
from app.schemas.user import UserPublic
|
|
|
|
|
|
async def search_directory(
|
|
db: AsyncSession,
|
|
class_id: int,
|
|
search: str | None = None,
|
|
industry: str | None = None,
|
|
company: str | None = None,
|
|
page: int = 1,
|
|
page_size: int = 20,
|
|
) -> tuple[list[User], int]:
|
|
"""Search approved members in a class."""
|
|
query = select(User).where(
|
|
User.class_id == class_id, User.status == "approved"
|
|
)
|
|
count_query = select(func.count(User.id)).where(
|
|
User.class_id == class_id, User.status == "approved"
|
|
)
|
|
|
|
if search:
|
|
search_term = f"%{search}%"
|
|
query = query.where(
|
|
or_(
|
|
User.name.ilike(search_term),
|
|
User.company.ilike(search_term),
|
|
User.position.ilike(search_term),
|
|
)
|
|
)
|
|
count_query = count_query.where(
|
|
or_(
|
|
User.name.ilike(search_term),
|
|
User.company.ilike(search_term),
|
|
User.position.ilike(search_term),
|
|
)
|
|
)
|
|
|
|
if industry:
|
|
query = query.where(User.industry == industry)
|
|
count_query = count_query.where(User.industry == industry)
|
|
|
|
if company:
|
|
company_term = f"%{company}%"
|
|
query = query.where(User.company.ilike(company_term))
|
|
count_query = count_query.where(User.company.ilike(company_term))
|
|
|
|
total_result = await db.execute(count_query)
|
|
total = total_result.scalar() or 0
|
|
|
|
result = await db.execute(
|
|
query.order_by(User.name)
|
|
.offset((page - 1) * page_size)
|
|
.limit(page_size)
|
|
)
|
|
users = list(result.scalars().all())
|
|
return users, total
|
|
|
|
|
|
def user_to_public(user: User, include_contact: bool = True) -> UserPublic:
|
|
"""Convert User model to public profile, optionally hiding contact info."""
|
|
return UserPublic(
|
|
id=user.id,
|
|
name=user.name,
|
|
student_id=user.student_id,
|
|
industry=user.industry,
|
|
company=user.company,
|
|
position=user.position,
|
|
wechat_id=user.wechat_id if include_contact else None,
|
|
phone=user.phone if include_contact else None,
|
|
avatar_url=user.avatar_url,
|
|
bio=user.bio,
|
|
)
|