from django.contrib.auth.models import User
from django.db import models
from project.models import Project


class Employee(models.Model):
    user = models.ForeignKey(User)
    phone = models.CharField(max_length=15)
    salary = models.FloatField()
    emp_id = models.CharField(max_length=15)
    start_date = models.DateField()
    end_date = models.DateField(null=True, blank=True)
    projects = models.ManyToManyField(Project)

    def __str__(self):
        return self.name