- 商品详情页「AI 生成三视图」原走独立生图老链路(纯文生图),只凭商品名脑补, 出来通用图不像真品。改:前端传 reference_product;后端 run_standalone_image_task 据此取商品主图走 image_edit(复用 build_product_triview_prompt_refs 锁包装), 取不到主图优雅回落文生图。图片创作页(自由创作)不带标记,不受影响。 - 商品详情页素材区改为只显示「该商品」AI 素材(asset.product 归属)而非全团队。 - 首登水合带重试,失败不再静默(否则页面卡在全 0,要刷新才好)。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
from django.conf import settings
|
|
from rest_framework import serializers
|
|
|
|
from .models import Asset, AssetFile
|
|
from .storage import TosStorage
|
|
|
|
|
|
_tos_storage = None
|
|
|
|
|
|
def _tos():
|
|
global _tos_storage
|
|
if _tos_storage is None:
|
|
_tos_storage = TosStorage()
|
|
return _tos_storage
|
|
|
|
|
|
class AssetFileSerializer(serializers.ModelSerializer):
|
|
preview_url = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = AssetFile
|
|
fields = [
|
|
"id",
|
|
"object_key",
|
|
"bucket",
|
|
"content_type",
|
|
"size_bytes",
|
|
"width",
|
|
"height",
|
|
"duration_ms",
|
|
"preview_url",
|
|
"is_primary",
|
|
]
|
|
read_only_fields = [
|
|
"id",
|
|
"object_key",
|
|
"bucket",
|
|
"content_type",
|
|
"size_bytes",
|
|
"width",
|
|
"height",
|
|
"duration_ms",
|
|
"is_primary",
|
|
]
|
|
|
|
def get_preview_url(self, obj):
|
|
# 存储字段优先(如外部已写入绝对 URL);否则用 object_key 实时签发 TOS 预签名 GET URL
|
|
if obj.preview_url:
|
|
return obj.preview_url
|
|
if not obj.object_key or not settings.TOS.get("endpoint"):
|
|
return ""
|
|
try:
|
|
return _tos().presigned_get_url(object_key=obj.object_key)
|
|
except Exception:
|
|
return ""
|
|
|
|
|
|
class AssetSerializer(serializers.ModelSerializer):
|
|
files = AssetFileSerializer(many=True, read_only=True)
|
|
# 资产归属商品(只读):商品详情页据此只展示「该商品」的 AI 素材,而非全团队
|
|
product = serializers.SerializerMethodField()
|
|
|
|
def get_product(self, obj):
|
|
"""解析资产所属商品:
|
|
1) 独立生图(图片创作/模特图/平台套图):生图时已写入 metadata.product_id;
|
|
2) 项目内生成(基础资产/分镜图等):回溯 origin_task → project → product。
|
|
两条路都拿不到则归属 None(如纯手动上传到素材库、与任何商品无关的图)。"""
|
|
pid = (obj.metadata or {}).get("product_id")
|
|
if pid:
|
|
return str(pid)
|
|
task = obj.origin_task
|
|
if task and task.project_id and task.project.product_id:
|
|
return str(task.project.product_id)
|
|
return None
|
|
|
|
class Meta:
|
|
model = Asset
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"asset_type",
|
|
"source",
|
|
"category",
|
|
"description",
|
|
"metadata",
|
|
"is_deleted",
|
|
"origin_task",
|
|
"product",
|
|
"files",
|
|
"review_status",
|
|
"review_error",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
read_only_fields = ["id", "review_status", "review_error", "created_at", "updated_at"]
|
|
|
|
|
|
class AssetUploadSerializer(serializers.Serializer):
|
|
file = serializers.FileField()
|
|
name = serializers.CharField(max_length=255, required=False, allow_blank=True)
|
|
asset_type = serializers.ChoiceField(choices=Asset.Type.choices)
|
|
category = serializers.ChoiceField(choices=Asset.Category.choices, default=Asset.Category.UPLOAD)
|
|
description = serializers.CharField(required=False, allow_blank=True)
|
|
|