|
1 | | -from django.views.generic import ListView, DetailView |
| 1 | +from django.views.generic import ListView |
| 2 | +from django.views.generic import DetailView |
2 | 3 | from .models import Link, UserProfile |
3 | 4 | from .forms import UserProfileForm |
| 5 | +from .forms import LinkForm |
| 6 | + |
4 | 7 | from django.contrib.auth import get_user_model |
| 8 | +from django.views.generic.edit import CreateView |
5 | 9 | from django.views.generic.edit import UpdateView |
6 | | -from django.core.urlresolvers import reverse |
| 10 | +from django.views.generic.edit import DeleteView |
| 11 | +from django.core.urlresolvers import reverse, reverse_lazy |
| 12 | + |
| 13 | +from django.contrib.comments.models import Comment |
7 | 14 |
|
8 | | -class LinkListView(ListView): |
| 15 | +class RandomGossipMixin(object): |
| 16 | + def get_context_data(self, **kwargs): |
| 17 | + context = super(RandomGossipMixin, self).get_context_data(**kwargs) |
| 18 | + context[u"randomquip"] = Comment.objects.order_by('?')[0] |
| 19 | + return context |
| 20 | + |
| 21 | +class LinkListView(RandomGossipMixin, ListView): |
9 | 22 | model = Link |
10 | 23 | queryset = Link.with_votes.all() |
11 | 24 | paginate_by = 5 |
12 | 25 |
|
| 26 | +class LinkDetailView(DetailView): |
| 27 | + model = Link |
| 28 | + |
| 29 | +class LinkCreateView(CreateView): |
| 30 | + model = Link |
| 31 | + form_class = LinkForm |
| 32 | + |
| 33 | + def form_valid(self, form): |
| 34 | + f = form.save(commit=False) |
| 35 | + f.rank_score = 0.0 |
| 36 | + f.submitter = self.request.user |
| 37 | + f.save() |
| 38 | + return super(LinkCreateView, self).form_valid(form) |
| 39 | + |
| 40 | +class LinkUpdateView(UpdateView): |
| 41 | + model = Link |
| 42 | + form_class = LinkForm |
| 43 | + |
| 44 | +class LinkDeleteView(DeleteView): |
| 45 | + model = Link |
| 46 | + success_url = reverse_lazy("home") |
13 | 47 |
|
14 | 48 | class UserProfileDetailView(DetailView): |
15 | 49 | model = get_user_model() |
|
0 commit comments