From 5f0251bd87433a06e717d7672c6257be23ef2227 Mon Sep 17 00:00:00 2001 From: Corban-Lee Jones Date: Fri, 11 Oct 2024 10:35:03 +0100 Subject: [PATCH] ChannelsView - 404 if server not found --- apps/home/views.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/apps/home/views.py b/apps/home/views.py index dd5c417..54db893 100644 --- a/apps/home/views.py +++ b/apps/home/views.py @@ -1,14 +1,12 @@ # -*- encoding: utf-8 -*- -import json import logging import httpx from django.conf import settings -from django.utils import timezone from asgiref.sync import sync_to_async from django.shortcuts import redirect -from django.http import JsonResponse, HttpResponse +from django.http import JsonResponse, HttpResponseNotFound from django.views.generic import TemplateView, View from django.core.exceptions import PermissionDenied @@ -109,7 +107,10 @@ class ChannelsView(View): return redirect("/oauth2/login") guild_id = request.GET.get("guild") - server = await Server.objects.aget(pk=guild_id) + try: + server = await Server.objects.aget(pk=guild_id) + except Server.DoesNotExist: + return HttpResponseNotFound("Server not found.") if not ServerMember.objects.filter(server=server, user=request.user).aexists(): raise PermissionDenied("You aren't a member of this server.")