Mercurial > public > sg101

annotate downloads/views.py @ 1150:0e0cd152b86d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
WIP convert downloads to V3 design.
author Brian Neal <bgneal@gmail.com>
date Tue, 13 Dec 2016 20:20:24 -0600
parents e1c03da72818
children f56798c96ec2
rev   line source
gremmie@1 1 """
gremmie@1 2 Views for the downloads application.
gremmie@1 3 """
bgneal@679 4 import json
gremmie@1 5
bgneal@1031 6 from django.shortcuts import render, get_object_or_404
gremmie@1 7 from django.contrib.auth.decorators import login_required
gremmie@1 8 from django.http import Http404
gremmie@1 9 from django.http import HttpResponse
gremmie@1 10 from django.http import HttpResponseRedirect
gremmie@1 11 from django.http import HttpResponseForbidden
gremmie@1 12 from django.http import HttpResponseBadRequest
bgneal@404 13 from django.http import HttpResponseNotFound
bgneal@92 14 from django.core.paginator import InvalidPage
gremmie@1 15 from django.core.urlresolvers import reverse
gremmie@1 16 from django.views.decorators.http import require_POST
gremmie@1 17
gremmie@1 18 from core.paginator import DiggPaginator
gremmie@1 19 from core.functions import email_admins
bgneal@241 20 from core.functions import get_page
gremmie@1 21 from downloads.models import Category
gremmie@1 22 from downloads.models import Download
gremmie@1 23 from downloads.models import VoteRecord
gremmie@1 24 from downloads.forms import AddDownloadForm
gremmie@1 25
gremmie@1 26 #######################################################################
gremmie@1 27
bgneal@242 28 DLS_PER_PAGE = 10
gremmie@1 29
gremmie@1 30 def create_paginator(dls):
gremmie@1 31 return DiggPaginator(dls, DLS_PER_PAGE, body=5, tail=3, margin=3, padding=2)
gremmie@1 32
gremmie@1 33 #######################################################################
gremmie@1 34
gremmie@1 35 @login_required
gremmie@1 36 def index(request):
gremmie@1 37 categories = Category.objects.all()
gremmie@1 38 total_dls = Download.public_objects.all().count()
bgneal@1031 39 return render(request, 'downloads/index.html', {
gremmie@1 40 'categories': categories,
gremmie@1 41 'total_dls': total_dls,
bgneal@1150 42 'top_menu': 'categories',
bgneal@1150 43 'V3_DESIGN': True,
bgneal@1031 44 })
gremmie@1 45
gremmie@1 46 #######################################################################
gremmie@1 47 # Maps URL component to database field name for the Download table:
gremmie@1 48
gremmie@1 49 DOWNLOAD_FIELD_MAP = {
bgneal@404 50 'title': 'title',
gremmie@1 51 'date': '-date_added',
gremmie@1 52 'rating': '-average_score',
gremmie@1 53 'hits': '-hits'
gremmie@1 54 }
gremmie@1 55
gremmie@1 56 @login_required
bgneal@241 57 def category(request, slug, sort='title'):
bgneal@480 58
bgneal@480 59 cat = get_object_or_404(Category, slug=slug)
gremmie@1 60
gremmie@1 61 if sort not in DOWNLOAD_FIELD_MAP:
gremmie@1 62 sort = 'title'
gremmie@1 63 order_by = DOWNLOAD_FIELD_MAP[sort]
gremmie@1 64
bgneal@190 65 downloads = Download.public_objects.filter(category=cat.pk).order_by(
bgneal@190 66 order_by)
gremmie@1 67 paginator = create_paginator(downloads)
bgneal@241 68 page = get_page(request.GET)
gremmie@1 69 try:
bgneal@241 70 the_page = paginator.page(page)
gremmie@1 71 except InvalidPage:
gremmie@1 72 raise Http404
gremmie@1 73
bgneal@1031 74 return render(request, 'downloads/download_list.html', {
gremmie@1 75 's' : sort,
gremmie@1 76 'category' : cat,
bgneal@404 77 'page' : the_page,
bgneal@1031 78 })
gremmie@1 79
gremmie@1 80 #######################################################################
gremmie@1 81
gremmie@1 82 @login_required
gremmie@1 83 def new(request):
bgneal@243 84 """Display new downloads with pagination."""
bgneal@243 85
bgneal@243 86 downloads = Download.public_objects.order_by('-date_added')
bgneal@243 87
bgneal@243 88 paginator = create_paginator(downloads)
bgneal@243 89 page = get_page(request.GET)
bgneal@243 90 try:
bgneal@243 91 the_page = paginator.page(page)
bgneal@243 92 except InvalidPage:
bgneal@243 93 raise Http404
bgneal@243 94
bgneal@1031 95 return render(request, 'downloads/download_summary.html', {
bgneal@243 96 'page': the_page,
bgneal@243 97 'title': 'Newest Downloads',
bgneal@1031 98 })
gremmie@1 99
gremmie@1 100 #######################################################################
gremmie@1 101
gremmie@1 102 @login_required
gremmie@1 103 def popular(request):
bgneal@243 104 """Display popular downloads with pagination."""
bgneal@243 105
bgneal@243 106 downloads = Download.public_objects.order_by('-hits')
bgneal@243 107
bgneal@243 108 paginator = create_paginator(downloads)
bgneal@243 109 page = get_page(request.GET)
bgneal@243 110 try:
bgneal@243 111 the_page = paginator.page(page)
bgneal@243 112 except InvalidPage:
bgneal@243 113 raise Http404
bgneal@243 114
bgneal@1031 115 return render(request, 'downloads/download_summary.html', {
bgneal@243 116 'page': the_page,
bgneal@243 117 'title': 'Popular Downloads',
bgneal@1031 118 })
gremmie@1 119
gremmie@1 120 #######################################################################
gremmie@1 121
gremmie@1 122 @login_required
gremmie@1 123 def rating(request):
bgneal@243 124 """Display downloads by rating with pagination."""
bgneal@243 125
bgneal@243 126 downloads = Download.public_objects.order_by('-average_score')
bgneal@243 127 paginator = create_paginator(downloads)
bgneal@243 128 page = get_page(request.GET)
bgneal@243 129 try:
bgneal@243 130 the_page = paginator.page(page)
bgneal@243 131 except InvalidPage:
bgneal@243 132 raise Http404
bgneal@243 133
bgneal@1031 134 return render(request, 'downloads/download_summary.html', {
bgneal@243 135 'page': the_page,
bgneal@243 136 'title': 'Highest Rated Downloads',
bgneal@1031 137 })
gremmie@1 138
gremmie@1 139 #######################################################################
gremmie@1 140
gremmie@1 141 @login_required
bgneal@23 142 def details(request, id):
bgneal@480 143 download = get_object_or_404(Download.public_objects, pk=id)
bgneal@845 144 if not download.is_public:
bgneal@845 145 raise Http404
bgneal@1031 146 return render(request, 'downloads/download_detail.html', {
gremmie@1 147 'download' : download,
bgneal@1031 148 })
gremmie@1 149
gremmie@1 150 #######################################################################
gremmie@1 151
gremmie@1 152 @login_required
gremmie@1 153 def add(request):
gremmie@1 154 if request.method == 'POST':
gremmie@1 155 form = AddDownloadForm(request.POST, request.FILES)
gremmie@1 156 if form.is_valid():
gremmie@1 157 dl = form.save(commit=False)
gremmie@1 158 dl.user = request.user
gremmie@1 159 dl.ip_address = request.META.get('REMOTE_ADDR', None)
bgneal@1023 160 dl.save(html=form.html)
gremmie@1 161 email_admins('New download for approval', """Hello,
gremmie@1 162
gremmie@1 163 A user has uploaded a new download for your approval.
gremmie@1 164 """)
gremmie@1 165 return HttpResponseRedirect(reverse('downloads-add_thanks'))
gremmie@1 166 else:
gremmie@1 167 form = AddDownloadForm()
gremmie@1 168
bgneal@1031 169 return render(request, 'downloads/add.html', {
gremmie@1 170 'add_form': form,
bgneal@1031 171 })
gremmie@1 172
gremmie@1 173 #######################################################################
gremmie@1 174
gremmie@1 175 @login_required
gremmie@1 176 def thanks(request):
bgneal@1031 177 return render(request, 'downloads/thanks.html', {
bgneal@1031 178 })
gremmie@1 179
gremmie@1 180 #######################################################################
gremmie@1 181
gremmie@1 182 @require_POST
gremmie@1 183 def rate_download(request):
gremmie@1 184 """This function is called by AJAX to rate a download."""
gremmie@1 185 if request.user.is_authenticated():
gremmie@1 186 id = request.POST.get('id', None)
gremmie@1 187 rating = request.POST.get('rating', None)
gremmie@1 188 if id is None or rating is None:
gremmie@1 189 return HttpResponseBadRequest('Missing id or rating.')
bgneal@404 190
gremmie@1 191 try:
gremmie@1 192 rating = int(rating)
gremmie@1 193 except ValueError:
gremmie@1 194 return HttpResponseBadRequest('Invalid rating.')
gremmie@1 195
gremmie@1 196 # rating will be from 0-4
gremmie@1 197 rating = min(5, max(1, rating))
gremmie@1 198
bgneal@480 199 download = get_object_or_404(Download.public_objects, pk=id)
gremmie@1 200
gremmie@1 201 # prevent multiple votes from the same user
bgneal@241 202 vote_record, created = VoteRecord.objects.get_or_create(
bgneal@241 203 download=download, user=request.user)
gremmie@1 204 if created:
gremmie@1 205 new_score = download.vote(rating)
gremmie@1 206 download.save()
gremmie@1 207 return HttpResponse(str(new_score))
gremmie@1 208 else:
gremmie@1 209 return HttpResponse('-1')
gremmie@1 210
gremmie@1 211 return HttpResponseForbidden('You must be logged in to rate a download.')
bgneal@404 212
bgneal@404 213 #######################################################################
bgneal@404 214
bgneal@404 215 @require_POST
bgneal@404 216 def request_download(request):
bgneal@404 217 """
bgneal@404 218 This function is called by AJAX to request a download. We update the hit
bgneal@404 219 count and then return a JSON object of the form:
bgneal@404 220 { id: download-id, 'url': link-to-download }
bgneal@404 221
bgneal@404 222 """
bgneal@404 223 if request.user.is_authenticated():
bgneal@404 224 dl_id = request.POST.get('id')
bgneal@404 225 if dl_id:
bgneal@404 226 try:
bgneal@404 227 dl = Download.public_objects.get(pk=dl_id)
bgneal@404 228 except Download.DoesNotExist:
bgneal@404 229 return HttpResponseNotFound("Download not found")
bgneal@404 230
bgneal@404 231 dl.hits += 1
bgneal@404 232 dl.save()
bgneal@404 233
bgneal@404 234 s = json.dumps({'id': dl_id, 'url': dl.file.url})
bgneal@404 235 return HttpResponse(s, content_type='application/json')
bgneal@404 236
bgneal@404 237 return HttpResponseForbidden('An error occurred.')