from django.shortcuts import render, redirect
from .models import Notice
from django.utils import timezone
from .forms import NoticeForm
from django.contrib.auth.decorators import login_required


@login_required
def new(request):
    if request.method == 'POST':
        form = NoticeForm(request.POST)
        if form.is_valid():
            notice = Notice()
            notice.notice_title = form.cleaned_data['notice_title']
            notice.notice_text = form.cleaned_data['notice_text']
            notice.pub_start = form.cleaned_data['pub_start']
            notice.pub_end = form.cleaned_data['pub_end']
            notice.save()
            return redirect('index')
    context = {'form': NoticeForm()}
    return render(request, 'polls/edit.html', context)

@login_required
def delete(request, deleteId=None):
    if deleteId is not None:
        try:
            notice = Notice.objects.get(pk=deleteId)
            if request.user.id == notice.user_id:
                notice.delete()
        except:
            pass
    return redirect('index')

def index(request):
    notices = Notice.objects.all()
    notices = notices.filter(pub_start__lte=timezone.now())
    notices = notices.filter(pub_end__gte=timezone.now())
    context = {'notices': notices}
    return render(request, 'polls/index.html', context)