didier/cogs/school.py

117 lines
4.2 KiB
Python
Raw Normal View History

2021-08-08 18:24:53 +02:00
from data import schedule
from data.courses import find_course_from_name
2021-11-29 21:37:04 +01:00
from data.embeds.deadlines import Deadlines
from data.embeds.food import Menu
from decorators import help
import discord
from discord.ext import commands
from enums.help_categories import Category
2021-11-29 21:37:04 +01:00
from functions import config, les
2021-08-08 18:33:32 +02:00
from functions.stringFormatters import capitalize
2021-11-29 21:37:04 +01:00
from functions.timeFormatters import skip_weekends
from functions.utils import reply_to_reference
class School(commands.Cog):
def __init__(self, client):
self.client = client
# Don't allow any commands to work when locked
def cog_check(self, ctx):
return not self.client.locked
@commands.command(name="Eten", aliases=["Food", "Menu"], usage="[Dag]*")
# @commands.check(checks.allowedChannels)
@help.Category(category=Category.School)
2021-09-26 13:10:57 +02:00
async def eten(self, ctx, day: str = None):
2021-10-04 20:15:01 +02:00
if day is not None:
day = day.lower()
embed = Menu(day).to_embed()
2021-09-26 13:10:57 +02:00
await ctx.reply(embed=embed, mention_author=False)
2021-09-20 01:42:13 +02:00
@commands.command(name="Les", aliases=["Class", "Classes", "Sched", "Schedule"], usage="[Dag]*")
# @commands.check(checks.allowedChannels)
2021-09-20 01:42:13 +02:00
@help.Category(category=Category.School)
async def les(self, ctx, day=None):
2021-10-04 20:15:01 +02:00
if day is not None:
day = day.lower()
2021-08-08 18:24:11 +02:00
date = les.find_target_date(day)
2021-08-08 18:33:32 +02:00
# Person explicitly requested a weekend-day
if day is not None and day.lower() in ("morgen", "overmorgen") and date.weekday() > 4:
return await ctx.send(f"{capitalize(day)} is het weekend.")
date = skip_weekends(date)
2021-08-08 18:12:16 +02:00
s = schedule.Schedule(date, int(config.get("year")), int(config.get("semester")), day is not None)
2021-08-08 19:32:41 +02:00
if s.semester_over:
return await ctx.send("Het semester is afgelopen.")
2021-09-26 21:27:52 +02:00
# DM only shows user's own minor
if ctx.guild is None:
minor_roles = [*schedule.find_minor(self.client, ctx.author.id)]
return await ctx.send(embed=s.create_schedule(minor_roles=minor_roles).to_embed())
2021-08-08 18:12:16 +02:00
return await ctx.send(embed=s.create_schedule().to_embed())
2022-01-05 19:37:59 +01:00
@commands.command(name="Pin", usage="[Message]*")
@help.Category(category=Category.Other)
async def pin(self, ctx, message: discord.Message = None):
# In case people abuse, check if they're blacklisted
blacklist = []
if ctx.author.id in blacklist:
return
# Support replying to the message that should be pinned
if message is None:
reference = ctx.message.reference
if reference is None:
return await ctx.reply("Controleer je argumenten.")
# If the message is cached, avoid sending an API call
if not reference.cached_message:
# Message is always in the current channel because we came from a reply
message = await ctx.channel.fetch_message(reference.message_id)
else:
message = reference.cached_message
if message.is_system():
return await ctx.send("Dus jij wil system messages pinnen?\nMag niet.")
await message.pin(reason=f"Didier Pin door {ctx.author.display_name}")
await ctx.message.add_reaction("")
@commands.command(name="Fiche", usage="[Vak]", aliases=["guide", "studiefiche"])
@help.Category(category=Category.School)
async def study_guide(self, ctx, name: str):
"""
Send links to study guides
"""
# Find code corresponding to the search query
course = find_course_from_name(name)
# Code not found
if course is None:
return await ctx.reply(f"Onbekend vak: \"{name}\".", mention_author=False, delete_after=15)
# Get the guide for the current year
year = 2018 + int(config.get("year"))
link = f"https://studiekiezer.ugent.be/studiefiche/nl/{course.code}/{year}"
return await reply_to_reference(ctx, content=link)
2021-11-29 21:37:04 +01:00
@commands.command(name="Deadlines", aliases=["dl"])
@help.Category(category=Category.School)
async def deadlines(self, ctx):
await ctx.send(embed=Deadlines().to_embed())
def setup(client):
client.add_cog(School(client))