Django. Как правильно сохранить сгенерированный файл в FileField?

Question

[Egegey]

Привет!

Я крайне неопытен в Django и мне нужна помощь.
Я попытался переопределить метод save(), чтобы вместе с ним происходила генерация QR кода и последующее сохранение.
Однако при cохранении записи в папке media у меня появляются два файла.

Один файл ("10.png") состоит из поля slug модели и расширения.
Файл создаётся после выполнения этой строки:

self.qr.save(self.slug+'.png', BytesIO(qr), save=False)

Имя другого файла генерирует сам Django (уже после создания первого) и его же сохраняет в поле qr модели.

Как сделать так, чтобы создавался только один файл ( только "10.png") и в поле сохранялся именно он?

Model.py

class Url(models.Model):
    slug = models.CharField(max_length=50, unique=True)
    qr = models.FileField(upload_to='', blank=True, null=True)

    def save(self, *args, **kwargs):
        qr = self.qr_generate(self.slug)
        self.qr.save(self.slug+'.png', BytesIO(qr), save=False)
        super(Url, self).save(*args, **kwargs)

    def __str__(self):
        return self.slug

    @staticmethod
    def qr_generate(slug):
        qr = qrcode.QRCode(
            version=None,
            error_correction=qrcode.constants.ERROR_CORRECT_L,
            box_size=10,
            border=4,
        )
        qr.add_data("somedata" + str(slug))
        qr.make(fit=True)

        img = qr.make_image(fill_color="black", back_color="white")
        qrByte = BytesIO()
        img.save(qrByte)
        return qrByte.getvalue()

Comments

[Модератор]

Egegey дорогой пользователь, настоятельно рекомендуем еще раз обратить самое пристальное внимание на п. 3.1 регламента работы сервиса (и, в особенности, на его последний абзац). В противном случае, ваши вопросы будут удаляться по причине тег-спама, а систематические нарушения приведут к блокировке учетной записи.

Answer 1

[Вадим]

А если убрать super(Url, self).save(*args, **kwargs) ?

Comments

[Egegey]

В этом случае в базу не сохраняется ничего.
Но так ведь и должно быть, там же в переопределённом методе много чего ещё выполняется.

Ещё я сделал вот так и в итоге работает.
В папке только один файл и в поле лежит нужный файл.

def save(self, *args, **kwargs):
        if self.qr == None:
            img = self.qr_generate(self.slug)
            self.qr.save(self.slug + '.png', BytesIO(img), save=False)
        super(Url, self).save(*args, **kwargs)

Только всё равно не совсем понимаю принцип работы.
И с объектом FieldFile (не поле!) не понимаю как работать. Доку читал.

[Вадим]

А посмотри саму базу, какие там поля? Что стоит в столбце qr ?

[Egegey]

Нашёл ошибку у себя в коде.
Я также переопределял метод form_valid и там дважды вызывал save.
В итоге один вызов убрал и стало норм, указанный выше models.py работает как и хотел

[Вадим]

Интересно FieldFile существует как поле в базе данных... после миграции?

[Egegey]

Не совсем понял вопроса. Я профан в этом деле.

https://docs.djangoproject.com/en/3.0/ref/models/f...

[Вадим]

Я профан в этом деле - я тоже пока )) Просто было интересно, сохраняется ли файл в самой базе данных как binary field или как ссылка. Но из твоего отрывка похоже что это просто текст с названием файла.

Какой DB viewer используешь?

Answer 2

[JRazor]

Можно QR вынести в отдельную модель и работать внутри нее с qr кодом. А в URL уже подвязывать объект через OneToOne

Answer 3

[sanchemba]

У меня была такая же проблема. Дело в том, что при самостоятельной привязке файла к полю, видимо, нужно отключить генератор имен файлов в поле FileField и/или включать перезапись файлов. Как это сделать, и можно ли вообще, я пока не разобрался. Очевидно, при сохранении экземпляра модели и, соответственно, поля (в Вашем случае qr), проверяется наличие файла по пути указанному в path (а Вы, если я правильно понимаю, это делаете вызывая self.qr.save()). Если указанный путь "занят", файл не перезаписывается, а генерируется новое имя файла.
Примерно так это работает. Если кто-то подскажет как изменить поведение или переопределить метод save поля FileField, буду очень рад.
Я же сделал довольно топорно:
У меня есть отдельный метод, отвечающий за составление имён файлов, который передаётся в поле upload_to и он то и будет определять конечное имя файла.
В переопределённом методе save() модели я запоминаю путь к файлу, который привязываю к полю FileField. После вызова super(Url, self).save(*args, **kwargs), я просто удаляю временный файл (именно тот, что сам привязывал, а путь к тому, что реально остался определился в методе, передаваемом в upload_to при сохранении модели).
Я делаю сервис для себя, поэтому мне пока норм. Но при большом потоке файлов это может быть довольно накладно. Так что продолжаю искать лучшие пути...