# Create your tasks here
from __future__ import absolute_import, unicode_literals
from apps.hrm.models import Employee
from apps.utils.task import CustomTask
from apps.vm.models import Visit, Vpeople
from celery import shared_task


@shared_task(base=CustomTask)
def visit_audit_end(ticket_id):
    visit = Visit.objects.get(ticket__id=ticket_id)
    if visit.state == Visit.V_AUDIT:
        visit.state = Visit.V_ENTER
        visit.save()
    # 更新企业访客人员库
    for i in Vpeople.objects.filter(visit=visit):
        visitor = i.visitor
        ep = Employee.objects.filter(id_number=visitor.id_number).first()
        if ep:
            pass
        else:
            ep = Employee()
        ep.name = visitor.name
        ep.phone = visitor.phone
        ep.photo = visitor.photo
        ep.id_number = visitor.id_number
        ep.type = 'visitor'
        ep.save()
        visitor.employee = ep
        visitor.save()