from django.shortcuts import render, redirect

from exam_23_06.web.forms import ProfileCreateForm, FruitCreateForm, FruitEditForm, FruitDeleteForm, \
    ProfileEditForm
from exam_23_06.web.models import Profile, Fruit


def get_profile():
    users = Profile.objects.all()
    if users:
        return users[0]
    return None


def index(request):
    profile = get_profile()
    context = {
        'profile': profile,
    }
    return render(request, 'index.html', context)


def dashboard(request):
    profile = get_profile()
    fruits = Fruit.objects.all()
    context = {
        'profile': profile,
        'fruits': fruits
    }
    return render(request, 'dashboard.html', context)


def create_fruit(request):
    profile = get_profile()
    if request.method == 'POST':
        form = FruitCreateForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect('dashboard')
    else:
        form = FruitCreateForm()
    context = {
        'profile': profile,
        'form': form
    }

    return render(request, 'create-fruit.html', context)


def fruit_details(request, pk):
    profile = get_profile()
    fruit = Fruit.objects.filter(pk=pk).get()
    context = {
        'fruit': fruit,
        'profile': profile
    }
    return render(request, 'details-fruit.html', context)


def fruit_edit(request, pk):
    profile = get_profile()
    fruit = Fruit.objects.filter(pk=pk).get()
    if request.method == 'POST':
        form = FruitEditForm(request.POST, instance=fruit)
        if form.is_valid():
            form.save()
            return redirect('dashboard')

    else:
        form = FruitEditForm(instance=fruit)

    context = {
        'form': form,
        'profile': profile,
        'fruit': fruit
    }
    return render(request, 'edit-fruit.html', context)


def fruit_delete(request, pk):
    fruit = Fruit.objects.filter(pk=pk).get()
    form = FruitDeleteForm(instance=fruit)
    if request.method == 'POST':
        fruit.delete()
        return redirect('dashboard')

    context = {
        'form': form,
        'fruit': fruit
    }
    return render(request, 'delete-fruit.html', context)


def profile_create(request):
    profile = get_profile()
    if request.method == 'POST':
        form = ProfileCreateForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect('dashboard')
    else:
        form = ProfileCreateForm()
    context = {
        'form': form,
        'profile': profile
    }
    return render(request, 'create-profile.html', context)


def profile_details(request):
    profile = get_profile()
    fruits = Fruit.objects.all()
    all_posts = None
    if len(fruits) == 0:
        all_posts = 0
    elif len(fruits) > 0:
        all_posts = len(fruits)
    context = {
        'profile': profile,
        'all_posts': all_posts
    }
    return render(request, 'details-profile.html', context)


def profile_edit(request):
    user = get_profile()
    if request.method == "POST":
        form = ProfileEditForm(request.POST, instance=user)
        if form.is_valid():
            form.save()
            return redirect('profile details')
    else:
        form = ProfileEditForm(instance=user)

    context = {
        'form': form,
        'user': user
    }
    return render(request, 'edit-profile.html', context)


def profile_delete(request):
    return render(request, 'delete-profile.html')