views.py

from django.http import HttpResponseRedirect
from django.utils import timezone
from django.shortcuts import get_object_or_404
from django.core.urlresolvers import reverse
from django.views import generic

from .models import Links

# Create your views here.
everything = False


class RedirView(generic.RedirectView):
    def get_context_data(self, **kwargs):
        self.permanent = True
        # Call the base implementation first to get a context
        c = super(RedirView, self).get_context_data(**kwargs)
        # add the request to the context
        c.update({'request': self.request})
        return c

    def get_redirect_url(self):
        filler = self.request.COOKIES['sessionid']
        return reverse('polls:index', args=(filler,))


class IndexView(generic.ListView):
    def get_context_data(self, **kwargs):
        # Call the base implementation first to get a context
        c = super(IndexView, self).get_context_data(**kwargs)
        # add the request to the context
        c.update({'request': self.request})
        return c

    template_name = 'polls/index.html'
    context_object_name = 'latest_links_list'

    def get_queryset(self):
        global everything
        tmp = Links.objects.order_by('-date_pub')
        return [[l for l in tmp if l.own == self.request.COOKIES['sessionid']][:5], tmp][everything]


class DetailView(generic.DetailView):
    model = Links
    template_name = 'polls/detail.html'
    context_object_name = 'link'


class ResultsView(generic.DetailView):
    model = Links
    template_name = 'polls/result.html'


def owner(request, link_id):
    filler = request.COOKIES['sessionid']
    link_obj = get_object_or_404(Links, pk=link_id)
    if "youtu" in request.POST['link']:
        link_obj.link = request.POST['link']
        link_obj.save()
    return HttpResponseRedirect(reverse('polls:index', args=(filler,)))


def adder(request):
    filler = request.COOKIES['sessionid']
    global everything
    link_text = request.POST['link']
    p = Links(link=link_text, own=request.COOKIES['sessionid'], date_pub=timezone.now())
    if "i want to see everything" == link_text.lower():
        everything = True
        return HttpResponseRedirect(reverse('polls:index', args=(filler,)))
    elif "i dont want to see everything" == link_text.lower():
        everything = False
        return HttpResponseRedirect(reverse('polls:index', args=(filler,)))
    if "youtu" in link_text:
        p.save()
    return HttpResponseRedirect(reverse('polls:index', args=(filler,)))