satnogs-db/db/base/views.py

from django.shortcuts import render, redirect
from django.contrib.auth.models import User
from django.views.decorators.http import require_POST
from django.contrib.auth.decorators import login_required
from django.contrib import messages
from django.core.urlresolvers import reverse
from django.http import HttpResponseNotFound, HttpResponseServerError
from django.shortcuts import render

from db.base.models import Transponder, Satellite, Suggestion, MODE_CHOICES
from db.base.forms import SatelliteSearchForm, SuggestionForm


def home(request):
    """View to render home page."""
    satellites = Satellite.objects.all()

    if request.method == 'POST':
        satellite_form = SatelliteSearchForm(request.POST)
        if satellite_form.is_valid():
            term = satellite_form.cleaned_data['term']
            norad_cat_id = term.split()[0]
            try:
                satellite = Satellite.objects.get(norad_cat_id=norad_cat_id)
            except:
                messages.error(request, 'Please select one of the available Satellites')
                return redirect(reverse('home'))

            suggestions = Suggestion.objects.filter(satellite=satellite).count()

            return render(request, 'base/suggest.html', {'satellite': satellite,
                                                         'satellites': satellites,
                                                         'suggestions': suggestions,
                                                         'satellite_form': satellite_form,
                                                         'modes': MODE_CHOICES})

    transponders = Transponder.objects.all().count()
    suggestions = Suggestion.objects.all().count()
    contributors = User.objects.filter(is_active=1).count()
    return render(request, 'base/home.html', {'satellites': satellites,
                                              'transponders': transponders,
                                              'contributors': contributors,
                                              'suggestions': suggestions})


def custom_404(request):
    """Custom 404 error handler."""
    return HttpResponseNotFound(render(request, '404.html'))


def custom_500(request):
    """Custom 500 error handler."""
    return HttpResponseServerError(render(request, '500.html'))


@login_required
@require_POST
def suggestion(request):
    """View to process suggestion form"""
    suggestion_form = SuggestionForm(request.POST)
    if suggestion_form.is_valid():
        suggestion = suggestion_form.save(commit=False)
        suggestion.user = request.user
        suggestion.save()

        messages.success(request, ('Your suggestion was stored successfully. '
                                   'Thanks for contibuting!'))
        return redirect(reverse('home'))

    messages.error(request, 'We are sorry, but some error occured :(')
    return redirect(reverse('home'))


def about(request):
    """View to render about page."""
    return render(request, 'base/about.html')


def faq(request):
    """View to render faq page."""
    return render(request, 'base/faq.html')