29 lines
815 B
Python
29 lines
815 B
Python
import os
|
|
|
|
from django.db import models
|
|
from django.db.models.signals import post_delete
|
|
from django.dispatch import receiver
|
|
|
|
# Create your models here.
|
|
|
|
|
|
class FaceDetection(models.Model):
|
|
company_id = models.OneToOneField(
|
|
"base.Company", related_name="face_detection", on_delete=models.CASCADE
|
|
)
|
|
start = models.BooleanField(default=False)
|
|
|
|
|
|
class EmployeeFaceDetection(models.Model):
|
|
employee_id = models.OneToOneField(
|
|
"employee.Employee", related_name="face_detection", on_delete=models.CASCADE
|
|
)
|
|
image = models.ImageField()
|
|
|
|
|
|
@receiver(post_delete, sender=EmployeeFaceDetection)
|
|
def delete_image_file(sender, instance, **kwargs):
|
|
if instance.image and instance.image.path:
|
|
if os.path.isfile(instance.image.path):
|
|
os.remove(instance.image.path)
|