User.view

from django.shortcuts import render, redirect, get_object_or_404, HttpResponseRedirect
from django.contrib.auth import login, logout, authenticate
from .forms import RegisterCustomerForm, RegisterAgentForm
from home.models import Property
from .models import Customer, User, Agent
#from django.contrib.auth.decorators import login_required


def favourite_add(request, id):
    property = get_object_or_404(Property, id=id)
    if property.favourites.filter(id=request.customer.id).exists():
        property.favourites.remove(request.customer)
    else:
        property.favourites.add(request.customer)
        return HttpResponseRedirect(request.META['HTTP_REFERER'])


def login_user(request):
    if request.method == "POST":
        email = request.POST["email"]
        password = request.POST["password"]
        user = authenticate(request, email=email, password=password)
        if user is not None:
            login(request, user)
            return redirect("home")
        else:
            return redirect("login")
    else:
        return render(request, "authenticate/login.html")


def register_customer(request):
    if request.method == "POST":
        form = RegisterCustomerForm(request.POST)
        if form.is_valid():
            form.save()
            username = form.cleaned_data["email"]
            password = form.cleaned_data["password1"]
            user = authenticate(username=username, password=password)
            login(request, user)
            return redirect("home")
    else:
        form = RegisterCustomerForm()

    return render(request, "authenticate/register_customer.html", {"form": form})


def register_agent(request):
    if request.method == "POST":
        form = RegisterAgentForm(request.POST)
        if form.is_valid():
            form.save()
            username = form.cleaned_data["email"]
            password = form.cleaned_data["password1"]
            user = authenticate(username=username, password=password)
            login(request, user)
            return redirect("home")
    else:
        form = RegisterAgentForm()

    return render(request, "authenticate/register_agent.html", {"form": form})


def logout_user(request):
    logout(request)
    return redirect("home")


def profile_page(request):
    agent = User.objects.all()
    customer = Customer.objects.all()
    properties = Property.objects.all()
    property = Property.objects.filter(favourites=request.user.id)

    context = {
        "agent": agent,
        "customer": customer,
        "properties": properties,
        "property":property
        }

    return render(request, "authenticate/profile.html", context)