Код на салфетке

Генерация кода для файла.
Отвлечёмся от основной темы поста.
Не уходя из файла models.py добавим генерацию кода для файла.

В классе PostFilesModel создадим метод генерации уникального кода и переопределим метод сохранения модели. Должно получиться так:

import random


    def save(self, *args, **kwargs):
        if not self.code:
            self.code = self.generate_unique_code()
        super().save(*args, **kwargs)

    @staticmethod
    def generate_unique_code():
        code = random.randint(100000, 999999)
        while PostFilesModel.objects.filter(code=code).exists():
            code = random.randint(100000, 999999)
        return code

Не обязательно, но поле code можно сделать полем "только для чтения".
Перейдём в файл admin.py и найдём класс PostFilesAdmin. Добавим в него всего одно поле:

    readonly_fields = ('code',)

Теперь код будет генерироваться при создании объекта, а в панели администратора нельзя будет его изменить.

Применим миграции:

python manage.py makemigrations
python manage.py migrate

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

<h2 class="mb-3">Дополнительные материалы</h2>  
<a href="{{ post.file.file.path }}" class="mb-1">Скачать материалы и исходный код с сайта.</a>  
<p class="mt-1">Или в <a href="https://t.iss.one/press_any_button_bot">Telegram-боте</a> по коду: <b>{{ post.file.code }}</b></p>

Изменение сериализатора.
Изменим сериализатор, чтобы он возвращал не только название файла и путь до него, но и поля из поста, например, ссылку на пост.
Добавим два новых поля:
- post_url - возвращающий ссылку на пост, для этого задействуем метод SerializerMethodField, поскольку получение полного URL до поста, это метод get_absolute_url. Параметр read_only=True позволяет, в случае если у файла нет связанного поста, получить None вместо исключения.
- post_telegram_link - возвращающий ссылку на пост в Telegram.

И метод, получающий URL поста, и добавляющий к нему UTM-метку для сбора статистики переходов.
В методе используем try-except, также чтобы избежать исключения, если у файла нет связанного поста.

Полный код сериализатора:

class FileModelSerializer(serializers.ModelSerializer):
    file_path = serializers.SerializerMethodField()
    post_url = serializers.SerializerMethodField()
    post_telegram_link = serializers.StringRelatedField(source='post.telegram_link', read_only=True)

    class Meta:
        model = models.PostFilesModel
        fields = ['title', 'file_path', 'post_url', 'post_telegram_link']

    @staticmethod
    def get_file_path(obj):
        return obj.file.path if obj.file else None

    @staticmethod
    def get_post_url(obj: models.PostFilesModel):
        try:
            url = obj.post.get_absolute_url()
            utm_params = '?utm_source=telegram&utm_medium=bot&utm_campaign=get_file'
            return get_current_site(None).domain + url + utm_params
        except ObjectDoesNotExist:
            return None

Следующим постом внесём изменения в наш бот.

Файлы к посту, можно получить в боте по коду: 187343

❤1👍1

195 viewsedited 10:53