diff --git a/extend/models.py b/extend/models.py index 82aca54..bcbdde0 100644 --- a/extend/models.py +++ b/extend/models.py @@ -1,7 +1,8 @@ -from datetime import datetime from django.db import models from django.contrib.auth.models import User from django.utils.safestring import mark_safe +from django.utils import timezone +from django.core.validators import FileExtensionValidator from mptt.models import MPTTModel, TreeForeignKey from pytils.translit import slugify from model_utils import Choices @@ -77,9 +78,15 @@ class Photo(models.Model): user = models.ForeignKey(User, null=True, on_delete=models.SET_NULL, verbose_name="Пользователь") title = models.CharField("Название", blank=True, max_length=255) image = ThumbnailerImageField( - "Фото", upload_to="photos/", max_length=255, default="", resize_source=dict(size=(825, 825), sharpen=True) + "Фото", + upload_to="photos/", + max_length=255, + default="", + resize_source=dict(size=(825, 825), sharpen=True), + validators=[FileExtensionValidator( + allowed_extensions=["png", "jpg", "jpeg", "gif"])] ) - created = models.DateTimeField("Создано", default=datetime.now) + created = models.DateTimeField("Создано", default=timezone.now) updated = models.DateTimeField("Изменено", auto_now=True) class Meta: