Mercurial > public > sg101
view downloads/views.py @ 870:ee56c8c8cf98
More WIP on ssl_images management command and tests.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Mon, 15 Dec 2014 21:16:47 -0600 |
parents | 41d0389fc85a |
children | a5ebc74dc3f3 |
line wrap: on
line source
""" Views for the downloads application. """ import json from django.shortcuts import render_to_response, get_object_or_404 from django.template import RequestContext from django.contrib.auth.decorators import login_required from django.http import Http404 from django.http import HttpResponse from django.http import HttpResponseRedirect from django.http import HttpResponseForbidden from django.http import HttpResponseBadRequest from django.http import HttpResponseNotFound from django.core.paginator import InvalidPage from django.core.urlresolvers import reverse from django.views.decorators.http import require_POST from core.paginator import DiggPaginator from core.functions import email_admins from core.functions import get_page from downloads.models import Category from downloads.models import Download from downloads.models import VoteRecord from downloads.forms import AddDownloadForm ####################################################################### DLS_PER_PAGE = 10 def create_paginator(dls): return DiggPaginator(dls, DLS_PER_PAGE, body=5, tail=3, margin=3, padding=2) ####################################################################### @login_required def index(request): categories = Category.objects.all() total_dls = Download.public_objects.all().count() return render_to_response('downloads/index.html', { 'categories': categories, 'total_dls': total_dls, }, context_instance = RequestContext(request)) ####################################################################### # Maps URL component to database field name for the Download table: DOWNLOAD_FIELD_MAP = { 'title': 'title', 'date': '-date_added', 'rating': '-average_score', 'hits': '-hits' } @login_required def category(request, slug, sort='title'): cat = get_object_or_404(Category, slug=slug) if sort not in DOWNLOAD_FIELD_MAP: sort = 'title' order_by = DOWNLOAD_FIELD_MAP[sort] downloads = Download.public_objects.filter(category=cat.pk).order_by( order_by) paginator = create_paginator(downloads) page = get_page(request.GET) try: the_page = paginator.page(page) except InvalidPage: raise Http404 return render_to_response('downloads/download_list.html', { 's' : sort, 'category' : cat, 'page' : the_page, }, context_instance = RequestContext(request)) ####################################################################### @login_required def new(request): """Display new downloads with pagination.""" downloads = Download.public_objects.order_by('-date_added') paginator = create_paginator(downloads) page = get_page(request.GET) try: the_page = paginator.page(page) except InvalidPage: raise Http404 return render_to_response('downloads/download_summary.html', { 'page': the_page, 'title': 'Newest Downloads', }, context_instance = RequestContext(request)) ####################################################################### @login_required def popular(request): """Display popular downloads with pagination.""" downloads = Download.public_objects.order_by('-hits') paginator = create_paginator(downloads) page = get_page(request.GET) try: the_page = paginator.page(page) except InvalidPage: raise Http404 return render_to_response('downloads/download_summary.html', { 'page': the_page, 'title': 'Popular Downloads', }, context_instance = RequestContext(request)) ####################################################################### @login_required def rating(request): """Display downloads by rating with pagination.""" downloads = Download.public_objects.order_by('-average_score') paginator = create_paginator(downloads) page = get_page(request.GET) try: the_page = paginator.page(page) except InvalidPage: raise Http404 return render_to_response('downloads/download_summary.html', { 'page': the_page, 'title': 'Highest Rated Downloads', }, context_instance = RequestContext(request)) ####################################################################### @login_required def details(request, id): download = get_object_or_404(Download.public_objects, pk=id) if not download.is_public: raise Http404 return render_to_response('downloads/download_detail.html', { 'download' : download, }, context_instance = RequestContext(request)) ####################################################################### @login_required def add(request): if request.method == 'POST': form = AddDownloadForm(request.POST, request.FILES) if form.is_valid(): dl = form.save(commit=False) dl.user = request.user dl.ip_address = request.META.get('REMOTE_ADDR', None) dl.save() email_admins('New download for approval', """Hello, A user has uploaded a new download for your approval. """) return HttpResponseRedirect(reverse('downloads-add_thanks')) else: form = AddDownloadForm() return render_to_response('downloads/add.html', { 'add_form': form, }, context_instance=RequestContext(request)) ####################################################################### @login_required def thanks(request): return render_to_response('downloads/thanks.html', { }, context_instance=RequestContext(request)) ####################################################################### @require_POST def rate_download(request): """This function is called by AJAX to rate a download.""" if request.user.is_authenticated(): id = request.POST.get('id', None) rating = request.POST.get('rating', None) if id is None or rating is None: return HttpResponseBadRequest('Missing id or rating.') try: rating = int(rating) except ValueError: return HttpResponseBadRequest('Invalid rating.') # rating will be from 0-4 rating = min(5, max(1, rating)) download = get_object_or_404(Download.public_objects, pk=id) # prevent multiple votes from the same user vote_record, created = VoteRecord.objects.get_or_create( download=download, user=request.user) if created: new_score = download.vote(rating) download.save() return HttpResponse(str(new_score)) else: return HttpResponse('-1') return HttpResponseForbidden('You must be logged in to rate a download.') ####################################################################### @require_POST def request_download(request): """ This function is called by AJAX to request a download. We update the hit count and then return a JSON object of the form: { id: download-id, 'url': link-to-download } """ if request.user.is_authenticated(): dl_id = request.POST.get('id') if dl_id: try: dl = Download.public_objects.get(pk=dl_id) except Download.DoesNotExist: return HttpResponseNotFound("Download not found") dl.hits += 1 dl.save() s = json.dumps({'id': dl_id, 'url': dl.file.url}) return HttpResponse(s, content_type='application/json') return HttpResponseForbidden('An error occurred.')