from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
from django.dispatch import receiver

class WorkerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    university = models.CharField(max_length=30, blank=True)
    birth_date = models.DateField(null=True, blank=True)

    role = models.CharField(max_length = 10, default = 'USER')


    def __str__(self):
        return self.user.username

@receiver(post_save, sender=User)
def create_worker_profile(sender, instance, created, **kwargs):
    if created:
        WorkerProfile.objects.create(user=instance)

@receiver(post_save, sender=User)
def save_worker_profile(sender, instance, **kwargs):
    instance.workerprofile.save()







class CustomerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    university = models.CharField(max_length=30, blank=True)
    birth_date = models.DateField(null=True, blank=True)

    role = models.CharField(max_length = 10, default = 'CUSTOMER')
    needLaundryDone = models.BooleanField(default = False)

    def __str__(self):
        return self.user.username

@receiver(post_save, sender=User)
def create_customer_profile(sender, instance, created, **kwargs):
    if created:
        CustomerProfile.objects.create(user=instance)

@receiver(post_save, sender=User)
def save_customer_profile(sender, instance, **kwargs):
    instance.customerprofile.save()

from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User

class WorkerSignUpForm(UserCreationForm):
    #birth_date and university fields need to be declared seperately because they are not apart of User:
    birth_date = forms.DateField(help_text='Required. Format: YYYY-MM-DD')
    university = forms.CharField()


    class Meta:
        model = User
        fields = ('username',
                  'email',
                  'first_name',
                  'last_name',
                  'birth_date',
                  'university',
                  'password1',
                  'password2', )


class CustomerSignUpForm(UserCreationForm):
    #birth_date and university fields need to be declared seperately because they are not apart of User:
    birth_date = forms.DateField(help_text='Required. Format: YYYY-MM-DD')
    university = forms.CharField()


    class Meta:
        model = User
        fields = ('username',
                  'email',
                  'first_name',
                  'last_name',
                  'birth_date',
                  'university',
                  'password1',
                  'password2', )

def signup(request):
    if request.method == 'POST':
        form_worker = WorkerSignUpForm(request.POST)
        form_customer = CustomerSignUpForm(request.POST)
        if form_worker.is_valid():
            user = form_worker.save()
            user.refresh_from_db()  # load the profile instance created by the signal
            user.workerprofile.birth_date = form_worker.cleaned_data.get('birth_date')
            user.workerprofile.university = form_worker.cleaned_data.get('university')
            user.save()  # explicitly save custom fields not in User model
            raw_password = form_worker.cleaned_data.get('password1')
            user = authenticate(username=user.username, password=raw_password)
            login(request, user)  # login user after signup
            return redirect('home')
        elif form_customer.is_valid():
            user = form_customer.save()
            user.refresh_from_db()  # load the profile instance created by the signal
            user.customerprofile.birth_date = form_customer.cleaned_data.get('birth_date')
            user.customerprofile.university = form_customer.cleaned_data.get('university')
            user.save()  # explicitly save custom fields not in User model
            raw_password = form_customer.cleaned_data.get('password1')
            user = authenticate(username=user.username, password=raw_password)
            login(request, user)  # login user after signup
            return redirect('home')
    else:
        form_worker = WorkerSignUpForm(request.POST)
        form_customer = CustomerSignUpForm(request.POST)
    return render(request, 'core/signup.html', {'form_worker': form_worker,'form_customer': form_customer })

{%  extends 'core/base.html' %}

{% block head %}
    <title> Sign Up</title>
{% endblock %}
{% block body %}


  <h3>Sign Up As Worker</h3>
  <form method="post">
    {% csrf_token %}
    {{ form_worker.as_p }}
    <button type="submit">Sign up</button>
  </form>

   <h3>Sign Up As Customer</h3>
  <form method="post" action = "home.html">
    {% csrf_token %}
    {{ form_customer.as_p }}
    <button type="submit">Sign up</button>
  </form>
{% endblock %}