Compare commits

..

No commits in common. "3bc9c15af7113766c4b8346d4547aabcd0631078" and "c1721b951cc29eebda8b067203416f60a6fc320b" have entirely different histories.

3 changed files with 11 additions and 34 deletions

View File

@ -1,4 +1,3 @@
import datetime
from typing import Optional
from zoneinfo import ZoneInfo
@ -25,14 +24,12 @@ async def add_deadline(session: AsyncSession, course_id: int, name: str, date_st
await session.commit()
async def get_deadlines(
session: AsyncSession, *, after: Optional[datetime.date] = None, course: Optional[UforaCourse] = None
) -> list[Deadline]:
"""Get a list of all upcoming deadlines"""
statement = select(Deadline)
async def get_deadlines(session: AsyncSession, *, course: Optional[UforaCourse] = None) -> list[Deadline]:
"""Get a list of all deadlines that are currently known
if after is not None:
statement = statement.where(Deadline.deadline > after)
This includes deadlines that have passed already
"""
statement = select(Deadline)
if course is not None:
statement = statement.where(Deadline.course_id == course.course_id)

View File

@ -15,7 +15,7 @@ from didier.data.embeds.schedules import Schedule, get_schedule_for_day
from didier.exceptions import HTTPException, NotInMainGuildException
from didier.utils.discord.converters.time import DateTransformer
from didier.utils.discord.flags.school import StudyGuideFlags
from didier.utils.discord.users import has_course, to_main_guild_member
from didier.utils.discord.users import to_main_guild_member
from didier.utils.types.datetime import skip_weekends, tz_aware_today
@ -30,15 +30,11 @@ class School(commands.Cog):
@commands.hybrid_command(name="deadlines")
async def deadlines(self, ctx: commands.Context):
"""Show upcoming deadlines."""
async with ctx.typing():
async with self.client.postgres_session as session:
deadlines = await get_deadlines(session, after=tz_aware_today())
async with self.client.postgres_session as session:
deadlines = await get_deadlines(session)
member = to_main_guild_member(self.client, ctx.author)
deadlines = list(filter(lambda d: has_course(member, d.course), deadlines))
embed = Deadlines(deadlines).to_embed()
await ctx.reply(embed=embed, mention_author=False, ephemeral=False)
embed = Deadlines(deadlines).to_embed()
await ctx.reply(embed=embed, mention_author=False, ephemeral=False)
@commands.hybrid_command(name="les", aliases=["sched", "schedule"])
@app_commands.rename(day_dt="date")

View File

@ -2,26 +2,10 @@ from typing import Union
import discord
from database.schemas import UforaCourse
from didier import Didier
from didier.exceptions import NotInMainGuildException
__all__ = ["has_course", "to_main_guild_member"]
def has_course(member: discord.Member, course: UforaCourse) -> bool:
"""Check if a member is taking a Ufora course"""
for role in member.roles:
if role.id == course.role_id:
return True
if course.overarching_role_id is not None and course.overarching_role_id == role.id:
return True
if course.alternative_overarching_role_id is not None and course.alternative_overarching_role_id == role.id:
return True
return False
__all__ = ["to_main_guild_member"]
def to_main_guild_member(client: Didier, user: Union[discord.User, discord.Member]) -> discord.Member: