From f674709a84e24a2ae9f2c32e0f937432b269af50 Mon Sep 17 00:00:00 2001
From: ks129 <45097959+ks129@users.noreply.github.com>
Date: Sun, 20 Sep 2020 18:14:08 +0300
Subject: Add handling for invalid category for category guide

---
 pydis_site/apps/guides/views/guide.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'pydis_site/apps/guides/views/guide.py')

diff --git a/pydis_site/apps/guides/views/guide.py b/pydis_site/apps/guides/views/guide.py
index cb30a525..deb1b8c4 100644
--- a/pydis_site/apps/guides/views/guide.py
+++ b/pydis_site/apps/guides/views/guide.py
@@ -21,6 +21,9 @@ class GuideView(View):
             category_name = None
         else:
             dir_path = os.path.join(settings.BASE_DIR, "pydis_site", "apps", "guides", "resources", "guides", category)
+            if not os.path.exists(dir_path) or not os.path.isdir(dir_path):
+                raise Http404("Category not found.")
+
             path = os.path.join(dir_path, f"{guide}.md")
             with open(os.path.join(dir_path, "_info.yml")) as f:
                 category_name = yaml.load(f.read())["name"]
-- 
cgit v1.2.3