from django.db import models from django.utils.text import slugify from django.utils import timezone from django.core.validators import MinValueValidator, MaxValueValidator from django.conf import settings import os class Category(models.Model): """作品分类模型""" name = models.CharField('分类名称', max_length=100) slug = models.SlugField('URL标识', max_length=100, unique=True, blank=True) created_at = models.DateTimeField('创建时间', auto_now_add=True) updated_at = models.DateTimeField('更新时间', auto_now=True) class Meta: verbose_name = '作品分类' verbose_name_plural = '作品分类' ordering = ['name'] def __str__(self): return self.name def save(self, *args, **kwargs): # 生成唯一的 slug if not self.slug: base_slug = slugify(self.name) if not base_slug: # 如果 slugify 返回空字符串 base_slug = f'category-{self.id}' if self.id else 'category' # 确保 slug 唯一 slug = base_slug counter = 1 while Category.objects.filter(slug=slug).exclude(pk=self.pk).exists(): slug = f'{base_slug}-{counter}' counter += 1 self.slug = slug super().save(*args, **kwargs) class Artwork(models.Model): """摄影作品模型""" title = models.CharField('作品标题', max_length=200) description = models.TextField('作品描述', blank=True) slug = models.SlugField('URL标识', max_length=200, unique=True, blank=True) # 图片字段 image = models.ImageField('作品图片', upload_to='artworks/%Y/%m/%d/') thumbnail = models.ImageField('缩略图', upload_to='thumbnails/%Y/%m/%d/', blank=True, null=True) # 关联字段 category = models.ForeignKey( Category, on_delete=models.SET_NULL, verbose_name='作品分类', null=True, blank=True ) # 元数据字段 order = models.IntegerField('排序序号', default=0, help_text='数字越小越靠前') # 时间字段 created_at = models.DateTimeField('创建时间', default=timezone.now) updated_at = models.DateTimeField('更新时间', auto_now=True) # 统计字段 view_count = models.PositiveIntegerField('浏览次数', default=0) class Meta: verbose_name = '摄影作品' verbose_name_plural = '摄影作品' ordering = ['order', '-created_at'] def __str__(self): return self.title def save(self, *args, **kwargs): # 生成唯一的 slug if not self.slug: base_slug = slugify(self.title) if not base_slug: # 如果 slugify 返回空字符串 base_slug = f'artwork-{self.id}' if self.id else 'artwork' # 确保 slug 唯一 slug = base_slug counter = 1 while Artwork.objects.filter(slug=slug).exclude(pk=self.pk).exists(): slug = f'{base_slug}-{counter}' counter += 1 self.slug = slug # 如果是新对象或图片被更新，生成缩略图 if self.pk is None or 'image' in kwargs.get('update_fields', []): super().save(*args, **kwargs) self.generate_thumbnail() else: super().save(*args, **kwargs) def generate_thumbnail(self): """生成缩略图""" from PIL import Image from io import BytesIO from django.core.files.base import ContentFile import os if not self.image: return # 打开原图 img = Image.open(self.image) # 计算目标像素总量：约90万像素（如1280x720） target_pixels = 900000 # 获取原图尺寸 width, height = img.size current_pixels = width * height # 计算缩放比例，使像素总量接近90万（但限制最大边不超过1600px） max_dimension = 1600 if current_pixels <= 0: # 如果图片尺寸无效，使用原尺寸但限制最大边 if width > max_dimension or height > max_dimension: scale = max_dimension / max(width, height) new_width = int(width * scale) new_height = int(height * scale) img = img.resize((new_width, new_height), Image.Resampling.LANCZOS) else: # 首先计算缩放至目标像素的比例 scale_to_target = (target_pixels / current_pixels) ** 0.5 # 然后检查缩放后是否超过最大边限制 new_width_temp = int(width * scale_to_target) new_height_temp = int(height * scale_to_target) # 如果缩放后的尺寸超过最大边限制，使用最大边限制的比例 if new_width_temp > max_dimension or new_height_temp > max_dimension: scale_to_max = max_dimension / max(new_width_temp, new_height_temp) scale = scale_to_target * scale_to_max else: scale = scale_to_target new_width = int(width * scale) new_height = int(height * scale) img = img.resize((new_width, new_height), Image.Resampling.LANCZOS) # 保存到内存 thumb_io = BytesIO() # 保持原格式，如果是JPEG则优化 # 如果img.format为空，根据文件扩展名判断格式 format_to_use = img.format if not format_to_use: # 根据文件扩展名判断格式 filename = os.path.basename(self.image.name) name, ext = os.path.splitext(filename) ext = ext.lower() if ext in ['.jpg', '.jpeg', '.jpe', '.jfif']: format_to_use = 'JPEG' elif ext in ['.png']: format_to_use = 'PNG' elif ext in ['.gif']: format_to_use = 'GIF' elif ext in ['.webp']: format_to_use = 'WEBP' else: # 默认使用JPEG格式 format_to_use = 'JPEG' if format_to_use == 'JPEG': img.save(thumb_io, format='JPEG', quality=85, optimize=True) else: img.save(thumb_io, format=format_to_use) # 生成缩略图文件名 filename = os.path.basename(self.image.name) name, ext = os.path.splitext(filename) thumb_filename = f'{name}_thumb{ext}' # 保存缩略图 self.thumbnail.save( thumb_filename, ContentFile(thumb_io.getvalue()), save=False ) super().save(update_fields=['thumbnail']) def increment_view_count(self): """增加浏览次数""" self.view_count += 1 self.save(update_fields=['view_count']) def get_dynamic_grid_class(self): """根据图片宽高比动态返回网格CSS类""" try: if not self.thumbnail: return 'col-span-1 row-span-1' # 尝试从缩略图获取尺寸 from PIL import Image import os thumb_path = self.thumbnail.path if not os.path.exists(thumb_path): return 'col-span-1 row-span-1' with Image.open(thumb_path) as img: width, height = img.size aspect_ratio = width / height # 根据宽高比返回不同的网格类 if aspect_ratio > 1.5: # 很宽的图片 return 'col-span-2 row-span-1' # 占2列宽 elif aspect_ratio < 0.67: # 很高的图片 return 'col-span-1 row-span-2' # 占2行高 elif aspect_ratio > 1.2: # 中等宽度 return 'col-span-2 row-span-1' # 占2列宽 elif aspect_ratio < 0.83: # 中等高度 return 'col-span-1 row-span-2' # 占2行高 else: # 接近正方形 return 'col-span-1 row-span-1' except Exception as e: # 发生错误时返回默认 print(f"Error getting grid class for artwork {self.id}: {e}") return 'col-span-1 row-span-1' class About(models.Model): """关于页面模型""" title = models.CharField('标题', max_length=200) content = models.TextField('内容') image = models.ImageField('图片', upload_to='about/', blank=True, null=True) created_at = models.DateTimeField('创建时间', auto_now_add=True) updated_at = models.DateTimeField('更新时间', auto_now=True) class Meta: verbose_name = '关于页面' verbose_name_plural = '关于页面' def __str__(self): return self.title def save(self, *args, **kwargs): # 确保只有一个关于页面 if not self.pk and About.objects.exists(): # 如果已经存在关于页面，更新它而不是创建新的 existing = About.objects.first() existing.title = self.title existing.content = self.content if self.image: existing.image = self.image existing.save() return existing super().save(*args, **kwargs) class Comment(models.Model): """评论模型""" artwork = models.ForeignKey(Artwork, on_delete=models.CASCADE, verbose_name='作品') user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, verbose_name='用户') text = models.TextField('评论内容', blank=True) image = models.ImageField('评论图片', upload_to='comments/%Y/%m/%d/', blank=True, null=True) created_at = models.DateTimeField('创建时间', auto_now_add=True) updated_at = models.DateTimeField('更新时间', auto_now=True) is_active = models.BooleanField('是否有效', default=True) class Meta: verbose_name = '评论' verbose_name_plural = '评论' ordering = ['-created_at'] def __str__(self): return f"{self.user.username} 对 {self.artwork.title} 的评论"