Ch18: Refactor NewsLink views with new URLs.

organizer/models.py

@@ -62,6 +62,11 @@ def get_delete_url(self):
        return reverse('organizer_startup_delete',
                       kwargs={'slug': self.slug})

    def get_newslink_create_url(self):
        return reverse(
            'organizer_newslink_create',
            kwargs={'startup_slug': self.slug})

    def get_update_url(self):
        return reverse('organizer_startup_update',
                       kwargs={'slug': self.slug})
@@ -90,9 +95,13 @@ def get_absolute_url(self):
    def get_delete_url(self):
        return reverse(
            'organizer_newslink_delete',
            kwargs={'pk': self.pk})
            kwargs={
                'startup_slug': self.startup.slug,
                'newslink_slug': self.slug})

    def get_update_url(self):
        return reverse(
            'organizer_newslink_update',
            kwargs={'pk': self.pk})
            kwargs={
                'startup_slug': self.startup.slug,
                'newslink_slug': self.slug})

organizer/templates/organizer/newslink_form.html

@@ -8,7 +8,7 @@
  <div class="row">
    <div class="offset-by-two eight columns">
      <form
          action="{% url 'organizer_newslink_create' %}"
          action="{{ startup.get_newslink_create_url }}"
          method="post">
        {% csrf_token %}
        {{ form.as_p }}

organizer/templates/organizer/startup_detail.html

@@ -53,7 +53,7 @@ <h2>{{ startup.name }}</h2>
            <h3>Recent News</h3>
              <p>
                <a
                    href="{% url 'organizer_newslink_create' %}"
                    href="{{ startup.get_newslink_create_url }}"
                    class="button">
                  Add Article</a></p>
            <ul>

organizer/urls/startup.py

@@ -1,8 +1,9 @@
from django.conf.urls import url

from ..views import (
    StartupCreate, StartupDelete, StartupDetail,
    StartupList, StartupUpdate)
    NewsLinkCreate, NewsLinkDelete,
    NewsLinkUpdate, StartupCreate, StartupDelete,
    StartupDetail, StartupList, StartupUpdate)

urlpatterns = [
    url(r'^$',
@@ -14,10 +15,24 @@
    url(r'^(?P<slug>[\w\-]+)/$',
        StartupDetail.as_view(),
        name='organizer_startup_detail'),
    url(r'^(?P<startup_slug>[\w\-]+)/'
        r'add_article_link/$',
        NewsLinkCreate.as_view(),
        name='organizer_newslink_create'),
    url(r'^(?P<slug>[\w\-]+)/delete/$',
        StartupDelete.as_view(),
        name='organizer_startup_delete'),
    url(r'^(?P<slug>[\w\-]+)/update/$',
        StartupUpdate.as_view(),
        name='organizer_startup_update'),
    url(r'^(?P<startup_slug>[\w\-]+)/'
        r'(?P<newslink_slug>[\w\-]+)/'
        r'delete/$',
        NewsLinkDelete.as_view(),
        name='organizer_newslink_delete'),
    url(r'^(?P<startup_slug>[\w\-]+)/'
        r'(?P<newslink_slug>[\w\-]+)/'
        r'update/$',
        NewsLinkUpdate.as_view(),
        name='organizer_newslink_update'),
]

organizer/utils.py

@@ -1,3 +1,8 @@
from django.shortcuts import get_object_or_404

from .models import Startup


class PageLinksMixin:
    page_kwarg = 'page'

@@ -49,3 +54,20 @@ def get_context_data(self, **kwargs):
                    self.last_page(page),
            })
        return context


class StartupContextMixin():
    startup_slug_url_kwarg = 'startup_slug'
    startup_context_object_name = 'startup'

    def get_context_data(self, **kwargs):
        startup_slug = self.kwargs.get(
            self.startup_slug_url_kwarg)
        startup = get_object_or_404(
            Startup, slug__iexact=startup_slug)
        context = {
            self.startup_context_object_name:
                startup,
        }
        context.update(kwargs)
        return super().get_context_data(**context)

organizer/views.py

@@ -7,25 +7,31 @@
from .forms import (
    NewsLinkForm, StartupForm, TagForm)
from .models import NewsLink, Startup, Tag
from .utils import PageLinksMixin
from .utils import (
    PageLinksMixin, StartupContextMixin)


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


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

    def get_success_url(self):
        return (self.object.startup
                .get_absolute_url())


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


class StartupCreate(CreateView):

suorganizer/urls.py

@@ -21,7 +21,6 @@
from blog import urls as blog_urls
from contact import urls as contact_urls
from organizer.urls import (
    newslink as newslink_urls,
    startup as startup_urls, tag as tag_urls)

urlpatterns = [
@@ -36,7 +35,6 @@
    url(r'^admin/', include(admin.site.urls)),
    url(r'^blog/', include(blog_urls)),
    url(r'^contact/', include(contact_urls)),
    url(r'^newslink/', include(newslink_urls)),
    url(r'^startup/', include(startup_urls)),
    url(r'^tag/', include(tag_urls)),
]