mirror of
https://github.com/zhtyyx/ioe.git
synced 2026-06-03 21:02:59 +08:00
149 lines
3.8 KiB
Python
149 lines
3.8 KiB
Python
"""
|
||
图片处理工具函数
|
||
"""
|
||
from io import BytesIO
|
||
from PIL import Image, ImageOps
|
||
import os
|
||
|
||
|
||
def generate_thumbnail(image_file, size=(300, 300), format='JPEG', quality=85):
|
||
"""
|
||
生成图片缩略图
|
||
|
||
参数:
|
||
image_file: 原始图片文件对象(Django UploadedFile或文件路径)
|
||
size: 缩略图尺寸，格式为(宽, 高)
|
||
format: 图片格式('JPEG', 'PNG'等)
|
||
quality: 图片质量(1-100)
|
||
|
||
返回:
|
||
PIL.Image: 处理后的缩略图对象
|
||
"""
|
||
# 如果是文件路径，打开文件
|
||
if isinstance(image_file, str):
|
||
img = Image.open(image_file)
|
||
# 如果是Django的InMemoryUploadedFile或TemporaryUploadedFile
|
||
elif hasattr(image_file, 'read'):
|
||
if hasattr(image_file, 'seek'):
|
||
image_file.seek(0)
|
||
img = Image.open(image_file)
|
||
else:
|
||
# 已经是PIL.Image对象
|
||
img = image_file
|
||
|
||
# 转换为RGB模式（去除透明通道）
|
||
if img.mode != 'RGB':
|
||
img = img.convert('RGB')
|
||
|
||
# 生成缩略图
|
||
img.thumbnail(size, Image.Resampling.LANCZOS)
|
||
|
||
# 确保缩略图是指定的尺寸（通过填充）
|
||
thumb = ImageOps.fit(img, size, Image.Resampling.LANCZOS)
|
||
|
||
return thumb
|
||
|
||
|
||
def save_thumbnail(image, path, format='JPEG', quality=85):
|
||
"""
|
||
保存缩略图到指定路径
|
||
|
||
参数:
|
||
image: PIL.Image对象
|
||
path: 保存路径
|
||
format: 图片格式
|
||
quality: 图片质量
|
||
|
||
返回:
|
||
str: 保存后的路径
|
||
"""
|
||
# 确保目录存在
|
||
directory = os.path.dirname(path)
|
||
if directory and not os.path.exists(directory):
|
||
os.makedirs(directory)
|
||
|
||
# 保存图片
|
||
image.save(path, format=format, quality=quality)
|
||
|
||
return path
|
||
|
||
|
||
def image_to_base64(image, format='JPEG', quality=85):
|
||
"""
|
||
将图片转换为base64编码
|
||
|
||
参数:
|
||
image: PIL.Image对象
|
||
format: 图片格式
|
||
quality: 图片质量
|
||
|
||
返回:
|
||
str: base64编码的图片数据
|
||
"""
|
||
import base64
|
||
|
||
buffered = BytesIO()
|
||
image.save(buffered, format=format, quality=quality)
|
||
img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
|
||
|
||
return f"data:image/{format.lower()};base64,{img_str}"
|
||
|
||
|
||
def resize_image(image_file, size, format='JPEG', quality=85):
|
||
"""
|
||
调整图片大小
|
||
|
||
参数:
|
||
image_file: 原始图片文件对象或路径
|
||
size: 新尺寸，格式为(宽, 高)
|
||
format: 图片格式
|
||
quality: 图片质量
|
||
|
||
返回:
|
||
PIL.Image: 调整大小后的图片
|
||
"""
|
||
# 如果是文件路径，打开文件
|
||
if isinstance(image_file, str):
|
||
img = Image.open(image_file)
|
||
# 如果是Django的InMemoryUploadedFile或TemporaryUploadedFile
|
||
elif hasattr(image_file, 'read'):
|
||
if hasattr(image_file, 'seek'):
|
||
image_file.seek(0)
|
||
img = Image.open(image_file)
|
||
else:
|
||
# 已经是PIL.Image对象
|
||
img = image_file
|
||
|
||
# 转换为RGB模式（去除透明通道）
|
||
if img.mode != 'RGB':
|
||
img = img.convert('RGB')
|
||
|
||
# 调整大小
|
||
resized_img = img.resize(size, Image.Resampling.LANCZOS)
|
||
|
||
return resized_img
|
||
|
||
|
||
def get_image_dimensions(image_file):
|
||
"""
|
||
获取图片尺寸
|
||
|
||
参数:
|
||
image_file: 图片文件对象或路径
|
||
|
||
返回:
|
||
tuple: (宽, 高)
|
||
"""
|
||
# 如果是文件路径，打开文件
|
||
if isinstance(image_file, str):
|
||
img = Image.open(image_file)
|
||
# 如果是Django的InMemoryUploadedFile或TemporaryUploadedFile
|
||
elif hasattr(image_file, 'read'):
|
||
if hasattr(image_file, 'seek'):
|
||
image_file.seek(0)
|
||
img = Image.open(image_file)
|
||
else:
|
||
# 已经是PIL.Image对象
|
||
img = image_file
|
||
|
||
return img.size |