242 lines
15 KiB
Python
242 lines
15 KiB
Python
import os
|
||
import logging
|
||
from io import BytesIO
|
||
from django.apps import AppConfig
|
||
from django.core.files.base import ContentFile
|
||
from django.db.models.signals import pre_save
|
||
from PIL import Image as PILImage
|
||
import pillow_heif
|
||
|
||
from lpon_site.settings import (
|
||
THUMBNAIL_WEBP_QUALITY,
|
||
FILER_ENABLE_PERMISSIONS,
|
||
FILER_UPLOADER_MAX_FILE_SIZE,
|
||
FILER_WHITELIST_FOR_PATH_ACCESS,
|
||
MIME_TYPE_WHITELIST,
|
||
FILE_VALIDATORS,
|
||
)
|
||
|
||
# Регистрируем плагин HEIF в Pillow
|
||
pillow_heif.register_heif_opener()
|
||
|
||
# Получаем логгер для текущего модуля
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# Кэш для размеров изображений больше не нужен!
|
||
# Filer сам распознает WebP по mime_type и создаст Image запись
|
||
# PIL автоматически получит размеры из WebP файла
|
||
|
||
|
||
# ==============================================================================
|
||
# Конфигурация приложения для фронтенда lpon
|
||
# ==============================================================================
|
||
class FrontendConfig(AppConfig):
|
||
name = 'frontend'
|
||
verbose_name = 'Сайт lpon.ru'
|
||
default_auto_field = 'django.db.models.AutoField'
|
||
|
||
def ready(self):
|
||
from django.contrib import admin
|
||
admin.site.site_header = 'Управление LPON'
|
||
admin.site.site_title = 'LPON Administrator'
|
||
admin.site.index_title = 'Добро пожаловать в LPON'
|
||
|
||
# ==============================================================================
|
||
# Кастомная конфигурация для django-filer
|
||
# - Патчинг MultiStorageFieldFile.save() для автоматической конвертации в WebP
|
||
# ==============================================================================
|
||
class CustomFilerConfig(AppConfig):
|
||
name = 'filer'
|
||
verbose_name = 'Медиафайлы'
|
||
|
||
# ========================================================================
|
||
# Атрибуты конфигурации Django-Filer (импортированы из settings.py)
|
||
# ========================================================================
|
||
FILER_ENABLE_PERMISSIONS = FILER_ENABLE_PERMISSIONS
|
||
FILER_UPLOADER_MAX_FILE_SIZE = FILER_UPLOADER_MAX_FILE_SIZE
|
||
FILER_WHITELIST_FOR_PATH_ACCESS = FILER_WHITELIST_FOR_PATH_ACCESS
|
||
MIME_TYPE_WHITELIST = MIME_TYPE_WHITELIST
|
||
FILE_VALIDATORS = FILE_VALIDATORS
|
||
|
||
|
||
@staticmethod
|
||
def _convert_to_webp_if_needed(name: str, content):
|
||
"""
|
||
Преобразует загруженное изображение в WebP формат.
|
||
Поддерживает JPEG, PNG, BMP, TIFF и HEIC/HEIF.
|
||
|
||
Возвращает кортеж: (content, new_name, was_converted, image_width, image_height)
|
||
|
||
КРИТИЧНО: Функция ДОЛЖНА получать размеры из УЖЕ ПРЕОБРАЗОВАННОГО WebP файла, а не из исходника!
|
||
Почему:
|
||
- PIL может открыть HEIC (благодаря pillow_heif) но размеры исходного могут отличаться
|
||
- После конвертации HEIC→WebP размеры могут измениться (масштабирование, кроппинг)
|
||
- Filer использует размеры для определения каких миниатюр генерировать
|
||
- Размеры ДОЛЖНЫ совпадать с реальным WebP файлом в хранилище!
|
||
"""
|
||
_, original_ext = os.path.splitext(name)
|
||
image_width = None
|
||
image_height = None
|
||
|
||
# Список расширений, которые конвертируем в WebP
|
||
convertible_extensions = [".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".heic", ".heif"]
|
||
|
||
if original_ext.lower() in convertible_extensions:
|
||
try:
|
||
# Шаг 1: Открываем исходный файл с помощью PIL (поддерживает HEIC благодаря pillow_heif)
|
||
content.seek(0)
|
||
img = PILImage.open(BytesIO(content.read()))
|
||
logger.debug(f"Открыт файл '{name}' с расширением '{original_ext}'. Размер: {img.size}")
|
||
|
||
# Шаг 2: Конвертируем CMYK в RGB если требуется (WebP не поддерживает CMYK)
|
||
if img.mode == 'CMYK':
|
||
img = img.convert('RGB')
|
||
logger.debug("Конвертировано CMYK → RGB")
|
||
|
||
# Шаг 3: Сохраняем исходное изображение в WebP формат в памяти (BytesIO)
|
||
webp_buffer = BytesIO()
|
||
img.save(webp_buffer, format="WEBP", quality=THUMBNAIL_WEBP_QUALITY)
|
||
webp_buffer.seek(0)
|
||
logger.debug(f"Изображение конвертировано в WebP (качество={THUMBNAIL_WEBP_QUALITY})")
|
||
|
||
# Шаг 4: КРИТИЧНО - Переоткрываем WebP из памяти и получаем его РЕАЛЬНЫЕ размеры
|
||
# Это гарантирует что размеры совпадают с тем что РЕАЛЬНО будет сохранено в хранилище!
|
||
webp_buffer.seek(0)
|
||
webp_img = PILImage.open(BytesIO(webp_buffer.read()))
|
||
image_width, image_height = webp_img.size
|
||
logger.info(f"Получены размеры CONVERTED WebP: {image_width}x{image_height}px")
|
||
|
||
# Шаг 5: Возвращаем WebP контент в начало буфера для последующего сохранения
|
||
webp_buffer.seek(0)
|
||
|
||
# Шаг 6: Изменяем расширение на .webp
|
||
new_name = os.path.splitext(name)[0] + ".webp"
|
||
logger.info(f"Конвертировано '{name}' → '{new_name}' (WebP). "
|
||
f"Размеры: {image_width}x{image_height}px")
|
||
|
||
# Возвращаем WebP контент и размеры из уже преобразованного файла
|
||
return ContentFile(webp_buffer.read()), new_name, True, image_width, image_height
|
||
|
||
except Exception:
|
||
logger.error(f"Ошибка при конвертации '{name}' в WebP.", exc_info=True)
|
||
# При ошибке возвращаем исходный файл без размеров
|
||
content.seek(0)
|
||
return content, name, False, None, None
|
||
|
||
# Если файл не нуждается в конвертации, возвращаем как есть
|
||
content.seek(0)
|
||
return content, name, False, None, None
|
||
|
||
def ready(self):
|
||
"""
|
||
Готовимся к работе: патчим MultiStorageFieldFile.save() для WebP конвертации.
|
||
"""
|
||
# КРИТИЧЕСКИ ВАЖНО: Переопределяем IMAGE_EXTENSIONS и IMAGE_MIME_TYPES в filer
|
||
# Это гарантирует что filer распознает WebP файлы как изображения, а не как обычные файлы
|
||
#
|
||
# Почему это нужно делать здесь в apps.py:
|
||
# - IMAGE_EXTENSIONS и IMAGE_MIME_TYPES в filer это ГЛОБАЛЬНЫЕ КОНСТАНТЫ (не getattr настройки)
|
||
# - Они импортируются как: from filer.settings import IMAGE_EXTENSIONS
|
||
# - Нет встроенного механизма для переопределения через settings.py
|
||
# - Единственный способ - модифицировать их после импорта, в ready() методе AppConfig
|
||
#
|
||
# Это стандартная практика для переопределения внутренних настроек третьих пакетов в Django
|
||
from filer import settings as filer_settings
|
||
|
||
# Добавляем WebP и HEIC/HEIF расширения к списку расширений изображений
|
||
if '.webp' not in filer_settings.IMAGE_EXTENSIONS:
|
||
filer_settings.IMAGE_EXTENSIONS = list(filer_settings.IMAGE_EXTENSIONS) + ['.webp']
|
||
if '.heic' not in filer_settings.IMAGE_EXTENSIONS:
|
||
filer_settings.IMAGE_EXTENSIONS = list(filer_settings.IMAGE_EXTENSIONS) + ['.heic']
|
||
if '.heif' not in filer_settings.IMAGE_EXTENSIONS:
|
||
filer_settings.IMAGE_EXTENSIONS = list(filer_settings.IMAGE_EXTENSIONS) + ['.heif']
|
||
|
||
# Добавляем WebP и HEIC/HEIF mime-типы
|
||
if 'webp' not in filer_settings.IMAGE_MIME_TYPES:
|
||
filer_settings.IMAGE_MIME_TYPES = list(filer_settings.IMAGE_MIME_TYPES) + ['webp']
|
||
if 'heic' not in filer_settings.IMAGE_MIME_TYPES:
|
||
filer_settings.IMAGE_MIME_TYPES = list(filer_settings.IMAGE_MIME_TYPES) + ['heic']
|
||
if 'heif' not in filer_settings.IMAGE_MIME_TYPES:
|
||
filer_settings.IMAGE_MIME_TYPES = list(filer_settings.IMAGE_MIME_TYPES) + ['heif']
|
||
|
||
logger.info(f"Обновлены filer IMAGE_EXTENSIONS: {filer_settings.IMAGE_EXTENSIONS}")
|
||
logger.info(f"Обновлены filer IMAGE_MIME_TYPES: {filer_settings.IMAGE_MIME_TYPES}")
|
||
|
||
from filer.fields.multistorage_file import MultiStorageFieldFile
|
||
|
||
logger.info("Patching MultiStorageFieldFile.save() for WebP conversion...")
|
||
original_save = MultiStorageFieldFile.save
|
||
|
||
def patched_save(self_instance, name, content, save=True):
|
||
"""
|
||
Патчированная версия MultiStorageFieldFile.save() для конвертации HEIC/HEIF в WebP.
|
||
|
||
Процесс:
|
||
1. Конвертирует загруженное изображение в WebP (JPEG, PNG, BMP, TIFF, HEIC, HEIF)
|
||
2. Устанавливает правильный mime_type=image/webp
|
||
3. Вызывает оригинальный save() с преобразованным контентом
|
||
4. Filer автоматически распознает WebP как Image, читает размеры и создает Image запись
|
||
"""
|
||
# Шаг 1: Преобразуем загруженный файл в WebP если требуется
|
||
# Метод вернет размеры ИЗ ПРЕОБРАЗОВАННОГО WebP (не исходного файла)
|
||
new_content, new_name, converted, img_width, img_height = CustomFilerConfig._convert_to_webp_if_needed(name, content)
|
||
|
||
# Шаг 2: Если файл был конвертирован в WebP, обновляем метаданные
|
||
if converted:
|
||
# Важно: Устанавливаем mime_type ПЕРЕД сохранением
|
||
# Filer использует mime_type чтобы определить тип файла (Image vs File)
|
||
self_instance.instance.mime_type = "image/webp"
|
||
logger.info(f"[WebP] Конвертировано '{name}' → '{new_name}'. "
|
||
f"Size: {img_width}x{img_height}px, mime_type=image/webp")
|
||
|
||
# Обновляем original_filename если есть
|
||
if hasattr(self_instance.instance, 'original_filename') and self_instance.instance.original_filename:
|
||
original_filename, _ = os.path.splitext(self_instance.instance.original_filename)
|
||
self_instance.instance.original_filename = f"{original_filename}.webp"
|
||
|
||
# Шаг 3: Вызываем оригинальный save() метод с new_name и new_content
|
||
result = original_save(self_instance, new_name, new_content, save)
|
||
|
||
# Шаг 4: После original_save() гарантируем что mime_type=image/webp сохранен в БД
|
||
# (на случай если решение о типе файла уже было принято)
|
||
if converted and new_name.lower().endswith('.webp'):
|
||
# Если instance имеет id после save, обновляем mime_type в БД
|
||
if self_instance.instance.id:
|
||
# Обновляем только mime_type
|
||
self_instance.instance.__class__.objects.filter(id=self_instance.instance.id).update(
|
||
mime_type='image/webp'
|
||
)
|
||
logger.info(f"[WebP] Updated mime_type in DB for id={self_instance.instance.id}")
|
||
|
||
return result
|
||
|
||
MultiStorageFieldFile.save = patched_save
|
||
logger.info("MultiStorageFieldFile.save() patched successfully.")
|
||
logger.info("Все остальное сделает filer - он распознает image/webp по mime_type и создаст Image запись!")
|
||
|
||
# Регистрируем pre_save сигнал для изменения mime_type HEIC/HEIF файлов ДО их сохранения
|
||
# Это гарантирует что filer распознает их как Image, а не как File
|
||
from filer.models import File as FilerFile
|
||
|
||
def fix_heic_mime_type_before_save(sender, instance, **kwargs):
|
||
"""
|
||
Перехватывает HEIC/HEIF файлы ДО сохранения и восстанавливает mime_type='image/webp'.
|
||
|
||
Проблема: Браузер отправляет mime_type='image/heic' для HEIC файлов.
|
||
Filer видит расширение/имя и может неправильно определить тип.
|
||
Решение: Устанавливаем mime_type='image/webp' ДО сохранения.
|
||
Это гарантирует что filer распознает как Image, а не как File.
|
||
"""
|
||
if instance.file and hasattr(instance.file, 'name'):
|
||
file_name = instance.file.name.lower() if instance.file.name else ""
|
||
|
||
# Если это WebP файл или конвертированный HEIC/HEIF, устанавливаем правильный mime_type
|
||
if file_name.endswith('.webp') or file_name.endswith('.heic') or file_name.endswith('.heif'):
|
||
old_mime = instance.mime_type
|
||
instance.mime_type = 'image/webp'
|
||
logger.info(f"[PRE-SAVE] Fixed mime_type for {file_name}: '{old_mime}' → 'image/webp'")
|
||
|
||
pre_save.connect(fix_heic_mime_type_before_save, sender=FilerFile, dispatch_uid="fix_heic_mime_before_save")
|
||
logger.info("Pre-save signal handler for HEIC mime_type fixing registered.")
|
||
|