from django.db import models
from apps.hrm.models import Employee
from apps.utils.models import CommonAModel, CommonBModel, BaseModel
from apps.system.models import User


# Create your models here.
class Visit(CommonBModel):
    """
    访问项目
    """
    VISIT_PURPOSE_CHOICES = (
        (10, '参观'),
        (20, '拜访'),
        (30, '面试'),
        (40, '开会')
    )
    purpose = models.PositiveSmallIntegerField('来访事由')
    description = models.CharField('来访详述', max_length=200)
    visit_time = models.DateTimeField('来访时间')
    leave_time = models.DateTimeField('离开时间')
    receptionist = models.ForeignKey(User, verbose_name='接待人', on_delete=models.CASCADE)


class VisitPeople(BaseModel):
    """
    访客项目人员
    """
    visit = models.ForeignKey(Visit, verbose_name='关联访问项目', on_delete=models.CASCADE)
    visitor = models.ForeignKey(Employee, verbose_name='访客', on_delete=models.CASCADE)
    is_manager = models.BooleanField('是否主访人', default=False)