diff --git a/classquiz/emails/__init__.py b/classquiz/emails/__init__.py index 0227245..cd7cd02 100644 --- a/classquiz/emails/__init__.py +++ b/classquiz/emails/__init__.py @@ -37,8 +37,7 @@ def _sendMail(template: str, to: str, subject: str): server.sendmail(settings.mail_address, to, msg.as_string()) -async def send_register_email(email: str): - user = await User.objects.get_or_none(email=email, verified=False) +async def send_register_email(user: User): if user is None: raise ValueError("User not found") if settings.skip_email_verification: @@ -48,7 +47,7 @@ async def send_register_email(email: str): else: template = jinja.get_template("register.jinja2") template = await template.render_async(base_url=settings.root_address, token=user.verify_key) - _sendMail(template=template, to=email, subject="Verify your email") + _sendMail(template=template, to=user.email, subject="Verify your email") async def send_forgotten_password_email(email: str): diff --git a/classquiz/routers/users.py b/classquiz/routers/users.py index 5147589..ceae893 100644 --- a/classquiz/routers/users.py +++ b/classquiz/routers/users.py @@ -73,7 +73,7 @@ async def create_user(user: RouteUser, background_task: BackgroundTasks) -> User if len(user.username) == 32: return JSONResponse({"details": "Username mustn't be 32 characters long"}, 400) await user.save() - background_task.add_task(send_register_email, email=user.email) + background_task.add_task(send_register_email, user) await redis.delete("global_user_count") return user