Ch18: NewsLink views use startup_slug URL segment.

jambonrose · Apr 30, 2015 · f4e337f96e46c76e8b6cf93ccf7ab9b31439b987 · f4e337f
1 parent 7dcdf22
commit f4e337f96e46c76e8b6cf93ccf7ab9b31439b987
Unified Split

Showing with 32 additions and 11 deletions.

+7 −6 organizer/forms.py

+14 −1 organizer/utils.py

+11 −4 organizer/views.py
diff --git a/organizer/forms.py b/organizer/forms.py
@@ -4,12 +4,6 @@
 from .models import NewsLink, Startup, Tag


-class NewsLinkForm(forms.ModelForm):
-    class Meta:
-        model = NewsLink
-        fields = '__all__'
-
-
 class SlugCleanMixin:
    """Mixin class for slug cleaning method."""

@@ -22,6 +16,13 @@ def clean_slug(self):
        return new_slug


+class NewsLinkForm(
+        SlugCleanMixin, forms.ModelForm):
+    class Meta:
+        model = NewsLink
+        fields = '__all__'
+
+
 class StartupForm(
        SlugCleanMixin, forms.ModelForm):
    class Meta:
diff --git a/organizer/utils.py b/organizer/utils.py
@@ -1,6 +1,19 @@
 from django.shortcuts import get_object_or_404

-from .models import Startup
+from .models import NewsLink, Startup
+
+
+class NewsLinkGetObjectMixin():
+
+    def get_object(self, queryset=None):
+        startup_slug = self.kwargs.get(
+            self.startup_slug_url_kwarg)
+        newslink_slug = self.kwargs.get(
+            self.slug_url_kwarg)
+        return get_object_or_404(
+            NewsLink,
+            slug__iexact=newslink_slug,
+            startup__slug__iexact=startup_slug)


 class PageLinksMixin:
diff --git a/organizer/views.py b/organizer/views.py
@@ -8,17 +8,22 @@
    NewsLinkForm, StartupForm, TagForm)
 from .models import NewsLink, Startup, Tag
 from .utils import (
-    PageLinksMixin, StartupContextMixin)
+    NewsLinkGetObjectMixin, PageLinksMixin,
+    StartupContextMixin)


 class NewsLinkCreate(
-        StartupContextMixin, CreateView):
+        NewsLinkGetObjectMixin,
+        StartupContextMixin,
+        CreateView):
    form_class = NewsLinkForm
    model = NewsLink


 class NewsLinkDelete(
-        StartupContextMixin, DeleteView):
+        NewsLinkGetObjectMixin,
+        StartupContextMixin,
+        DeleteView):
    model = NewsLink
    slug_url_kwarg = 'newslink_slug'

@@ -28,7 +33,9 @@ def get_success_url(self):


 class NewsLinkUpdate(
-        StartupContextMixin, UpdateView):
+        NewsLinkGetObjectMixin,
+        StartupContextMixin,
+        UpdateView):
    form_class = NewsLinkForm
    model = NewsLink
    slug_url_kwarg = 'newslink_slug'