view bandmap/views.py @ 978:a828e80223d2

Prevent 500 error when previewing no text. Don't call image_check() when there is no data.
author Brian Neal <bgneal@gmail.com>
date Mon, 05 Oct 2015 20:26:14 -0500
parents 09ed84a7394c
children bb7e2fc24690
line wrap: on
line source
"""Views for the bandmap application.

"""
import json

from django.contrib.auth.decorators import login_required
from django.shortcuts import redirect, render
from django.contrib import messages as django_messages
from django.http import HttpResponse
from django.core.cache import cache
from django.conf import settings

from bandmap.forms import BandForm
from bandmap.models import BandEntry


SUCCESS = u"Successfully submitted {} for admin review. Thanks!"


def map_view(request):
    return render(request, 'bandmap/map.html')


@login_required
def add_band(request):
    """
    Provides the ability for a user to submit a new band to the map.

    """
    if request.method == 'POST':
        form = BandForm(request.POST)
        if form.is_valid():
            band = form.save(commit=False)
            band.user = request.user
            band.save()
            django_messages.success(request, SUCCESS.format(band.name))
            return redirect('bandmap-add')
    else:
        form = BandForm()

    return render(request, 'bandmap/add.html', {
        'form': form,
        })


def query(request):
    """Retrieves the band map entries and returns them as JSON.

    """
    show_param = request.GET.get('show', 'all')
    if show_param not in ['all', 'active', 'inactive']:
        show_param = 'all'

    # Do we have the results cached already?
    cache_key = 'bandmap-results-{}'.format(show_param)
    results = cache.get(cache_key)
    if results and not settings.DEBUG:
        return HttpResponse(results, content_type='application/json')

    # Generate results
    qs = BandEntry.objects.filter(is_approved=True)
    if show_param == 'active':
        qs = qs.filter(is_active=True)
    elif show_param == 'inactive':
        qs = qs.filter(is_active=False)

    bands = []
    for band in qs.iterator():
        bands.append({
            'name': band.name,
            'lat': band.lat,
            'lon': band.lon,
            'is_active': band.is_active,
            'note': band.html,
        })
    results = json.dumps(bands)

    # Store in cache
    cache.set(cache_key, results, 600)

    return HttpResponse(results, content_type='application/json')