Displaying deadlines

database/crud/deadlines.py

@@ -0,0 +1,31 @@
+from zoneinfo import ZoneInfo
+
+from dateutil.parser import parse
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy.orm import selectinload
+
+from database.schemas.relational import Deadline
+
+__all__ = ["add_deadline", "get_deadlines"]
+
+
+async def add_deadline(session: AsyncSession, course_id: int, name: str, date_str: str):
+    """Add a new deadline"""
+    date_dt = parse(date_str).replace(tzinfo=ZoneInfo("Europe/Brussels"))
+
+    if date_dt.hour == date_dt.minute == date_dt.second == 0:
+        date_dt.replace(hour=23, minute=59, second=59)
+
+    deadline = Deadline(course_id=course_id, name=name, deadline=date_dt)
+    session.add(deadline)
+    await session.commit()
+
+
+async def get_deadlines(session: AsyncSession) -> list[Deadline]:
+    """Get a list of all deadlines that are currently known
+
+    This includes deadlines that have passed already
+    """
+    statement = select(Deadline).options(selectinload(Deadline.course))
+    return (await session.execute(statement)).scalars().all()