Untitled

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.shortcuts import get_object_or_404, render
from django.views import generic
from django.http import HttpResponseRedirect
from .models import Colour, Mode,Led
from .forms import LoginForm, LedForm
from django.contrib.auth import authenticate,login
from django.contrib.auth.decorators import login_required


# Create your views here.
@login_required(login_url='/')
def index(request):
    ledlist= Led.objects.all()
    colourList = Colour.objects.all()
    modeList = Mode.objects.all()

    if 'butt' in request.POST:
        # print (request.POST.get('mod'))
        # print (request.POST.get('clr'))
        led = Led.objects.all()[0]
        mode = Mode.objects.get(name=request.POST.get('mod'))
        colour = Colour.objects.get(name=request.POST.get('clr'))
        led.mode = mode
        led.colour = colour
        # mode = Mode.objects.filter()
        led.save()

    return render(request, 'LEDs/index.html', {'ledlist': ledlist, 'colourList': colourList, 'modeList': modeList})


def home(request):
    title = "Welcome to LED Control Site"
    form = LoginForm(request.POST or None)
    state = "Please login:"
    if form.is_valid():
        user = authenticate(username=form.cleaned_data.get("username"), password=form.cleaned_data.get("password"))
        if user is not None:
            login(request, user)

            return HttpResponseRedirect('/leds')
        else:
            return HttpResponseRedirect('/')

    return render(request, 'home.html', {'template_title': title,'form': form,'state': state})