Как установить изображение по умолчанию в зависимости от пола в django-rest-framework?

Я знаю, что здесь есть несколько похожих вопросов, но ни один из них, похоже, не решает мою проблему. Я использую Django-Rest-Framework.

Я создаю профиль пользователя одновременно с созданием пользователя с помощью сигналов.

Поскольку мой вопрос довольно самоочевиден, вот мой код

models.py

from django.contrib.auth.models import AbstractUser
from django.utils.translation import ugettext_lazy as _
from django.db import models
from django.conf import settings
from PIL import Image

GENDER_SELECTION = (
    ('Male', 'Male'),
    ('Female', 'Female'),
)

class CustomUser(AbstractUser):
    username = models.CharField(max_length=100, blank=True, null=True)
    email = models.EmailField(_('email address'), unique=True)
    gender = models.CharField(max_length=20, choices=GENDER_SELECTION)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username', 'first_name', 'last_name', 'gender']

    def __str__(self):
        return self.email


class UserProfile(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name='profile')
    profile_pic = models.ImageField(upload_to='profile/', default='default.png', blank=True)

    def __str__(self):
        return f'{self.user.first_name} Profile'

    def save(self, *args, **kwargs):
        super(UserProfile, self).save(*args, **kwargs)

        uploaded_image = Image.open(self.profile_pic.path)

        if uploaded_image.height > 300 or uploaded_image.width > 300:
            output_size = (300, 300)
            uploaded_image.thumbnail(output_size)
            uploaded_image.save(self.profile_pic.path)

Вот что я пробовал

В models.py

# ...

def save(self, *args, **kwargs):
    super(Profile, self).save(*args, **kwargs)

    if self.profile_pic == 'default.png':
        if CustomUser.gender == 'Male':
            self.profile_pic = 'user_default_m.png'
            return self.profile_pic
        else:
            self.profile_pic = 'user_default_f.png'
            return self.profile_pic
    else:
        uploaded_image = Image.open(self.profile_pic.path)
        if uploaded_image.height > 300 or uploaded_image.width > 300:
            output_size = (300, 300)
            uploaded_image.thumbnail(output_size)
            uploaded_image.save(self.profile_pic.path)

Обновление - Вставка signals.py файла в соответствии с запросом

signals.py

@receiver(post_save, sender=CustomUser)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        Profile.objects.create(user=instance)


@receiver(post_save, sender=CustomUser)
def save_user_profile(sender, instance, **kwargs):
    instance.profile.save()

Вот что я заметил, вы пытаетесь изменить размер загруженного изображения и пытаетесь установить изображение по умолчанию в одном и том же действии. Также вы пытаетесь сделать это post_save.

Я предполагаю, что вы разрешаете пользователю загружать изображение, когда он редактирует свой профиль, что я предполагаю, поскольку вы переопределили метод сохранения.

Я бы постарался избежать объединения этих двух операций. Хотя они и относятся к операциям с изображениями, по сути, это разные операции.

Сохраните ваш файл models.py и восстановите его в исходный код.

Теперь, поскольку вы уже используете сигналы, вы можете попробовать pre_save в вашем signals.py файле. Отредактируйте ваш signals.py файл и добавьте следующий фрагмент кода

@receiver(pre_save, sender=UserProfile)
def set_profile_image(sender, instance, *args, **kwargs):

    # Obtain the gender of the just created user
    gender = CustomUser.objects.all().last().gender

    # Check for the gender and assign the file
    if gender == 'Male' and instance.profile_pic == 'default.png':
        instance.profile_pic = 'default_m.png'
    elif gender == 'Female' and instance.profile_pic == 'default.png':
        instance.profile_pic = 'default_f.png'

Это должно работать.

Вернуться на верх