from django.contrib.auth.models import User
from django.db import models
from phonenumber_field.modelfields import PhoneNumberField
from project.models import Project


class Employee(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    employee_id = models.CharField(max_length=15)
    phone = PhoneNumberField()
    salary = models.DecimalField()
    start_date = models.DateField()
    end_date = models.DateField(null=True, blank=True)
    projects = models.ManyToManyField(Project)

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