from django.views.generic import CreateView, UpdateView
from django.urls import reverse_lazy

from .models import ClientProfile

from localauth.mixins import LoginRequiredMixin

class ClientRegistrationView(LoginRequiredMixin, CreateView):
    model = ClientProfile
    exclude = ["user"]
    template_name = "clients/signup.html"
    fields = ["company", "vat_id", "first_name", "last_name", "street", "city", "zip", "state", "country"]

    def form_valid(self, form):
        form.instance.user = self.request.user
        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy("clients:profile")

    def get_initial(self):
        try:
            partner = self.request.user.partnerprofile
            return {
                "company": partner.company,
                "vat_id": partner.vat_id,
                "first_name": partner.first_name,
                "last_name": partner.last_name,
                "street": partner.street,
                "city": partner.city,
                "zip": partner.zip,
                "state": partner.state,
                "country": partner.country
            }

        except:
            return {}

class ClientProfileView(LoginRequiredMixin, UpdateView):
    model = ClientProfile
    exclude = ["user"]
    template_name = "clients/profile.html"
    fields = ["company", "vat_id", "first_name", "last_name", "street", "city", "zip", "state", "country"]

    def get_success_url(self):
        return reverse_lazy("clients:profile")