mirror of
https://github.com/zhinianboke/xianyu-auto-reply.git
synced 2025-08-30 01:27:35 +08:00
217 lines
8.9 KiB
Python
217 lines
8.9 KiB
Python
"""
|
||
图片上传器 - 负责将图片上传到闲鱼CDN
|
||
"""
|
||
import aiohttp
|
||
import asyncio
|
||
import json
|
||
import os
|
||
import tempfile
|
||
from typing import Optional, Dict, Any
|
||
from loguru import logger
|
||
from PIL import Image
|
||
import io
|
||
|
||
|
||
class ImageUploader:
|
||
"""图片上传器 - 上传图片到闲鱼CDN"""
|
||
|
||
def __init__(self, cookies_str: str):
|
||
self.cookies_str = cookies_str
|
||
self.upload_url = "https://stream-upload.goofish.com/api/upload.api?floderId=0&appkey=xy_chat&_input_charset=utf-8"
|
||
self.session = None
|
||
|
||
async def create_session(self):
|
||
"""创建HTTP会话"""
|
||
if not self.session:
|
||
connector = aiohttp.TCPConnector(limit=100, limit_per_host=30)
|
||
timeout = aiohttp.ClientTimeout(total=30)
|
||
self.session = aiohttp.ClientSession(
|
||
connector=connector,
|
||
timeout=timeout,
|
||
headers={
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
|
||
}
|
||
)
|
||
|
||
async def close_session(self):
|
||
"""关闭HTTP会话"""
|
||
if self.session:
|
||
await self.session.close()
|
||
self.session = None
|
||
|
||
def _compress_image(self, image_path: str, max_size: int = 5 * 1024 * 1024, quality: int = 85) -> Optional[str]:
|
||
"""压缩图片"""
|
||
try:
|
||
with Image.open(image_path) as img:
|
||
# 转换为RGB模式（如果是RGBA或其他模式）
|
||
if img.mode in ('RGBA', 'LA', 'P'):
|
||
# 创建白色背景
|
||
background = Image.new('RGB', img.size, (255, 255, 255))
|
||
if img.mode == 'P':
|
||
img = img.convert('RGBA')
|
||
background.paste(img, mask=img.split()[-1] if img.mode in ('RGBA', 'LA') else None)
|
||
img = background
|
||
elif img.mode != 'RGB':
|
||
img = img.convert('RGB')
|
||
|
||
# 获取原始尺寸
|
||
original_width, original_height = img.size
|
||
|
||
# 如果图片太大，调整尺寸
|
||
max_dimension = 1920
|
||
if original_width > max_dimension or original_height > max_dimension:
|
||
if original_width > original_height:
|
||
new_width = max_dimension
|
||
new_height = int((original_height * max_dimension) / original_width)
|
||
else:
|
||
new_height = max_dimension
|
||
new_width = int((original_width * max_dimension) / original_height)
|
||
|
||
img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
|
||
logger.info(f"图片尺寸调整: {original_width}x{original_height} -> {new_width}x{new_height}")
|
||
|
||
# 创建临时文件
|
||
temp_fd, temp_path = tempfile.mkstemp(suffix='.jpg')
|
||
os.close(temp_fd)
|
||
|
||
# 保存压缩后的图片
|
||
img.save(temp_path, 'JPEG', quality=quality, optimize=True)
|
||
|
||
# 检查文件大小
|
||
file_size = os.path.getsize(temp_path)
|
||
if file_size > max_size:
|
||
# 如果还是太大，降低质量
|
||
quality = max(30, quality - 20)
|
||
img.save(temp_path, 'JPEG', quality=quality, optimize=True)
|
||
file_size = os.path.getsize(temp_path)
|
||
logger.info(f"图片质量调整为 {quality}%，文件大小: {file_size / 1024:.1f}KB")
|
||
|
||
logger.info(f"图片压缩完成: {file_size / 1024:.1f}KB")
|
||
return temp_path
|
||
|
||
except Exception as e:
|
||
logger.error(f"图片压缩失败: {e}")
|
||
return None
|
||
|
||
async def upload_image(self, image_path: str) -> Optional[str]:
|
||
"""上传图片到闲鱼CDN"""
|
||
temp_path = None
|
||
try:
|
||
if not self.session:
|
||
await self.create_session()
|
||
|
||
# 压缩图片
|
||
temp_path = self._compress_image(image_path)
|
||
if not temp_path:
|
||
logger.error("图片压缩失败")
|
||
return None
|
||
|
||
# 读取压缩后的图片数据
|
||
with open(temp_path, 'rb') as f:
|
||
image_data = f.read()
|
||
|
||
# 构造文件名
|
||
filename = os.path.basename(image_path)
|
||
if not filename.lower().endswith(('.jpg', '.jpeg')):
|
||
filename = os.path.splitext(filename)[0] + '.jpg'
|
||
|
||
# 构造请求头
|
||
headers = {
|
||
'cookie': self.cookies_str,
|
||
'Referer': 'https://www.goofish.com/',
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
||
'x-requested-with': 'XMLHttpRequest',
|
||
'Accept': 'application/json, text/javascript, */*; q=0.01',
|
||
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
||
'Accept-Encoding': 'gzip, deflate, br',
|
||
'Connection': 'keep-alive',
|
||
'Sec-Fetch-Dest': 'empty',
|
||
'Sec-Fetch-Mode': 'cors',
|
||
'Sec-Fetch-Site': 'same-site'
|
||
}
|
||
|
||
# 构造multipart/form-data
|
||
data = aiohttp.FormData()
|
||
data.add_field('file', image_data, filename=filename, content_type='image/jpeg')
|
||
|
||
# 发送上传请求
|
||
logger.info(f"开始上传图片到闲鱼CDN: {filename}")
|
||
async with self.session.post(self.upload_url, data=data, headers=headers) as response:
|
||
if response.status == 200:
|
||
response_text = await response.text()
|
||
logger.debug(f"上传响应: {response_text}")
|
||
|
||
# 解析响应获取图片URL
|
||
image_url = self._parse_upload_response(response_text)
|
||
if image_url:
|
||
logger.info(f"图片上传成功: {image_url}")
|
||
return image_url
|
||
else:
|
||
logger.error("解析上传响应失败")
|
||
return None
|
||
else:
|
||
logger.error(f"图片上传失败: HTTP {response.status}")
|
||
return None
|
||
|
||
except Exception as e:
|
||
logger.error(f"图片上传异常: {e}")
|
||
return None
|
||
finally:
|
||
# 清理临时文件
|
||
if temp_path and os.path.exists(temp_path):
|
||
try:
|
||
os.remove(temp_path)
|
||
except:
|
||
pass
|
||
|
||
def _parse_upload_response(self, response_text: str) -> Optional[str]:
|
||
"""解析上传响应获取图片URL"""
|
||
try:
|
||
# 尝试解析JSON响应
|
||
response_data = json.loads(response_text)
|
||
|
||
# 方式1: 标准响应格式
|
||
if 'data' in response_data and 'url' in response_data['data']:
|
||
return response_data['data']['url']
|
||
|
||
# 方式2: 在object字段中（闲鱼CDN的响应格式）
|
||
if 'object' in response_data and isinstance(response_data['object'], dict):
|
||
obj = response_data['object']
|
||
if 'url' in obj:
|
||
logger.info(f"从object.url提取到图片URL: {obj['url']}")
|
||
return obj['url']
|
||
|
||
# 方式3: 直接在根级别
|
||
if 'url' in response_data:
|
||
return response_data['url']
|
||
|
||
# 方式4: 在result中
|
||
if 'result' in response_data and 'url' in response_data['result']:
|
||
return response_data['result']['url']
|
||
|
||
# 方式5: 检查是否有文件信息
|
||
if 'data' in response_data and isinstance(response_data['data'], dict):
|
||
data = response_data['data']
|
||
if 'fileUrl' in data:
|
||
return data['fileUrl']
|
||
if 'file_url' in data:
|
||
return data['file_url']
|
||
|
||
logger.error(f"无法从响应中提取图片URL: {response_data}")
|
||
return None
|
||
|
||
except json.JSONDecodeError:
|
||
# 如果不是JSON格式，尝试其他解析方式
|
||
logger.error(f"响应不是有效的JSON格式: {response_text}")
|
||
return None
|
||
except Exception as e:
|
||
logger.error(f"解析上传响应异常: {e}")
|
||
return None
|
||
|
||
async def __aenter__(self):
|
||
await self.create_session()
|
||
return self
|
||
|
||
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
||
await self.close_session()
|