mirror of https://github.com/stijndcl/didier
31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
from typing import Optional
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from database.models import UforaCourse, UforaCourseAlias
|
|
|
|
|
|
async def get_all_courses(session: AsyncSession) -> list[UforaCourse]:
|
|
"""Get a list of all courses in the database"""
|
|
statement = select(UforaCourse)
|
|
return list((await session.execute(statement)).scalars().all())
|
|
|
|
|
|
async def get_course_by_name(session: AsyncSession, query: str) -> Optional[UforaCourse]:
|
|
"""Try to find a course by its name
|
|
|
|
This checks for regular name first, and then aliases
|
|
"""
|
|
# Search case-insensitively
|
|
query = query.lower()
|
|
|
|
statement = select(UforaCourse).where(UforaCourse.name.ilike(f"%{query}%"))
|
|
result = (await session.execute(statement)).scalars().first()
|
|
if result:
|
|
return result
|
|
|
|
statement = select(UforaCourseAlias).where(UforaCourseAlias.alias.ilike(f"%{query}%"))
|
|
result = (await session.execute(statement)).scalars().first()
|
|
return result.course if result else None
|