context_processors

akundev · Oct 13, 2023 · 6f1dc59 · 6f1dc59
1 parent a1055fd
commit 6f1dc59
Show file tree

Hide file tree

Showing 18 changed files with 125 additions and 221 deletions.
diff --git a/apps/articles/tests.py b/apps/articles/tests.py
@@ -153,14 +153,6 @@ def test_home(self):
         self.assertTemplateUsed(response, "articles/home.html")
         self.assertEqual(response.status_code, 200)
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=normal_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("home"))
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestArticleListView(TestCase):
     def setUp(self):
@@ -181,14 +173,6 @@ def test_articles(self):
         self.assertTemplateUsed(response, "articles/articles.html")
         self.assertEqual(response.status_code, 200)
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=normal_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("articles"))
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestArticleDetailView(TestCase):
     def setUp(self):
@@ -207,16 +191,6 @@ def test_article_detail(self):
         self.assertEqual(response.status_code, 200)
         self.assertTemplateUsed(response, "articles/article_detail.html")
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=normal_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(
-            reverse("article_detail", kwargs={"slug": test_article["slug"]})
-        )
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestTagDetailView(TestCase):
     """Test Tag detail view.
@@ -270,13 +244,25 @@ def test_tag_detail(self):
         self.assertEqual(str(response.context["products"]), str(products))
         self.assertTemplateUsed(response, "articles/tag_detail.html")
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
 
-        response = self.client.get(
-            reverse("tag_detail", kwargs={"slug": self.test_tag["slug"]})
+class TestContextProcessors(TestCase):
+    def setUp(self):
+        CustomUser.objects.create_user(**normal_user)
+        test_article["author"] = CustomUser.objects.get(
+            username=normal_user["username"]
         )
+        Article.objects.create(**test_article)
 
-        self.assertEqual(response.context["main_author"], main_author)
+    def test_context_processors(self):
+        key_list = [
+            "github_link",
+            "linkedin_link",
+            "cv_link",
+            "email_link",
+        ]
+        response = self.client.get(
+            reverse("article_detail", kwargs={"slug": test_article["slug"]})
+        )
+        context = response.context
+        for key in key_list:
+            self.assertTrue(key in context)
diff --git a/apps/articles/views.py b/apps/articles/views.py
@@ -2,7 +2,6 @@
 
 from apps.shop.models import Product
 from apps.tools.models import Tool
-from apps.users.models import CustomUser
 
 from .models import Article, Tag
 
@@ -15,7 +14,6 @@ class HomepageListView(ListView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
         context["tools"] = Tool.objects.all()[:5]
         context["products"] = Product.objects.all()[:5]
         return context
@@ -27,11 +25,6 @@ class ArticleListView(ListView):
     template_name = "articles/articles.html"
     queryset = Article.objects.all().is_published()
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
-
 
 class ArticleDetailView(DetailView):
     model = Article
@@ -44,11 +37,6 @@ def get_object(self):
         obj.content = obj.content_to_markdown()
         return obj
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
-
 
 class TagDetailView(DetailView):
     model = Tag
@@ -57,7 +45,6 @@ class TagDetailView(DetailView):
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
         context["articles"] = self.object.article_set.all()
         context["tools"] = self.object.tool_set.all()
         context["products"] = self.object.product_set.all()

diff --git a/apps/orders/tests.py b/apps/orders/tests.py
@@ -57,14 +57,6 @@ def test_order_create_view(self):
         self.assertTemplateUsed(response, "orders/order_form.html")
         self.assertEqual(response.status_code, 200)
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("order_create"))
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestSuccessView(TestCase):
     """Test Success View"""
@@ -78,11 +70,3 @@ def test_order_success_view(self):
 
         self.assertTemplateUsed(response, "orders/success_created.html")
         self.assertEqual(response.status_code, 200)
-
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("success_created"))
-
-        self.assertEqual(response.context["main_author"], main_author)
diff --git a/apps/orders/views.py b/apps/orders/views.py
@@ -2,7 +2,6 @@
 from django.views.generic import TemplateView
 from django.views.generic.edit import CreateView
 
-from apps.users.models import CustomUser
 
 from .models import Order
 
@@ -11,11 +10,6 @@
 class SuccessView(TemplateView):
     template_name = "orders/success_created.html"
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
-
 
 class OrderCreateView(CreateView):
     model = Order
@@ -28,8 +22,3 @@ def form_valid(self, form):
         response = super().form_valid(form)
         self.object.send_email()
         return response
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
diff --git a/apps/shop/tests.py b/apps/shop/tests.py
@@ -71,14 +71,6 @@ def test_product_list_view(self):
         self.assertTemplateUsed(response, "shop/products_list.html")
         self.assertEqual(response.status_code, 200)
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("products_list"))
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestProductsDetailViews(TestCase):
     def setUp(self):
@@ -111,11 +103,3 @@ def test_product_detail_view(self):
             response.context["product"].tags.all()[0].tag_name,
             self.test_tag["tag_name"],
         )
-
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("products_list"))
-
-        self.assertEqual(response.context["main_author"], main_author)
diff --git a/apps/shop/views.py b/apps/shop/views.py
@@ -1,6 +1,5 @@
 from django.views.generic import DetailView, ListView
 
-from apps.users.models import CustomUser
 
 from .models import Product
 
@@ -11,18 +10,8 @@ class ProductListView(ListView):
     context_object_name = "products"
     template_name = "shop/products_list.html"
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
-
 
 class ProductDetailView(DetailView):
     model = Product
     context_object_name = "product"
     template_name = "shop/product_detail.html"
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
diff --git a/apps/tools/tests.py b/apps/tools/tests.py
@@ -63,14 +63,6 @@ def test_tools_list_view(self):
         self.assertTemplateUsed(response, "tools/tools_list.html")
         self.assertEqual(response.status_code, 200)
 
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("tools_list"))
-
-        self.assertEqual(response.context["main_author"], main_author)
-
 
 class TestToolsDetailViews(TestCase):
     def setUp(self):
@@ -101,11 +93,3 @@ def test_tools_detail_view(self):
         self.assertEqual(
             response.context["tool"].tags.all()[0].tag_name, self.test_tag["tag_name"]
         )
-
-    def test_main_author(self):
-        main_author = CustomUser.objects.get(username=self.test_user["username"])
-        main_author.main_user = True
-        main_author.save()
-        response = self.client.get(reverse("tools_list"))
-
-        self.assertEqual(response.context["main_author"], main_author)
diff --git a/apps/tools/views.py b/apps/tools/views.py
@@ -1,6 +1,5 @@
 from django.views.generic import DetailView, ListView
 
-from apps.users.models import CustomUser
 
 from .models import Tool
 
@@ -13,18 +12,8 @@ class ToolListView(ListView):
     template_name = "tools/tools_list.html"
     queryset = Tool.objects.all()[:5]
 
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
-
 
 class ToolDetailView(DetailView):
     model = Tool
     context_object_name = "tool"
     template_name = "tools/tool_detail.html"
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
diff --git a/apps/users/tests.py b/apps/users/tests.py
@@ -38,7 +38,6 @@ def setUp(self):
     def test_about(self):
         self.assertTemplateUsed(self.response, "users/about.html")
         self.assertEqual(self.response.status_code, 200)
-        self.assertEqual(self.response.context["main_author"], self.main_author)
 
 
 class UsersManagersTests(TestCase):

diff --git a/apps/users/views.py b/apps/users/views.py
@@ -1,12 +1,5 @@
 from django.views.generic import TemplateView
 
-from .models import CustomUser
-
 
 class AboutTemplateView(TemplateView):
     template_name = "users/about.html"
-
-    def get_context_data(self, **kwargs):
-        context = super().get_context_data(**kwargs)
-        context["main_author"] = CustomUser.objects.filter(main_user=True).first()
-        return context
diff --git a/settings/context_processors.py b/settings/context_processors.py
@@ -0,0 +1,10 @@
+from django.conf import settings
+
+
+def about_links(request):
+    return {
+        "github_link": settings.AB_GITHUB_LINK,
+        "linkedin_link": settings.AB_LINKEDIN_LINK,
+        "cv_link": settings.AB_CV_LINK,
+        "email_link": settings.AB_EMAIL_LINK,
+    }
diff --git a/settings/settings.py b/settings/settings.py
@@ -73,6 +73,7 @@
                 "django.template.context_processors.request",
                 "django.contrib.auth.context_processors.auth",
                 "django.contrib.messages.context_processors.messages",
+                "settings.context_processors.about_links",
             ],
         },
     },
@@ -186,3 +187,10 @@
         "level": "DEBUG",
     },
 }
+
+
+# About links
+AB_GITHUB_LINK = os.environ.get("AB_GITHUB_LINK", "")
+AB_LINKEDIN_LINK = os.environ.get("AB_LINKEDIN_LINK", "")
+AB_CV_LINK = os.environ.get("AB_CV_LINK", "")
+AB_EMAIL_LINK = os.environ.get("AB_EMAIL_LINK", "")