mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-15 13:26:45 +03:00
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from passfinder.common.tasks import crop_model_image
|
||
|
|
{ % - if cookiecutter.use_celery != "y" %}
|
||
|
|
from passfinder.utils.files import crop_image
|
||
|
|
{ % - endif %}
|
||
|
|
|
||
|
|
|
||
|
|
def create_cropped_model_image(sender, instance, created, **kwargs):
|
||
|
|
model = sender
|
||
|
|
if created:
|
||
|
|
if instance.image:
|
||
|
|
{ % - if cookiecutter.use_celery == "y" %}
|
||
|
|
crop_model_image.apply_async(
|
||
|
|
kwargs={
|
||
|
|
"pk": instance.pk,
|
||
|
|
"app_label": model._meta.app_label,
|
||
|
|
"model_name": model._meta.model_name,
|
||
|
|
},
|
||
|
|
countdown=2,
|
||
|
|
)
|
||
|
|
{ % - else - %}
|
||
|
|
instance.image_cropped.save(
|
||
|
|
instance.image.path.split(".")[0].split("/")[-1] + ".png",
|
||
|
|
File(crop_image(instance.image.path, length=250)),
|
||
|
|
save=False,
|
||
|
|
)
|
||
|
|
instance.save(update_fields=["image_cropped"])
|
||
|
|
{ % - endif %}
|
||
|
|
|
||
|
|
|
||
|
|
def update_cropped_model_image(sender, instance, **kwargs):
|
||
|
|
model = sender
|
||
|
|
if instance.id:
|
||
|
|
previous = model.objects.get(id=instance.id)
|
||
|
|
if previous.image != instance.image:
|
||
|
|
# delete previous cropped image
|
||
|
|
if instance.image_cropped:
|
||
|
|
if os.path.isfile(instance.image_cropped.path):
|
||
|
|
os.remove(instance.image_cropped.path)
|
||
|
|
# run task to create new cropped image
|
||
|
|
if kwargs["update_fields"] != frozenset({"image_cropped"}) and instance:
|
||
|
|
if instance.image:
|
||
|
|
{ % - if cookiecutter.use_celery == "y" %}
|
||
|
|
crop_model_image.apply_async(
|
||
|
|
kwargs={
|
||
|
|
"pk": instance.pk,
|
||
|
|
"app_label": model._meta.app_label,
|
||
|
|
"model_name": model._meta.model_name,
|
||
|
|
},
|
||
|
|
countdown=2,
|
||
|
|
)
|
||
|
|
{ % - else - %}
|
||
|
|
instance.image_cropped.save(
|
||
|
|
instance.image.path.split(".")[0].split("/")[-1] + ".png",
|
||
|
|
File(crop_image(instance.image.path, length=250)),
|
||
|
|
save=False,
|
||
|
|
)
|
||
|
|
instance.save(update_fields=["image_cropped"])
|
||
|
|
{ % - endif %}
|
||
|
|
else:
|
||
|
|
instance.image_cropped = None
|
||
|
|
|
||
|
|
|
||
|
|
def delete_cropped_model_image(sender, instance, **kwargs):
|
||
|
|
if instance.image_cropped:
|
||
|
|
if os.path.isfile(instance.image_cropped.path):
|
||
|
|
os.remove(instance.image_cropped.path)
|