mirror of
https://github.com/spbleadersofdigtal/backend.git
synced 2024-11-27 13:23:44 +03:00
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
from django.db import models
|
|
|
|
from pitch_deck_generator.common.signals import (
|
|
create_cropped_model_image,
|
|
delete_cropped_model_image,
|
|
update_cropped_model_image,
|
|
)
|
|
from pitch_deck_generator.utils.files import user_file_upload_mixin
|
|
from pitch_deck_generator.utils.generators import generate_charset
|
|
|
|
|
|
class BaseImageModel(models.Model):
|
|
"""
|
|
stores user's images in their media folder, creates, updates and deletes preview
|
|
"""
|
|
|
|
image = models.ImageField(upload_to=user_file_upload_mixin, blank=True)
|
|
image_cropped = models.ImageField(upload_to="cropped/", blank=True)
|
|
|
|
@classmethod
|
|
def __init_subclass__(cls, **kwargs):
|
|
super().__init_subclass__(**kwargs)
|
|
models.signals.pre_save.connect(update_cropped_model_image, sender=cls)
|
|
models.signals.post_save.connect(create_cropped_model_image, sender=cls)
|
|
models.signals.post_delete.connect(delete_cropped_model_image, sender=cls)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
def create_model_slug(sender, instance, **kwargs):
|
|
def _generate_charset():
|
|
if private:
|
|
return generate_charset(private_slug_length)
|
|
return generate_charset(slug_length)
|
|
|
|
if instance.id is None:
|
|
model = sender
|
|
slug_length = 5
|
|
private_slug_length = 20
|
|
private = False
|
|
|
|
if hasattr(model, "SlugMeta"):
|
|
if hasattr(model.SlugMeta, "slug_length"):
|
|
slug_length = model.SlugMeta.slug_length
|
|
if hasattr(model.SlugMeta, "private_slug_length"):
|
|
private_slug_length = model.SlugMeta.private_slug_length
|
|
if hasattr(instance, "private"):
|
|
if instance.private:
|
|
private = True
|
|
if hasattr(instance, "public"):
|
|
if not instance.public:
|
|
private = True
|
|
|
|
slug = _generate_charset()
|
|
while model.objects.filter(slug=slug).exists():
|
|
slug = _generate_charset()
|
|
instance.slug = slug
|
|
|
|
|
|
class SlugModel(models.Model):
|
|
"""
|
|
model to store and generate slug for model instances
|
|
for custom slug length use: slug_length, private_slug_length SlugMeta options
|
|
"""
|
|
|
|
slug = models.SlugField(max_length=20, blank=True, unique=True)
|
|
|
|
@classmethod
|
|
def __init_subclass__(cls, **kwargs):
|
|
super().__init_subclass__(**kwargs)
|
|
models.signals.pre_save.connect(create_model_slug, sender=cls)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
class SlugMeta:
|
|
slug_length = 5
|
|
private_slug_length = 20
|