from django.shortcuts import render
from django.http import JsonResponse, HttpResponse
from django.shortcuts import redirect

from django.views.generic import ListView
from tickets.models import Ticket, User
from tickets.forms import UserForm

def index(request):
    return render(request, "tickets/main_tick.html")

class TicketListView(ListView):
    model = Ticket
    template_name = 'tickets/main_tick.html'

    def get(self, request, *args, **kwargs):
        if request.is_ajax():
            return self.get_for_ajax()
        else:
            return super(TicketListView, self)\
                .get(request, *args, **kwargs)

    def get_for_ajax(self):
        go_from = self.request.GET.get('go_from', None)
        go_to = self.request.GET.get('go_to', None)
        filtered_obj = Ticket.objects.filter(go_from=go_from,
                                             go_to=go_to)
        return render(self.request, "tickets/tick_list.html",
                      {'object_list': filtered_obj})

    def get_context_data(self, **kwargs):
        context = super(TicketListView, self)\
            .get_context_data(**kwargs)
        context['from_places'] = [ticket.go_from for ticket in Ticket.objects.all()]
        context['to_places'] = [ticket.go_to for ticket in Ticket.objects.all()]

        return context

def CreateTicket(request):
    if request.method == 'POST':
        form = UserForm(data = request.POST)
        if form.is_valid():
            form.save()

            return redirect('/index')
        else:
            return render(request, 'tickets/main_tick.html', {'form':form})
    else:
        return render_to_response('tickets/main_tick.html', {},)